#include"TextureFileCreater.h"
#include"ILImage.h"
#include<hgl/log/LogInfo.h>

class TextureFileCreaterRGB:public TextureFileCreater
{
    ILuint type;

public:

    using TextureFileCreater::TextureFileCreater;

public:

    bool InitFormat(ILImage* img) override
    {
        image = img;

        if(pixel_format->format==ColorFormat::RGB32U
         ||pixel_format->format==ColorFormat::RGB32I
         ||pixel_format->format==ColorFormat::RGB32F)
        {
            if(!ToILType(type,pixel_format->bits[0],pixel_format->type))
                return(false);
        }
        else if(pixel_format->format==ColorFormat::RGB565)
        {
            type=IL_UNSIGNED_BYTE;
        }
        else if(pixel_format->format==ColorFormat::B10GR11UF)
        {
            type=IL_HALF;
        }
        else
        {
            LOG_ERROR(OS_TEXT("Don't support this RGB format"));
            return(false);
        }

        return image->ConvertToRGB(type);
    }

    void RGB8toRGB565(uint16 *target,uint8 *src,uint size)
    {
        for(uint i=0;i<size;i++)
        {
            *target=((src[0]<<8)&0xF800)
                   |((src[1]<<3)&0x7E0)
                   | (src[2]>>3);

            ++target;
            src+=3;
        }
    }

    // Bit depth    Sign bit present    Exponent bits   Mantissa bits
    //  32              Yes                 8               23
    //  16              Yes                 5               10
    //  11              No                  5               6
    //  10              No                  5               5

    void RGB16FtoB10GR11UF(uint32 *target,uint16 *src,uint size)
    {
        for(uint i=0;i<size;i++)
        {
            *target=((src[2]&0x7FE0)<<17)
                   |((src[1]&0x7FF0)<<7)
                   | (src[0]&0x7FF0)>>4;

            ++target;
            src+=3;
        }
    }

public:
 
    uint32 Write() override
    {
        const uint total_bytes=(pixel_format->total_bits*image->pixel_total())>>3;
        
        std::cout<<"Convert Image To: "<<image->width()<<"x"<<image->height()<<" "<<total_bytes<<" bytes."<<std::endl;

        if(pixel_format->format==ColorFormat::RGB32U
         ||pixel_format->format==ColorFormat::RGB32I
         ||pixel_format->format==ColorFormat::RGB32F)
        {
            void *origin_rgb=image->GetRGB(type);

            return TextureFileCreater::Write(origin_rgb,total_bytes);
        }
        else if(pixel_format->format==ColorFormat::RGB565)
        {
            void *origin_rgb=image->GetRGB(IL_UNSIGNED_BYTE);

            AutoDeleteArray<uint16> rgb565(image->pixel_total());

            RGB8toRGB565(rgb565,(uint8 *)origin_rgb,image->pixel_total());

            return TextureFileCreater::Write(rgb565,total_bytes);
        }
        else if(pixel_format->format==ColorFormat::B10GR11UF)
        {
            void *origin_rgb=image->GetRGB(IL_HALF);

            AutoDeleteArray<uint32> b10gr11(image->pixel_total());

            RGB16FtoB10GR11UF(b10gr11,(uint16 *)origin_rgb,image->pixel_total());

            return TextureFileCreater::Write(b10gr11,total_bytes);
        }
        else
        {
            LOG_ERROR(OS_TEXT("Don't support this RGB format"));
            return(0);
        }
    }
};//class TextureFileCreaterRGB:public TextureFileCreater

TextureFileCreater *CreateTextureFileCreaterRGB(const PixelFormat *pf)
{
    return(new TextureFileCreaterRGB(pf));
}