CeXDib.cpp
上传用户:dengkfang
上传日期:2008-12-30
资源大小:5233k
文件大小:9k
源码类别:

CA认证

开发平台:

Visual C++

  1. #include "stdafx.h"
  2. #include "CeXDib.h"
  3. #ifndef _MFC_VER
  4. #include <windows.h>
  5. #include <tchar.h>
  6. #pragma message("    compiling for Win32")
  7. #endif
  8. /*
  9. #ifdef _DEBUG
  10. #undef THIS_FILE
  11. static char THIS_FILE[]=__FILE__;
  12. #define new DEBUG_NEW
  13. #endif
  14. */
  15. CCeXDib::CCeXDib()
  16. {
  17. m_hDib = NULL;
  18. m_dwLineWidth = 0;
  19. m_wColors = 0;
  20. m_hMemDC = NULL;
  21. m_hBitmap = NULL;
  22. m_lpBits = NULL;
  23. FreeResources();
  24. }
  25. CCeXDib::~CCeXDib()
  26. {
  27. FreeResources();
  28. }
  29. void CCeXDib::FreeResources()
  30. {
  31. if (m_hMemDC)
  32. ::DeleteDC(m_hMemDC);
  33. if (m_hBitmap)
  34. ::DeleteObject(m_hBitmap);
  35. if (m_hDib)
  36. delete m_hDib;
  37. m_hDib = NULL;
  38. m_hMemDC = NULL;
  39. m_hBitmap = NULL;
  40. m_lpBits = NULL;
  41. memset(&m_bi, 0, sizeof(m_bi));
  42. } // End of FreeResources
  43. HDIB CCeXDib::Create(DWORD dwWidth, DWORD dwHeight, WORD wBitCount)
  44. {
  45.     LPBITMAPINFOHEADER  lpbi = NULL; // Pointer to BITMAPINFOHEADER
  46.     DWORD               dwLen = 0; // Size of memory block
  47. FreeResources();
  48. // Following <switch> is taken from
  49. // CDIBSectionLite class by Chris Maunder
  50.     switch (wBitCount) 
  51.     {
  52.     case 1:  m_wColors = 2;   break;
  53. #ifdef _WIN32_WCE
  54.         case 2:  m_wColors = 4;   break;   // winCE only       
  55. #endif
  56.         case 4:  m_wColors = 16;  break;
  57.         case 8:  m_wColors = 256; break;
  58.         case 16:
  59.         case 24:
  60.         case 32: m_wColors = 0;   break;   // 16,24 or 32 bpp have no color table
  61.         default:
  62.            m_wColors = 0;
  63.     } // switch
  64. /*
  65.     // Make sure bits per pixel is valid
  66.     if (wBitCount <= 1) wBitCount = 1;
  67.     else if (wBitCount <= 4) wBitCount = 4;
  68.     else if (wBitCount <= 8) wBitCount = 8;
  69.     else         wBitCount = 24;
  70.     switch (wBitCount)
  71. {
  72.         case 1:
  73.             m_wColors = 2;
  74. break;
  75.         case 4:
  76.             m_wColors = 16;
  77. break;
  78.         case 8:
  79.             m_wColors = 256;
  80. break;
  81.         default:
  82.             m_wColors = 0;
  83. break;
  84.     } // switch
  85. */
  86.     m_dwLineWidth = WIDTHBYTES(wBitCount * dwWidth);
  87.     // Initialize BITMAPINFOHEADER
  88.     m_bi.biSize = sizeof(BITMAPINFOHEADER);
  89.     m_bi.biWidth = dwWidth;         // fill in width from parameter
  90.     m_bi.biHeight = dwHeight;       // fill in height from parameter
  91.     m_bi.biPlanes = 1;              // must be 1
  92.     m_bi.biBitCount = wBitCount;    // from parameter
  93.     m_bi.biCompression = BI_RGB;    
  94.     m_bi.biSizeImage = m_dwLineWidth * dwHeight;
  95.     m_bi.biXPelsPerMeter = 0;
  96.     m_bi.biYPelsPerMeter = 0;
  97.     m_bi.biClrUsed = 0;
  98.     m_bi.biClrImportant = 0;
  99.     // Calculate size of memory block required to store the DIB.  This
  100.     // block should be big enough to hold the BITMAPINFOHEADER, the color
  101.     // table, and the bits.
  102.     dwLen = GetSize();
  103. m_hDib = new HDIB[dwLen]; // Allocate memory block to store our bitmap
  104.     if (m_hDib == NULL) return NULL;
  105.     // Use our bitmap info structure to fill in first part of
  106.     // our DIB with the BITMAPINFOHEADER
  107. lpbi = (LPBITMAPINFOHEADER)(m_hDib);
  108.     *lpbi = m_bi;
  109.     return m_hDib; // Return handle to the DIB
  110. } // End of Create
  111. DWORD CCeXDib::GetSize()
  112. {
  113. return m_bi.biSize + m_bi.biSizeImage + GetPaletteSize();
  114. } // End of GetSize
  115. DWORD CCeXDib::GetPaletteSize()
  116. {
  117. return (m_wColors * sizeof(RGBQUAD));
  118. } // End of GetPaletteSize
  119. LPBYTE CCeXDib::GetBits()
  120. {
  121. if (m_hDib)
  122. return ((LPBYTE)m_hDib + *(LPDWORD)m_hDib + GetPaletteSize()); 
  123. return NULL;
  124. } // End of GetBits
  125. DWORD CCeXDib::GetWidth()
  126. {
  127. return m_bi.biWidth;
  128. } // End of GetWidth
  129. DWORD CCeXDib::GetHeight()
  130. {
  131. return m_bi.biHeight;
  132. } // End of GetHeight
  133. DWORD CCeXDib::GetLineWidth()
  134. {
  135. return m_dwLineWidth;
  136. } // End of GetLineWidth
  137. void CCeXDib::BlendPalette(COLORREF crColor, DWORD dwPerc)
  138. {
  139. if (m_hDib == NULL || m_wColors == 0) 
  140. return;
  141. LPBYTE iDst = (LPBYTE)(m_hDib) + sizeof(BITMAPINFOHEADER);
  142. long i,r,g,b;
  143. RGBQUAD* pPal = (RGBQUAD*)iDst;
  144. r = GetRValue(crColor);
  145. g = GetGValue(crColor);
  146. b = GetBValue(crColor);
  147. if (dwPerc > 100) 
  148. dwPerc = 100;
  149. for (i = 0; i < m_wColors; i++)
  150. {
  151. pPal[i].rgbBlue = (BYTE)((pPal[i].rgbBlue * (100 - dwPerc) + b * dwPerc) / 100);
  152. pPal[i].rgbGreen = (BYTE)((pPal[i].rgbGreen * (100 - dwPerc) + g * dwPerc) / 100);
  153. pPal[i].rgbRed = (BYTE)((pPal[i].rgbRed * (100 - dwPerc) + r * dwPerc) / 100);
  154. } // for
  155. } // End of BlendPalette
  156. void CCeXDib::Clear(BYTE byVal)
  157. {
  158. if (m_hDib) 
  159. memset(GetBits(), byVal, m_bi.biSizeImage);
  160. } // End of Clear
  161. void CCeXDib::SetPixelIndex(DWORD dwX, DWORD dwY, BYTE byI)
  162. {
  163. if ((m_hDib == NULL) || (m_wColors == 0) ||
  164. ((long)dwX < 0) || ((long)dwY < 0) || (dwX >= (DWORD)m_bi.biWidth) || (dwY >= (DWORD)m_bi.biHeight)) return;
  165. LPBYTE iDst = GetBits();
  166. iDst[(m_bi.biHeight - dwY - 1) * m_dwLineWidth + dwX] = byI;
  167. } // End of SetPixelIndex
  168. void CCeXDib::Clone(CCeXDib* src)
  169. {
  170. Create(src->GetWidth(), src->GetHeight(), src->GetBitCount());
  171. if (m_hDib) 
  172. memcpy(m_hDib, src->m_hDib, GetSize());
  173. } // End of Clone
  174. WORD CCeXDib::GetBitCount()
  175. {
  176. return m_bi.biBitCount;
  177. } // End of GetBitCount
  178. void CCeXDib::SetPaletteIndex(BYTE byIdx, BYTE byR, BYTE byG, BYTE byB)
  179. {
  180. if (m_hDib && m_wColors)
  181. {
  182. LPBYTE iDst = (LPBYTE)(m_hDib) + sizeof(BITMAPINFOHEADER);
  183. if ((byIdx >= 0) && (byIdx < m_wColors))
  184. {
  185. long ldx = byIdx * sizeof(RGBQUAD);
  186. iDst[ldx++] = (BYTE)byB;
  187. iDst[ldx++] = (BYTE)byG;
  188. iDst[ldx++] = (BYTE)byR;
  189. iDst[ldx] = (BYTE)0;
  190. } // if
  191. } // if
  192. } // End of SetPaletteIndex
  193. void CCeXDib::Draw(HDC hDC, DWORD dwX, DWORD dwY)
  194. {
  195. HBITMAP hBitmap = NULL;
  196. HBITMAP hOldBitmap = NULL;
  197. HDC hMemDC = NULL;
  198. if (m_hBitmap == NULL)
  199. {
  200. m_hBitmap = CreateDIBSection(hDC, (BITMAPINFO*)m_hDib, DIB_RGB_COLORS, &m_lpBits, NULL, 0);
  201. if (m_hBitmap == NULL) return;
  202. if (m_lpBits == NULL)
  203. {
  204. ::DeleteObject(m_hBitmap);
  205. m_hBitmap = NULL;
  206. return;
  207. } // if
  208. } // if
  209.     memcpy(m_lpBits, GetBits(), m_bi.biSizeImage);
  210. if (m_hMemDC == NULL)
  211. {
  212. m_hMemDC = CreateCompatibleDC(hDC);
  213. if (m_hMemDC == NULL) return;
  214. } // if
  215. hOldBitmap = (HBITMAP)SelectObject(m_hMemDC, m_hBitmap);
  216. BitBlt(hDC, dwX, dwY, m_bi.biWidth, m_bi.biHeight, m_hMemDC, 0, 0, SRCCOPY);
  217. SelectObject(m_hMemDC, hOldBitmap);
  218. } // End of Draw
  219. void CCeXDib::Copy(HDC hDC, DWORD dwX, DWORD dwY)
  220. {
  221. if (m_hBitmap == NULL)
  222. {
  223. m_hBitmap = CreateDIBSection(hDC, (BITMAPINFO*)m_hDib, DIB_RGB_COLORS, &m_lpBits, NULL, 0);
  224. if (m_hBitmap == NULL)
  225. return;
  226. if (m_lpBits == NULL)
  227. {
  228. ::DeleteObject(m_hBitmap);
  229. m_hBitmap = NULL;
  230. return;
  231. } // if
  232. } // if
  233. HDC hMemDC = ::CreateCompatibleDC(hDC);
  234. HBITMAP hOldBitmap = (HBITMAP)::SelectObject(hMemDC, m_hBitmap);
  235. ::BitBlt(hMemDC, 0, 0, m_bi.biWidth, m_bi.biHeight, hDC, dwX, dwY, SRCCOPY);
  236. ::SelectObject(hMemDC, hOldBitmap);
  237. } // End of Copy
  238. void CCeXDib::SetGrayPalette()
  239. {
  240. RGBQUAD pal[256];
  241. RGBQUAD* ppal;
  242. LPBYTE iDst;
  243. int ni;
  244. if (m_hDib == NULL || m_wColors == 0) return;
  245. ppal = (RGBQUAD*)&pal[0];
  246. iDst = (LPBYTE)(m_hDib) + sizeof(BITMAPINFOHEADER);
  247. for (ni = 0; ni < m_wColors; ni++)
  248. {
  249. pal[ni] = RGB2RGBQUAD(RGB(ni,ni,ni));
  250. } // for
  251. pal[0] = RGB2RGBQUAD(RGB(0,0,0));
  252. pal[m_wColors-1] = RGB2RGBQUAD(RGB(255,255,255));
  253. memcpy(iDst, ppal, GetPaletteSize());
  254. } // End of SetGrayPalette
  255. RGBQUAD CCeXDib::RGB2RGBQUAD(COLORREF cr)
  256. {
  257. RGBQUAD c;
  258. c.rgbRed = GetRValue(cr); /* get R, G, and B out of DWORD */
  259. c.rgbGreen = GetGValue(cr);
  260. c.rgbBlue = GetBValue(cr);
  261. c.rgbReserved=0;
  262. return c;
  263. } // End of RGB2RGBQUAD
  264. WORD CCeXDib::GetNumColors()
  265. {
  266. return m_wColors;
  267. } // End of GetNumColors
  268. BOOL CCeXDib::WriteBMP(LPCTSTR bmpFileName)
  269. {
  270. BITMAPFILEHEADER hdr;
  271. HANDLE hFile;
  272. DWORD nByteWrite;
  273. if (*bmpFileName == _T('') || m_hDib == 0) return 0;
  274. hFile=CreateFile( // open if exist ini file
  275. bmpFileName, // pointer to name of the file 
  276. GENERIC_WRITE, // access mode 
  277. 0, // share mode 
  278. NULL, // pointer to security descriptor 
  279. CREATE_ALWAYS, // how to create 
  280. FILE_ATTRIBUTE_NORMAL, // file attributes 
  281. NULL   // handle to file with attributes to copy  
  282. );
  283. if (hFile == INVALID_HANDLE_VALUE) return FALSE;
  284.     // Fill in the fields of the file header
  285. hdr.bfType = BFT_BITMAP;
  286. hdr.bfSize = GetSize() + sizeof(BITMAPFILEHEADER);
  287. hdr.bfReserved1 = hdr.bfReserved2 = 0;
  288. hdr.bfOffBits = (DWORD) sizeof(BITMAPFILEHEADER)+
  289. m_bi.biSize + GetPaletteSize();
  290.     // Write the file header
  291. WriteFile( // write ini (sync mode <-> no overlapped)
  292. hFile, // handle of file to write 
  293. (LPSTR) &hdr, // address of buffer that contains data  
  294. sizeof(BITMAPFILEHEADER), // number of bytes to write 
  295. &nByteWrite, // address of number of bytes written 
  296. NULL   // address of structure for data 
  297. );
  298.     // Write the DIB header and the bits
  299. WriteFile( // write ini (sync mode <-> no overlapped)
  300. hFile, // handle of file to write 
  301. (LPSTR) m_hDib, // address of buffer that contains data  
  302. GetSize(), // number of bytes to write 
  303. &nByteWrite, // address of number of bytes written 
  304. NULL   // address of structure for data 
  305. );
  306. CloseHandle(hFile); // free file handle
  307. return TRUE;
  308. } // End of WriteBMP