开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > 2D图形编程 > 查看源码
cdib.cpp
资源名称：Canny边缘检测.rar　[点击查看]
上传用户：wzyhdz
上传日期：2010-04-06
资源大小：164k
文件大小：38k
源码类别：

2D图形编程

开发平台：

Visual C++

cdib.cpp：源码内容
  1. // cdib.cpp
  3. #include "stdafx.h"
  4. #include "math.h"
  5. #include "process.h"
  7. #include "cdib.h"
  8. #include "GlobalApi.h"
  10. #ifdef _DEBUG
  11. #define new DEBUG_NEW
  12. #undef THIS_FILE
  13. static char THIS_FILE[] = __FILE__;
  14. #endif
  16. // 声明串行化过程
  17. IMPLEMENT_SERIAL(CDib, CObject, 0);
  20. /*************************************************************************
  21.  *
  22.  * 函数名称：
  23.  *   CDib()
  24.  *
  25.  * 输入参数:
  26.  * 无
  27.  *
  28.  * 返回值:
  29.  *   无
  30.  *
  31.  * 说明:
  32.  *   构造函数
  33.  *
  34.  ************************************************************************
  35.  */
  36. CDib::CDib()
  37. {
  38. m_hFile      = NULL;
  39. m_hBitmap    = NULL;
  40. m_hPalette   = NULL;
  41. m_nBmihAlloc = m_nImageAlloc = noAlloc;
  42. Empty();
  43. }
  46. /*************************************************************************
  47.  *
  48.  * 函数名称：
  49.  *   CDib()
  50.  *
  51.  * 输入参数:
  52.  * CSize size - 位图尺寸
  53.  * int nBitCount - 象素位数
  54.  *
  55.  * 返回值:
  56.  *   无
  57.  *
  58.  * 说明:
  59.  *   构造函数
  60.  *   根据给定的位图尺寸和象素位数构造CDib对象，并对信息头和调色板分配内存
  61.  *   但并没有给位图数据分配内存
  62.  *
  63.  ************************************************************************
  64.  */
  65. CDib::CDib(CSize size, int nBitCount)
  66. {
  67. m_hFile      = NULL;
  68. m_hBitmap    = NULL;
  69. m_hPalette   = NULL;
  70. m_nBmihAlloc = m_nImageAlloc = noAlloc;
  71. Empty();
  73. // 根据象素位数计算调色板尺寸
  74. ComputePaletteSize(nBitCount);
  76. // 分配DIB信息头和调色板的内存
  77. m_lpBMIH = (LPBITMAPINFOHEADER) new 
  78. char[sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD) * m_nColorTableEntries];
  80. // 设置信息头内存分配状态
  81. m_nBmihAlloc = crtAlloc;
  83. // 设置信息头中的信息
  84. m_lpBMIH->biSize = sizeof(BITMAPINFOHEADER);
  85. m_lpBMIH->biWidth = size.cx;
  86. m_lpBMIH->biHeight = size.cy;
  87. m_lpBMIH->biPlanes = 1;
  88. m_lpBMIH->biBitCount = nBitCount;
  89. m_lpBMIH->biCompression = BI_RGB;
  90. m_lpBMIH->biSizeImage = 0;
  91. m_lpBMIH->biXPelsPerMeter = 0;
  92. m_lpBMIH->biYPelsPerMeter = 0;
  93. m_lpBMIH->biClrUsed = m_nColorTableEntries;
  94. m_lpBMIH->biClrImportant= m_nColorTableEntries;
  96. // 计算图象数据内存的大小，并设置此DIB的调色板的指针
  97. ComputeMetrics();
  99. // 将此DIB的调色板初始化为0
  100. memset(m_lpvColorTable, 0, sizeof(RGBQUAD) * m_nColorTableEntries);
  102. // 暂时不分配图象数据内存
  103. m_lpImage = NULL; 
  104. }
  107. /*************************************************************************
  108.  *
  109.  * 函数名称：
  110.  *   ~CDib()
  111.  *
  112.  * 输入参数:
  113.  * 无
  114.  *
  115.  * 返回值:
  116.  *   无
  117.  *
  118.  * 说明:
  119.  *   析构函数，并释放所有分配的DIB内存
  120.  *
  121.  ************************************************************************
  122.  */
  123. CDib::~CDib()
  124. {
  125. Empty();
  126. }
  129. /*************************************************************************
  130.  *
  131.  * 函数名称：
  132.  *   GetDimensions()
  133.  *
  134.  * 输入参数:
  135.  *   无
  136.  *
  137.  * 返回值:
  138.  *   CSize - DIB的宽度和高度
  139.  *
  140.  * 说明:
  141.  *   返回以象素表示的DIB的宽度和高度
  142.  *
  143.  ************************************************************************
  144.  */
  145. CSize CDib::GetDimensions()
  146. {
  147. if(m_lpBMIH == NULL) return CSize(0, 0);
  148. return CSize((int) m_lpBMIH->biWidth, (int) m_lpBMIH->biHeight);
  149. }
  152. /*************************************************************************
  153.  *
  154.  * 函数名称：
  155.  *   AttachMapFile()
  156.  *
  157.  * 输入参数:
  158.  *   const char* strPathname - 映射文件的路径名
  159.  *   BOOL bShare - 如果文件以共享形式打开，设置为TRUE
  160.  * - 默认值为FALSE
  161.  *
  162.  * 返回值:
  163.  *   BOOL - 如果成功，则为TRUE 
  164.  *
  165.  * 说明:
  166.  *   以读模式打开内存映射文件，并将其与CDib对象进行关联。因为在文件使用之前并没有读入内存，
  167.  *   所以它立即返回。不过，当访问这个DIB的时候，可能会有一些延迟，因为文件是分页的。
  168.  *   DetachMapFile函数可以释放现有的已分配的内存，并关闭以前关联的任何内存映射文件。
  169.  *   用内存中的DIB与已有的CDib对象关联。此内存可能是程序的资源，或者是可能是剪贴板
  170.  *   或者OLE数据对象内存。内存可能已经由CRT堆栈用new运算符分配了，或者可能已经由
  171.  *   Windows堆栈用GlobalAlloc分配了。
  172.  *   如果打开相同的文件两次，则Windows以另一个文件来对待 
  173.  *
  174.  ************************************************************************
  175.  */
  176. BOOL CDib::AttachMapFile(const char* strPathname, BOOL bShare) // for reading
  177. {
  178. // 获取文件句柄，并设置打开模式为共享
  179. HANDLE hFile = ::CreateFile(strPathname, GENERIC_WRITE | GENERIC_READ,
  180. bShare ? FILE_SHARE_READ : 0,
  181. NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
  182. ASSERT(hFile != INVALID_HANDLE_VALUE);
  184. // 获取文件的尺寸
  185. DWORD dwFileSize = ::GetFileSize(hFile, NULL);
  187. // 创建文件映射对象，并设置文件映射的模式为读写
  188. HANDLE hMap = ::CreateFileMapping(hFile, NULL, PAGE_READWRITE, 0, 0, NULL);
  189. DWORD dwErr = ::GetLastError();
  190. if(hMap == NULL) {
  191. AfxMessageBox("Empty bitmap file");
  192. return FALSE;
  193. }
  195. // 映射整个文件，注意FILE_MAP_WRITE为读写模式
  196. LPVOID lpvFile = ::MapViewOfFile(hMap, FILE_MAP_WRITE, 0, 0, 0); // map whole file
  197. ASSERT(lpvFile != NULL);
  199. // 确认为bmp格式文件
  200. if(((LPBITMAPFILEHEADER) lpvFile)->bfType != 0x4d42) {
  201. AfxMessageBox("Invalid bitmap file");
  202. DetachMapFile();
  203. return FALSE;
  204. }
  206. // 将内存中的DIB与已有的CDib对象关联
  207. AttachMemory((LPBYTE) lpvFile + sizeof(BITMAPFILEHEADER));
  209. // 将这些有用的句柄设置为类数据成员
  210. m_lpvFile = lpvFile;
  211. m_hFile   = hFile;
  212. m_hMap    = hMap;
  214. // 返回
  215. return TRUE;
  216. }
  219. /*************************************************************************
  220.  *
  221.  * 函数名称：
  222.  *   CopyToMapFile()
  223.  *
  224.  * 输入参数:
  225.  *   const char* strPathname - 映射文件的路径名
  226.  *
  227.  * 返回值:
  228.  *   BOOL - 如果成功，则为TRUE 
  229.  *
  230.  * 说明:
  231.  *   该函数可以创建一个新的内存映射文件，并将现有的CDib数据复制到该文件的内存
  232.  *   释放以前的所有内存。并关闭现有的所有内存映射文件。实际上，直到新文件
  233.  *   关闭的时候，才将这个数据写到磁盘，但是，当CDib对象被重复使用或被破坏
  234.  *   时，也会发生写磁盘操作
  235.  *
  236.  ************************************************************************
  237.  */
  238. BOOL CDib::CopyToMapFile(const char* strPathname)
  239. {
  240. BITMAPFILEHEADER bmfh;
  242. // 设置文件头信息
  243. bmfh.bfType = 0x4d42;  
  244. bmfh.bfSize = m_dwSizeImage + sizeof(BITMAPINFOHEADER) +
  245. sizeof(RGBQUAD) * m_nColorTableEntries + sizeof(BITMAPFILEHEADER);
  246. bmfh.bfReserved1= bmfh.bfReserved2 = 0;
  247. bmfh.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) +
  248. sizeof(RGBQUAD) * m_nColorTableEntries;
  250. // 创建接收数据的文件
  251. HANDLE hFile = ::CreateFile(strPathname, GENERIC_WRITE | GENERIC_READ, 0, NULL,
  252. CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
  253. ASSERT(hFile != INVALID_HANDLE_VALUE);
  255. // 计算文件的大小尺寸
  256. int nSize = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) +
  257. sizeof(RGBQUAD) * m_nColorTableEntries +  m_dwSizeImage;
  259. // 创建内存映射文件对象
  260. HANDLE hMap = ::CreateFileMapping(hFile, NULL, PAGE_READWRITE, 0, nSize, NULL);
  261. DWORD dwErr = ::GetLastError();
  262. ASSERT(hMap != NULL);
  264. // 映射整个文件
  265. LPVOID lpvFile = ::MapViewOfFile(hMap, FILE_MAP_WRITE, 0, 0, 0); 
  266. ASSERT(lpvFile != NULL);
  268. // 临时文件指针
  269. LPBYTE lpbCurrent = (LPBYTE) lpvFile;
  271. // 拷贝文件头信息到内存映射文件中
  272. memcpy(lpbCurrent, &bmfh, sizeof(BITMAPFILEHEADER)); 
  274. // 计算信息头在文件中的地址，并拷贝信息头信息
  275. lpbCurrent += sizeof(BITMAPFILEHEADER);
  276. LPBITMAPINFOHEADER lpBMIH = (LPBITMAPINFOHEADER) lpbCurrent;
  277. memcpy(lpbCurrent, m_lpBMIH,
  278. sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD) * m_nColorTableEntries); 
  280. // 计算调色板在文件中的地址，并拷贝调色板
  281. lpbCurrent += sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD) * m_nColorTableEntries;
  282. memcpy(lpbCurrent, m_lpImage, m_dwSizeImage); 
  284. // 暂时存放图象数据尺寸变量
  285. DWORD dwSizeImage = m_dwSizeImage;
  287. // 释放一起分配的所有内存
  288. Empty();
  290. // 设置图象数据尺寸并设置内存分配状态
  291. m_dwSizeImage = dwSizeImage;
  292. m_nBmihAlloc  = m_nImageAlloc = noAlloc;
  294. // 信息头指针重新指向文件中的位置
  295. m_lpBMIH = lpBMIH;
  297. // 图象数据指针重新指向文件中的数据地址
  298. m_lpImage = lpbCurrent;
  300. // 设置文件句柄
  301. m_hFile = hFile;
  303. // 设置映射对象句柄
  304. m_hMap = hMap;
  306. // 设置映射文件指针
  307. m_lpvFile = lpvFile;
  309. // 重新计算得到调色板尺寸
  310. ComputePaletteSize(m_lpBMIH->biBitCount);
  312. // 重新计算图象数据块大小，并设置调色板指针
  313. ComputeMetrics();
  315. // 如果调色板存在的话，读取并创建一个Windows调色板
  316. MakePalette();
  318. // 返回
  319. return TRUE;
  320. }
  323. /*************************************************************************
  324.  *
  325.  * 函数名称：
  326.  *   AttachMemory()
  327.  *
  328.  * 输入参数:
  329.  *   LPVOID lpvMem - 要关联的内存地址
  330.  *   BOOL bMustDelete - 如果CDib类负责删除这个内存，标记为TRUE
  331.  * - 默认值为FALSE
  332.  *   HGLOBAL hGlobal - 如果内存是通过Win32 GlobalAlloc得到的，
  333.  * - 则CDib对象必须保存该句柄，这样，以后
  334.  * - 可以释放句柄。这里假设bMustDelete设置为TRUE
  335.  *
  336.  * 返回值:
  337.  *   BOOL - 如果成功，则为TRUE 
  338.  *
  339.  * 说明:
  340.  *   用内存中的DIB与已有的CDib对象关联。此内存可能是程序的资源，或者是可能是剪贴板
  341.  *   或者OLE数据对象内存。内存可能已经由CRT堆栈用new运算符分配了，或者可能已经由
  342.  *   Windows堆栈用GlobalAlloc分配了。
  343.  *
  344.  ************************************************************************
  345.  */
  346. BOOL CDib::AttachMemory(LPVOID lpvMem, BOOL bMustDelete, HGLOBAL hGlobal)
  347. {
  348. // 首先释放已经分配的内存
  349. Empty();
  351. m_hGlobal = hGlobal;
  353. // bMustDelete为TRUE表示此CDib类分配的内存，负责删除
  354. // 否则的设置信息头分配状态为noAlloc
  355. if(bMustDelete == FALSE) {
  356. m_nBmihAlloc = noAlloc;
  357. }
  358. else {
  359. m_nBmihAlloc = ((hGlobal == NULL) ? crtAlloc : heapAlloc);
  360. }
  361. try {
  362. // 设置信息头指针
  363. m_lpBMIH = (LPBITMAPINFOHEADER) lpvMem;
  365. // 重新计算得到图象数据块的大小，并设置调色板的指针
  366. ComputeMetrics();
  368. // 计算调色板的尺寸
  369. ComputePaletteSize(m_lpBMIH->biBitCount);
  371. // 设置图象数据指针
  372. m_lpImage = (LPBYTE) m_lpvColorTable + sizeof(RGBQUAD) * m_nColorTableEntries;
  374. // 如果调色板存在的话，读取它，并创建一个Windows调色板，
  375. // 并将调色板的句柄存放在数据成员中
  376. MakePalette();
  377. }
  379. // 错误处理
  380. catch(CException* pe) {
  381. AfxMessageBox("AttachMemory error");
  382. pe->Delete();
  383. return FALSE;
  384. }
  386. // 返回
  387. return TRUE;
  388. }
  391. /*************************************************************************
  392.  *
  393.  * 函数名称：
  394.  *   UsePalette()
  395.  *
  396.  * 输入参数:
  397.  *   CDC* pDC - 要实现调色板的设备上下文指针
  398.  *   BOOL bBackground - 如果标记为FALSE（默认值），并且应用
  399.  * - 程序正在前台运行，则Windows将把该调
  400.  * - 色板作为前台调色板来实现（向系统调色
  401.  * - 板中复制尽可能多的颜色）。如果标记为
  402.  * - TURE，则Windows将把该调色板作为后台
  403.  * - 调色板来实现（尽可能相系统调色板映射
  404.  * - 逻辑调色板）
  405.  *
  406.  * 返回值:
  407.  *   UINT - 如果成功，则返回映射到系统调色板的逻
  408.  * - 辑调色板的表项数，否则返回GDI_ERROR 
  409.  *
  410.  * 说明:
  411.  *   该函数将CDib对象的逻辑调色板选入设备上下文，然后实现该调色板。Draw成员函
  412.  *   数在绘制DIB之前调用UsePalette。
  413.  *
  414.  ************************************************************************
  415.  */
  416. UINT CDib::UsePalette(CDC* pDC, BOOL bBackground /* = FALSE */)
  417. {
  418. // 判断是否存在调色板
  419. if(m_hPalette == NULL) return 0;
  421. // 得到设备上下文句柄
  422. HDC hdc = pDC->GetSafeHdc();
  424. // 选择调色板到设备上下文
  425. ::SelectPalette(hdc, m_hPalette, bBackground);
  427. // 实现该调色板
  428. return ::RealizePalette(hdc);
  429. }
  432. /*************************************************************************
  433.  *
  434.  * 函数名称：
  435.  *   Draw()
  436.  *
  437.  * 输入参数:
  438.  *   CDC* pDC - 指向将要接收DIB图象的设备上下文指针
  439.  *   CPoint origin - 显示DIB的逻辑坐标
  440.  *   CSize size - 显示矩形的宽度和高度
  441.  *
  442.  * 返回值:
  443.  *   BOOL - 如果成功，则为TRUE，
  444.  *
  445.  * 说明:
  446.  *   通过调用Win32 SDK的StretchDIBits函数将CDib对象输出到显示器（或者打印机）。
  447.  *   为了适合指定的矩形，位图可以进行必要的拉伸
  448.  *
  449.  ************************************************************************
  450.  */
  451. BOOL CDib::Draw(CDC* pDC, CPoint origin, CSize size)
  452. {
  453. // 如果信息头为空，表示尚未有数据，返回FALSE
  454. if(m_lpBMIH == NULL) return FALSE;
  456. // 如果调色板不为空，则将调色板选入设备上下文
  457. if(m_hPalette != NULL) {
  458. ::SelectPalette(pDC->GetSafeHdc(), m_hPalette, TRUE);
  459. }
  461. // 设置显示模式
  462. pDC->SetStretchBltMode(COLORONCOLOR);
  464. // 在设备的origin位置上画出大小为size的图象
  465. ::StretchDIBits(pDC->GetSafeHdc(), origin.x, origin.y,size.cx,size.cy,
  466. 0, 0, m_lpBMIH->biWidth, m_lpBMIH->biHeight,
  467. m_lpImage, (LPBITMAPINFO) m_lpBMIH, DIB_RGB_COLORS, SRCCOPY);
  469. // 返回
  470. return TRUE;
  471. }
  474. /*************************************************************************
  475.  *
  476.  * 函数名称：
  477.  *   CreateSection()
  478.  *
  479.  * 输入参数:
  480.  *   CDC* pDC - 设备上下文指针
  481.  *
  482.  * 返回值:
  483.  *   HBITMAP - 到GDI位图的句柄。如果不成功，则为NULL。
  484.  * - 该句柄也是作为公共数据成员存储的
  485.  *
  486.  * 说明:
  487.  *   通过调用Win32 SDK的CreateDIBSection函数创建一个DIB段。图象内存将不被初始化
  488.  *
  489.  ************************************************************************
  490.  */
  491. HBITMAP CDib::CreateSection(CDC* pDC /* = NULL */)
  492. {
  493. // 如果信息头为空，不作任何处理
  494. if(m_lpBMIH == NULL) return NULL;
  496. // 如果图象数据不存在，不作任何处理
  497. if(m_lpImage != NULL) return NULL; 
  499. // 创建一个DIB段
  500. m_hBitmap = ::CreateDIBSection(pDC->GetSafeHdc(), (LPBITMAPINFO) m_lpBMIH,
  501. DIB_RGB_COLORS, (LPVOID*) &m_lpImage, NULL, 0);
  502. ASSERT(m_lpImage != NULL);
  504. // 返回HBIMAP句柄
  505. return m_hBitmap;
  506. }
  509. /*************************************************************************
  510.  *
  511.  * 函数名称：
  512.  *   MakePalette()
  513.  *
  514.  * 输入参数:
  515.  *   无
  516.  *
  517.  * 返回值:
  518.  *   BOOL - 如果成功，则为TRUE
  519.  *
  520.  * 说明:
  521.  *   如果颜色表存在的话，该函数将读取它，并创建一个Windows调色板。
  522.  *   HPALETTE存储在一个数据成员中。
  523.  *
  524.  ************************************************************************
  525.  */
  526. BOOL CDib::MakePalette()
  527. {
  528. // 如果不存在调色板，则返回FALSE
  529. if(m_nColorTableEntries == 0) return FALSE;
  531. if(m_hPalette != NULL) ::DeleteObject(m_hPalette);
  532. TRACE("CDib::MakePalette -- m_nColorTableEntries = %dn", m_nColorTableEntries);
  534. // 给逻辑调色板分配内存
  535. LPLOGPALETTE pLogPal = (LPLOGPALETTE) new char[2 * sizeof(WORD) +
  536. m_nColorTableEntries * sizeof(PALETTEENTRY)];
  538. // 设置逻辑调色板的信息
  539. pLogPal->palVersion = 0x300;
  540. pLogPal->palNumEntries = m_nColorTableEntries;
  542. // 拷贝DIB中的颜色表到逻辑调色板
  543. LPRGBQUAD pDibQuad = (LPRGBQUAD) m_lpvColorTable;
  544. for(int i = 0; i < m_nColorTableEntries; i++) {
  545. pLogPal->palPalEntry[i].peRed = pDibQuad->rgbRed;
  546. pLogPal->palPalEntry[i].peGreen = pDibQuad->rgbGreen;
  547. pLogPal->palPalEntry[i].peBlue = pDibQuad->rgbBlue;
  548. pLogPal->palPalEntry[i].peFlags = 0;
  549. pDibQuad++;
  550. }
  552. // 创建逻辑调色板
  553. m_hPalette = ::CreatePalette(pLogPal);
  555. // 删除临时变量并返回TRUE
  556. delete pLogPal;
  557. return TRUE;
  558. }
  561. /*************************************************************************
  562.  *
  563.  * 函数名称：
  564.  *   SetSystemPalette()
  565.  *
  566.  * 输入参数:
  567.  *   CDC* pDC - 设备上下文指针
  568.  *
  569.  * 返回值:
  570.  *   BOOL - 如果成功，则为TRUE，
  571.  *
  572.  * 说明:
  573.  *   如果16bpp、24bpp或32bppDIB不具备调色板，则该函数可以为CDib对象创建一个逻辑调色板，
  574.  *   它与由CreatehalftonePalette函数返回的调色板相匹配。如果程序在256色调色板显示器上
  575.  *   运行，而你又没有调用SetSystemPalette，那么，你将不具有任何调色板，只有20中标准的
  576.  *   Windows颜色出现在DIB中
  577.  *
  578.  ************************************************************************
  579.  */
  580. BOOL CDib::SetSystemPalette(CDC* pDC)
  581. {
  582. // 如果DIB不具备调色板，则需要利用系统的调色板
  583. if(m_nColorTableEntries != 0) return FALSE;
  585. // 为设备上下文创建中间调色板，并将其与CPalette对象连接
  586. m_hPalette = ::CreateHalftonePalette(pDC->GetSafeHdc());
  588. // 返回
  589. return TRUE;
  590. }
  593. /*************************************************************************
  594.  *
  595.  * 函数名称：
  596.  *   CreateBitmap()
  597.  *
  598.  * 输入参数:
  599.  *   CDC* pDC - 设备上下文指针
  600.  *
  601.  * 返回值:
  602.  *   HBITMAP - 到GDI位图的句柄；如果不成功，则为NULL
  603.  * - 该句柄不是作为公共数据成员存储的
  604.  *
  605.  * 说明:
  606.  *   从已有的DIB中创建DDB位图。不要将这个函数与CreateSection
  607.  *   弄混了，后者的作用是生成DIB并保存句柄
  608.  *
  609.  ************************************************************************
  610.  */
  611. HBITMAP CDib::CreateBitmap(CDC* pDC)
  612. {
  613. // 如果不存在图象数据，则返回NULL
  614. if (m_dwSizeImage == 0) return NULL;
  616. // 用指定的DIB来创建DDB，并用DIB信息初始化位图的图象位
  617. HBITMAP hBitmap = ::CreateDIBitmap(pDC->GetSafeHdc(), m_lpBMIH,
  618. CBM_INIT, m_lpImage, (LPBITMAPINFO) m_lpBMIH, DIB_RGB_COLORS);
  619. ASSERT(hBitmap != NULL);
  621. // 返回DDB位图句柄
  622. return hBitmap;
  623. }
  626. /*************************************************************************
  627.  *
  628.  * 函数名称：
  629.  *   ConvertDDBToDIB()
  630.  *
  631.  * 输入参数:
  632.  *   HBITMAP hBitmap - 指向源数据的BITMAP句柄
  633.  *   CDib* pDibDst - 指向转换目标的CDib对象指针
  634.  *
  635.  * 返回值:
  636.  *   BOOL - 如果操作成功，则返回TRUE
  637.  *
  638.  * 说明:
  639.  *   该函数将源BITMAP类pDibSrc中的数据拷贝到pDibDst中，并对相应的数据成员赋值
  640.  *
  641.  *************************************************************************
  642.  */
  644. BOOL CDib::ConvertFromDDB(HBITMAP hBitmap, HPALETTE hPal)
  645. {
  646. // 声明一个BITMAP结构
  647. BITMAP bm;
  649. // 设备上下文
  650. HDC hDC;
  652. // 象素位数
  653. WORD biBitCount;
  655. // 调色板表项数
  656. int nColorTableEntries;
  658. // 如果hBitmap句柄无效，则返回
  659. if(!hBitmap){
  660. return FALSE;
  661. }
  664. // 释放已分配的内存
  665. Empty();
  667. // 填充图象数据到bm中，其中最后一个参数表示接收这个指定的对象的指针
  668. if(!GetObject(hBitmap,sizeof(BITMAP),(LPBYTE)&bm)){
  669. return FALSE;
  670. }
  672. // 计算象素位数
  673. biBitCount=bm.bmPlanes*bm.bmBitsPixel;
  674. if(biBitCount<=1)
  675. biBitCount=1;
  676. else if(biBitCount<=4)
  677. biBitCount=4;
  678. else if(biBitCount<=8)
  679. biBitCount=8;
  680. else 
  681. biBitCount=24;
  683. // 计算调色板的尺寸
  684. // 如果biClrUsed为零，则用到的颜色数为2的biBitCount次方
  685. switch(biBitCount) {
  686. case 1:
  687. nColorTableEntries = 2;
  688. break;
  689. case 4:
  690. nColorTableEntries = 16;
  691. break;
  692. case 8:
  693. nColorTableEntries = 256;
  694. break;
  695. case 16:
  696. case 24:
  697. case 32:
  698. nColorTableEntries = 0;
  699. break;
  700. default:
  701. ASSERT(FALSE);
  702. }
  703. ASSERT((nColorTableEntries >= 0) && (nColorTableEntries <= 256)); 
  704. m_nColorTableEntries = nColorTableEntries;
  706. // 分配DIB信息头和调色板的内存
  707. m_lpBMIH = (LPBITMAPINFOHEADER) new char
  708. [sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD) * nColorTableEntries];
  709. m_nBmihAlloc = m_nImageAlloc = crtAlloc;
  711. m_lpBMIH->biSize = sizeof(BITMAPINFOHEADER); 
  712. m_lpBMIH->biWidth = bm.bmWidth; 
  713. m_lpBMIH->biHeight = bm.bmHeight; 
  714. m_lpBMIH->biPlanes = 1; 
  715. m_lpBMIH->biBitCount = biBitCount; 
  716. m_lpBMIH->biCompression = BI_RGB; 
  717. m_lpBMIH->biSizeImage = 0; 
  718. m_lpBMIH->biXPelsPerMeter = 0; 
  719. m_lpBMIH->biYPelsPerMeter = 0; 
  720. m_lpBMIH->biClrUsed = nColorTableEntries;
  721. m_lpBMIH->biClrImportant = nColorTableEntries;
  723. // 获得设备上下文句柄
  724. hDC=GetDC(NULL);
  725.    
  726. // 如果没有指定调色板，则从系统中获得当前的系统调色板
  727. if(hPal==NULL){
  728. hPal = GetSystemPalette();
  729. }
  730. hPal = SelectPalette(hDC, hPal, FALSE); 
  731. RealizePalette(hDC); 
  732.  
  734. // 调用GetDIBits填充信息头，并获得图象数据的尺寸。注意这里图象数据指针为NULL
  735. GetDIBits( hDC, hBitmap, 0, (UINT)m_lpBMIH->biHeight, NULL, (LPBITMAPINFO)m_lpBMIH, DIB_RGB_COLORS);
  737. // 如果没有正确的获得图象数据尺寸，则重新计算
  738. if( m_lpBMIH->biSizeImage == 0 ){
  739. m_lpBMIH->biSizeImage=(((bm.bmWidth*biBitCount) + 31) / 32 * 4)*bm.bmHeight;
  740. }
  742. // 分配存放图象数据的内存
  743. m_lpImage = (LPBYTE) new char[m_lpBMIH->biSizeImage];
  745. // 调用GetDIBits加载图象数据，注意这里给出了图象数据指针
  747. // 如果加载图象数据不成功，则释放已经分配的内存，并返回FALSE
  748. if( GetDIBits( hDC, hBitmap, 0, (UINT)m_lpBMIH->biHeight, (LPBYTE)m_lpImage,
  749. (LPBITMAPINFO)m_lpBMIH, DIB_RGB_COLORS) == 0 ){
  750. //clean up and return NULL
  751. Empty();
  753. SelectPalette( hDC, hPal, TRUE );
  754. RealizePalette( hDC );
  755. ReleaseDC( NULL, hDC );
  757. return FALSE;
  758. }
  760. // 删除临时变量
  761. SelectPalette(hDC, hPal, TRUE); 
  762. RealizePalette(hDC); 
  763. ReleaseDC(NULL, hDC); 
  764. return TRUE;
  765. }
  767. /*************************************************************************
  768.  *
  769.  * 函数名称：
  770.  *   Compress()
  771.  *
  772.  * 输入参数:
  773.  *   CDC* pDC - 设备上下文指针
  774.  *   BOOL bCompress - TRUE对应于压缩的DIB，FALSE对应于不压缩的DIB
  775.  *
  776.  * 返回值:
  777.  *   BOOL - 如果成功，则返回TRUE
  778.  *
  779.  * 说明:
  780.  *   该函数将DIB重新生成为压缩或者不压缩的DIB。在内部，它转换已有的DIB为DDB位图
  781.  *   然后生成一个新的压缩或者不压缩的DIB。压缩仅为4bpp和8bpp的DIB所支持。不能
  782.  *   压缩DIB段
  783.  *
  784.  ************************************************************************
  785.  */
  786. BOOL CDib::Compress(CDC* pDC, BOOL bCompress /* = TRUE */)
  787. {
  788. // 判断是否为4bpp或者8bpp位图，否则，不进行压缩，返回FALSE
  789. if((m_lpBMIH->biBitCount != 4) && (m_lpBMIH->biBitCount != 8)) return FALSE;
  791. // 如果为DIB段，也不能支持压缩，返回FALSE
  792. if(m_hBitmap) return FALSE; 
  794. TRACE("Compress: original palette size = %dn", m_nColorTableEntries); 
  796. // 获得设备上下文句柄
  797. HDC hdc = pDC->GetSafeHdc();
  799. // 将此DIB的调色板选入设备上下文，并保存以前的调色板句柄
  800. HPALETTE hOldPalette = ::SelectPalette(hdc, m_hPalette, FALSE);
  802. HBITMAP hBitmap;  
  804. // 创建一个DDB位图，如果不成功，则返回FALSE
  805. if((hBitmap = CreateBitmap(pDC)) == NULL) return FALSE;
  807. // 计算信息头加上调色板的大小尺寸，并给它们分配内存
  808. int nSize = sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD) * m_nColorTableEntries;
  809. LPBITMAPINFOHEADER lpBMIH = (LPBITMAPINFOHEADER) new char[nSize];
  811. // 将信息头和调色板拷贝到内存中
  812. memcpy(lpBMIH, m_lpBMIH, nSize);  // new header
  814. // 如果需要进行压缩，设置相应的信息，并创建压缩格式的DIB
  815. if(bCompress) {
  816. switch (lpBMIH->biBitCount) {
  817. case 4:
  818. lpBMIH->biCompression = BI_RLE4;
  819. break;
  820. case 8:
  821. lpBMIH->biCompression = BI_RLE8;
  822. break;
  823. default:
  824. ASSERT(FALSE);
  825. }
  827. // 设置位图数据指针为NULL，调用GetDIBits来得到压缩格式的DIB的尺寸
  828. // 如果不能创建DIB，则进行相应的错误处理。
  829. if(!::GetDIBits(pDC->GetSafeHdc(), hBitmap, 0, (UINT) lpBMIH->biHeight,
  830. NULL, (LPBITMAPINFO) lpBMIH, DIB_RGB_COLORS)) {
  831. AfxMessageBox("Unable to compress this DIB");
  833. // 删除临时变量，并释放已分配内存
  834.   ::DeleteObject(hBitmap);
  835. delete [] lpBMIH;
  837. // 重新将以前的调色板选入，并返回FALSE
  838. ::SelectPalette(hdc, hOldPalette, FALSE);
  839. return FALSE; 
  840. }
  842. // 如果位图数据为空，则进行相应的错误处理
  843. if (lpBMIH->biSizeImage == 0) {
  844. AfxMessageBox("Driver can't do compression");
  846. // 删除临时变量，并释放已分配内存
  847.   ::DeleteObject(hBitmap);
  848. delete [] lpBMIH;
  850. // 重新将以前的调色板选入，并返回FALSE
  851. ::SelectPalette(hdc, hOldPalette, FALSE);
  852. return FALSE; 
  853. }
  855. // 将位图数据尺寸赋值给类的成员变量
  856. else {
  857. m_dwSizeImage = lpBMIH->biSizeImage;
  858. }
  859. }
  861. // 如果是解压缩，进行相应的处理
  862. else {
  864. // 设置压缩格式为不压缩
  865. lpBMIH->biCompression = BI_RGB; 
  867. // 根据位图的宽度和高度计算位图数据内存的大小
  868. DWORD dwBytes = ((DWORD) lpBMIH->biWidth * lpBMIH->biBitCount) / 32;
  869. if(((DWORD) lpBMIH->biWidth * lpBMIH->biBitCount) % 32) {
  870. dwBytes++;
  871. }
  872. dwBytes *= 4;
  874. // 将得到位图数据的大小尺寸保存在类的成员变量中
  875. m_dwSizeImage = dwBytes * lpBMIH->biHeight; 
  877. // 将位图数据内存的大小赋值给临时的信息头中的相应的变量
  878. lpBMIH->biSizeImage = m_dwSizeImage;
  879. } 
  881. // 再次调用GetDIBits来生成DIB数据
  883. // 分配临时存放位图数据
  884. LPBYTE lpImage = (LPBYTE) new char[m_dwSizeImage];
  886. // 再次调用GetDIBits来生成DIB数据，注意此时位图数据指针不为空
  887. VERIFY(::GetDIBits(pDC->GetSafeHdc(), hBitmap, 0, (UINT) lpBMIH->biHeight,
  888.      lpImage, (LPBITMAPINFO) lpBMIH, DIB_RGB_COLORS));
  889. TRACE("dib successfully created - height = %dn", lpBMIH->biHeight);
  891. // 压缩转换完毕，进行相应的其他处理
  893. // 删除临时的DDB位图
  894. ::DeleteObject(hBitmap);
  896. // 释放原来的DIB分配的内存
  897. Empty();
  899. // 重新设置图象信息头和图象数据内存分配状态
  900. m_nBmihAlloc = m_nImageAlloc = crtAlloc;
  902. // 重新定位信息头和图象数据指针
  903. m_lpBMIH = lpBMIH;
  904. m_lpImage = lpImage;
  906. // 计算图象数据尺寸，并设置DIB中调色板的指针
  907. ComputeMetrics();
  909. // 计算DIB中调色板的尺寸
  910. ComputePaletteSize(m_lpBMIH->biBitCount);
  912. // 如果DIB中调色板存在的话，读取并创建一个Windows调色板
  913. MakePalette();
  915. // 恢复以前的调色板
  916. ::SelectPalette(hdc, hOldPalette, FALSE);
  917. TRACE("Compress: new palette size = %dn", m_nColorTableEntries); 
  919. // 返回
  920. return TRUE;
  921. }
  924. /*************************************************************************
  925.  *
  926.  * 函数名称：
  927.  *   Read()
  928.  *
  929.  * 输入参数:
  930.  *   CFile* pFile - 指向CFile对象的指针
  931.  *
  932.  * 返回值:
  933.  *   BOOL - 如果成功，则返回TRUE
  934.  *
  935.  * 说明:
  936.  *   该函数DIB从一个文件读入CDib对象。该文件必须成功打开。如果该文件是BMP文件
  937.  *   读取工作从文件头开始。如果该文件是一个文档，读取工作则从当前文件指针处开始 
  938.  *
  939.  ************************************************************************
  940.  */
  941. BOOL CDib::Read(CFile* pFile)
  942. {
  943. // 释放已经分配的内存
  944. Empty();
  946. // 临时存放信息的变量
  947. int nCount, nSize;
  948. BITMAPFILEHEADER bmfh;
  950. // 进行读操作
  951. try 
  952. {
  953. // 读取文件头
  954. nCount = pFile->Read((LPVOID) &bmfh, sizeof(BITMAPFILEHEADER));
  955. if(nCount != sizeof(BITMAPFILEHEADER)) {
  956. throw new CException;
  957. }
  959. // 如果文件类型部位"BM"，则返回并进行相应错误处理
  960. if(bmfh.bfType != 0x4d42) {
  961. throw new CException;
  962. }
  964. // 计算信息头加上调色板的大小，并分配相应的内存
  965. nSize = bmfh.bfOffBits - sizeof(BITMAPFILEHEADER);
  966. m_lpBMIH = (LPBITMAPINFOHEADER) new char[nSize];
  967. m_nBmihAlloc = m_nImageAlloc = crtAlloc;
  969. // 读取信息头和调色板
  970. nCount = pFile->Read(m_lpBMIH, nSize); 
  972. // 计算图象数据大小并设置调色板指针
  973. ComputeMetrics();
  975. // 计算调色板的表项数
  976. ComputePaletteSize(m_lpBMIH->biBitCount);
  978. // 如果DIB中存在调色板，则创建一个Windows调色板
  979. MakePalette();
  981. // 分配图象数据内存，并从文件中读取图象数据
  982. m_lpImage = (LPBYTE) new char[m_dwSizeImage];
  983. nCount = pFile->Read(m_lpImage, m_dwSizeImage); 
  984. }
  986. // 错误处理
  987. catch(CException* pe) 
  988. {
  989. AfxMessageBox("Read error");
  990. pe->Delete();
  991. return FALSE;
  992. }
  994. // 返回
  995. return TRUE;
  996. }
  999. /*************************************************************************
  1000.  *
  1001.  * 函数名称：
  1002.  *   ReadSection()
  1003.  *
  1004.  * 输入参数:
  1005.  *   CFile* pFile - 指向CFile对象的指针；对应的磁盘
  1006.  * - 文件中包含DIB
  1007.  *   CDC* pDC - 设备上下文指针
  1008.  *
  1009.  * 返回值:
  1010.  *   BOOL - 如果成功，则返回TRUE
  1011.  *
  1012.  * 说明:
  1013.  *   该函数从BMP文件中读取信息头，调用CreateDIBSection来分配图象内存，然后将
  1014.  *   图象从该文件读入刚才分配的内存。如果你想从磁盘读取一个DIB，然后通过调用
  1015.  *   GDI函数编辑它的话，可以使用该函数。你可以用Write或CopyToMapFile将DIB写
  1016.  *   回到磁盘
  1017.  *
  1018.  ************************************************************************
  1019.  */
  1020. BOOL CDib::ReadSection(CFile* pFile, CDC* pDC /* = NULL */)
  1021. {
  1023. // 释放已经分配的内存
  1024. Empty();
  1026. // 临时变量
  1027. int nCount, nSize;
  1028. BITMAPFILEHEADER bmfh;
  1030. // 从文件中读取数据
  1031. try {
  1032. // 读取文件头
  1033. nCount = pFile->Read((LPVOID) &bmfh, sizeof(BITMAPFILEHEADER));
  1034. if(nCount != sizeof(BITMAPFILEHEADER)) {
  1035. throw new CException;
  1036. }
  1038. // 如果文件类型部位"BM"，则返回并进行相应错误处理
  1039. if(bmfh.bfType != 0x4d42) {
  1040. throw new CException;
  1041. }
  1043. // 计算信息头加上调色板的大小，并分配相应的内存
  1044. nSize = bmfh.bfOffBits - sizeof(BITMAPFILEHEADER);
  1045. m_lpBMIH = (LPBITMAPINFOHEADER) new char[nSize];
  1046. m_nBmihAlloc = crtAlloc;
  1047. m_nImageAlloc = noAlloc;
  1049. // 读取信息头和调色板
  1050. nCount = pFile->Read(m_lpBMIH, nSize); 
  1052. // 如果图象为压缩格式，则不进行后续处理
  1053. if(m_lpBMIH->biCompression != BI_RGB) {
  1054. throw new CException;
  1055. }
  1057. // 计算图象数据大小并设置调色板指针
  1058. ComputeMetrics();
  1060. // 计算调色板的表项数
  1061. ComputePaletteSize(m_lpBMIH->biBitCount);
  1063. // 如果DIB中存在调色板，则创建一个Windows调色板
  1064. MakePalette();
  1066. // 将CDib对象的逻辑调色板选入设备上下文
  1067. UsePalette(pDC);
  1069. // 创建一个DIB段，并分配图象内存
  1070. m_hBitmap = ::CreateDIBSection(pDC->GetSafeHdc(), (LPBITMAPINFO) m_lpBMIH,
  1071. DIB_RGB_COLORS, (LPVOID*) &m_lpImage, NULL, 0);
  1072. ASSERT(m_lpImage != NULL);
  1074. // 从文件中读取图象数据
  1075. nCount = pFile->Read(m_lpImage, m_dwSizeImage); 
  1076. }
  1078. // 错误处理
  1079. catch(CException* pe) {
  1080. AfxMessageBox("ReadSection error");
  1081. pe->Delete();
  1082. return FALSE;
  1083. }
  1084. return TRUE;
  1085. }
  1088. /*************************************************************************
  1089.  *
  1090.  * 函数名称：
  1091.  *   Write()
  1092.  *
  1093.  * 输入参数:
  1094.  *   CFile* pFile - 指向CFile对象的指针
  1095.  *
  1096.  * 返回值:
  1097.  *   BOOL - 如果成功，则返回TRUE
  1098.  *
  1099.  * 说明:
  1100.  *   该函数把DIB从CDib对象写进文件中。该文件必须成功打开或者创建
  1101.  *
  1102.  ************************************************************************
  1103.  */
  1104. BOOL CDib::Write(CFile* pFile)
  1105. {
  1106. BITMAPFILEHEADER bmfh;
  1108. // 设置文件头中文件类型为"BM"
  1109. bmfh.bfType = 0x4d42;  
  1111. // 计算信息头和调色板的大小尺寸
  1112. int nSizeHdr = sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD) * m_nColorTableEntries;
  1114. // 设置文件头信息
  1115. bmfh.bfSize = sizeof(BITMAPFILEHEADER) + nSizeHdr + m_dwSizeImage;
  1116. bmfh.bfReserved1 = bmfh.bfReserved2 = 0;
  1117. bmfh.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) +
  1118. sizeof(RGBQUAD) * m_nColorTableEntries;
  1120. // 进行写操作
  1121. try {
  1122. pFile->Write((LPVOID) &bmfh, sizeof(BITMAPFILEHEADER));
  1123. pFile->Write((LPVOID) m_lpBMIH,  nSizeHdr);
  1124. pFile->Write((LPVOID) m_lpImage, m_dwSizeImage);
  1125. }
  1127. // 错误处理
  1128. catch(CException* pe) {
  1129. pe->Delete();
  1130. AfxMessageBox("write error");
  1131. return FALSE;
  1132. }
  1134. // 返回
  1135. return TRUE;
  1136. }
  1139. /*************************************************************************
  1140.  *
  1141.  * 函数名称：
  1142.  *   Serialize()
  1143.  *
  1144.  * 输入参数:
  1145.  *   CArchive& ar - 指向应用程序归档对象
  1146.  *
  1147.  * 返回值:
  1148.  *   无
  1149.  *
  1150.  * 说明:
  1151.  *   该函数进行串行化过程，将CDib数据进行读入或者写出
  1152.  *
  1153.  ************************************************************************
  1154.  */
  1155. void CDib::Serialize(CArchive& ar)
  1156. {
  1157. DWORD dwPos;
  1159. // 获得此归档文件的CFile对象指针
  1160. dwPos = ar.GetFile()->GetPosition();
  1161. TRACE("CDib::Serialize -- pos = %dn", dwPos);
  1163. // 从归档文件缓冲区中冲掉未写入数据
  1164. ar.Flush();
  1166. // 重新获得此归档文件的CFile对象指针
  1167. dwPos = ar.GetFile()->GetPosition();
  1168. TRACE("CDib::Serialize -- pos = %dn", dwPos);
  1170. // 确定归档文件是否被存储，是则进行存储
  1171. if(ar.IsStoring()) {
  1172. Write(ar.GetFile());
  1173. }
  1175. // 否则进行加载
  1176. else {
  1177. Read(ar.GetFile());
  1178. }
  1179. }
  1182. /*************************************************************************
  1183.  *
  1184.  * 函数名称：
  1185.  *   ComputePaletteSize()
  1186.  *
  1187.  * 输入参数:
  1188.  *   int nBitCount - 指向CFile对象的指针
  1189.  *
  1190.  * 返回值:
  1191.  *   无
  1192.  *
  1193.  * 说明:
  1194.  *   该函数根据位图象素位数计算调色板的尺寸
  1195.  *
  1196.  ************************************************************************
  1197.  */
  1198. void CDib::ComputePaletteSize(int nBitCount)
  1199. {
  1200. // 如果biClrUsed为零，则用到的颜色数为2的biBitCount次方
  1201. if((m_lpBMIH == NULL) || (m_lpBMIH->biClrUsed == 0)) {
  1202. switch(nBitCount) {
  1203. case 1:
  1204. m_nColorTableEntries = 2;
  1205. break;
  1206. case 4:
  1207. m_nColorTableEntries = 16;
  1208. break;
  1209. case 8:
  1210. m_nColorTableEntries = 256;
  1211. break;
  1212. case 16:
  1213. case 24:
  1214. case 32:
  1215. m_nColorTableEntries = 0;
  1216. break;
  1217. default:
  1218. ASSERT(FALSE);
  1219. }
  1220. }
  1222. // 否则调色板的表项数就是用到的颜色数目
  1223. else {
  1224. m_nColorTableEntries = m_lpBMIH->biClrUsed;
  1225. }
  1227. ASSERT((m_nColorTableEntries >= 0) && (m_nColorTableEntries <= 256)); 
  1228. }
  1231. /*************************************************************************
  1232.  *
  1233.  * 函数名称：
  1234.  *   ComputeMetrics()
  1235.  *
  1236.  * 输入参数:
  1237.  *   无
  1238.  *
  1239.  * 返回值:
  1240.  *   无
  1241.  *
  1242.  * 说明:
  1243.  *   该函数计算图象位图的尺寸，并对DIB中的调色板的指针进行赋值
  1244.  *
  1245.  ************************************************************************
  1246.  */
  1247. void CDib::ComputeMetrics()
  1248. {
  1249. // 如果结构的长度不对，则进行错误处理
  1250. if(m_lpBMIH->biSize != sizeof(BITMAPINFOHEADER)) {
  1251. TRACE("Not a valid Windows bitmap -- probably an OS/2 bitmapn");
  1252. throw new CException;
  1253. }
  1255. // 保存图象数据内存大小到CDib对象的数据成员中
  1256. m_dwSizeImage = m_lpBMIH->biSizeImage;
  1258. // 如果图象数据内存大小为0，则重新计算
  1259. if(m_dwSizeImage == 0) {
  1260. DWORD dwBytes = ((DWORD) m_lpBMIH->biWidth * m_lpBMIH->biBitCount) / 32;
  1261. if(((DWORD) m_lpBMIH->biWidth * m_lpBMIH->biBitCount) % 32) {
  1262. dwBytes++;
  1263. }
  1264. dwBytes *= 4;
  1265. m_dwSizeImage = dwBytes * m_lpBMIH->biHeight;
  1266. }
  1268. // 设置DIB中的调色板指针
  1269. m_lpvColorTable = (LPBYTE) m_lpBMIH + sizeof(BITMAPINFOHEADER);
  1270. }
  1273. /*************************************************************************
  1274.  *
  1275.  * 函数名称：
  1276.  *   Empty()
  1277.  *
  1278.  * 输入参数:
  1279.  *   无
  1280.  *
  1281.  * 返回值:
  1282.  *   无
  1283.  *
  1284.  * 说明:
  1285.  *   该函数清空DIB，释放已分配的内存，并且必要的时候关闭映射文件
  1286.  *
  1287.  ************************************************************************
  1288.  */
  1289. void CDib::Empty()
  1290. {
  1291. // 关闭内存映射文件的连接
  1292. DetachMapFile();
  1294. // 根据内存分配的状态，调用相应的函数释放信息头
  1295. if(m_nBmihAlloc == crtAlloc) {
  1296. delete [] m_lpBMIH;
  1297. }
  1298. else if(m_nBmihAlloc == heapAlloc) {
  1299. ::GlobalUnlock(m_hGlobal);
  1300. ::GlobalFree(m_hGlobal);
  1301. }
  1303. // 释放图象数据内存
  1304. if(m_nImageAlloc == crtAlloc) delete [] m_lpImage;
  1306. // 释放调色板句柄
  1307. if(m_hPalette != NULL) ::DeleteObject(m_hPalette);
  1309. // 如果创建了BITMAP，则进行释放
  1310. if(m_hBitmap != NULL) ::DeleteObject(m_hBitmap);
  1312. // 重新设置内存分配状态
  1313. m_nBmihAlloc = m_nImageAlloc = noAlloc;
  1315. // 释放内存后，还需要将指针设置为NULL并将相应的数据设置为0
  1316. m_hGlobal = NULL;
  1317. m_lpBMIH = NULL;
  1318. m_lpImage = NULL;
  1319. m_lpvColorTable = NULL;
  1320. m_nColorTableEntries = 0;
  1321. m_dwSizeImage = 0;
  1322. m_lpvFile = NULL;
  1323. m_hMap = NULL;
  1324. m_hFile = NULL;
  1325. m_hBitmap = NULL;
  1326. m_hPalette = NULL;
  1327. }
  1330. /*************************************************************************
  1331.  *
  1332.  * 函数名称：
  1333.  *   DetachMapFile()
  1334.  *
  1335.  * 输入参数:
  1336.  *   无
  1337.  *
  1338.  * 返回值:
  1339.  *   无
  1340.  *
  1341.  * 说明:
  1342.  *   函数可以释放现有的已分配的内存，并关闭以前关联的任何内存映射文件。
  1343.  *
  1344.  ************************************************************************
  1345.  */
  1346. void CDib::DetachMapFile()
  1347. {
  1348. // 如果没有进行内存映射，则不进行处理
  1349. if(m_hFile == NULL) return;
  1351. // 关闭内存映射
  1352. ::UnmapViewOfFile(m_lpvFile);
  1354. // 关闭内存映射对象和文件
  1355. ::CloseHandle(m_hMap);
  1356. ::CloseHandle(m_hFile);
  1357. m_hFile = NULL;
  1358. }
  1361. /*************************************************************************
  1362.  *
  1363.  * 函数名称：
  1364.  *   PaletteSize()
  1365.  *
  1366.  * 输入参数:
  1367.  *   无
  1368.  *
  1369.  * 返回值:
  1370.  *   DWORD - 返回调色板的尺寸
  1371.  *
  1372.  * 说明:
  1373.  *   该函数计算机调色板所需的尺寸
  1374.  *
  1375.  ************************************************************************
  1376.  */
  1377. WORD CDib::PaletteSize()
  1378. {
  1379. // 临时变量
  1380. WORD        NumColors;
  1381. LPBITMAPINFOHEADER lpbi=m_lpBMIH;
  1383. // 如果biClrUsed为零，且图象象素位数小于8，则计算调色板用到的表项数
  1384. NumColors =  ((lpbi)->biClrUsed == 0 && (lpbi)->biBitCount <= 8 
  1385.                                     ? (int)(1 << (int)(lpbi)->biBitCount)          
  1386.                                     : (int)(lpbi)->biClrUsed);
  1388. // 根据颜色表示的字节数计算调色板的尺寸
  1389. if (lpbi->biSize == sizeof(BITMAPCOREHEADER))
  1390. return NumColors * sizeof(RGBTRIPLE);
  1391. else 
  1392. return NumColors * sizeof(RGBQUAD);
  1394. }
  1397. /*************************************************************************
  1398.  *
  1399.  * 函数名称：
  1400.  *   IsEmpty()
  1401.  *
  1402.  * 输入参数:
  1403.  *   无
  1404.  *
  1405.  * 返回值:
  1406.  *   BOOL - 如果信息头和图象数据为空，则返回TRUE
  1407.  *
  1408.  * 说明:
  1409.  *   判断信息头和图象数据是否为空
  1410.  *
  1411.  ************************************************************************
  1412.  */
  1413. BOOL CDib::IsEmpty()
  1414. {
  1416. if( m_lpBMIH == NULL&&m_lpImage == NULL)
  1417. return TRUE;
  1418. else
  1419. return FALSE;
  1421. }
  1424. /*************************************************************************
  1425.  *
  1426.  * 函数名称：
  1427.  *   GetDibSaveDim()
  1428.  *
  1429.  * 输入参数:
  1430.  *   无
  1431.  *
  1432.  * 返回值:
  1433.  *   CSize - DIB实际存储的高度和宽度
  1434.  *
  1435.  * 说明:
  1436.  *   该函数函数用来得到dib的实际存储宽度（DWORD对齐）
  1437.  *
  1438.  ************************************************************************
  1439.  */
  1441. CSize CDib::GetDibSaveDim()
  1442. {
  1443. CSize sizeSaveDim;
  1444. sizeSaveDim.cx = ( m_lpBMIH->biWidth * m_lpBMIH->biBitCount + 31)/32 * 4;
  1445. sizeSaveDim.cy = m_lpBMIH->biHeight; 
  1446. return sizeSaveDim;
  1448. }
  1451. /*************************************************************************
  1452.  *
  1453.  * 函数名称：
  1454.  *   GetPixelOffset()
  1455.  *
  1456.  * 输入参数:
  1457.  *   int x - 象素在X轴的坐标
  1458.  *   int y - 象素在Y轴的坐标
  1459.  *
  1460.  * 返回值:
  1461.  *   int - 返回象素在图象数据块中的真实地址
  1462.  *
  1463.  * 说明:
  1464.  *   该函数得到坐标为(x,y)的象素点的真实地址。由于DIB结构中对图象数据排列的
  1465.  *   方式为从下到上，从左到右的，所以需要进行转换。
  1466.  *
  1467.  ************************************************************************
  1468.  */
  1469. LONG CDib::GetPixelOffset(int  x, int y)
  1470. {
  1471. CSize sizeSaveDim;
  1472. sizeSaveDim = GetDibSaveDim();
  1474. LONG lOffset = (LONG) (sizeSaveDim.cy - y - 1) * sizeSaveDim.cx +
  1475. x  / (8 / m_lpBMIH->biBitCount);
  1476. return lOffset;
  1477. }
  1480. /*************************************************************************
  1481.  *
  1482.  * 函数名称：
  1483.  *   GetPixel()
  1484.  *
  1485.  * 输入参数:
  1486.  *   int x - 象素在X轴的坐标
  1487.  *   int y - 象素在Y轴的坐标
  1488.  *
  1489.  * 返回值:
  1490.  *   RGBQUAD - 返回DIB在该点真实的颜色
  1491.  *
  1492.  * 说明:
  1493.  *   该函数得到DIB图象在该点真是的颜色。
  1494.  *
  1495.  ************************************************************************
  1496.  */
  1497. RGBQUAD CDib::GetPixel(int x, int y)
  1498. {
  1499. // 颜色结构
  1500. RGBQUAD cColor;
  1502. // 根据每象素比特数得到此点的象素值
  1503. switch (m_lpBMIH->biBitCount)
  1504. {
  1505. case 1 :
  1506. if (1<<(7-x%8) & *(LPBYTE)(m_lpImage+GetPixelOffset(x, y)))
  1507. {
  1508. cColor.rgbBlue  = 255;
  1509. cColor.rgbGreen = 255;
  1510. cColor.rgbRed   = 255;
  1511. cColor.rgbReserved =0;
  1512. }
  1513. else
  1514. {
  1515. cColor.rgbBlue  = 0;
  1516. cColor.rgbGreen = 0;
  1517. cColor.rgbRed   = 0;
  1518. cColor.rgbReserved =0;
  1519. }
  1520. break;
  1521. case 4 :
  1522. {
  1523. int nIndex = (*(LPBYTE)(m_lpImage+GetPixelOffset(x, y)) & 
  1524.    (x%2 ? 0x0f : 0xf0)) >> (x%2 ? 0 : 4);
  1525. LPRGBQUAD pDibQuad = (LPRGBQUAD) (m_lpvColorTable) + nIndex;
  1526. cColor.rgbBlue  = pDibQuad->rgbBlue;
  1527. cColor.rgbGreen = pDibQuad->rgbGreen;
  1528. cColor.rgbRed   = pDibQuad->rgbRed;
  1529. cColor.rgbReserved =0;
  1530. }
  1531. break;
  1532. case 8 :
  1533. {
  1534. int nIndex = *(BYTE*)(m_lpImage+GetPixelOffset(x, y));
  1535. LPRGBQUAD pDibQuad = (LPRGBQUAD) (m_lpvColorTable) + nIndex;
  1536. cColor.rgbBlue  = pDibQuad->rgbBlue;
  1537. cColor.rgbGreen = pDibQuad->rgbGreen;
  1538. cColor.rgbRed   = pDibQuad->rgbRed;
  1539. cColor.rgbReserved =0;
  1540. }
  1541. break;
  1542. default:
  1543. int nIndex = *(BYTE*)(m_lpImage+GetPixelOffset(x, y));
  1544. cColor.rgbRed   = m_lpImage[nIndex];
  1545. cColor.rgbGreen = m_lpImage[nIndex + 1];
  1546. cColor.rgbBlue  = m_lpImage[nIndex + 2];
  1547. cColor.rgbReserved =0;
  1548. break;
  1549. }
  1551. // 返回颜色结构
  1552. return cColor;
  1553. }