Image: added option to use custom memory allocator

DiligentGraphics · Nov 5, 2024 · 598fc8a · 598fc8a
1 parent 58a9a83
commit 598fc8a
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 11 deletions.
diff --git a/TextureLoader/interface/Image.h b/TextureLoader/interface/Image.h
@@ -84,6 +84,9 @@ struct ImageLoadInfo
     ///
     /// \note This flag is only used if PermultiplyAlpha is true.
     bool IsSRGB DEFAULT_INITIALIZER(false);
+
+    /// Memory allocator
+    struct IMemoryAllocator* pAllocator DEFAULT_INITIALIZER(nullptr);
 };
 typedef struct ImageLoadInfo ImageLoadInfo;
 
@@ -133,15 +136,16 @@ struct Image : public ObjectBase<IObject>
 
     struct EncodeInfo
     {
-        Uint32            Width       = 0;
-        Uint32            Height      = 0;
-        TEXTURE_FORMAT    TexFormat   = TEX_FORMAT_UNKNOWN;
-        bool              KeepAlpha   = false;
-        bool              FlipY       = false;
-        const void*       pData       = nullptr;
-        Uint32            Stride      = 0;
-        IMAGE_FILE_FORMAT FileFormat  = IMAGE_FILE_FORMAT_JPEG;
-        int               JpegQuality = 95;
+        Uint32                   Width       = 0;
+        Uint32                   Height      = 0;
+        TEXTURE_FORMAT           TexFormat   = TEX_FORMAT_UNKNOWN;
+        bool                     KeepAlpha   = false;
+        bool                     FlipY       = false;
+        const void*              pData       = nullptr;
+        Uint32                   Stride      = 0;
+        IMAGE_FILE_FORMAT        FileFormat  = IMAGE_FILE_FORMAT_JPEG;
+        int                      JpegQuality = 95;
+        struct IMemoryAllocator* pAllocator  = nullptr;
     };
     static void Encode(const EncodeInfo& Info, IDataBlob** ppEncodedData);
 

diff --git a/TextureLoader/src/Image.cpp b/TextureLoader/src/Image.cpp
@@ -331,7 +331,7 @@ Image::Image(IReferenceCounters*  pRefCounters,
              IDataBlob*           pFileData,
              const ImageLoadInfo& LoadInfo) :
     TBase{pRefCounters},
-    m_pData{DataBlobImpl::Create()}
+    m_pData{DataBlobImpl::Create(LoadInfo.pAllocator)}
 {
     if (LoadInfo.Format == IMAGE_FILE_FORMAT_TIFF)
     {
@@ -476,7 +476,7 @@ std::vector<Uint8> Image::ConvertImageData(Uint32         Width,
 
 void Image::Encode(const EncodeInfo& Info, IDataBlob** ppEncodedData)
 {
-    auto pEncodedData = DataBlobImpl::Create();
+    auto pEncodedData = DataBlobImpl::Create(Info.pAllocator);
     if (Info.FileFormat == IMAGE_FILE_FORMAT_JPEG)
     {
         auto RGBData = ConvertImageData(Info.Width, Info.Height, reinterpret_cast<const Uint8*>(Info.pData), Info.Stride, Info.TexFormat, TEX_FORMAT_RGBA8_UNORM, false, Info.FlipY);

diff --git a/TextureLoader/src/TextureLoaderImpl.cpp b/TextureLoader/src/TextureLoaderImpl.cpp
@@ -155,6 +155,7 @@ TextureLoaderImpl::TextureLoaderImpl(IReferenceCounters*        pRefCounters,
         }
         ImgLoadInfo.IsSRGB           = TexLoadInfo.IsSRGB;
         ImgLoadInfo.PermultiplyAlpha = TexLoadInfo.PermultiplyAlpha;
+        ImgLoadInfo.pAllocator       = TexLoadInfo.pAllocator;
         RefCntAutoPtr<Image> pImage;
         Image::CreateFromDataBlob(m_pDataBlob, ImgLoadInfo, &pImage);
         LoadFromImage(pImage, TexLoadInfo);