DS_HuffmanEncodingTree.h

Go to the documentation of this file.

#ifndef __HUFFMAN_ENCODING_TREE
#define __HUFFMAN_ENCODING_TREE

#include "RakMemoryOverride.h"
#include "DS_HuffmanEncodingTreeNode.h"
#include "BitStream.h"
#include "Export.h"
#include "DS_LinkedList.h" 

namespace RakNet
{

class RAK_DLL_EXPORT HuffmanEncodingTree
{

public:
    HuffmanEncodingTree();
    ~HuffmanEncodingTree();

    void EncodeArray( unsigned char *input, size_t sizeInBytes, RakNet::BitStream * output );

    // \brief Decodes an array encoded by EncodeArray().
    unsigned DecodeArray( RakNet::BitStream * input, BitSize_t sizeInBits, size_t maxCharsToWrite, unsigned char *output );
    void DecodeArray( unsigned char *input, BitSize_t sizeInBits, RakNet::BitStream * output );

    void GenerateFromFrequencyTable( unsigned int frequencyTable[ 256 ] );

    void FreeMemory( void );

private:


    HuffmanEncodingTreeNode *root;



    struct CharacterEncoding
    {
        unsigned char* encoding;
        unsigned short bitLength;
    };

    CharacterEncoding encodingTable[ 256 ];

    void InsertNodeIntoSortedList( HuffmanEncodingTreeNode * node, DataStructures::LinkedList<HuffmanEncodingTreeNode *> *huffmanEncodingTreeNodeList ) const;
};

} // namespace RakNet

#endif