huffman_h.h

huffman编码的c++实现源码

#ifndef _HUFFMAN_H_HEADER_001_
#define _HUFFMAN_H_HEADER_001_

#include "huffman_base.h"

// 在huffman_f的基础上，增加限制码长的功能
// 限制码长的算法参考了zlib-1.1.4中构造限制码长的Huffman编码的源代码
// zlib的源代码见http://www.gzip.org/zlib/
// 其中限制长度的算法在tree.c的gen_bitlen()函数中

// 但zlib源码中的限制码长算法只适用于所有超长码长只比最大码长多1的情形
// 对于类似{1, 3, 7, 15, 31, 63, 127}，限最大码长3的例子，就不能正确处理
// huffman_h对此做了调整，可适应各类超长的情形

// zlib的方法之所以不会出错，是因为对于zlib的最大码长设置（15）、
// 最大窗口大小（32KB）、元素数量（256），不会出现上面那种超长码长比
// 最大码长大许多的情形

class huffman_h :
	public huffman_base
{
public:
	huffman_h(int max_code_len = sizeof(unsigned long)) 
	{
		this->max_code_len = max_code_len;
	}
	virtual ~huffman_h(void) {}

public:
	void generate_codes(int num, const unsigned long* weights);

protected:
	int max_code_len;
};

#endif