#include "stdio.h"
#include "Huffman.h"

void main(){
	HuffmanTree HT;
	HuffmanCode HC;
	
	FILE *fp1;
	char filename[20];
	printf("請輸入需要進行Huffman編碼的文件名：  ");
	scanf("%s",filename);
	if((fp1=fopen(filename,"r"))==NULL) exit(0);

	int *w;
	w=(int *)malloc((100)*sizeof(int));

	printf("進行Huffman編碼...\n\n");

/*****************************************************************************************************************
                       統計ASC碼從32——122的字符以及換行符的出現次數，存入數組w（共92個）
*****************************************************************************************************************/
	int i;
	char findch;
	findch=' ';
	for(i=0;i<=90;i++){
		w[i]=emergetime(filename,findch);
		findch++;
	}
	w[91]=emergetime(filename,char(10));//記錄換行符

	HuffmanCoding(HT,HC,w,92);

	printf("統計結果及Huffman編碼：\n\n");

	int j;
	for(j=0;j<=90;j++){
	    printf("字符 %c 出現的次數為：%d   \t",' '+j,w[j]);
		printf("Huffman編碼:  %s\n", HC[j+1]);
	}
	printf("換行符出現的次數為：%d   \t",w[91]);
	printf("Huffman編碼:  %s\n", HC[92]);

	printf("\n");

/*****************************************************************************************************************
                                          將編碼表存入myfile2文件中
*****************************************************************************************************************/
	FILE *fp2;
	if((fp2=fopen("myfile2.txt","w"))==NULL) exit(0);
	fprintf(fp2,"%s\n","編碼表:");
	for(j=1;j<=91;j++){
		fprintf(fp2,"%c   ",' '+j-1);
		fprintf(fp2,"%s\n",HC[j]);
	}


	fprintf(fp2,"%s   ","換行符");
	fprintf(fp2,"%s\n",HC[92]);

/*****************************************************************************************************************
                                   對原文件中英文文章編碼,存入文件myfile2中
*****************************************************************************************************************/

	fprintf(fp2,"%s\n","編碼后的文章為：");
	char ch;
	ch=fgetc(fp1);
	while(ch!=EOF){
		for(j=1;j<=91;j++){
			if(ch==' '+j-1) fprintf(fp2,"%s",HC[j]);	
		}
	    if(ch==char(10)) fprintf(fp2,"%s",HC[92]);
		ch=fgetc(fp1);
	}

	fclose(fp1);
	fclose(fp2);
	printf("編碼表和編碼文章已存入文件myfile2中。\n\n");


/*****************************************************************************************************************
                                 對編碼文件解碼，生成原文件存入文件myfile3中
*****************************************************************************************************************/

	printf("下面將編碼文件譯為原文...\n");
	system("pause");
	system("cls");

/*************************************讀編碼表********************************************/
	
	if((fp2=fopen("myfile2.txt","r"))==NULL) exit(0);
	char str[30];
        fgets(str,30,fp2); //讀取一行（使文件當前指針跳過“編碼表：”字樣） 
	for(i=0;i<91;i++){
		fgets(str,30,fp2);
		HCode[i].ch=str[0]; 
		strcpy(HCode[i].str,&str[4]);//Huffman編碼從第四個字符開始
	}

	fgets(str,30,fp2);  //讀取換行符編碼
	HCode[91].ch=char(10); 
	strcpy(HCode[91].str,&str[9]);

/***************************根據編碼表建立二叉樹，用于將Huffman編碼譯為原文****************/
	
	BiTree T;
	T=(BiTree)malloc(sizeof(BiTNode));
	T->lchild=NULL;
	T->rchild=NULL;
	for(i=0;i<92;i++){
		CreateTree(T,HCode[i]);		
	}


/**********************下面開始讀取編碼文件進行譯碼并寫入文件********************************/

	FILE *fp3;
	if((fp3=fopen("myfile3.txt","w"))==NULL) exit(0);
	
	fgets(str,30,fp2); //讀取一行（使文件當前指針跳過“編碼文件如下：”字樣）

	BiTree F;
	F=T;  //先指向建立的二叉樹T的根
	ch=fgetc(fp2);
	while(ch!=EOF){
		if(ch=='0')
		{
			if(F->lchild)  {F=F->lchild;ch=fgetc(fp2);}
			else 
			{
				printf("%c",F->data);          //F已經是葉子節點，則輸出該節點字符.
			                                   //此時的ch已經屬于下一個字符的編碼，故不需要ch=fgetc(fp);
				fprintf(fp3,"%c",F->data);

				F=T;                           //再讓F指向T的根，翻譯下一個字符
			}

		}
		else 
		{
			if(F->rchild)  {F=F->rchild;ch=fgetc(fp2);}
			else 
			{
				printf("%c",F->data); 
				fprintf(fp3,"%c",F->data);
				F=T;
			}
		}
	}

	printf("該文章已存入文件myfile3中。");
	fclose(fp2);
	fclose(fp3);

}



/*****************************************************************************************************************
                                                   end...
*****************************************************************************************************************/