CompressedStreamZlib.h

 
#pragma once
 
#include <Foundation/Basics.h>
#include <Foundation/Containers/DynamicArray.h>
#include <Foundation/IO/Stream.h>
 
#ifdef BUILDSYSTEM_ENABLE_ZLIB_SUPPORT
 
struct z_stream_s;
 
class PL_FOUNDATION_DLL plCompressedStreamReaderZip : public plStreamReader
{
public:
  plCompressedStreamReaderZip();
  ~plCompressedStreamReaderZip();
 
  void SetInputStream(plStreamReader* pInputStream, plUInt64 uiInputSize);
 
  virtual plUInt64 ReadBytes(void* pReadBuffer, plUInt64 uiBytesToRead) override; // [tested]
 
private:
  plUInt64 m_uiRemainingInputSize = 0;
  bool m_bReachedEnd = false;
  plDynamicArray<plUInt8> m_CompressedCache;
  plStreamReader* m_pInputStream = nullptr;
  z_stream_s* m_pZLibStream = nullptr;
};
 
 
class PL_FOUNDATION_DLL plCompressedStreamReaderZlib : public plStreamReader
{
public:
  plCompressedStreamReaderZlib(plStreamReader* pInputStream); // [tested]
 
  ~plCompressedStreamReaderZlib();                            // [tested]
 
  virtual plUInt64 ReadBytes(void* pReadBuffer, plUInt64 uiBytesToRead) override; // [tested]
 
private:
  bool m_bReachedEnd = false;
  plDynamicArray<plUInt8> m_CompressedCache;
  plStreamReader* m_pInputStream = nullptr;
  z_stream_s* m_pZLibStream = nullptr;
};
 
class PL_FOUNDATION_DLL plCompressedStreamWriterZlib : public plStreamWriter
{
public:
  enum Compression
  {
    Uncompressed = 0,
    Fastest = 1,
    Fast = 3,
    Average = 5,
    High = 7,
    Highest = 9,
    Default = Fastest 
  };
 
  plCompressedStreamWriterZlib(plStreamWriter* pOutputStream, Compression ratio = Compression::Default); // [tested]
 
  ~plCompressedStreamWriterZlib(); // [tested]
 
  virtual plResult WriteBytes(const void* pWriteBuffer, plUInt64 uiBytesToWrite) override; // [tested]
 
  plResult CloseStream(); // [tested]
 
  plUInt64 GetUncompressedSize() const { return m_uiUncompressedSize; } // [tested]
 
  plUInt64 GetCompressedSize() const { return m_uiCompressedSize; } // [tested]
 
  virtual plResult Flush() override;
 
private:
  plUInt64 m_uiUncompressedSize = 0;
  plUInt64 m_uiCompressedSize = 0;
 
  plStreamWriter* m_pOutputStream = nullptr;
  z_stream_s* m_pZLibStream = nullptr;
 
  plDynamicArray<plUInt8> m_CompressedCache;
};
 
#endif // BUILDSYSTEM_ENABLE_ZLIB_SUPPORT