#include <stdio.h>
#include <iostream.h>
#include "config.h"
#include "HTree.h"

HTree::HTree(int n_freq, char *n_objet)
{
	freq = n_freq;
	objet = n_objet;
	left = right = NULL;
}

HTree::HTree(HTree * n_left, HTree * n_right)
{
	left = n_left;
	right = n_right;
	freq = n_left->freq + n_right->freq;
	objet = NULL;
}

HTree::~HTree()
{
	if (left)
		delete left;

	if (right)
		delete right;
}

ostream & HTree::Trace(ostream & os, int d)
{
	static char cmpr[MAX_HDEPTH + 1];
	static int tsize;
	static int rsize;
	static int dsize;

	if (!d)
		tsize = rsize = dsize = 0;

	if (objet) {
		cmpr[d] = '\0';
		os << objet << " (" << freq << ") = " << cmpr << endl;
		tsize += freq * strlen(cmpr);
		rsize += freq * strlen(objet);
		dsize += ((strlen(cmpr) >> 3) + (strlen(cmpr) & 7 ? 1 : 0)) + strlen(objet) + 2;
	} else {
		if (left) {
			cmpr[d] = '0';
			left->Trace(os, d + 1);
		}
		if (right) {
			cmpr[d] = '1';
			right->Trace(os, d + 1);
		}
	}

	if (!d) {
		os << _("Bitstream length           : ") << tsize << _(" bits (= ") << ((tsize >> 3) +
											(tsize & 7 ? 1 : 0)) <<
			_(" bytes)\n");
		os << _("Real size input            : ") << (rsize << 3) << _(" bits (= ") << rsize << _(" bytes)\n");
		os << _("Size squeezed by           : ") << 100.0 - 100.0 * tsize / (rsize << 3) << _(" percents\n");
		os << _("Dictionnary size           : ") << (dsize << 3) << _(" bits (= ") << dsize << _(" bytes)\n");
		os << _("Total bitstream length     : ") << tsize + (dsize << 3) << _(" bits (= ") << ((tsize >> 3) +
												       (tsize & 7 ? 1 :
													0) +
												       dsize) <<
			_(" bytes)\n");
		os << _("Real gain (4 bytes header) : ") << 100.0 - 100.0 * ((tsize >> 3) + (tsize & 7 ? 1 : 0) +
									     dsize + 4) / rsize << _(" percents\n");
	}
	return os;
}

int HTree::ReadFreq(void)
{
	return freq;
}

char *HTree::ReadObj(void)
{
	return objet;
}