// Handle command line arguments

if (argc != 3) {

std::cerr << "Usage: " << argv[0] << " InputFile OutputFile" << std::endl;

return EXIT_FAILURE;

}

const char *inputFile = argv[1];

const char *outputFile = argv[2];

// Perform file compression

std::ifstream in(inputFile, std::ios::binary);

std::ofstream out(outputFile, std::ios::binary);

BitOutputStream bout(out);

try {

const std::vector<uint32_t> initFreqs(257, 1);

FrequencyTable freqs(initFreqs);

HuffmanEncoder enc(bout);

CodeTree tree = freqs.buildCodeTree(); // Don't need to make canonical code because we don't transmit the code tree

enc.codeTree = &tree;

uint32_t count = 0; // Number of bytes read from the input file

while (true) {

// Read and encode one byte

int symbol = in.get();

if (symbol == EOF)

break;

if (symbol < 0 || symbol > 255)

throw std::logic_error("Assertion error");

enc.write(static_cast<uint32_t>(symbol));

count++;

// Update the frequency table and possibly the code tree

freqs.increment(static_cast<uint32_t>(symbol));

if ((count < 262144 && isPowerOf2(count)) || count % 262144 == 0) // Update code tree

tree = freqs.buildCodeTree();

if (count % 262144 == 0) // Reset frequency table

freqs = FrequencyTable(initFreqs);

}

enc.write(256); // EOF

bout.finish();

return EXIT_SUCCESS;

} catch (const char *msg) {

std::cerr << msg << std::endl;

return EXIT_FAILURE;

}