Code/Engine/RendererCore/Textures/TextureUtils.cpp - dev - redgpu/ezEngine

Code/Engine/RendererCore/Textures/TextureUtils.cpp
476 строк17 KB
Sanakan8472
Copy / Readback fixes (#1963)
11 июн 2026, 18:07
11 июн 2026, 18:07b50d6f1
#include <RendererCore/RendererCorePCH.h>

#include <Foundation/Memory/MemoryUtils.h>
#include <Foundation/Reflection/ReflectionUtils.h>
#include <RendererCore/RenderContext/RenderContext.h>
#include <RendererCore/Textures/TextureUtils.h>

bool ezTextureUtils::s_bForceFullQualityAlways = false;

namespace
{
  ezUInt32 GetMipSize(ezUInt32 uiSize, ezUInt32 uiMipLevel)
  {
    for (ezUInt32 i = 0; i < uiMipLevel; i++)
    {
      uiSize = uiSize / 2;
    }
    return ezMath::Max(1u, uiSize);
  }
} // namespace

ezGALResourceFormat::Enum ezTextureUtils::ImageFormatToGalFormat(ezImageFormat::Enum format, bool bSRGB)
{
  switch (format)
  {
    case ezImageFormat::R8G8B8A8_UNORM:
      if (bSRGB)
        return ezGALResourceFormat::RGBAUByteNormalizedsRGB;
      else
        return ezGALResourceFormat::RGBAUByteNormalized;

      // case ezImageFormat::R8G8B8A8_TYPELESS:
    case ezImageFormat::R8G8B8A8_UNORM_SRGB:
      return ezGALResourceFormat::RGBAUByteNormalizedsRGB;

    case ezImageFormat::R8G8B8A8_UINT:
      return ezGALResourceFormat::RGBAUInt;

    case ezImageFormat::R8G8B8A8_SNORM:
      return ezGALResourceFormat::RGBAByteNormalized;

    case ezImageFormat::R8G8B8A8_SINT:
      return ezGALResourceFormat::RGBAInt;

    case ezImageFormat::B8G8R8A8_UNORM:
      if (bSRGB)
        return ezGALResourceFormat::BGRAUByteNormalizedsRGB;
      else
        return ezGALResourceFormat::BGRAUByteNormalized;

    case ezImageFormat::B8G8R8X8_UNORM:
      if (bSRGB)
        return ezGALResourceFormat::BGRAUByteNormalizedsRGB;
      else
        return ezGALResourceFormat::BGRAUByteNormalized;

      // case ezImageFormat::B8G8R8A8_TYPELESS:
    case ezImageFormat::B8G8R8A8_UNORM_SRGB:
      return ezGALResourceFormat::BGRAUByteNormalizedsRGB;

      // case ezImageFormat::B8G8R8X8_TYPELESS:
    case ezImageFormat::B8G8R8X8_UNORM_SRGB:
      return ezGALResourceFormat::BGRAUByteNormalizedsRGB;

      // case ezImageFormat::B8G8R8_UNORM:

      // case ezImageFormat::BC1_TYPELESS:
    case ezImageFormat::BC1_UNORM:
      if (bSRGB)
        return ezGALResourceFormat::BC1sRGB;
      else
        return ezGALResourceFormat::BC1;

    case ezImageFormat::BC1_UNORM_SRGB:
      return ezGALResourceFormat::BC1sRGB;

      // case ezImageFormat::BC2_TYPELESS:
    case ezImageFormat::BC2_UNORM:
      if (bSRGB)
        return ezGALResourceFormat::BC2sRGB;
      else
        return ezGALResourceFormat::BC2;

    case ezImageFormat::BC2_UNORM_SRGB:
      return ezGALResourceFormat::BC2sRGB;

      // case ezImageFormat::BC3_TYPELESS:
    case ezImageFormat::BC3_UNORM:
      if (bSRGB)
        return ezGALResourceFormat::BC3sRGB;
      else
        return ezGALResourceFormat::BC3;

    case ezImageFormat::BC3_UNORM_SRGB:
      return ezGALResourceFormat::BC3sRGB;

      // case ezImageFormat::BC4_TYPELESS:
    case ezImageFormat::BC4_UNORM:
      return ezGALResourceFormat::BC4UNormalized;

    case ezImageFormat::BC4_SNORM:
      return ezGALResourceFormat::BC4Normalized;

      // case ezImageFormat::BC5_TYPELESS:
    case ezImageFormat::BC5_UNORM:
      return ezGALResourceFormat::BC5UNormalized;

    case ezImageFormat::BC5_SNORM:
      return ezGALResourceFormat::BC5Normalized;

      // case ezImageFormat::BC6H_TYPELESS:
    case ezImageFormat::BC6H_UF16:
      return ezGALResourceFormat::BC6UFloat;

    case ezImageFormat::BC6H_SF16:
      return ezGALResourceFormat::BC6Float;

      // case ezImageFormat::BC7_TYPELESS:
    case ezImageFormat::BC7_UNORM:
      if (bSRGB)
        return ezGALResourceFormat::BC7UNormalizedsRGB;
      else
        return ezGALResourceFormat::BC7UNormalized;

    case ezImageFormat::BC7_UNORM_SRGB:
      return ezGALResourceFormat::BC7UNormalizedsRGB;

    case ezImageFormat::B5G6R5_UNORM:
      return ezGALResourceFormat::B5G6R5UNormalized; /// \todo Not supported by some GPUs ?

    case ezImageFormat::R16_FLOAT:
      return ezGALResourceFormat::RHalf;

    case ezImageFormat::R32_FLOAT:
      return ezGALResourceFormat::RFloat;

    case ezImageFormat::R16G16_FLOAT:
      return ezGALResourceFormat::RGHalf;

    case ezImageFormat::R32G32_FLOAT:
      return ezGALResourceFormat::RGFloat;

    case ezImageFormat::R32G32B32_FLOAT:
      return ezGALResourceFormat::RGBFloat;

    case ezImageFormat::R16G16B16A16_FLOAT:
      return ezGALResourceFormat::RGBAHalf;

    case ezImageFormat::R32G32B32A32_FLOAT:
      return ezGALResourceFormat::RGBAFloat;

    case ezImageFormat::R16G16B16A16_UNORM:
      return ezGALResourceFormat::RGBAUShortNormalized;

    case ezImageFormat::R8_UNORM:
      return ezGALResourceFormat::RUByteNormalized;

    case ezImageFormat::R8G8_UNORM:
      return ezGALResourceFormat::RGUByteNormalized;

    case ezImageFormat::R16G16_UNORM:
      return ezGALResourceFormat::RGUShortNormalized;

    case ezImageFormat::R11G11B10_FLOAT:
      return ezGALResourceFormat::RG11B10Float;

    default:
      EZ_ASSERT_NOT_IMPLEMENTED;
      break;
  }

  return ezGALResourceFormat::Invalid;
}

ezImageFormat::Enum ezTextureUtils::GalFormatToImageFormat(ezGALResourceFormat::Enum format)
{
  switch (format)
  {
    case ezGALResourceFormat::RGBAFloat:
      return ezImageFormat::R32G32B32A32_FLOAT;
    case ezGALResourceFormat::RGBAUInt:
      return ezImageFormat::R32G32B32A32_UINT;
    case ezGALResourceFormat::RGBAInt:
      return ezImageFormat::R32G32B32A32_SINT;
    case ezGALResourceFormat::RGBFloat:
      return ezImageFormat::R32G32B32_FLOAT;
    case ezGALResourceFormat::RGBUInt:
      return ezImageFormat::R32G32B32_UINT;
    case ezGALResourceFormat::RGBInt:
      return ezImageFormat::R32G32B32_SINT;
    case ezGALResourceFormat::B5G6R5UNormalized:
      return ezImageFormat::B5G6R5_UNORM;
    case ezGALResourceFormat::BGRAUByteNormalized:
      return ezImageFormat::B8G8R8A8_UNORM;
    case ezGALResourceFormat::BGRAUByteNormalizedsRGB:
      return ezImageFormat::B8G8R8A8_UNORM_SRGB;
    case ezGALResourceFormat::RGBAHalf:
      return ezImageFormat::R16G16B16A16_FLOAT;
    case ezGALResourceFormat::RGBAUShort:
      return ezImageFormat::R16G16B16A16_UINT;
    case ezGALResourceFormat::RGBAUShortNormalized:
      return ezImageFormat::R16G16B16A16_UNORM;
    case ezGALResourceFormat::RGBAShort:
      return ezImageFormat::R16G16B16A16_SINT;
    case ezGALResourceFormat::RGBAShortNormalized:
      return ezImageFormat::R16G16B16A16_SNORM;
    case ezGALResourceFormat::RGFloat:
      return ezImageFormat::R32G32_FLOAT;
    case ezGALResourceFormat::RGUInt:
      return ezImageFormat::R32G32_UINT;
    case ezGALResourceFormat::RGInt:
      return ezImageFormat::R32G32_SINT;
    case ezGALResourceFormat::RG11B10Float:
      return ezImageFormat::R11G11B10_FLOAT;
    case ezGALResourceFormat::RGBAUByteNormalized:
      return ezImageFormat::R8G8B8A8_UNORM;
    case ezGALResourceFormat::RGBAUByteNormalizedsRGB:
      return ezImageFormat::R8G8B8A8_UNORM_SRGB;
    case ezGALResourceFormat::RGBAUByte:
      return ezImageFormat::R8G8B8A8_UINT;
    case ezGALResourceFormat::RGBAByteNormalized:
      return ezImageFormat::R8G8B8A8_SNORM;
    case ezGALResourceFormat::RGBAByte:
      return ezImageFormat::R8G8B8A8_SINT;
    case ezGALResourceFormat::RGHalf:
      return ezImageFormat::R16G16_FLOAT;
    case ezGALResourceFormat::RGUShort:
      return ezImageFormat::R16G16_UINT;
    case ezGALResourceFormat::RGUShortNormalized:
      return ezImageFormat::R16G16_UNORM;
    case ezGALResourceFormat::RGShort:
      return ezImageFormat::R16G16_SINT;
    case ezGALResourceFormat::RGShortNormalized:
      return ezImageFormat::R16G16_SNORM;
    case ezGALResourceFormat::RGUByte:
      return ezImageFormat::R8G8_UINT;
    case ezGALResourceFormat::RGUByteNormalized:
      return ezImageFormat::R8G8_UNORM;
    case ezGALResourceFormat::RGByte:
      return ezImageFormat::R8G8_SINT;
    case ezGALResourceFormat::RGByteNormalized:
      return ezImageFormat::R8G8_SNORM;
    case ezGALResourceFormat::DFloat:
      return ezImageFormat::R32_FLOAT;
    case ezGALResourceFormat::RFloat:
      return ezImageFormat::R32_FLOAT;
    case ezGALResourceFormat::RUInt:
      return ezImageFormat::R32_UINT;
    case ezGALResourceFormat::RInt:
      return ezImageFormat::R32_SINT;
    case ezGALResourceFormat::RHalf:
      return ezImageFormat::R16_FLOAT;
    case ezGALResourceFormat::RUShort:
      return ezImageFormat::R16_UINT;
    case ezGALResourceFormat::RUShortNormalized:
      return ezImageFormat::R16_UNORM;
    case ezGALResourceFormat::RShort:
      return ezImageFormat::R16_SINT;
    case ezGALResourceFormat::RShortNormalized:
      return ezImageFormat::R16_SNORM;
    case ezGALResourceFormat::RUByte:
      return ezImageFormat::R8_UINT;
    case ezGALResourceFormat::RUByteNormalized:
      return ezImageFormat::R8_UNORM;
    case ezGALResourceFormat::RByte:
      return ezImageFormat::R8_SINT;
    case ezGALResourceFormat::RByteNormalized:
      return ezImageFormat::R8_SNORM;
    case ezGALResourceFormat::AUByteNormalized:
      return ezImageFormat::R8_UNORM;
    case ezGALResourceFormat::D16:
      return ezImageFormat::R16_UINT;
    case ezGALResourceFormat::BC1:
      return ezImageFormat::BC1_UNORM;
    case ezGALResourceFormat::BC1sRGB:
      return ezImageFormat::BC1_UNORM_SRGB;
    case ezGALResourceFormat::BC2:
      return ezImageFormat::BC2_UNORM;
    case ezGALResourceFormat::BC2sRGB:
      return ezImageFormat::BC2_UNORM_SRGB;
    case ezGALResourceFormat::BC3:
      return ezImageFormat::BC3_UNORM;
    case ezGALResourceFormat::BC3sRGB:
      return ezImageFormat::BC3_UNORM_SRGB;
    case ezGALResourceFormat::BC4UNormalized:
      return ezImageFormat::BC4_UNORM;
    case ezGALResourceFormat::BC4Normalized:
      return ezImageFormat::BC4_SNORM;
    case ezGALResourceFormat::BC5UNormalized:
      return ezImageFormat::BC5_UNORM;
    case ezGALResourceFormat::BC5Normalized:
      return ezImageFormat::BC5_SNORM;
    case ezGALResourceFormat::BC6UFloat:
      return ezImageFormat::BC6H_UF16;
    case ezGALResourceFormat::BC6Float:
      return ezImageFormat::BC6H_SF16;
    case ezGALResourceFormat::BC7UNormalized:
      return ezImageFormat::BC7_UNORM;
    case ezGALResourceFormat::BC7UNormalizedsRGB:
      return ezImageFormat::BC7_UNORM_SRGB;
    case ezGALResourceFormat::RGB10A2UInt:
    case ezGALResourceFormat::RGB10A2UIntNormalized:
    case ezGALResourceFormat::D24S8:
    default:
    {
#if EZ_ENABLED(EZ_COMPILE_FOR_DEBUG)
      ezStringBuilder sFormat;
      EZ_ASSERT_DEBUG(ezReflectionUtils::EnumerationToString(ezGetStaticRTTI<ezGALResourceFormat>(), format, sFormat, ezReflectionUtils::EnumConversionMode::ValueNameOnly), "Cannot convert GAL format '{}' to string", format);
      EZ_ASSERT_DEBUG(false, "The GL format: '{}' does not have a matching image format.", sFormat);
#endif
    }
  }
  return ezImageFormat::UNKNOWN;
}

ezImageFormat::Enum ezTextureUtils::GalFormatToImageFormat(ezGALResourceFormat::Enum format, bool bRemoveSRGB)
{
  ezImageFormat::Enum imageFormat = GalFormatToImageFormat(format);
  if (bRemoveSRGB)
  {
    imageFormat = ezImageFormat::AsLinear(imageFormat);
  }
  return imageFormat;
}

void ezTextureUtils::ConfigureSampler(ezTextureFilterSetting::Enum filter, ezGALSamplerStateCreationDescription& out_sampler)
{
  out_sampler.m_MinFilter = ezGALTextureFilterMode::Linear;
  out_sampler.m_MagFilter = ezGALTextureFilterMode::Linear;
  out_sampler.m_MipFilter = ezGALTextureFilterMode::Linear;
  out_sampler.m_uiMaxAnisotropy = 1;

  if (filter >= ezTextureFilterSetting::LowestQuality)
  {
    out_sampler.m_useTextureQualitySlot = static_cast<ezGALTextureQualitySlot::Enum>(filter - ezTextureFilterSetting::LowestQuality);
  }
  else
  {
    out_sampler.m_useTextureQualitySlot = ezGALTextureQualitySlot::None;
  }

  switch (filter)
  {
    case ezTextureFilterSetting::FixedNearest:
      out_sampler.m_MinFilter = ezGALTextureFilterMode::Point;
      out_sampler.m_MagFilter = ezGALTextureFilterMode::Point;
      out_sampler.m_MipFilter = ezGALTextureFilterMode::Point;
      break;

    case ezTextureFilterSetting::FixedBilinear:
      out_sampler.m_MipFilter = ezGALTextureFilterMode::Point;
      break;

    case ezTextureFilterSetting::FixedTrilinear:
      break;

    case ezTextureFilterSetting::FixedAnisotropic2x:
      out_sampler.m_MinFilter = ezGALTextureFilterMode::Anisotropic;
      out_sampler.m_MagFilter = ezGALTextureFilterMode::Anisotropic;
      out_sampler.m_uiMaxAnisotropy = 2;
      break;

    case ezTextureFilterSetting::FixedAnisotropic4x:
      out_sampler.m_MinFilter = ezGALTextureFilterMode::Anisotropic;
      out_sampler.m_MagFilter = ezGALTextureFilterMode::Anisotropic;
      out_sampler.m_uiMaxAnisotropy = 4;
      break;

    case ezTextureFilterSetting::FixedAnisotropic8x:
      out_sampler.m_MinFilter = ezGALTextureFilterMode::Anisotropic;
      out_sampler.m_MagFilter = ezGALTextureFilterMode::Anisotropic;
      out_sampler.m_uiMaxAnisotropy = 8;
      break;

    case ezTextureFilterSetting::FixedAnisotropic16x:
      out_sampler.m_MinFilter = ezGALTextureFilterMode::Anisotropic;
      out_sampler.m_MagFilter = ezGALTextureFilterMode::Anisotropic;
      out_sampler.m_uiMaxAnisotropy = 16;
      break;

    default:
      break;
  }
}

void ezTextureUtils::CopySubResourceToImage(const ezGALTextureCreationDescription& desc, const ezGALTextureSubresource& subResource, const ezGALSystemMemoryDescription& memory, ezImage& out_image, bool bRemoveSRGB)
{
  ezImageHeader headerTemp;
  headerTemp.SetImageFormat(ezTextureUtils::GalFormatToImageFormat(desc.m_Format, bRemoveSRGB));
  headerTemp.SetWidth(desc.m_uiWidth);
  headerTemp.SetHeight(desc.m_uiHeight);
  // GetWidth/Height assert that the requested mip level is within range, so the helper header must know the texture's full mip chain.
  headerTemp.SetNumMipLevels(desc.m_uiMipLevelCount);

  ezImageHeader header;
  header.SetImageFormat(ezTextureUtils::GalFormatToImageFormat(desc.m_Format, bRemoveSRGB));
  header.SetWidth(headerTemp.GetWidth(subResource.m_uiMipLevel));
  header.SetHeight(headerTemp.GetHeight(subResource.m_uiMipLevel));

  out_image.ResetAndAlloc(header);

  if (header.GetRowPitch() == memory.m_uiRowPitch)
  {
    const void* pSource = memory.m_pData.GetPtr();
    ezUInt8* pDest = out_image.GetPixelPointer<ezUInt8>();
    ezUInt32 uiSize = static_cast<ezUInt32>(header.GetDepthPitch());
    EZ_ASSERT_DEBUG(uiSize <= memory.m_pData.GetCount(), "Not enough data in the buffer to create image");
    memcpy(pDest, pSource, uiSize);
  }
  else
  {
    // Copy row by row
    const ezUInt32 uiHeight = header.GetNumBlocksY();
    const ezUInt32 uiRowSize = static_cast<ezUInt32>(header.GetRowPitch());
    for (ezUInt32 y = 0; y < uiHeight; ++y)
    {
      const void* pSource = ezMemoryUtils::AddByteOffset(memory.m_pData.GetPtr(), y * memory.m_uiRowPitch);
      ezUInt8* pDest = out_image.GetPixelPointer<ezUInt8>(0, 0, 0, 0, y);
      memcpy(pDest, pSource, uiRowSize);
    }
  }
}

ezImageView ezTextureUtils::MakeImageViewFromSubResource(const ezGALTextureCreationDescription& desc, const ezGALTextureSubresource& subResource, const ezGALSystemMemoryDescription& memory, ezImage& ref_tempImage, bool bRemoveSRGB)
{
  ezImageView view;
  ezImageHeader headerTemp;
  headerTemp.SetImageFormat(ezTextureUtils::GalFormatToImageFormat(desc.m_Format, bRemoveSRGB));
  headerTemp.SetWidth(desc.m_uiWidth);
  headerTemp.SetHeight(desc.m_uiHeight);
  headerTemp.SetNumMipLevels(desc.m_uiMipLevelCount);

  ezImageHeader header;
  header.SetImageFormat(ezTextureUtils::GalFormatToImageFormat(desc.m_Format, bRemoveSRGB));
  header.SetWidth(headerTemp.GetWidth(subResource.m_uiMipLevel));
  header.SetHeight(headerTemp.GetHeight(subResource.m_uiMipLevel));

  if (header.GetRowPitch() == memory.m_uiRowPitch)
  {
    view.ResetAndViewExternalStorage(header, memory.m_pData);
  }
  else
  {
    CopySubResourceToImage(desc, subResource, memory, ref_tempImage, bRemoveSRGB);
    view = ref_tempImage.GetSubImageView();
  }
  return view;
}

void ezTextureUtils::CopySubResourceToMemory(const ezGALTextureCreationDescription& desc, const ezGALTextureSubresource& subResource, const ezGALSystemMemoryDescription& sourceMemory, ezArrayPtr<ezUInt8> targetData, ezUInt32 uiTargetRowPitch)
{
  if (sourceMemory.m_uiRowPitch == uiTargetRowPitch)
  {
    const ezUInt32 uiMemorySize = ezGALResourceFormat::GetBitsPerElement(desc.m_Format) *
                                  GetMipSize(desc.m_uiWidth, subResource.m_uiMipLevel) *
                                  GetMipSize(desc.m_uiHeight, subResource.m_uiMipLevel) / 8;
    EZ_ASSERT_DEBUG(uiMemorySize <= sourceMemory.m_pData.GetCount(), "");
    EZ_ASSERT_DEBUG(uiMemorySize <= targetData.GetCount(), "");
    memcpy(targetData.GetPtr(), sourceMemory.m_pData.GetPtr(), uiMemorySize);
  }
  else
  {
    // Copy row by row
    const ezUInt32 uiHeight = GetMipSize(desc.m_uiHeight, subResource.m_uiMipLevel);
    for (ezUInt32 y = 0; y < uiHeight; ++y)
    {
      const ezUInt8* pSource = ezMemoryUtils::AddByteOffset(sourceMemory.m_pData.GetPtr(), y * sourceMemory.m_uiRowPitch);
      ezUInt8* pDest = ezMemoryUtils::AddByteOffset(targetData.GetPtr(), y * uiTargetRowPitch);

      const ezUInt32 uiCopySize = ezGALResourceFormat::GetBitsPerElement(desc.m_Format) * GetMipSize(desc.m_uiWidth, subResource.m_uiMipLevel) / 8;
      EZ_ASSERT_DEBUG(pDest + uiCopySize <= targetData.GetEndPtr(), "");
      EZ_ASSERT_DEBUG(pSource + uiCopySize <= sourceMemory.m_pData.GetEndPtr(), "");
      memcpy(pDest, pSource, uiCopySize);
    }
  }
}