StringDeduplicationContext.h

 
#pragma once
 
#include <Foundation/Containers/DynamicArray.h>
#include <Foundation/Containers/Map.h>
#include <Foundation/IO/MemoryStream.h>
#include <Foundation/IO/SerializationContext.h>
#include <Foundation/Strings/String.h>
 
class plStreamWriter;
class plStreamReader;
 
class PL_FOUNDATION_DLL plStringDeduplicationWriteContext : public plSerializationContext<plStringDeduplicationWriteContext>
{
  PL_DECLARE_SERIALIZATION_CONTEXT(plStringDeduplicationWriteContext);
 
public:
  plStringDeduplicationWriteContext(plStreamWriter& ref_originalStream);
  ~plStringDeduplicationWriteContext();
 
  plStreamWriter& Begin();
 
  plResult End();
 
  void SerializeString(const plStringView& sString, plStreamWriter& ref_writer);
 
  plUInt32 GetUniqueStringCount() const;
 
  plStreamWriter& GetOriginalStream() { return m_OriginalStream; }
 
protected:
  plStreamWriter& m_OriginalStream;
 
  plDefaultMemoryStreamStorage m_TempStreamStorage;
  plMemoryStreamWriter m_TempStreamWriter;
 
  plMap<plHybridString<64>, plUInt32> m_DeduplicatedStrings;
};
 
class PL_FOUNDATION_DLL plStringDeduplicationReadContext : public plSerializationContext<plStringDeduplicationReadContext>
{
  PL_DECLARE_SERIALIZATION_CONTEXT(plStringDeduplicationReadContext);
 
public:
  plStringDeduplicationReadContext(plStreamReader& inout_stream);
  ~plStringDeduplicationReadContext();
 
  plStringView DeserializeString(plStreamReader& ref_reader);
 
protected:
  plDynamicArray<plHybridString<64>> m_DeduplicatedStrings;
};