赫夫曼.cpp

哈夫曼编码器

#include "huffman.h"

bool main ()
{
	string filename;
	char bd;
	int chfm=0;
	huffman *a=NULL;
	while(1)
	{
		system("cls");
		cout<<"*************赫夫曼编译器*************"<<endl;
		cout<<"**1.读取赫夫曼树                    **"<<endl;
		cout<<"**2.赫夫曼编码                      **"<<endl;
		cout<<"**3.赫夫曼译码                      **"<<endl;
		cout<<"**4.打印文件                        **"<<endl;
		cout<<"**5.退出                            **"<<endl;
		cout<<"**************************************"<<endl;
		do
		{
			bd=getch();
		}while(bd<'1'||bd>'5');
		cout<<endl;
		switch(bd)
		{
		case '1' :
			system("cls");
			cout<<"*************赫夫曼编译器*************"<<endl;
			cout<<"*************读取赫夫曼树*************"<<endl;
			cout<<"**1.新输入赫夫曼树                  **"<<endl;
			cout<<"**2.从文件读取赫夫曼树              **"<<endl;
			cout<<"**3.返回                            **"<<endl;
			cout<<"**************************************"<<endl;
			do
			{
				bd=getch();
			}while((bd<'1'||bd>'3')&&bd!=27);
			cout<<endl;
			system("cls");
			if(a)
				delete a;
			a=new huffman;
			cout<<"*******赫夫曼编译器*******"<<endl;
			switch(bd)
			{
			case '1' :
				a->creathuffmantree();
				break;
			case '2' :
				cout<<"请输入赫夫曼树文件名：";
				cin>>filename;
				a->creathuffmantree(filename);
				break;
			case '3' :
				break;
			case 27 :
				break;
			};
			chfm=1;
			break;
		case '2' : 
			if(chfm)
			{
				cout<<"请输入要编码的文件名：";
				cin>>filename;
				a->encoding(filename);
			}
			else
			{
				cout<<"您尚未创建赫夫曼树！！"<<endl;
				bd=getch();
			}
			break;
		case '3' :
			if(chfm)
			{
				cout<<"请输入要译码的文件名：";
				cin>>filename;
				a->decoding(filename);
			}
			else
			{
				cout<<"您尚未创建赫夫曼树！！"<<endl;
				bd=getch();
			}
			break;
		case '4' :
			cout<<"请输入要打印的文件名：";
			cin>>filename;
			fileprint(filename);
			break;
		case '5' :
			cout<<"谢谢使用！"<<endl;
			bd=getch();
			exit(0);
		}
	};
	return true;
}