40 lines
501 B
C++
40 lines
501 B
C++
#pragma once
|
|
#include "Buffer.hpp"
|
|
|
|
class BitWriter
|
|
{
|
|
public:
|
|
int k = 0;
|
|
Buffer buffer;
|
|
char x = 0;
|
|
|
|
void writeByte(uint8_t x)
|
|
{
|
|
// f.write((char*)&x, 1);
|
|
buffer.add_end(&x, sizeof(uint8_t));
|
|
}
|
|
|
|
void writeInt(int x)
|
|
{
|
|
buffer.add_end((uint8_t *)&x, sizeof(x));
|
|
}
|
|
|
|
void writeBit(bool b)
|
|
{
|
|
if (k == 8)
|
|
{
|
|
writeByte(x);
|
|
k = 0;
|
|
x = 0;
|
|
}
|
|
x ^= (-b ^ x) & (1 << k);
|
|
k++;
|
|
}
|
|
|
|
void finish()
|
|
{
|
|
if (k > 0)
|
|
writeByte(x);
|
|
}
|
|
};
|