// main.cpp

#include "Huffman.h"

int main( int argc, char* argv[] )
{
	// check number of arguments
  	if ( argc < 2 )
    {
    	cerr << "Missing argument." << endl;
      	cerr << "usage: Huffman [-d] filename" << endl;
      	return 1;
    }

	Huffman<byte> lCompressor;

	try
	{
		if ( argc == 2 )
			lCompressor.compress( argv[1] );
		else
		{
			string lOption(argv[1]);
			if ( lOption != "-d" )
			{
    			cerr << "Illegal option: " << lOption << endl;
      			cerr << "usage: Huffman [-d] filename" << endl;
      			return 1;
			}
			else
				lCompressor.uncompress( argv[2] );
		}			
	}
	catch( HuffmanException& e )
	{
		cout << "Error: " << e.getMessage() << endl; 	
	}

	return 0;
}