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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 中间件编程 > 查看源码
Dib.cpp
资源名称：08.zip　[点击查看]
上传用户：ynjin1970
上传日期：2014-10-13
资源大小：6438k
文件大小：9k
源码类别：

中间件编程

开发平台：

Visual C++

Dib.cpp：源码内容
  1. // Dib.cpp: implementation of the CDib class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #include "stdafx.h"
  6. #include "Cloud.h"
  7. #include "Dib.h"
  9. #include "math.h"
  11. #ifdef _DEBUG
  12. #undef THIS_FILE
  13. static char THIS_FILE[]=__FILE__;
  14. #define new DEBUG_NEW
  15. #endif
  17. #define PI  3.14159265358979
  18. //////////////////////////////////////////////////////////////////////
  19. // Construction/Destruction
  20. //////////////////////////////////////////////////////////////////////
  22. CDib::CDib()
  23. {
  24. m_Height=0;
  25. m_Width=0;
  27. hDIB=NULL;
  28. bmInfoHeader=NULL;
  29. bmpInfo =NULL;
  30. lpDIBBits=NULL;
  33. }
  35. CDib::~CDib()
  36. {
  37. ClearMemory();
  38. }
  39. /////////////////////////////////////////////////////////////////////////
  40. //打开并读取图像文件
  41. void CDib::ReadDIBFile(LPTSTR lpFilename)
  42. {
  44. BITMAPFILEHEADER bmfHeader;//图像文件头
  45. CFile File;
  46. CFileException Err;
  47. if(!File.Open(lpFilename,CFile::modeRead|CFile::shareDenyNone,&Err))
  48. {
  49. AfxMessageBox(Err.m_cause);
  50. return;
  51. }
  54. File.Read(&bmfHeader,sizeof(bmfHeader));
  55. if(bmfHeader.bfType!=0x4d42)
  56. {
  57. AfxMessageBox("此文件不是位图文件");
  58. return;
  59. }
  62. DWORD Filelen=File.GetLength();
  63. if(bmfHeader.bfSize!=Filelen)
  64. {
  65. AfxMessageBox("该文件已经损坏！");
  66. return;
  67. }
  68. hDIB=NULL;
  69. hDIB=new BYTE[Filelen-sizeof(BITMAPFILEHEADER)];
  70. if(!hDIB)
  71. {
  72. AfxMessageBox("内存不够！");
  73. return;
  75. }
  76. File.Read(hDIB,Filelen-sizeof(BITMAPFILEHEADER));
  77. File.Close();
  79. }
  81. /////////////////////////////////////////////////////////////////////////
  82. //装载DIB
  83. void CDib::LoadDIB(LPTSTR lpFilename)
  84. {
  85. ReadDIBFile(lpFilename);
  87. bmInfoHeader=(LPBITMAPINFOHEADER)hDIB;
  88. bmpInfo=(LPBITMAPINFO)hDIB;
  90. int ColorsNum=GetDIBColorsNum(bmInfoHeader);
  91. lpDIBBits=(BYTE*)hDIB+ColorsNum*sizeof(RGBQUAD)+sizeof(BITMAPINFOHEADER);
  93. m_Height=bmInfoHeader->biHeight;
  94. m_Width=bmInfoHeader->biWidth;
  97. }
  99. long CDib::GetImageHeight()
  100. {
  101. return m_Height;
  102. }
  104. long CDib::GetImageWidth()
  105. {
  106. return m_Width;
  107. }
  109. /////////////////////////////////////////////////////////////////////////
  110. //从得到图像颜色数
  111. int CDib::GetDIBColorsNum(BITMAPINFOHEADER* bmInfoHeader)
  112. {
  113. int ColorsNum;//图像颜色数
  115. switch(bmInfoHeader->biBitCount)
  116. {
  117. case 1:
  118. ColorsNum=2;
  119. break;
  120. case 4:
  121. ColorsNum=16;
  122. break;
  123. case 8:
  124. ColorsNum=256;
  125. break;
  126. default:
  127. ColorsNum=0;
  128. break;
  129. }
  131. return ColorsNum;
  132. }
  133. /////////////////////////////////////////////////////////////////////////
  134. //把DIB选入到指定设备hDC
  135. BOOL CDib::PaintDIB(HDC hDC,long width,long height)
  136. {
  138. int bSuccess;
  139. HPALETTE hPal=NULL;
  140. HPALETTE hOldPal;
  141. if(!hDIB)
  142. return FALSE;
  144. if(!hPal)
  145. CreateDIBPalette();
  147. if(hPal)
  148. {
  149. hOldPal=SelectPalette(hDC,hPal,TRUE);
  150. RealizePalette(hDC);
  151. }
  153. SetStretchBltMode(hDC,COLORONCOLOR);
  155. bSuccess=StretchDIBits(hDC,0,0,width,
  156.                    height,0,0,m_Width,m_Height,lpDIBBits, 
  157.                        bmpInfo,DIB_RGB_COLORS,SRCCOPY);
  160. if(hOldPal)
  161. SelectPalette(hDC,hOldPal,FALSE);
  162. return bSuccess;
  163. }
  165. /////////////////////////////////////////////////////////////////////////
  166. //生成自己的调色板
  167. HPALETTE CDib::CreateDIBPalette()
  168. {
  169. LOGPALETTE* lpPal;
  170. HPALETTE hPal=NULL;
  171. int wNumColors=GetDIBColorsNum(bmInfoHeader);
  173. lpPal=(LOGPALETTE*)new BYTE[wNumColors*sizeof(PALETTEENTRY)+2*sizeof(WORD)];
  174. lpPal->palNumEntries=wNumColors;
  175. lpPal->palVersion=1;
  176. for(int i=0;i<wNumColors;i++)
  177. {
  178.      lpPal->palPalEntry[i].peBlue=bmpInfo->bmiColors[i].rgbBlue;
  179.      lpPal->palPalEntry[i].peGreen=bmpInfo->bmiColors[i].rgbGreen;
  180.      lpPal->palPalEntry[i].peRed=bmpInfo->bmiColors[i].rgbRed;
  181.      lpPal->palPalEntry[i].peFlags=bmpInfo->bmiColors[i].rgbReserved=(BYTE)0;
  182. }
  183. hPal=CreatePalette(lpPal);
  184. return hPal;
  185. }
  187. /////////////////////////////////////////////////////////////////////////
  188. //释放内存
  189. void CDib::ClearMemory()
  190. {
  191. delete[] hDIB;
  192. delete[] bmInfoHeader;
  193. delete[] bmpInfo;
  194. delete[] lpDIBBits;
  197. }
  198. /////////////////////////////////////////////////////////////////////////
  199. //获得DIB数据指针
  201. HANDLE CDib::GetDIBHandle()
  202. {
  203. return hDIB;
  204. }
  206. /////////////////////////////////////////////////////////////////////////
  207. //从DIB获得DDB
  212. //////////////////////////////////////////
  213. //////////图像灰度线性变换函数
  214. BOOL CDib::OnChangeBrightness(float fa,float fb)
  215. {
  216. if(!lpDIBBits)
  217. {
  218. AfxMessageBox("请先打开图像");
  219. return FALSE;
  220. }
  222. BYTE* temDIBits;
  223. if(bmInfoHeader->biBitCount!=24)
  224. {
  225. AfxMessageBox("只能对24位真彩色的图像进行变换");
  226. return FALSE;
  227. }
  228. int w=m_Width;
  229. int h=m_Height;
  230. int sign=w%2;
  231. long total=w*h*3;
  232. long k=0;
  233. if(sign)
  234. w+=1;
  235. temDIBits=new BYTE[total];
  236. memcpy(temDIBits,lpDIBBits,total);
  237. for(int i=0;i<h;i++)
  238. {
  239. for(int j=0;j<w;j++)
  240. {
  241. float fTem;
  243. fTem=fa*temDIBits[k]+fb;//红色通道
  244. if(fTem>255)
  245. {
  246. lpDIBBits[k]=(BYTE)255;
  247. }else
  248. if(fTem<0)
  249. {
  250. lpDIBBits[k]=(BYTE)0;
  251. }else
  252. {
  253. lpDIBBits[k]=(BYTE)(fTem+0.5);
  254. }
  257. fTem=fa*temDIBits[k+1]+fb;//绿色通道
  258. if(fTem>255)
  259. {
  260. lpDIBBits[k+1]=(BYTE)255;
  261. }else
  262. if(fTem<0)
  263. {
  264. lpDIBBits[k+1]=(BYTE)0;
  265. }else
  266. {
  267. lpDIBBits[k+1]=(BYTE)(fTem+0.5);
  268. }
  271. fTem=fa*temDIBits[k+2]+fb;//蓝色通道
  272. if(fTem>255)
  273. {
  274. lpDIBBits[k+2]=(BYTE)255;
  275. }else
  276. if(fTem<0)
  277. {
  278. lpDIBBits[k+2]=(BYTE)0;
  279. }else
  280. {
  281. lpDIBBits[k+2]=(BYTE)(fTem+0.5);
  282. }
  285. if(k>=total)
  286. break;
  287. k+=3;
  288. }
  289. }
  292. return TRUE;
  293. }
  296. /////////////////////////////////////////////////////////////////
  297. //////////////////灰度拉伸函数
  298. BOOL CDib::OnGrayStretch(BYTE x1,BYTE x2,BYTE y1,BYTE y2)
  299. {
  300. if(!lpDIBBits)
  301. {
  302. AfxMessageBox("请先打开图像");
  303. return FALSE;
  304. }
  306. if(bmInfoHeader->biBitCount!=24)
  307. {
  308. AfxMessageBox("只能对24位真彩色的图像进行变换");
  309. return FALSE;
  310. }
  312. int w=m_Width;
  313. int h=m_Height;
  314. int sign=w%2;
  315. long total=w*h*3;
  316. long k=0;
  317. if(sign)
  318. w+=1;
  320. BYTE A=(y2-y1)/(x2-x1);
  321. for(int i=0;i<h;i++)
  322. {
  323. for(int j=0;j<w;j++)
  324. {
  325. float fTem;
  326. if(lpDIBBits[k]>=x1&&lpDIBBits[k]<=x2)
  327. {
  328. fTem=lpDIBBits[k];
  329. lpDIBBits[k]=(BYTE)fTem*A+y1;
  330. lpDIBBits[k+1]=(BYTE)fTem*A+y1;
  331. lpDIBBits[k+2]=(BYTE)fTem*A+y1;
  332. }else if(lpDIBBits[k]<x1)
  333. {
  334. lpDIBBits[k]=y1;
  335. lpDIBBits[k+1]=y1;
  336. lpDIBBits[k+2]=y1;
  337. }else
  338. {
  339. lpDIBBits[k]=y2;
  340. lpDIBBits[k+1]=y2;
  341. lpDIBBits[k+2]=y2;
  343. }
  344. if(k>=total)
  345. break;
  346. k+=3;
  348. }
  349. }
  352. return TRUE;
  353. }
  354. /////////////////////////////////////////////////////////////////
  355. //////////////////灰度化图像函数
  356. BOOL CDib::OnChangeToGray()
  357. {
  358. if(!lpDIBBits)
  359. {
  360. AfxMessageBox("请先打开图像");
  361. return FALSE;
  362. }
  364. if(bmInfoHeader->biBitCount!=24)
  365. {
  366. AfxMessageBox("只能对24位真彩色的图像进行变换");
  367. return FALSE;
  368. }
  369. int w=m_Width;
  370. int h=m_Height;
  371. int sign=w%2;
  372. long total=w*h*3;
  373. long k=0;
  374. if(sign)
  375. w+=1;
  377. for(int i=0;i<h;i++)
  378. {
  379. for(int j=0;j<w;j++)
  380. {
  381. float fTem;
  382. fTem=(float)(lpDIBBits[k]+lpDIBBits[k+1]+lpDIBBits[k+2])/3;//灰度值为三色RGB的平均值
  383. lpDIBBits[k]=(BYTE)fTem;
  384. lpDIBBits[k+1]=(BYTE)fTem;
  385. lpDIBBits[k+2]=(BYTE)fTem;
  387. if(k>=total)
  388. break;
  389. k+=3;
  390. }
  391. }
  394. return TRUE;
  395. }
  396. ////////////////////////////////////////////////////////
  397. //直方图均衡化函数
  398. void CDib::OnHistogramEqualize()
  399. {
  400. if(!lpDIBBits)
  401. {
  402. AfxMessageBox("请先打开图像");
  403. return;
  404. }
  405. if(bmInfoHeader->biBitCount!=24)
  406. {
  407. AfxMessageBox("只能对24位真彩色的图像进行变换");
  408. return;
  409. }
  412. OnChangeToGray();//灰度化图像
  414. int w=m_Width;
  415. int h=m_Height;
  416. int sign=w%2;
  417. long total=w*h*3;
  418. long k=0;//像素lpBITBits的下标
  420. long Temp;
  421. BYTE bMap[256];//新的灰度表
  422. long iCount[256];//图像0－255每种灰度的点数
  423. for(int i=0;i<256;i++)
  424. {
  425. iCount[i]=0;
  426. }
  428. if(sign)
  429. w+=1;
  431. for(i=0;i<h;i++)//统计每一级灰度的点的数量
  432. {
  433. for(int j=0;j<w;j++)
  434. {
  435. iCount[lpDIBBits[k]]++;
  437. if(k>=total)
  438. break;
  439. k+=3;
  440. }
  441. }
  443. for(i=0;i<256;i++)//计算新的灰度表
  444. {
  445. Temp=0;
  446. for(int j=0;j<i;j++)
  447. {
  448. Temp+=iCount[j];
  450. }
  451. bMap[i]=(BYTE)(Temp*255/h/w);
  452. }
  455. k=0;
  456. for(i=0;i<h;i++)//给每个像素点付新值
  457. {
  458. for(int j=0;j<w;j++)
  459. {
  460. BYTE tem=lpDIBBits[k];
  461. lpDIBBits[k]=bMap[tem];
  462. lpDIBBits[k+1]=bMap[tem];
  463. lpDIBBits[k+2]=bMap[tem];
  465. if(k>=total)
  466. break;
  467. k+=3;
  468. }
  469. }
  471. }
  472. ////////////////////////////////////////////////////////////////////////
  473. ///傅里叶变换In输入，其长度是图像占有字节数的两倍，i=偶数为实部，
  474. //i=奇数为虚部；nn是一个数组代表图像的宽度和高度，
  475. //sign为1或－1，分别表示正、负傅里叶变换