Compressor.h

#ifndef COMPRESSOR_H_INCLUDED
#define COMPRESSOR_H_INCLUDED
 
#include <vector>
#include <map>
 
#include "MyEngine.h"
 
typedef unsigned char GLubyte;
 
namespace nta {
    class HuffmanNode {
    private:
        HuffmanNode *m_lChild, *m_rChild;
    protected:
        int m_freq;
    public:
        HuffmanNode();
        HuffmanNode(HuffmanNode* l, HuffmanNode* r);
        virtual ~HuffmanNode();
        auto getEncodings(crstring enc = "") const -> std::map<GLubyte,std::string>;
        HuffmanNode* getLeft() const;
        HuffmanNode* getRight() const;
        bool hasChildren() const;
        int getFrequency() const;
    };
    class HuffmanLeaf : public HuffmanNode {
    private:
        GLubyte m_data;
    public:
        HuffmanLeaf();
        HuffmanLeaf(GLubyte data, int freq);
        ~HuffmanLeaf();
        GLubyte getData() const;
    };
    class Compressor {
    private:
        static std::string toBinary(int num, int numBits = 8);
        static int fromBinary(crstring num);
        static void createTree(const std::vector<GLubyte>& data);
        static std::map<GLubyte,std::string> m_encodings;
        static HuffmanNode* m_root;
    public:
        static std::vector<GLubyte> decompress(const std::vector<GLubyte>& data);
        static std::vector<GLubyte> compress(const std::vector<GLubyte>& data);
    };
}
 
#endif // COMPRESSOR_H_INCLUDED