StringCompressor.h

Go to the documentation of this file.

#ifndef __STRING_COMPRESSOR_H
#define __STRING_COMPRESSOR_H

#include "Export.h"
#include "DS_Map.h"
#include "RakMemoryOverride.h"
#include "NativeTypes.h"

#ifdef _STD_STRING_COMPRESSOR
#include <string>
#endif

namespace RakNet
{
    class BitStream;
    class RakString;
};


namespace RakNet
{
class HuffmanEncodingTree;

class RAK_DLL_EXPORT StringCompressor
{
public:
    
    // Destructor   
    ~StringCompressor();
    
    static StringCompressor* Instance(void);

    void GenerateTreeFromStrings( unsigned char *input, unsigned inputLength, uint8_t languageId );
    
    void EncodeString( const char *input, int maxCharsToWrite, RakNet::BitStream *output, uint8_t languageId=0 );
    
    bool DecodeString( char *output, int maxCharsToWrite, RakNet::BitStream *input, uint8_t languageId=0 );

#ifdef _CSTRING_COMPRESSOR
    void EncodeString( const CString &input, int maxCharsToWrite, RakNet::BitStream *output, uint8_t languageId=0 );
    bool DecodeString( CString &output, int maxCharsToWrite, RakNet::BitStream *input, uint8_t languageId=0 );
#endif

#ifdef _STD_STRING_COMPRESSOR
    void EncodeString( const std::string &input, int maxCharsToWrite, RakNet::BitStream *output, uint8_t languageId=0 );
    bool DecodeString( std::string *output, int maxCharsToWrite, RakNet::BitStream *input, uint8_t languageId=0 );
#endif

    void EncodeString( const RakNet::RakString *input, int maxCharsToWrite, RakNet::BitStream *output, uint8_t languageId=0 );
    bool DecodeString( RakNet::RakString *output, int maxCharsToWrite, RakNet::BitStream *input, uint8_t languageId=0 );

    static void AddReference(void);
    
    static void RemoveReference(void);

    StringCompressor();

private:
    
    static StringCompressor *instance;
    
    DataStructures::Map<int, HuffmanEncodingTree *> huffmanEncodingTrees;
    
    static int referenceCount;
};

} // namespace RakNet

#endif