我正在编写霍夫曼编码,一切正常,直到我尝试将结果保存到存档文件中。我们的老师让我们用这样的功能来做这件事(每次都要花一点时间,然后把它们中的8个输出一个字节):
long buff=0;
int counter=0;
std::ofstream out("output", std::iostream::binary);
void putbit(bool b)
{
buff<<=1;
if (b) buff++;
counter++;
if (counter>=8)
{
out.put(buff);
counter=0;
buff=0;
}
}
我尝试了输入这样的位序列的例子:
0011001011001101111010010001000001010101101100
但二进制模式下的输出文件仅包括:1111111
由于buff
变量具有正确的数字(25 102 250 68 21 108
),我建议我在笔记本中错误地编写代码并且此行有问题:
out.put(buff);
我试图用这一行删除它:
out << buff;
但得到了:1111111111111111
另一种方式是:
out.write((char *) &buff, 8);
给出:
100000001000000010000000100000001000000010000000
它看起来最接近正确的答案,但仍然无法正常工作。
也许我对文件输出一无所知。
你能解释一下如何让它发挥作用以及为什么以前的变种是错误的吗?
UPD: 输入来自此功能:
void code(std::vector<bool> cur, std::vector<bool> sh, std::vector<bool>* codes, Node* r)
{
if (r->l)
{
cur.push_back(0);
if (r->l->symb)
{
putbit(0);
codes[(int)r->l->symb] = cur;
for (int i=7; i>=0; i--)
{
if ((int)r->l->symb & (1 << i))
putbit(1);
else putbit(0);
}
}
else
{
putbit(0);
code(cur, sh, codes, r->l);
}
cur.pop_back();
}
if (r->r)
{
cur.push_back(1);
if (r->r->symb)
{
putbit(1);
codes[(int)r->r->symb] = cur;
for (int i=7; i>=0; i--)
{
if ((int)r->r->symb & (1 << i))
putbit(1);
else putbit(0);
}
}
else
{
putbit(1);
code(cur, sh, codes, r->r);
}
cur.pop_back();
}
}
答案 0 :(得分:1)
问题是,你的putbit
函数正在运行(尽管它非常糟糕,你使用全局变量,你的缓冲区应该是char
)。
例如,这是我测试你的功能的方式。
out.open( "outfile", std::ios::binary );
if ( out.is_open() ) {
putbit(1);
putbit(1);
putbit(0);
putbit(1);
putbit(0);
putbit(1);
putbit(0);
putbit(0);
out.close();
}
这应该以十六进制输出1101 0100
或d4
。
我认为这是一个XY问题。您尝试解决的问题不在putbit
函数中,而在于您使用它和算法的方式。
您说在将数据放入输出文件之前,您拥有正确的值。在stackoverflow中有许多类似的问题,只需查找它们。
真正的问题是,putbit功能不足以解决您的问题。你依赖于它在你调用它8次后会写一个字节的事实。如果你写的少于8个字节怎么办?此外,您永远不会刷新您的文件(至少在您发布的代码中),因此无法保证所有数据都会被写入。
首先,您必须了解文件句柄(流)的工作方式。在本地打开您的文件,检查它是否打开并在完成后关闭它。 Closing还可以保证文件缓冲区中的所有数据都写入文件。
outfile.open( "output", std::ios::binary );
if ( outfile.is_open() ) {
// ... use file ...
outfile.close();
}
else {
// Couldnt open file!
}
其他问题通过编写或使用BitStream来解决这个问题。看起来有点像这样,
class OutBitstream {
public:
OutBitstream();
~OutBitstream(); // close file
bool isOpen();
void open( const std::string &file );
void close(); // close file, also write pending bits
void writeBit( bool b ); // or putbit, use the names you prefer
void writeByte( char c );
void writePendingBits(); // write bits in the buffer they may
// be less than 8 so you may have to do some padding
private:
std::ofstream _out;
char _bitBuffer; //or std::bitset<8>
int _numbits;
};
使用此接口,应该更容易处理位输入。没有全局变量。我希望有所帮助。