#ifndef HUFFMANSET_HPP
#define HUFFMANSET_HPP

#include <QtCore/QMap>
#include <QtCore/QString>
#include <QtCore/QStringList>

#include "BitArray.hpp"

class BitDecoder;
class FastBitDecoder;

class HuffmanSet {
public:
  typedef uint key_t;

private:
  QMap<key_t, QString> newStrings;
  QMap<key_t, BitArray> map;
  QMap<QString, BitArray> encoder;
  uint cutoff;
  uint numInserts;
  //FastBitDecoder* lut;
  BitDecoder* lut;


public:
  HuffmanSet();
  void setCutoff(uint cutoff);
  static QStringList chunks(const QString& str);
  BitDecoder* createLut(const QMap<QString, uint>& freqTable);
  QString decode(const BitArray& bits) const;
  static BitArray encode(const QString& string, const QMap<QString, BitArray>& encoder);
  void rebuild();
  bool contains(key_t key) const;
  uint totalElements() const;
  key_t hash(const QString& str);
  key_t insert(const QString& str);
  QString value(key_t key) const;
};

#endif //HUFFMANSET_HPP