You've already forked UnrealEngineUWP
mirror of
https://github.com/izzy2lost/UnrealEngineUWP.git
synced 2026-03-26 18:15:20 -07:00
d96ed2d532
add support for F32 float import without conversion fix incorrect quantizer in color conversion #rb fabian.giesen #preflight 627461351e4246d26acdde55 [CL 20073801 by charles bloom in ue5-main branch]
411 lines
7.9 KiB
C++
411 lines
7.9 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#include "ImageWrapperBase.h"
|
|
#include "ImageWrapperPrivate.h"
|
|
|
|
|
|
/* FImageWrapperBase structors
|
|
*****************************************************************************/
|
|
|
|
FImageWrapperBase::FImageWrapperBase()
|
|
: Format(ERGBFormat::Invalid)
|
|
, BitDepth(0)
|
|
, Width(0)
|
|
, Height(0)
|
|
{ }
|
|
|
|
|
|
/* FImageWrapperBase interface
|
|
*****************************************************************************/
|
|
|
|
void FImageWrapperBase::Reset()
|
|
{
|
|
LastError.Empty();
|
|
|
|
RawData.Empty();
|
|
CompressedData.Empty();
|
|
|
|
Format = ERGBFormat::Invalid;
|
|
BitDepth = 0;
|
|
Width = 0;
|
|
Height = 0;
|
|
}
|
|
|
|
|
|
void FImageWrapperBase::SetError(const TCHAR* ErrorMessage)
|
|
{
|
|
LastError = ErrorMessage;
|
|
}
|
|
|
|
|
|
/* IImageWrapper structors
|
|
*****************************************************************************/
|
|
|
|
TArray64<uint8> FImageWrapperBase::GetCompressed(int32 Quality)
|
|
{
|
|
LastError.Empty();
|
|
Compress(Quality);
|
|
|
|
return MoveTemp(CompressedData);
|
|
}
|
|
|
|
|
|
bool FImageWrapperBase::GetRaw(const ERGBFormat InFormat, int32 InBitDepth, TArray64<uint8>& OutRawData)
|
|
{
|
|
LastError.Empty();
|
|
Uncompress(InFormat, InBitDepth);
|
|
|
|
if ( ! LastError.IsEmpty())
|
|
{
|
|
UE_LOG(LogImageWrapper, Warning, TEXT("ImageWrapper GetRaw failed: %s"), *LastError);
|
|
return false;
|
|
}
|
|
|
|
if ( RawData.IsEmpty() )
|
|
{
|
|
return false;
|
|
}
|
|
|
|
OutRawData = MoveTemp(RawData);
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
bool FImageWrapperBase::SetCompressed(const void* InCompressedData, int64 InCompressedSize)
|
|
{
|
|
Reset();
|
|
RawData.Empty(); // Invalidates the raw data too
|
|
|
|
if(InCompressedSize > 0 && InCompressedData != nullptr)
|
|
{
|
|
// this is usually an unnecessary allocation and copy
|
|
// we should decompress directly from the source buffer
|
|
|
|
CompressedData.Empty(InCompressedSize);
|
|
CompressedData.AddUninitialized(InCompressedSize);
|
|
FMemory::Memcpy(CompressedData.GetData(), InCompressedData, InCompressedSize);
|
|
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
|
|
bool FImageWrapperBase::SetRaw(const void* InRawData, int64 InRawSize, const int32 InWidth, const int32 InHeight, const ERGBFormat InFormat, const int32 InBitDepth, const int32 InBytesPerRow)
|
|
{
|
|
check(InRawData != NULL);
|
|
check(InRawSize > 0);
|
|
check(InWidth > 0);
|
|
check(InHeight > 0);
|
|
check(InBytesPerRow >= 0);
|
|
|
|
Reset();
|
|
CompressedData.Empty(); // Invalidates the compressed data too
|
|
|
|
if ( ! CanSetRawFormat(InFormat,InBitDepth) )
|
|
{
|
|
UE_LOG(LogImageWrapper, Warning, TEXT("ImageWrapper unsupported format; check CanSetRawFormat; %d x %d"), (int)InFormat,InBitDepth);
|
|
return false;
|
|
}
|
|
|
|
Format = InFormat;
|
|
BitDepth = InBitDepth;
|
|
Width = InWidth;
|
|
Height = InHeight;
|
|
|
|
int BytesPerRow = GetBytesPerRow();
|
|
|
|
// stride not supported :
|
|
check( InBytesPerRow == 0 || BytesPerRow == InBytesPerRow );
|
|
|
|
check( InRawSize == BytesPerRow * Height );
|
|
|
|
// this is usually an unnecessary allocation and copy
|
|
// we should compress directly from the source buffer
|
|
|
|
RawData.Empty(InRawSize);
|
|
RawData.AddUninitialized(InRawSize);
|
|
FMemory::Memcpy(RawData.GetData(), InRawData, InRawSize);
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
int IImageWrapper::GetRGBFormatBytesPerPel(ERGBFormat RGBFormat,int BitDepth)
|
|
{
|
|
switch(RGBFormat)
|
|
{
|
|
case ERGBFormat::RGBA:
|
|
case ERGBFormat::BGRA:
|
|
if ( BitDepth == 8 )
|
|
{
|
|
return 4;
|
|
}
|
|
else if ( BitDepth == 16 )
|
|
{
|
|
return 8;
|
|
}
|
|
break;
|
|
|
|
case ERGBFormat::Gray:
|
|
if ( BitDepth == 8 )
|
|
{
|
|
return 1;
|
|
}
|
|
else if ( BitDepth == 16 )
|
|
{
|
|
return 2;
|
|
}
|
|
break;
|
|
|
|
case ERGBFormat::BGRE:
|
|
if ( BitDepth == 8 )
|
|
{
|
|
return 4;
|
|
}
|
|
break;
|
|
|
|
case ERGBFormat::RGBAF:
|
|
if ( BitDepth == 16 )
|
|
{
|
|
return 8;
|
|
}
|
|
else if ( BitDepth == 32 )
|
|
{
|
|
return 16;
|
|
}
|
|
break;
|
|
|
|
case ERGBFormat::GrayF:
|
|
if ( BitDepth == 16 )
|
|
{
|
|
return 2;
|
|
}
|
|
else if ( BitDepth == 32 )
|
|
{
|
|
return 4;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
UE_LOG(LogImageWrapper, Error, TEXT("GetRGBFormatBytesPerPel not handled : %d,%d"), (int)RGBFormat,BitDepth );
|
|
return 0;
|
|
}
|
|
|
|
|
|
ERawImageFormat::Type IImageWrapper::ConvertRGBFormat(ERGBFormat RGBFormat,int BitDepth,bool * bIsExactMatch)
|
|
{
|
|
bool bIsExactMatchDummy;
|
|
if ( ! bIsExactMatch )
|
|
{
|
|
bIsExactMatch = &bIsExactMatchDummy;
|
|
}
|
|
|
|
switch(RGBFormat)
|
|
{
|
|
case ERGBFormat::RGBA:
|
|
if ( BitDepth == 8 )
|
|
{
|
|
*bIsExactMatch = false; // needs RB swap
|
|
return ERawImageFormat::BGRA8;
|
|
}
|
|
else if ( BitDepth == 16 )
|
|
{
|
|
*bIsExactMatch = true;
|
|
return ERawImageFormat::RGBA16;
|
|
}
|
|
break;
|
|
|
|
case ERGBFormat::BGRA:
|
|
if ( BitDepth == 8 )
|
|
{
|
|
*bIsExactMatch = true;
|
|
return ERawImageFormat::BGRA8;
|
|
}
|
|
else if ( BitDepth == 16 )
|
|
{
|
|
*bIsExactMatch = false; // needs RB swap
|
|
return ERawImageFormat::RGBA16;
|
|
}
|
|
break;
|
|
|
|
case ERGBFormat::Gray:
|
|
if ( BitDepth == 8 )
|
|
{
|
|
*bIsExactMatch = true;
|
|
return ERawImageFormat::G8;
|
|
}
|
|
else if ( BitDepth == 16 )
|
|
{
|
|
*bIsExactMatch = true;
|
|
return ERawImageFormat::G16;
|
|
}
|
|
break;
|
|
|
|
case ERGBFormat::BGRE:
|
|
if ( BitDepth == 8 )
|
|
{
|
|
*bIsExactMatch = true;
|
|
return ERawImageFormat::BGRE8;
|
|
}
|
|
break;
|
|
|
|
case ERGBFormat::RGBAF:
|
|
if ( BitDepth == 16 )
|
|
{
|
|
*bIsExactMatch = true;
|
|
return ERawImageFormat::RGBA16F;
|
|
}
|
|
else if ( BitDepth == 32 )
|
|
{
|
|
*bIsExactMatch = true;
|
|
return ERawImageFormat::RGBA32F;
|
|
}
|
|
break;
|
|
|
|
case ERGBFormat::GrayF:
|
|
if ( BitDepth == 16 )
|
|
{
|
|
*bIsExactMatch = true;
|
|
return ERawImageFormat::R16F;
|
|
}
|
|
else if ( BitDepth == 32 )
|
|
{
|
|
*bIsExactMatch = true;
|
|
return ERawImageFormat::R32F;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
UE_LOG(LogImageWrapper, Warning, TEXT("ConvertRGBFormat not handled : %d,%d"), (int)RGBFormat,BitDepth );
|
|
|
|
*bIsExactMatch = false;
|
|
return ERawImageFormat::Invalid;
|
|
}
|
|
|
|
void IImageWrapper::ConvertRawImageFormat(ERawImageFormat::Type RawFormat, ERGBFormat & OutFormat,int & OutBitDepth)
|
|
{
|
|
switch(RawFormat)
|
|
{
|
|
case ERawImageFormat::G8:
|
|
OutFormat = ERGBFormat::Gray;
|
|
OutBitDepth = 8;
|
|
break;
|
|
case ERawImageFormat::BGRA8:
|
|
OutFormat = ERGBFormat::BGRA;
|
|
OutBitDepth = 8;
|
|
break;
|
|
case ERawImageFormat::BGRE8:
|
|
OutFormat = ERGBFormat::BGRE;
|
|
OutBitDepth = 8;
|
|
break;
|
|
case ERawImageFormat::RGBA16:
|
|
OutFormat = ERGBFormat::RGBA;
|
|
OutBitDepth = 16;
|
|
break;
|
|
case ERawImageFormat::RGBA16F:
|
|
OutFormat = ERGBFormat::RGBAF;
|
|
OutBitDepth = 16;
|
|
break;
|
|
case ERawImageFormat::RGBA32F:
|
|
OutFormat = ERGBFormat::RGBAF;
|
|
OutBitDepth = 32;
|
|
break;
|
|
case ERawImageFormat::G16:
|
|
OutFormat = ERGBFormat::Gray;
|
|
OutBitDepth = 16;
|
|
break;
|
|
case ERawImageFormat::R16F:
|
|
OutFormat = ERGBFormat::GrayF;
|
|
OutBitDepth = 16;
|
|
break;
|
|
case ERawImageFormat::R32F:
|
|
OutFormat = ERGBFormat::GrayF;
|
|
OutBitDepth = 32;
|
|
break;
|
|
default:
|
|
check(0);
|
|
break;
|
|
}
|
|
}
|
|
|
|
bool IImageWrapper::GetRawImage(FImage & OutImage)
|
|
{
|
|
TArray64<uint8> OutRawData;
|
|
if ( ! GetRaw(OutRawData) )
|
|
{
|
|
return false;
|
|
}
|
|
|
|
int Width = GetWidth();
|
|
int Height = GetHeight();
|
|
ERGBFormat RGBFormat = GetFormat();
|
|
int BitDepth = GetBitDepth();
|
|
|
|
bool bExactMatch;
|
|
ERawImageFormat::Type RawFormat = GetClosestRawImageFormat(&bExactMatch);
|
|
|
|
if ( RawFormat == ERawImageFormat::Invalid )
|
|
{
|
|
return false;
|
|
}
|
|
|
|
// ImageWrapper RGBFormat doesn't track if pixels are Gamma/sRGB or not
|
|
// just assume they are Default for now :
|
|
EGammaSpace GammaSpace = ERawImageFormat::GetDefaultGammaSpace(RawFormat);
|
|
|
|
if ( bExactMatch )
|
|
{
|
|
// no conversion required
|
|
|
|
OutImage.RawData = MoveTemp(OutRawData);
|
|
OutImage.SizeX = Width;
|
|
OutImage.SizeY = Height;
|
|
OutImage.NumSlices = 1;
|
|
OutImage.Format = RawFormat;
|
|
OutImage.GammaSpace = GammaSpace;
|
|
}
|
|
else
|
|
{
|
|
OutImage.Init( Width, Height, RawFormat, GammaSpace );
|
|
|
|
FImageView SrcImage = OutImage;
|
|
SrcImage.RawData = OutRawData.GetData();
|
|
|
|
switch(RGBFormat)
|
|
{
|
|
case ERGBFormat::RGBA:
|
|
{
|
|
// RGBA8 -> BGRA8
|
|
check( BitDepth == 8 );
|
|
check( RawFormat == ERawImageFormat::BGRA8 );
|
|
FImageCore::CopyImageRGBABGRA(SrcImage, OutImage );
|
|
break;
|
|
}
|
|
|
|
case ERGBFormat::BGRA:
|
|
{
|
|
// BGRA16 -> RGBA16
|
|
check( BitDepth == 16 );
|
|
check( RawFormat == ERawImageFormat::RGBA16 );
|
|
FImageCore::CopyImageRGBABGRA(SrcImage, OutImage );
|
|
break;
|
|
}
|
|
|
|
default:
|
|
check(0);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|