do the same for paletted and rgb textues

This commit is contained in:
Ricardo Luís Vaz Silva 2024-03-31 19:11:49 -03:00 committed by Rachael Alexanderson
parent 74e9d575e8
commit 3b6c8349da
2 changed files with 35 additions and 46 deletions

View file

@ -48,7 +48,7 @@ void AnimTexture::SetFrameSize(int format, int width, int height)
{ {
pixelformat = format; pixelformat = format;
FTexture::SetSize(width, height); FTexture::SetSize(width, height);
Image.Resize(width * height * (format == Paletted ? 1 : format == RGB ? 3 : 4)); Image.Resize(width * height * 4);
memset(Image.Data(), 0, Image.Size()); memset(Image.Data(), 0, Image.Size());
} }
@ -68,15 +68,16 @@ static inline void YUVtoRGB(uint8_t yi, uint8_t ui, uint8_t vi, uint8_t * rgb)
(rgb)[3] = 255; (rgb)[3] = 255;
} }
void AnimTexture::SetFrame(const uint8_t* palette, const void* data_) void AnimTexture::SetFrame(const uint8_t* Palette, const void* data_)
{ {
if (palette) memcpy(Palette, palette, 768);
if (data_) if (data_)
{ {
auto dpix = Image.Data();
if (pixelformat == YUV) if (pixelformat == YUV)
{ {
auto spix = (const uint8_t*)data_; const uint8_t * spix = reinterpret_cast<const uint8_t *>(data_);
auto dpix = Image.Data();
for (int i = 0; i < Width * Height; i++) for (int i = 0; i < Width * Height; i++)
{ {
YUVtoRGB(spix[0], spix[1], spix[2], dpix); YUVtoRGB(spix[0], spix[1], spix[2], dpix);
@ -97,8 +98,6 @@ void AnimTexture::SetFrame(const uint8_t* palette, const void* data_)
const int ustride = img->stride[VPX_PLANE_U]; const int ustride = img->stride[VPX_PLANE_U];
const int vstride = img->stride[VPX_PLANE_V]; const int vstride = img->stride[VPX_PLANE_V];
auto dpix = Image.Data();
if(img->fmt == VPX_IMG_FMT_I420) if(img->fmt == VPX_IMG_FMT_I420)
{ {
for (unsigned int y = 0; y < Height; y++) for (unsigned int y = 0; y < Height; y++)
@ -165,9 +164,35 @@ void AnimTexture::SetFrame(const uint8_t* palette, const void* data_)
} }
} }
} }
else else if(pixelformat == RGB)
{ {
memcpy(Image.Data(), data_, Width * Height * (pixelformat == Paletted ? 1 : 3)); const uint8_t *img = reinterpret_cast<const uint8_t *>(data_);
for (int i = 0; i < Width * Height; i++)
{
dpix[0] = img[2];
dpix[1] = img[1];
dpix[2] = img[0];
dpix[3] = 255;
dpix += 4;
}
}
else if (pixelformat == Paletted)
{
assert(Palette);
const uint8_t *img = reinterpret_cast<const uint8_t *>(data_);
for (int i = 0; i < Width * Height; i++)
{
int index = img[i];
dpix[0] = Palette[index * 3 + 2];
dpix[1] = Palette[index * 3 + 1];
dpix[2] = Palette[index * 3];
dpix[3] = 255;
dpix += 4;
}
} }
} }
} }
@ -180,42 +205,7 @@ void AnimTexture::SetFrame(const uint8_t* palette, const void* data_)
FBitmap AnimTexture::GetBgraBitmap(const PalEntry* remap, int* trans) FBitmap AnimTexture::GetBgraBitmap(const PalEntry* remap, int* trans)
{ {
if(pixelformat == YUV || pixelformat == VPX)
{
return FBitmap(Image.Data(), Width * 4, Width, Height); return FBitmap(Image.Data(), Width * 4, Width, Height);
}
FBitmap bmp;
bmp.Create(Width, Height);
auto spix = Image.Data();
auto dpix = bmp.GetPixels();
if (pixelformat == Paletted)
{
for (int i = 0; i < Width * Height; i++)
{
int p = i * 4;
int index = spix[i];
dpix[p + 0] = Palette[index * 3 + 2];
dpix[p + 1] = Palette[index * 3 + 1];
dpix[p + 2] = Palette[index * 3];
dpix[p + 3] = 255;
}
}
else if (pixelformat == RGB)
{
for (int i = 0; i < Width * Height; i++)
{
int p = i * 4;
int q = i * 3;
dpix[p + 0] = spix[q + 2];
dpix[p + 1] = spix[q + 1];
dpix[p + 2] = spix[q];
dpix[p + 3] = 255;
}
}
return bmp;
} }
//========================================================================== //==========================================================================

View file

@ -5,7 +5,6 @@
class AnimTexture : public FTexture class AnimTexture : public FTexture
{ {
uint8_t Palette[768];
TArray<uint8_t> Image; TArray<uint8_t> Image;
int pixelformat; int pixelformat;
public: public: