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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图象 > 查看源码
smooth.c
资源名称：mghe_0.23.zip　[点击查看]
上传用户：tealon
上传日期：2007-01-18
资源大小：24k
文件大小：14k
源码类别：

图形图象

开发平台：

Visual C++

smooth.c：源码内容
  1. //////////////////////////////////////////////////////////////
  2. //Name:smooth.c
  3. //Purpose: To perform smoothing and sharpening
  4. //Author: phoenix, CS, TshingHua, Beijing, P.R.C.
  5. //Email: bjlufengjun@www.163.net or lufengjun@hotmail.com
  6. //Date:April 3, 1998
  8. //header file
  9. #include "bmp.h"
  10. #include "memory.h"
  11. #include "math.h"
  12. #include "stdio.h"
  13. //macro definition
  14. #define WIDTHBYTES(i)    ((i+31)/32*4)
  16. //function declaration
  17. int PASCAL WinMain (HANDLE, HANDLE, LPSTR, int);
  18. LRESULT CALLBACK MainWndProc(HWND , UINT,WPARAM, LPARAM);
  19. BOOL LoadBmpFile (HWND hWnd,char *BmpFileName);
  20. BOOL TemplateOperation(HWND hWnd, int TemplateType);
  21. BOOL MedianFilter(HWND hWnd,BOOL Hori);
  22. //global variable declaration
  23. BITMAPFILEHEADER   bf;
  24. BITMAPINFOHEADER   bi;
  25. HPALETTE           hPalette=NULL;
  26. HBITMAP            hBitmap=NULL;
  27. HGLOBAL            hImgData=NULL;
  28. DWORD              NumColors;
  29. DWORD              LineBytes;
  30. HINSTANCE          ghInst;
  31. DWORD              ImgWidth=0 , ImgHeight=0;
  32. //template array
  33. float Template_Smooth_Box[9]={1.0f,1.0f,1.0f,1.0f,1.0f,1.0f,1.0f,1.0f,1.0f};
  34. float Template_Smooth_Gauss[9]={1.0f,2.0f,1.0f,2.0f,4.0f,2.0f,1.0f,2.0f,1.0f};
  35. float Template_Sharpen_Laplacian[9]={-1.0f,-1.0f,-1.0f,-1.0f,9.0f,-1.0f,-1.0f,-1.0f,-1.0f};
  36. ///////////////////////////////////////////////////////////
  37. int PASCAL WinMain (HANDLE hInstance, HANDLE hPrevInstance,
  38.     LPSTR lpszCmdLine, int nCmdShow)
  39. {
  40. MSG       msg;
  41. WNDCLASS  wndclass;
  42. HWND      hWnd;
  44. ghInst=hInstance;
  45. if ( ! hPrevInstance ){
  46. wndclass.style = CS_HREDRAW | CS_VREDRAW;
  47. wndclass.lpfnWndProc = MainWndProc;
  48. wndclass.cbClsExtra = 0;
  49. wndclass.cbWndExtra = 0;
  50. wndclass.hInstance = hInstance;
  51. wndclass.hIcon = LoadIcon (NULL, IDI_APPLICATION);
  52. wndclass.hCursor = LoadCursor (NULL, IDC_ARROW);
  53. wndclass.hbrBackground = GetStockObject (WHITE_BRUSH);
  54. wndclass.lpszMenuName =  "BMPMENU";
  55. wndclass.lpszClassName = "phoenix ip system";
  56.     }
  58.     if ( ! RegisterClass (&wndclass) )
  59.     return FALSE;
  61. hWnd = CreateWindow ("phoenix ip system","Smoothing and Sharpening",
  62.  WS_OVERLAPPEDWINDOW,CW_USEDEFAULT,CW_USEDEFAULT,
  63.  CW_USEDEFAULT, CW_USEDEFAULT, NULL,NULL,
  64.  hInstance, NULL);
  65. if (!hWnd)
  66. return FALSE;
  67. ShowWindow (hWnd, SW_SHOWMAXIMIZED);
  68. UpdateWindow (hWnd);
  70. while ( GetMessage (&msg, NULL, 0, 0) ){
  71. TranslateMessage (&msg);
  72. DispatchMessage (&msg);
  73.     }
  75. return msg.wParam;
  76. }
  77. ////////////////////////////////////////////////////////////////
  78. LRESULT CALLBACK MainWndProc(HWND hWnd, UINT message,WPARAM wParam, LPARAM lParam)
  79. {
  80.     static  HDC    hDC,hMemDC;
  81.     PAINTSTRUCT    ps;
  83.     switch (message){
  84.     case WM_PAINT:
  85. {         
  86. hDC = BeginPaint(hWnd, &ps);
  87. if (hBitmap)
  88. { 
  89. hMemDC = CreateCompatibleDC(hDC);
  90. if (hPalette)
  91. {           
  92. SelectPalette (hDC, hPalette, FALSE); 
  93. SelectPalette (hMemDC, hPalette, FALSE);
  94. RealizePalette (hDC);
  95. }   
  96. SelectObject(hMemDC, hBitmap); 
  97. BitBlt(hDC, 0, 0, ImgWidth,ImgHeight, hMemDC, 0, 0, SRCCOPY);
  98. DeleteDC(hMemDC);
  99. }
  100. EndPaint(hWnd, &ps);
  101. break;
  102. }
  103. case WM_DESTROY: //注意释放内存和位图，调色板句柄
  104.         if(hBitmap!=NULL)
  105.         DeleteObject(hBitmap);
  106.     
  107. if(hPalette!=NULL)                     
  108.         DeleteObject(hPalette);
  110. if(hImgData!=NULL){
  111. GlobalUnlock(hImgData);
  112. GlobalFree(hImgData);
  113. }         
  114.     PostQuitMessage (0);
  115.     return 0;
  116. case WM_COMMAND:
  117.     switch (wParam){
  118.     case IDM_LOADBMP: 
  119. //注意重新分配内存和调色板，位图句柄时，先释放原来的
  120.         if(hBitmap!=NULL){     
  121. DeleteObject(hBitmap);
  122.             hBitmap=NULL;
  123.         }
  124.         if(hPalette!=NULL){                     
  125. DeleteObject(hPalette);
  126.             hPalette=NULL;
  127.         }
  128. if(hImgData!=NULL){
  129. GlobalUnlock(hImgData);
  130. GlobalFree(hImgData);  
  131. hImgData=NULL;
  132. }         
  133. if(LoadBmpFile(hWnd,"c:\test.bmp")) //成功，则重画窗口
  134.                 InvalidateRect(hWnd,NULL,TRUE);
  135.         break;
  136. case IDM_SMOOTHING_BOX:
  137. if(hImgData!=NULL){
  138. if(TemplateOperation(hWnd,TEMPLATE_SMOOTH_BOX))
  139. InvalidateRect(hWnd,NULL,TRUE);
  140. }
  141. else
  142. MessageBox(hWnd,"File not loaded yet!","Error Message",MB_OK|MB_ICONEXCLAMATION);
  143. break;
  144. case IDM_SMOOTHING_GAUSS:
  145. if(hImgData!=NULL){
  146. if(TemplateOperation(hWnd,TEMPLATE_SMOOTH_GAUSS))
  147. InvalidateRect(hWnd,NULL,TRUE);
  148. }
  149. else
  150. MessageBox(hWnd,"File not loaded yet!","Error Message",MB_OK|MB_ICONEXCLAMATION);
  151. break;
  152. case IDM_SHARPENING_LAPLACIAN:
  153. if(hImgData!=NULL){
  154. if(TemplateOperation(hWnd,TEMPLATE_SHARPEN_LAPLACIAN))
  155. InvalidateRect(hWnd,NULL,TRUE);
  156. }
  157. else
  158. MessageBox(hWnd,"File not loaded yet!","Error Message",MB_OK|MB_ICONEXCLAMATION);
  159. break;
  160. case IDM_HMEDIAN:
  161. if(hImgData!=NULL){
  162. if(MedianFilter(hWnd,TRUE))
  163. InvalidateRect(hWnd,NULL,TRUE);
  164. }
  165. else
  166. MessageBox(hWnd,"File not loaded yet!","Error Message",MB_OK|MB_ICONEXCLAMATION);
  167. break;
  168. case IDM_VMEDIAN:
  169. if(hImgData!=NULL){
  170. if(MedianFilter(hWnd,FALSE))
  171. InvalidateRect(hWnd,NULL,TRUE);
  172. }
  173. else
  174. MessageBox(hWnd,"File not loaded yet!","Error Message",MB_OK|MB_ICONEXCLAMATION);
  175. break;
  176. case IDM_EXIT:
  177.         SendMessage(hWnd,WM_DESTROY,0,0L);
  178.         break;
  179.     }
  180. break;                
  181.     }
  182.     return DefWindowProc (hWnd, message, wParam, lParam);
  183. }
  184. ////////////////////////////////////////////////////////////////
  185. BOOL LoadBmpFile (HWND hWnd,char *BmpFileName)
  186. {   
  187.     HFILE              hf;
  188.     LPBITMAPINFOHEADER lpImgData;
  189.     LOGPALETTE         *pPal;
  190.     LPRGBQUAD          lpRGB;
  191.     HPALETTE           hPrevPalette; 
  192.     HDC                hDc;
  193. HLOCAL             hPal;
  194. DWORD         ImgSize;
  195. DWORD              i;
  197.     if((hf=_lopen(BmpFileName,OF_READ))==HFILE_ERROR){
  198.         MessageBox(hWnd,"File c:\test.bmp not found!","Error Message",MB_OK|MB_ICONEXCLAMATION);
  199.         return FALSE;
  200. }
  201. _lread(hf,(LPSTR)&bf,sizeof(BITMAPFILEHEADER)); 
  202. _lread(hf,(LPSTR)&bi,sizeof(BITMAPINFOHEADER));
  203. ImgWidth=bi.biWidth;
  204. ImgHeight=bi.biHeight;
  205. LineBytes=(DWORD)WIDTHBYTES(bi.biWidth*bi.biBitCount);
  206. ImgSize=(DWORD)LineBytes*bi.biHeight;
  207.     if(bi.biClrUsed!=0)
  208. NumColors=(DWORD)bi.biClrUsed;
  209. else
  210.         switch(bi.biBitCount){
  211.         case 1:
  212.              NumColors=2;
  213.              break;
  214.          case 4:
  215.              NumColors=16;
  216.              break;
  217.          case 8:
  218.              NumColors=256;
  219.              break;
  220.          case 24:
  221.              NumColors=0;
  222.              break;
  223.               default:
  224.                   MessageBox(hWnd,"Invalid color numbers!","Error Message",MB_OK|MB_ICONEXCLAMATION);
  225.                   _lclose(hf);
  226.                   return FALSE; 
  227.         }
  228. if(bf.bfOffBits!=(DWORD)(NumColors*sizeof(RGBQUAD)+sizeof(BITMAPFILEHEADER)
  229. +sizeof(BITMAPINFOHEADER)))
  230. {
  231.      MessageBox(hWnd,"Invalid color numbers!","Error Message" ,MB_OK|
  232.                MB_ICONEXCLAMATION);
  233. _lclose(hf);
  234. return FALSE; 
  235. }
  236. bf.bfSize=sizeof(BITMAPFILEHEADER)+sizeof(BITMAPINFOHEADER)+NumColors*sizeof(RGBQUAD)+ImgSize;
  237. if((hImgData=GlobalAlloc(GHND,(DWORD)(sizeof(BITMAPINFOHEADER)+
  238.      NumColors*sizeof(RGBQUAD)+ImgSize)))==NULL)
  239. {
  240.      MessageBox(hWnd,"Error alloc memory!","ErrorMessage",MB_OK|
  241.                    MB_ICONEXCLAMATION);
  242.     _lclose(hf);
  243. return FALSE;
  244. }
  245.   
  246. lpImgData=(LPBITMAPINFOHEADER)GlobalLock(hImgData); 
  247.     _llseek(hf,sizeof(BITMAPFILEHEADER),FILE_BEGIN);
  248. _hread(hf,(char *)lpImgData,(long)sizeof(BITMAPINFOHEADER)
  249.            +(long)NumColors*sizeof(RGBQUAD)+ImgSize);
  250. _lclose(hf);
  251.     if(NumColors!=0)
  252. {                    
  253.     hPal=LocalAlloc(LHND,sizeof(LOGPALETTE) + NumColors* sizeof(PALETTEENTRY));
  254.     pPal =(LOGPALETTE *)LocalLock(hPal);
  255.     pPal->palNumEntries =(WORD) NumColors;
  256. pPal->palVersion    = 0x300;
  257.     lpRGB = (LPRGBQUAD)((LPSTR)lpImgData + (DWORD)sizeof(BITMAPINFOHEADER));
  258. for (i = 0; i < NumColors; i++) {
  259.       pPal->palPalEntry[i].peRed=lpRGB->rgbRed;
  260. pPal->palPalEntry[i].peGreen=lpRGB->rgbGreen;
  261. pPal->palPalEntry[i].peBlue=lpRGB->rgbBlue;
  262. pPal->palPalEntry[i].peFlags=(BYTE)0;
  263. lpRGB++;
  264. }
  265. hPalette=CreatePalette(pPal);
  266. LocalUnlock(hPal);
  267. LocalFree(hPal);
  268. }
  269. hDc=GetDC(hWnd);
  270. if(hPalette){
  271.         hPrevPalette=SelectPalette(hDc,hPalette,FALSE);
  272. RealizePalette(hDc);
  273. }
  274. hBitmap=CreateDIBitmap(hDc, (LPBITMAPINFOHEADER)lpImgData, (LONG)CBM_INIT,
  275. (LPSTR)lpImgData+sizeof(BITMAPINFOHEADER) +NumColors*sizeof(RGBQUAD),
  276.     (LPBITMAPINFO)lpImgData, DIB_RGB_COLORS);
  277. if(hPalette && hPrevPalette){
  278. SelectPalette(hDc,hPrevPalette,FALSE);
  279. RealizePalette(hDc);
  280. }
  281.  
  282. ReleaseDC(hWnd,hDc);
  283. GlobalUnlock(hImgData);
  284. return TRUE; 
  285. }
  286. ////////////////////////////////////////////////////////////////
  287. BOOL TemplateOperation(HWND hWnd, int TemplateType)
  288. {
  289. DWORD              OffBits,BufSize;
  290.     LPBITMAPINFOHEADER lpImgData;
  291. LPSTR              lpPtr;
  292. HLOCAL             hTempImgData;
  293. LPBITMAPINFOHEADER lpTempImgData;
  294. LPSTR              lpTempPtr;
  295. HDC                hDc;
  296. HFILE              hf;
  297. LONG               x,y;
  298. float              coef;
  299. float              CoefArray[9];
  300. float              TempNum;
  301. char               filename[80];
  303. switch(TemplateType){
  304. case TEMPLATE_SMOOTH_BOX:
  305. coef=(float)(1.0/9.0);
  306. memcpy(CoefArray,Template_Smooth_Box,9*sizeof(float));
  307. strcpy(filename,"c:\smbox.bmp");
  308. break;
  309. case TEMPLATE_SMOOTH_GAUSS:
  310. coef=(float)(1.0/16.0);
  311. memcpy(CoefArray,Template_Smooth_Gauss,9*sizeof(float));
  312. strcpy(filename,"c:\smgauss.bmp");
  313. break;
  314. case TEMPLATE_SHARPEN_LAPLACIAN:
  315. coef=(float)1.0;
  316. memcpy(CoefArray,Template_Sharpen_Laplacian,9*sizeof(float));
  317. strcpy(filename,"c:\shlaplac.bmp");
  318. break;
  319. }
  321. OffBits=bf.bfOffBits-sizeof(BITMAPFILEHEADER);
  322. BufSize=bf.bfSize-sizeof(BITMAPFILEHEADER);
  324. if((hTempImgData=LocalAlloc(LHND,BufSize))==NULL)
  325.     {
  326.      MessageBox(hWnd,"Error alloc memory!","Error Message",MB_OK|MB_ICONEXCLAMATION);
  327.         return FALSE;
  328.     }
  330.     lpImgData=(LPBITMAPINFOHEADER)GlobalLock(hImgData);    
  331. lpTempImgData=(LPBITMAPINFOHEADER)LocalLock(hTempImgData);
  333. lpPtr=(char *)lpImgData;
  334. lpTempPtr=(char *)lpTempImgData;
  336. memcpy(lpTempPtr,lpPtr,BufSize);
  338. for(y=1;y<bi.biHeight-1;y++)
  339. for(x=1;x<bi.biWidth-1;x++){
  340. lpPtr=(char *)lpImgData+(BufSize-LineBytes-y*LineBytes)+x;
  341. lpTempPtr=(char *)lpTempImgData+(BufSize-LineBytes-y*LineBytes)+x;
  342. TempNum=(float)((unsigned char)*(lpPtr+LineBytes-1))*CoefArray[0];
  343. TempNum+=(float)((unsigned char)*(lpPtr+LineBytes))*CoefArray[1];
  344. TempNum+=(float)((unsigned char)*(lpPtr+LineBytes+1))*CoefArray[2];
  345. TempNum+=(float)((unsigned char)*(lpPtr-1))*CoefArray[3];
  346. TempNum+=(float)((unsigned char)*lpPtr)*CoefArray[4];
  347. TempNum+=(float)((unsigned char)*(lpPtr+1))*CoefArray[5];
  348. TempNum+=(float)((unsigned char)*(lpPtr-LineBytes-1))*CoefArray[6];
  349. TempNum+=(float)((unsigned char)*(lpPtr-LineBytes))*CoefArray[7];
  350. TempNum+=(float)((unsigned char)*(lpPtr-LineBytes+1))*CoefArray[8];
  351. TempNum*=coef;
  352. if(TempNum>255.0) *lpTempPtr=(BYTE)255;
  353. else if(TempNum<0.0) 
  354. *lpTempPtr=(unsigned char)fabs(TempNum);
  355. else *lpTempPtr=(BYTE)TempNum;
  356. }
  358. hDc=GetDC(hWnd);
  359.     if(hBitmap!=NULL)
  360.     DeleteObject(hBitmap);
  362. hBitmap=CreateDIBitmap(hDc, (LPBITMAPINFOHEADER)lpTempImgData, (LONG)CBM_INIT,
  363. (LPSTR)lpTempImgData+sizeof(BITMAPINFOHEADER) +NumColors*sizeof(RGBQUAD),
  364.     (LPBITMAPINFO)lpTempImgData, DIB_RGB_COLORS);
  366. hf=_lcreat(filename,0);
  367. _lwrite(hf,(LPSTR)&bf,sizeof(BITMAPFILEHEADER)); 
  368. _lwrite(hf,(LPSTR)lpTempImgData,BufSize);
  369. _lclose(hf);
  371.   ReleaseDC(hWnd,hDc);
  372. LocalUnlock(hTempImgData);
  373. LocalFree(hTempImgData);
  374. GlobalUnlock(hImgData);
  375. return TRUE;
  376. }
  377. ////////////////////////////////////////////////////////////////
  378. BOOL MedianFilter(HWND hWnd,BOOL Hori)
  379. {
  380. DWORD              OffBits,BufSize;
  381.     LPBITMAPINFOHEADER lpImgData;
  382. LPSTR              lpPtr;
  383. HLOCAL             hTempImgData;
  384. LPBITMAPINFOHEADER lpTempImgData;
  385. LPSTR              lpTempPtr;
  386. HDC                hDc;
  387. HFILE              hf;
  388. LONG               x,y;
  389. int                g,g1,g2,g3;
  391. OffBits=bf.bfOffBits-sizeof(BITMAPFILEHEADER);
  392. BufSize=bf.bfSize-sizeof(BITMAPFILEHEADER);
  394. if((hTempImgData=LocalAlloc(LHND,BufSize))==NULL)
  395.     {
  396.      MessageBox(hWnd,"Error alloc memory!","Error Message",MB_OK|MB_ICONEXCLAMATION);
  397.         return FALSE;
  398.     }
  400.     lpImgData=(LPBITMAPINFOHEADER)GlobalLock(hImgData);    
  401. lpTempImgData=(LPBITMAPINFOHEADER)LocalLock(hTempImgData);
  403. memcpy(lpTempImgData,lpImgData,BufSize);
  405. for(y=1;y<bi.biHeight-1;y++)
  406. for(x=1;x<bi.biWidth-1;x++){
  407. lpPtr=(char *)lpImgData+(BufSize-LineBytes-y*LineBytes)+x;
  408. lpTempPtr=(char *)lpTempImgData+(BufSize-LineBytes-y*LineBytes)+x;
  409. g2=(unsigned char)*(lpPtr);
  410. if(Hori){
  411. g1=(unsigned char)*(lpPtr-1);
  412. g3=(unsigned char)*(lpPtr+1);
  413. }
  414. else{
  415. g1=(unsigned char)*(lpPtr+LineBytes);
  416. g3=(unsigned char)*(lpPtr-LineBytes);
  417. }
  418. if(g1>g2){
  419. if(g2>g3) g=g2;
  420. else{
  421. if(g1>g3) g=g3;
  422. else g=g1;
  423. }
  424. }
  425. else{ //g1<=g2
  426. if(g1>g3) g=g1;
  427. else{ 
  428. if(g2>g3) g=g3;
  429. else g=g2;
  430. }
  431. }
  432. *lpTempPtr=(BYTE)g;
  433. }
  435. hDc=GetDC(hWnd);
  436.     if(hBitmap!=NULL)
  437.     DeleteObject(hBitmap);
  439. hBitmap=CreateDIBitmap(hDc, (LPBITMAPINFOHEADER)lpTempImgData, (LONG)CBM_INIT,
  440. (LPSTR)lpTempImgData+sizeof(BITMAPINFOHEADER) +NumColors*sizeof(RGBQUAD),
  441.     (LPBITMAPINFO)lpTempImgData, DIB_RGB_COLORS);
  443. if(Hori)
  444. hf=_lcreat("c:\hmedian.bmp",0);
  445. else
  446. hf=_lcreat("c:\vmedian.bmp",0);
  447. _lwrite(hf,(LPSTR)&bf,sizeof(BITMAPFILEHEADER)); 
  448. _lwrite(hf,(LPSTR)lpTempImgData,BufSize);
  449. _lclose(hf);
  451.   ReleaseDC(hWnd,hDc);
  452. LocalUnlock(hTempImgData);
  453. LocalFree(hTempImgData);
  454. GlobalUnlock(hImgData);
  455. return TRUE;
  456. }