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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 行业应用 > 交通/航空行业 > 查看源码
Hdib.cpp
资源名称：vehiclehsv.rar　[点击查看]
上传用户：cjd055
上传日期：2013-04-01
资源大小：608k
文件大小：13k
源码类别：

交通/航空行业

开发平台：

Visual C++

Hdib.cpp：源码内容
  1. // HDib.cpp: implementation of the CDib class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #include "stdafx.h"
  6. #include "HDib.h"
  8. #ifdef _DEBUG
  9. #undef THIS_FILE
  10. static char THIS_FILE[]=__FILE__;
  11. #endif
  13. //////////////////////////////////////////////////////////////////////
  14. // Construction/Destruction
  15. //////////////////////////////////////////////////////////////////////
  17. CDib::CDib()
  18. {
  19. m_BmpLoaded=FALSE;
  20. m_Buffer=0;
  21. m_Buffer1=0;
  22. m_BmpInfo=0;
  23. m_Quad=0;
  24. m_hPal=0;
  25. BmpMarker=('M'<<8)|'B';
  26. Width=0;
  27. Height=0;
  28. }
  30. CDib::~CDib()
  31. {
  32. Free();
  33. }
  35. //释放所有动态分配的内存：
  36. void CDib::Free()
  37. {
  38. if(m_Buffer)
  39. {
  40. delete [] m_Buffer;
  41. m_Buffer=0;
  42. }
  43. if(m_Buffer1)
  44. {
  45. delete [] m_Buffer1;
  46. m_Buffer1=0;
  47. }
  48. if(m_BmpInfo)
  49. {
  50. delete [] m_BmpInfo;
  51. m_BmpInfo=0;
  52. }
  53. if(m_Quad)
  54. {
  55. delete [] m_Quad;
  56. m_Quad=0;
  57. }
  58. }
  60. void CDib::Resize(int sz)
  61. {
  62. if(m_Buffer)
  63. {
  64. delete [] m_Buffer;
  65. m_Buffer=0;
  66. }
  67. if(m_Buffer1)
  68. {
  69. delete [] m_Buffer1;
  70. m_Buffer1=0;
  71. }
  72. m_Buffer=new BYTE[sz];
  73. m_Buffer1=new BYTE[sz];
  74. //位图每一行占据的字节数:
  75. Bpl=(DWORD)WIDTHBYTES(BPP*Width); //一种使bpl为4的整数倍的技巧
  76. //位图数据区尺寸（字节）:
  77. BufferSize=Bpl*Height;
  78. }
  80. //从文件中读取位图：
  81. BOOL CDib::LoadBmp(LPCTSTR fn)
  82. {
  83. int j;
  84. if(fn=="")
  85. {
  86. return FALSE;
  87. }
  88. Free();
  89.     arcFileName=fn;
  90. //打开bmp文件：
  91. ifstream files(fn,ios::binary|ios::nocreate);
  92. if(!files.is_open())
  93. {
  94. return FALSE;
  95. }
  97. //读入BITMAPFILEHEADER：
  98. files.read((LPSTR)&m_BmpFH, sizeof(BITMAPFILEHEADER));
  100. //判断是否是bmp文件:
  101. if (m_BmpFH.bfType != BmpMarker)
  102. {
  103. return FALSE;
  104. }
  106. //读入BITMAPINFO:
  107. DWORD bmpinfosz;
  108. bmpinfosz=m_BmpFH.bfOffBits-sizeof(BITMAPFILEHEADER) 
  109.       +256*sizeof(RGBQUAD);  //40+256*4
  110. m_BmpInfo=(LPBITMAPINFO) new BYTE[bmpinfosz];
  111. files.read((char *)m_BmpInfo,m_BmpFH.bfOffBits-sizeof(BITMAPFILEHEADER)); //m_BmpFH.bfOffBits-sizeof(BITMAPFILEHEADER)=40
  113. //计算位图的相关数据信息：
  114. CalBmpData();
  115. //只支持8位以上的位图：
  116. if(BPP>32)
  117. {
  118. Free();
  119. return FALSE;
  120. }
  121. //读入位图数据：
  122. m_Buffer=new BYTE[BufferSize];
  123. m_Buffer1=new BYTE[BufferSize];
  124. files.read((char *)m_Buffer,BufferSize);
  125. if(m_BmpInfo->bmiHeader.biCompression!=BI_RGB)
  126. {
  127. Free();
  128. return FALSE;
  129. }
  130. if(!CreatePalette())
  131. {
  132. Free();
  133. return FALSE;
  134. }
  135.     for ( j=0;j<(int)BufferSize;j++)
  136. {
  137.     *(m_Buffer1+j)=*(m_Buffer+j);
  138. }
  139. m_BmpLoaded=TRUE;
  140. return TRUE;
  141. }
  143. //将内存中的位图存入文件：
  144. BOOL CDib::SaveBmp(LPCTSTR fn)
  145. {
  146. if(fn=="")
  147. {
  148. return FALSE;
  149. }
  150. if(!m_BmpLoaded)
  151. {
  152. return FALSE;
  153. }
  154. if(m_BmpInfo==0||m_Buffer==0)
  155. {
  156. return FALSE;
  157. }
  159. //1. 更改m_BmpInfo->bmiHeader.biSizeImage:
  160. //1. biSizeImage=位图数据的大小：
  161. m_BmpInfo->bmiHeader.biSizeImage=BufferSize;
  163. //2. 填充bmpfh结构；
  164. BITMAPFILEHEADER bmpfh;
  165. memset(&bmpfh,0,sizeof(BITMAPFILEHEADER));
  166. bmpfh.bfType=BmpMarker;
  167. //bmpfh.bfSize=BITMAPFILEHEADER+BITMAPINFO+调色板+BufferSize:
  168. bmpfh.bfSize=sizeof(BITMAPFILEHEADER)+
  169.        sizeof(BITMAPINFO)+
  170.            (NumberOfColors-1)*sizeof(RGBQUAD)+
  171.    BufferSize;
  173. bmpfh.bfReserved1=0;
  174. bmpfh.bfReserved2=0;
  175. bmpfh.bfOffBits=(DWORD)sizeof(BITMAPFILEHEADER)+
  176.               m_BmpInfo->bmiHeader.biSize+
  177.   PaletteSize();
  178. //3. 写文件：
  179. FILE * fp;
  180. size_t sz;
  181. if((fp=fopen(fn,"wb"))==NULL)
  182. {
  183. return FALSE;
  184. }
  185. //1). 写入BITMAPFILEHEADER:
  186. sz=fwrite((void *)&bmpfh,sizeof(BITMAPFILEHEADER),1,fp);
  187. if(sz!=1)
  188. {
  189. fclose(fp);
  190. return FALSE;
  191. }
  192. //2). 写入BITMAPINFO:
  193. sz=fwrite((void *)m_BmpInfo,sizeof(BITMAPINFO)+(NumberOfColors-1)*sizeof(RGBQUAD),1,fp);
  194. if(sz!=1)
  195. {
  196. fclose(fp);
  197. return FALSE;
  198. }
  199. //3). 写入位图数据：
  200. sz=fwrite(m_Buffer,1,BufferSize,fp);
  201. if(sz!=BufferSize)
  202. {
  203. fclose(fp);
  204. return FALSE;
  205. }
  206. fclose(fp);
  207. return TRUE;
  208. }
  210. //获取位图内存地址：
  211. BYTE * CDib::GetBuffer()
  212. {
  213. return m_Buffer;
  214. }
  216. //根据像素在位图中的位置求像素在内存中的地址：
  217. BYTE * CDib::FindPos(DWORD x, DWORD y)
  218. {
  219. BYTE * pos;
  220. if(!m_BmpLoaded)
  221. {
  222. return 0;
  223. }
  224. pos=m_Buffer+Bpl*(Height-1-y)+x*BPP/8;
  225. return pos;
  226. }
  228. //将位图的子画面存入文件：
  229. BOOL CDib::PickBitmapToFile(LPCTSTR fn, DWORD xt, DWORD yt, 
  230. DWORD xb, DWORD yb)
  231. {
  232. BYTE * buf;
  233. DWORD w,h,bpl,bufsize,n0;
  234. if(!m_BmpLoaded)
  235. {
  236. return FALSE;
  237. }
  239. if(m_BmpInfo==0||m_Buffer==0)
  240. {
  241. return FALSE;
  242. }
  244. if(fn=="")
  245. {
  246. return FALSE;
  247. }
  249. //1. 计算基本参数：
  250. Sort(xt,xb);
  251. Sort(yt,yb);
  252. if(xt>Width)
  253. {
  254. return FALSE;
  255. }
  256. if(yt>Width)
  257. {
  258. return FALSE;
  259. }
  260. if(xb>Width)
  261. {
  262. xb=Width;
  263. }
  264. if(yb>Height)
  265. {
  266. yb=Height;
  267. }
  268. w=xb-xt;//子画面宽度；
  269. h=yb-yt;//子画面高度；
  270. bpl=(w*(BPP/8)+3)&~3;//每一行大小；
  271. n0=bpl-w*BPP/8;//需要在每一行末尾添加的0的数目；
  272. bufsize=bpl*h;//位图数据区大小；
  274. //2. 填写位图区：
  275. BYTE * pos,* bufpos;
  276. pos=FindPos(xt,yt);//找到(xt,yt)像素在内存中的位置；
  277. buf=new BYTE[bufsize];
  278. memset((void *)buf,0,bufsize);
  279. UINT i;
  280. bufpos=buf+bpl*(h-1);
  281. for(i=0;i<h;i++)
  282. {
  283. memcpy(bufpos,pos,w*BPP/8);
  284. bufpos-=bpl;
  285. pos-=Bpl;
  286. }
  288. //3. 填写位图结构：
  289. BITMAPFILEHEADER fh;
  290. BITMAPINFO       info;
  291. memset(&fh  ,0,sizeof(BITMAPFILEHEADER));
  292. memset(&info,0,sizeof(BITMAPINFO));
  293. //1. 填充BITMAPINFO结构:
  294. info.bmiHeader.biBitCount=(WORD)BPP;
  295. info.bmiHeader.biClrImportant=m_BmpInfo->bmiHeader.biClrImportant;
  296. info.bmiHeader.biClrUsed=m_BmpInfo->bmiHeader.biClrUsed;
  297. info.bmiHeader.biCompression=BI_RGB;
  298. info.bmiHeader.biHeight=h;
  299. info.bmiHeader.biPlanes=m_BmpInfo->bmiHeader.biPlanes;
  300. info.bmiHeader.biSize=m_BmpInfo->bmiHeader.biSize;
  301. info.bmiHeader.biSizeImage=bufsize;
  302. info.bmiHeader.biWidth=w;
  303. info.bmiHeader.biXPelsPerMeter=m_BmpInfo->bmiHeader.biXPelsPerMeter;
  304. info.bmiHeader.biYPelsPerMeter=m_BmpInfo->bmiHeader.biYPelsPerMeter;
  305. info.bmiColors[0]=m_BmpInfo->bmiColors[0];
  306. //2. 填充bmpfh结构；
  307. fh.bfType=BmpMarker;
  308. //bmpfh.bfSize=BITMAPFILEHEADER+BITMAPINFO+调色板+BufferSize:
  309. fh.bfSize=sizeof(BITMAPFILEHEADER)+
  310.        sizeof(BITMAPINFO)+
  311.            (NumberOfColors-1)*sizeof(RGBQUAD)+
  312.    bufsize;
  314. fh.bfReserved1=0;
  315. fh.bfReserved2=0;
  316. fh.bfOffBits=(DWORD)sizeof(BITMAPFILEHEADER)+
  317.               m_BmpInfo->bmiHeader.biSize+
  318.   PaletteSize();
  319. //4. 写文件：
  320. FILE * fp;
  321. if((fp=fopen(fn,"wb"))==NULL)
  322. {
  323. return FALSE;
  324. }
  325. //1). 写入BITMAPFILEHEADER:
  326. fwrite((char *)&fh,sizeof(BITMAPFILEHEADER),1,fp);
  327. //2). 写入BITMAPINFOHEADER部分:
  328. fwrite((char *)&info.bmiHeader,sizeof(BITMAPINFOHEADER),1,fp);
  329. //3). //写入彩色表部分：
  330. if(BPP<16)
  331. {
  332. fwrite((char *)&m_Quad[0],NumberOfColors*sizeof(RGBQUAD),1,fp);
  333. }
  334. //4). 写入位图数据：
  335. fwrite((char *)buf,bufsize,1,fp);
  336. delete buf;
  337. fclose(fp);
  338. return TRUE;
  339. }
  341. void CDib::CalBmpData()
  342. {
  343. if(m_BmpInfo==0)
  344. {
  345. return;
  346. }
  347. //1. 位图宽度（像素）:
  348. Width=m_BmpInfo->bmiHeader.biWidth;
  349. //位图高度（像素）:
  350. Height=m_BmpInfo->bmiHeader.biHeight;
  351. //位图色彩深度:
  352. BPP=m_BmpInfo->bmiHeader.biBitCount; //取值可为1,4,8，16,24,32，如取值16,24,32，则彩色表不是必须的
  353. //位图每一行占据的字节数:
  354. Bpl=(DWORD)WIDTHBYTES(BPP*Width); //一种使bpl为4的整数倍的技巧
  355. //位图数据区尺寸（字节）:
  356. BufferSize=Bpl*Height;
  357. //为m_Quad分配内存：
  358. //如果色彩数目大于8 则返回:
  359. if(m_BmpInfo->bmiHeader.biBitCount>=16)
  360. {
  361. NumberOfColors=0;
  362. return;
  363. }
  365. if(m_BmpInfo->bmiHeader.biClrUsed!=0)
  366. {
  367. NumberOfColors=m_BmpInfo->bmiHeader.biClrUsed;
  368. }
  369. else
  370. {
  371. switch(BPP){
  372. case 1:
  373. NumberOfColors=2;
  374. break;
  375. case 4:
  376. NumberOfColors=16;
  377. break;
  378. case 8:
  379. NumberOfColors=256;
  380. break;
  381. default:
  382. return;
  383. }
  384. }
  385. //读取 logical palette
  386. m_Quad=new RGBQUAD[NumberOfColors];
  387. for(DWORD i=0;i<NumberOfColors;i++)
  388. {
  389. m_Quad[i]=m_BmpInfo->bmiColors[i];
  390. }
  391. }
  393. void CDib::SetBmpMarker(DWORD b, DWORD m)
  394. {
  395. BmpMarker=(WORD)((m<<8)|b); //'m'左移8位 and 'b' 再强制类型转换
  396. }
  398. DWORD CDib::PaletteSize()
  399. {
  400. if (m_BmpInfo==0)
  401. {
  402. return 0;
  403. }
  404. return (DWORD)NumberOfColors*sizeof(RGBQUAD);//256*4
  405. }
  406. //using bmp file's QUAD to change phisical palette
  407. BOOL CDib::CreatePalette()
  408. {
  409. if (m_BmpInfo==0)
  410. {
  411. return FALSE;
  412. }
  413. //16位以下的位图均需要 logical palette
  414. if(BPP>=16)
  415. {
  416. return TRUE;
  417. }
  419. DWORD i;
  420. // allocate memory block for 逻辑彩色区logical palette
  421. LPLOGPALETTE lpPal = (LPLOGPALETTE) new BYTE[sizeof(LOGPALETTE) + sizeof(PALETTEENTRY)*NumberOfColors]; //sizeof(PALETTEENTRY)*NumberOfColors=4*256 bytes
  422. // if not enough memory, clean up and return NULL
  423. if (lpPal == 0)
  424. {
  425. return FALSE;
  426. }
  428. // set version and number of palette entries
  429. lpPal->palVersion = PALVERSION;
  430. lpPal->palNumEntries = (WORD)NumberOfColors;
  431. for (i=0;i<NumberOfColors;i++)
  432. {
  433. lpPal->palPalEntry[i].peRed   = m_Quad[i].rgbRed;
  434. lpPal->palPalEntry[i].peGreen = m_Quad[i].rgbGreen;
  435. lpPal->palPalEntry[i].peBlue  = m_Quad[i].rgbBlue;
  436. lpPal->palPalEntry[i].peFlags = 0;
  437. }
  439. /* according to bmp file palette table,create the palette and 
  440. get handle to it */
  441. if (m_hPal)
  442. {
  443. ::DeleteObject((HGDIOBJ)m_hPal); //删除老的逻辑调色对象
  444. }
  445. m_hPal=::CreatePalette(lpPal); //创建新的逻辑调色对象
  446. if(!m_hPal)
  447. {
  448. return FALSE;
  449. }
  451. delete [] lpPal; //作用完即destroy logical Palette
  452. return TRUE;
  453. }
  455. BOOL CDib::Draw(HDC hDC, LPRECT rcDest, LPRECT rcSrc)
  456. {
  457. if (!m_BmpLoaded)
  458. {
  459. return FALSE;
  460. }
  462. HPALETTE hOldPal = 0;        // Previous palette
  464. // Get the DIB's palette, then select it into DC
  465. if (m_hPal!=0)
  466. {
  467. // Select as background since we have
  468. // already realized in forground if needed
  469. hOldPal = ::SelectPalette(hDC,m_hPal,TRUE); //selects the specified logical palette into a device context
  470. }
  472. /* Make sure to use the stretching mode best for color pictures */
  473. ::SetStretchBltMode(hDC, COLORONCOLOR);
  475. /* Determine whether to call StretchDIBits() or SetDIBitsToDevice() */
  476. BOOL Ok;
  477. if ((RECTWIDTH(rcDest)  == RECTWIDTH(rcSrc)) &&
  478.     (RECTHEIGHT(rcDest) == RECTHEIGHT(rcSrc)))
  479. {
  480. Ok = ::SetDIBitsToDevice(hDC,                                 // hDC
  481.  rcDest->left,                        // DestX
  482.  rcDest->top,                         // DestY
  483.  RECTWIDTH(rcDest),                   // nDestWidth
  484.  RECTHEIGHT(rcDest),                  // nDestHeight
  485.  rcSrc->left,                         // SrcX
  486.  Height-rcSrc->top -RECTHEIGHT(rcSrc),// SrcY
  487.  0,                                   // nStartScan
  488.  (WORD)Height,                        // nNumScans
  489.  m_Buffer,                            // lpBits
  490.  m_BmpInfo,                           // lpBitsInfo
  491.  DIB_RGB_COLORS);                     // wUsage
  492. }
  493.    else
  494.    {
  495.   Ok = ::StretchDIBits(hDC,                            // hDC
  496.    rcDest->left,               // DestX
  497.    rcDest->top,                // DestY
  498.    RECTWIDTH(rcDest),          // nDestWidth
  499.    RECTHEIGHT(rcDest),         // nDestHeight
  500.    rcSrc->left,                // SrcX
  501.    rcSrc->top,                 // SrcY
  502.    RECTWIDTH(rcSrc),           // wSrcWidth
  503.    RECTHEIGHT(rcSrc),          // wSrcHeight
  504.    m_Buffer,                   // lpBits
  505.    m_BmpInfo,                   // lpBitsInfo
  506.    DIB_RGB_COLORS,             // wUsage
  507.    SRCCOPY);                   // dwROP
  508.    }
  509. /* Reselect old palette */
  510. if (hOldPal != 0)
  511. {
  512. ::SelectPalette(hDC, hOldPal, TRUE); //恢复老的调色板
  513. }
  514.    return Ok;
  515. }
  517. void CDib::Sort(DWORD &x0, DWORD &x1)
  518. {
  519. DWORD t;
  520. if(x0>x1)
  521. {
  522. t=x1;
  523. x1=x0;
  524. x0=t;
  525. }
  526. }
  528. BITMAPINFOHEADER CDib::GetImageInfo(LPCTSTR fn, DWORD &w, DWORD &h)
  529. {
  530. BITMAPFILEHEADER bmpfilehdr;
  531. BITMAPINFOHEADER bmpinfohdr;
  532. w=0;
  533. h=0;
  534. memset(&bmpfilehdr,0,sizeof(BITMAPFILEHEADER)); //初始化内存并得到内存指针；
  535. memset(&bmpinfohdr,0,sizeof(BITMAPINFOHEADER));
  537. ifstream file( fn, ios::binary | ios::nocreate ); //打开已存在的且二进制的文件；
  538. if (!file.is_open())
  539. {
  540. return bmpinfohdr;
  541. }
  543. file.read( (char*)&bmpfilehdr, sizeof(bmpfilehdr) );
  545. if (bmpfilehdr.bfType!=(WORD)BmpMarker)
  546. {
  547. return bmpinfohdr;
  548. }
  550. file.read( (char*)&bmpinfohdr, sizeof(bmpinfohdr) );
  552. w=bmpinfohdr.biWidth;
  553. h=bmpinfohdr.biHeight;
  555. return bmpinfohdr;
  556. }
  558. void CDib::restoreimage()
  559. {
  560. int j;
  561.     for ( j=0;j<(int)BufferSize;j++)
  562. {
  563.     *(m_Buffer+j)=*(m_Buffer1+j);
  564. }
  565. }
  567. BOOL CDib::GetBlueImg()
  568. {
  569.     if (BPP<=8)
  570.    return FALSE;
  571. int i,j;
  572. for(i=0;i<(int)Height;i++)
  573. {
  574. for ( j=0;j<(int)Bpl-2;j=j+3)
  575. {
  576.     *(m_Buffer+i*Bpl+j+1)=0;
  577. *(m_Buffer+i*Bpl+j+2)=0;
  578. }
  579. }
  580.     return TRUE;
  581. }
  584. BOOL CDib::GetGreenImg()
  585. {
  586.     if (BPP<=8)
  587.    return FALSE;
  588. int i,j;
  589. for(i=0;i<(int)Height;i++)
  590. {
  591. for ( j=0;j<(int)Bpl-2;j=j+3)
  592. {
  593.     *(m_Buffer+i*Bpl+j)=0;
  594. *(m_Buffer+i*Bpl+j+2)=0;
  595. }
  596. }
  597.     return TRUE;
  598. }
  600. BOOL CDib::GetRedImg()
  601. {
  602.     if (BPP<=8)
  603.    return FALSE;
  604. int i,j;
  605. for(i=0;i<(int)Height;i++)
  606. {
  607. for ( j=0;j<(int)Bpl-2;j=j+3)
  608. {
  609.     *(m_Buffer+i*Bpl+j)=0;
  610. *(m_Buffer+i*Bpl+j+1)=0;
  611. }
  612. }
  613.     return TRUE;
  614. }