main.h

huffman 压缩算法 支持文件夹的压缩与解压

#include <Windows.h>
#include <stdio.h>
#include <malloc.h>
#include <tchar.h>
#include <wchar.h>
#include <strsafe.h>
#include "huffman.h"
#define BUFSIZE MAX_PATH

int get_file_size(FILE *fp);
void CompressHuffman (_TCHAR*sour ,_TCHAR*dest);
void DecompressHuffman (_TCHAR*sour ,_TCHAR*dest);

int get_file_size(FILE *fp)
{
    fseek(fp, 0, SEEK_END);
    int size = ftell(fp);
    fseek(fp, 0, SEEK_SET);
    return size;
}

void CompressHuffman (_TCHAR*sourf ,_TCHAR*destf)
{
    FILE *source_file, *dest_file;
    if ((source_file = _wfopen(sourf, L"rb")) == NULL)
    {
        wprintf(L"bad file %s.\n", sourf);
        exit(1);
    }
    else if ((dest_file = _wfopen(destf, L"wb")) == NULL)
    {
        wprintf(L"bad file %s.\n", destf);
        exit(1);
    }

    unsigned char *sour, *dest;
    int usize, csize;


    Huffman huffman;
    usize = get_file_size(source_file);

    sour = new unsigned char[usize];
    dest = new unsigned char[usize];
    usize = (int)fread(sour, 1, usize, source_file);

    huffman.encode(dest, csize, sour, usize);

    fwrite(dest, 1, csize, dest_file);
    wprintf(L" cmpression ration: %.2f\n", (float)usize / (float)csize);

    fclose(source_file);
    fclose(dest_file);
}
void DecompressHuffman (_TCHAR *sourf ,_TCHAR*destf)
{
    FILE *source_file, *dest_file;
    if ((source_file = _wfopen(sourf, L"rb")) == NULL)
    {
        wprintf(L"bad file %s.\n", sourf);
        exit(1);
    }
    else if ((dest_file = _wfopen(destf, L"wb")) == NULL)
    {
        wprintf(L"bad file %s.\n", destf);
        exit(1);
    }
    unsigned char *sour, *dest;
    int usize, csize;

   
    Huffman huffman;
    csize = get_file_size(source_file);

    sour = new unsigned char[csize];
    csize = fread(sour, 1, csize, source_file);
	
    PHUFFHDR  header =(PHUFFHDR) sour;
	//test if its a compressed file
	
   // if (strlen(header->magic)!=5 ){  // wprintf(L"bad file");}
	//else
	{
    dest = new unsigned char[huffman.get_uncompressed_size(sour)];

    huffman.decode(dest, usize, sour);

    fwrite(dest, 1, usize, dest_file);

    fclose(source_file);
    fclose(dest_file);
	}
}