IMAGECODING.rar_C实现算术编码_mfc香农_霍夫曼压缩 mfc_香农 编码 MFC

// cdib.cpp #include "stdafx.h" #include "math.h" #include "process.h" #include "cdib.h" #include "GlobalApi.h" #ifdef _DEBUG #define new DEBUG_NEW #undef THIS_FILE static char THIS_FILE[] = __FILE__; #endif // 声明串行化过程 IMPLEMENT_SERIAL(CDib, CObject, 0); /************************************************************************* * * \函数名称： * CDib() * * \输入参数: * 无 * * \返回值: * 无 * * \说明: * 构造函数 * ************************************************************************ */ CDib::CDib() { m_hFile = NULL; m_hBitmap = NULL; m_hPalette = NULL; m_nBmihAlloc = m_nImageAlloc = noAlloc; Empty(); } /************************************************************************* * * \函数名称： * CDib() * * \输入参数: * CSize size - 位图尺寸 * int nBitCount - 象素位数 * * \返回值: * 无 * * \说明: * 构造函数 * 根据给定的位图尺寸和象素位数构造CDib对象，并对信息头和调色板分配内存 * 但并没有给位图数据分配内存 * ************************************************************************ */ CDib::CDib(CSize size, int nBitCount) { m_hFile = NULL; m_hBitmap = NULL; m_hPalette = NULL; m_nBmihAlloc = m_nImageAlloc = noAlloc; Empty(); // 根据象素位数计算调色板尺寸 ComputePaletteSize(nBitCount); // 分配DIB信息头和调色板的内存 m_lpBMIH = (LPBITMAPINFOHEADER) new char[sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD) * m_nColorTableEntries]; // 设置信息头内存分配状态 m_nBmihAlloc = crtAlloc; // 设置信息头中的信息 m_lpBMIH->biSize = sizeof(BITMAPINFOHEADER); m_lpBMIH->biWidth = size.cx; m_lpBMIH->biHeight = size.cy; m_lpBMIH->biPlanes = 1; m_lpBMIH->biBitCount = nBitCount; m_lpBMIH->biCompression = BI_RGB; m_lpBMIH->biSizeImage = 0; m_lpBMIH->biXPelsPerMeter = 0; m_lpBMIH->biYPelsPerMeter = 0; m_lpBMIH->biClrUsed = m_nColorTableEntries; m_lpBMIH->biClrImportant= m_nColorTableEntries; // 计算图象数据内存的大小，并设置此DIB的调色板的指针 ComputeMetrics(); // 将此DIB的调色板初始化为0 memset(m_lpvColorTable, 0, sizeof(RGBQUAD) * m_nColorTableEntries); // 暂时不分配图象数据内存 m_lpImage = NULL; } /************************************************************************* * * \函数名称： * ~CDib() * * \输入参数: * 无 * * \返回值: * 无 * * \说明: * 析构函数，并释放所有分配的DIB内存 * ************************************************************************ */ CDib::~CDib() { Empty(); } /************************************************************************* * * \函数名称： * GetDimensions() * * \输入参数: * 无 * * \返回值: * CSize - DIB的宽度和高度 * * \说明: * 返回以象素表示的DIB的宽度和高度 * ************************************************************************ */ CSize CDib::GetDimensions() { if(m_lpBMIH == NULL) return CSize(0, 0); return CSize((int) m_lpBMIH->biWidth, (int) m_lpBMIH->biHeight); } /************************************************************************* * * \函数名称： * AttachMapFile() * * \输入参数: * const char* strPathname - 映射文件的路径名 * BOOL bShare - 如果文件以共享形式打开，设置为TRUE * - 默认值为FALSE * * \返回值: * BOOL - 如果成功，则为TRUE * * \说明: * 以读模式打开内存映射文件，并将其与CDib对象进行关联。因为在文件使用之前并没有读入内存， * 所以它立即返回。不过，当访问这个DIB的时候，可能会有一些延迟，因为文件是分页的。 * DetachMapFile函数可以释放现有的已分配的内存，并关闭以前关联的任何内存映射文件。 * 用内存中的DIB与已有的CDib对象关联。此内存可能是程序的资源，或者是可能是剪贴板 * 或者OLE数据对象内存。内存可能已经由CRT堆栈用new运算符分配了，或者可能已经由 * Windows堆栈用GlobalAlloc分配了。 * 如果打开相同的文件两次，则Windows以另一个文件来对待 * ************************************************************************ */ BOOL CDib::AttachMapFile(const char* strPathname, BOOL bShare) // for reading { // 获取文件句柄，并设置打开模式为共享 HANDLE hFile = ::CreateFile(strPathname, GENERIC_WRITE | GENERIC_READ, bShare ? FILE_SHARE_READ : 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL); ASSERT(hFile != INVALID_HANDLE_VALUE); // 获取文件的尺寸 DWORD dwFileSize = ::GetFileSize(hFile, NULL); // 创建文件映射对象，并设置文件映射的模式为读写 HANDLE hMap = ::CreateFileMapping(hFile, NULL, PAGE_READWRITE, 0, 0, NULL); DWORD dwErr = ::GetLastError(); if(hMap == NULL) { AfxMessageBox("Empty bitmap file"); return FALSE; } // 映射整个文件，注意FILE_MAP_WRITE为读写模式 LPVOID lpvFile = ::MapViewOfFile(hMap, FILE_MAP_WRITE, 0, 0, 0); // map whole file ASSERT(lpvFile != NULL); // 确认为bmp格式文件 if(((LPBITMAPFILEHEADER) lpvFile)->bfType != 0x4d42) { AfxMessageBox("Invalid bitmap file"); DetachMapFile(); return FALSE; } // 将内存中的DIB与已有的CDib对象关联 AttachMemory((LPBYTE) lpvFile + sizeof(BITMAPFILEHEADER)); // 将这些有用的句柄设置为类数据成员 m_lpvFile = lpvFile; m_hFile = hFile; m_hMap = hMap; // 返回 return TRUE; } /************************************************************************* * * \函数名称： * CopyToMapFile() * * \输入参数: * const char* strPathname - 映射文件的路径名 * * \返回值: * BOOL - 如果成功，则为TRUE * * \说明: * 该函数可以创建一个新的内存映射文件，并将现有的CDib数据复制到该文件的内存 * 释放以前的所有内存。并关闭现有的所有内存映射文件。实际上，直到新文件 * 关闭的时候，才将这个数据写到磁盘，但是，当CDib对象被重复使用或被破坏 * 时，也会发生写磁盘操作 * ************************************************************************ */ BOOL CDib::CopyToMapFile(const char* strPathname) { BITMAPFILEHEADER bmfh; // 设置文件头信息 bmfh.bfType = 0x4d42; bmfh.bfSize = m_dwSizeImage + sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD) * m_nColorTableEntries + sizeof(BITMAPFILEHEADER); bmfh.bfReserved1= bmfh.bfReserved2 = 0; bmfh.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD) * m_nColorTableEntries; // 创建接收数据的文件 HANDLE hFile = ::CreateFile(strPathname, GENERIC_WRITE | GENERIC_READ, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); ASSERT(hFile != INVALID_HANDLE_VALUE); // 计算文件的大小尺寸 int nSize = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD) * m_nColorTableEntries + m_dwSizeImage; // 创建内存映射文件对象 HANDLE hMap = ::CreateFileMapping(hFile, NULL, PAGE_READWRITE, 0, nSize, NULL); DWORD dwErr = ::GetLastError(); ASSERT(hMap != NULL); // 映射整个文件 LPVOID lpvFile = ::MapViewOfFile(hMap, FILE_MAP_WRITE, 0, 0, 0); ASSERT(lpvFile != NULL); // 临时文件指针 LPBYTE lpbCurrent = (LPBYTE) lpvFile; // 拷贝文件头信息到内存映射文件中 memcpy(lpbCurrent, &bmfh, sizeof(BITMAPFILEHEADER)); // 计算信息头在文件中的地址，并拷贝信息头信息 lpbCurrent += sizeof(BITMAPFILEHEADER); LPBITMAPINFOHEADER lpBMIH = (LPBITMAPINFOHEADER) lpbCurrent; memcpy(lpbCurrent, m_lpBMIH, sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD) * m_nColorTableEntries); // 计算调色板在文件中的地址，并拷贝调色板 lpbCurrent += sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD) * m_nColorTableEntries; memcpy(lpbCurrent, m_lpImage, m_dwSizeImage); // 暂时存放图象数据尺寸变量 DWORD dwSizeImage = m_dwSizeImage; // 释放一起分配的所有内存 Empty(); // 设置图象数据尺寸并设置内存分配状态 m_dwSizeImage = dwSizeImage; m_nBmihAlloc = m_nImageAlloc = noAlloc; // 信息头指针重新指向文件中的位置 m_lpBMIH = lpBMIH; // 图象数据指针重新指向文件中的数据地址 m_lpImage = lpbCurrent; // 设置文件句柄 m_hFile = hFile; // 设置映射对象句柄 m_hMap = hMap; // 设置映射文件指针 m_lpvFile = lpvFile; // 重新计算得到调色板尺寸 ComputePaletteSize(m_lpBMI