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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > 图形图像处理 > 查看源码
FaceDetectDoc.cpp
资源名称：人脸识别系统设计—毕业设计.rar　[点击查看]
上传用户：sdsuchuang
上传日期：2013-01-12
资源大小：2228k
文件大小：15k
源码类别：

图形图像处理

开发平台：

Visual C++

FaceDetectDoc.cpp：源码内容
  1. // FaceDetectDoc.cpp : implementation of the CFaceDetectDoc class
  2. //
  4. #include "stdafx.h"
  5. #include "FaceDetect.h"
  6. #include "DIB.h"
  7. #include "FaceDetectDoc.h"
  8. #include "FaceDetectView.h"
  9. #include "LikelyHood.h"
  10. #include "math.h"
  11. #define WIDTHBYTES(bits)  ((bits+31)/32*4)
  12. //#include "ReplaceDoc.h"
  13. #ifdef _DEBUG
  14. #define new DEBUG_NEW
  15. #undef THIS_FILE
  16. static char THIS_FILE[] = __FILE__;
  17. #endif
  18. extern HANDLE hDIB;
  19. extern HANDLE hDIB1;
  20. extern HANDLE hDIBTemp;
  21. extern HANDLE hDIBTemp1;
  22. extern class DIB gDib;
  23. extern int gheight;
  24. extern int gwidth;
  25. extern WORD gwBytesPerLine;
  26. #define IDC_BMPSHOW                     1000
  28. /////////////////////////////////////////////////////////////////////////////
  29. // CFaceDetectDoc
  31. IMPLEMENT_DYNCREATE(CFaceDetectDoc, CDocument)
  33. BEGIN_MESSAGE_MAP(CFaceDetectDoc, CDocument)
  34. //{{AFX_MSG_MAP(CFaceDetectDoc)
  35. ON_COMMAND(ID_READY_GetFaceOutLline, OnREADYGetFaceOutLline)
  36. //}}AFX_MSG_MAP
  37. END_MESSAGE_MAP()
  39. /////////////////////////////////////////////////////////////////////////////
  40. // CFaceDetectDoc construction/destruction
  42. CFaceDetectDoc::CFaceDetectDoc()
  43. {
  44. // TODO: add one-time construction code here
  45.  m_bFaceOK = false;
  49. }
  51. CFaceDetectDoc::~CFaceDetectDoc()
  52. {
  53. }
  55. BOOL CFaceDetectDoc::OnNewDocument()
  56. {
  57. if (!CDocument::OnNewDocument())
  58. return FALSE;
  60. // TODO: add reinitialization code here
  61. // (SDI documents will reuse this document)
  63. return TRUE;
  64. }
  68. /////////////////////////////////////////////////////////////////////////////
  69. // CFaceDetectDoc serialization
  71. void CFaceDetectDoc::Serialize(CArchive& ar)
  72. {
  73. if (ar.IsStoring())
  74. {
  75. // TODO: add storing code here
  76. }
  77. else
  78. {
  79. // TODO: add loading code here
  80. }
  81. }
  83. /////////////////////////////////////////////////////////////////////////////
  84. // CFaceDetectDoc diagnostics
  86. #ifdef _DEBUG
  87. void CFaceDetectDoc::AssertValid() const
  88. {
  89. CDocument::AssertValid();
  90. }
  92. void CFaceDetectDoc::Dump(CDumpContext& dc) const
  93. {
  94. CDocument::Dump(dc);
  95. }
  96. #endif //_DEBUG
  98. /////////////////////////////////////////////////////////////////////////////
  99. // CFaceDetectDoc commands
  100. BOOL CFaceDetectDoc::OnOpenDocument(LPCTSTR lpszPathName) 
  101. {
  102. if (!CDocument::OnOpenDocument(lpszPathName))
  103. return FALSE;
  104. if(hDIB)
  105. GlobalFree(hDIB);
  106.    path=lpszPathName;
  107. hDIB = gDib.LoadDIB(lpszPathName);
  108. ASSERT(hDIB != NULL);
  109. hDIBTemp = gDib.CopyHandle(hDIB);
  110. LPBITMAPINFOHEADER lpbi;
  111. lpbi = (LPBITMAPINFOHEADER)GlobalLock(hDIB);
  112. gheight = lpbi->biHeight;
  113. gwidth = lpbi->biWidth;
  114.     m_nWndWidth = gwidth;
  115. m_nWndHeight=gheight ;
  116. gwBytesPerLine = gDib.BytePerLine(hDIB);
  117. GlobalUnlock(hDIB);
  119. return TRUE;
  120. }
  122. BOOL CFaceDetectDoc::OnOpenDocument1(LPCTSTR lpszPathName) 
  123. {
  124. if (!CDocument::OnOpenDocument(lpszPathName))
  125. return FALSE;
  126. if(hDIB1)
  127. GlobalFree(hDIB1);
  128.    path=lpszPathName;
  129. hDIB1 = gDib.LoadDIB(lpszPathName);
  130. ASSERT(hDIB1 != NULL);
  131. hDIBTemp1 = gDib.CopyHandle(hDIB1);
  132. LPBITMAPINFOHEADER lpbi;
  133. lpbi = (LPBITMAPINFOHEADER)GlobalLock(hDIB1);
  134. gheight = lpbi->biHeight;
  135. gwidth = lpbi->biWidth;
  136.     m_nWndWidth = gwidth;
  137. m_nWndHeight=gheight ;
  138. gwBytesPerLine = gDib.BytePerLine(hDIB1);
  139. GlobalUnlock(hDIB);
  141. return TRUE;
  142. }
  143. ///////////////////////获取图片数据//////////////////
  144. void CFaceDetectDoc::OnREADYGetFaceOutLline() 
  145. {
  146. // BYTE *colorTable;
  147. if(!hDIB)
  148. {
  149. ::AfxMessageBox("请确认是否有图片?");return ;
  150. }
  151. READYGetFaceOutLline1() ;
  153. }
  155. void CFaceDetectDoc::READYGetFaceOutLline1() 
  156. {
  157. DIB *m_pMainDib=new DIB();
  159.    //BYTE *m_pDib=m_pMainDib->GetBits();
  161. // colorTable = (BYTE *)m_pDib;
  163.    //////////////////////////////////////////////////////
  164.     HDIB hNewDIB = NULL;
  165. LPBITMAPINFOHEADER lpSrc,lpDest;
  166. LPBYTE lpS,lpD;
  167. DWORD dwBytesPerLine;
  168. DWORD dwImgSize;
  169. // WORD wBytesPerLine;
  170. unsigned i ,j,height,width;
  171. if(!hDIB)
  172. return ;
  174. lpSrc = (LPBITMAPINFOHEADER)GlobalLock(hDIB);
  176. dwBytesPerLine = WIDTHBYTES(24*(lpSrc->biWidth));
  177. dwImgSize = lpSrc->biHeight * dwBytesPerLine;
  178. //申请新的内存，大小等于原来图象的大小
  179. hNewDIB = GlobalAlloc(GHND,sizeof(BITMAPINFOHEADER)+dwImgSize);
  181. lpDest = (LPBITMAPINFOHEADER)GlobalLock(hNewDIB);
  182. //保存图片的长宽、颜色深度等信息
  184.     memcpy((void*)lpDest,(void*)lpSrc,sizeof(BITMAPINFOHEADER));
  186. DWORD dwSBytesPerLine;
  188. dwSBytesPerLine = (24*(lpSrc->biWidth)+31)/32*4;
  189. height = lpDest->biHeight;
  190. width = lpDest->biWidth;
  191. lpS = (LPBYTE)lpSrc;
  192. // wBytesPerLine = this->BytePerLine(hDIB);//
  193. lpD = (LPBYTE)lpDest;
  194. lpS = lpS + sizeof(BITMAPINFOHEADER);
  195. lpD = lpD + sizeof(BITMAPINFOHEADER);
  197.    ////////////////////////////////////////////////////
  198.   
  199. int byteBitCount  = m_pMainDib->GetBiBitCount()/8;
  201. m_tOriPixelArray  = new RGBQUAD*[height];
  202. m_tResPixelArray  = new RGBQUAD*[height];
  204. for(i=0; i<height; i++)
  205. {
  206. m_tOriPixelArray[i] = new RGBQUAD[width];
  207. m_tResPixelArray[i] = new RGBQUAD[width];
  209. }
  211.     LPBYTE lpData;
  212. long lOffset;
  213. lpData = gDib.FindDIBBits(hDIB);
  214.     WORD gwBytesPerLine = gDib.BytePerLine(hDIB);
  215. for(i=0; i<height; i++)
  216. {
  217. for(j=0; j<width; j++)
  219. {  
  220. lOffset = gDib.PixelOffset(i, j, gwBytesPerLine);
  222.             m_tOriPixelArray[i][j].rgbBlue =*(lpData + lOffset++);
  223.     m_tOriPixelArray[i][j].rgbGreen=*(lpData + lOffset++);
  224. m_tOriPixelArray[i][j].rgbRed  =*(lpData + lOffset++);
  226. m_tOriPixelArray[i][j].rgbReserved = 0;
  228. m_tResPixelArray[i][j]=m_tOriPixelArray[i][j];
  231. }
  233. }
  234. m_nWndWidth=width;
  235.     m_nWndHeight=height;
  237.   method1 =new CLikelyHood(m_tOriPixelArray,m_nWndWidth,m_nWndHeight);
  238.   
  239.   Likehood(m_nWndWidth,m_nWndHeight) ;
  240.   OnBtnBinary();
  241.   
  242.   HistogramV();
  243.   OnBtnHistogramH();
  244.   
  245.   OnBtnMarkFace();
  246.   ///////////////////////////////////////////////
  247.   /*
  248.   for(int k1=top;k1<bottom; k1++)
  249. for(int k=left;k<right; k++)
  251.             if(method1->m_pBinaryArray[k1][k]==1)
  252. {
  253.   lOffset = gDib.PixelOffset(k1, k, gwBytesPerLine);
  254.                *(lpData+lOffset)=255;
  255.                *(lpData+lOffset+1)=255;
  256.    *(lpData+lOffset+2)=255;
  257.    }
  258.  */
  259. ////////////////////////////////////////////////////
  260.   
  261. }
  263. ////////////////////////////////////////////////////////////////////////////////
  264. // 计算相似度
  265. ////////////////////////////////////////////////////////////////////////////////
  266. void CFaceDetectDoc::Likehood(int m_WndWidth,int m_WndHeight) 
  267. {
  268. int i,j;
  270. SetCursor(LoadCursor(NULL,IDC_WAIT));
  271.     
  272. method1->CalLikeHood();
  274. for( i=0; i<m_WndHeight; i++)
  275. for( j=0; j<m_WndWidth;  j++)
  276. {
  277. m_tResPixelArray[i][j].rgbBlue = m_tResPixelArray[i][j].rgbGreen =m_tResPixelArray[i][j].rgbRed=(int)(method1->m_pLikeliHoodArray[i][j]*255);
  278.     
  279. }
  280. //MakeBitMap();
  281.   
  282. SetCursor(LoadCursor(NULL,IDC_ARROW));
  283. }
  285. ////////////////////////////////////////////////////////////////////////////////
  286. // 二值化
  287. ////////////////////////////////////////////////////////////////////////////////
  288. void CFaceDetectDoc::OnBtnBinary() 
  289. {
  290. SetCursor(LoadCursor(NULL,IDC_WAIT));
  291. int i,j;
  292. //  long lOffset;
  293.  LPBYTE lpData = gDib.FindDIBBits(hDIB);
  294.     WORD gwBytesPerLine = gDib.BytePerLine(hDIB);
  295. if(!method1->CalBinary())
  296. {
  297. AfxMessageBox("请先计算相似度！");
  298. SetCursor(LoadCursor(NULL,IDC_ARROW));
  299. return;
  300. }
  302.     
  303. for(i=0; i<m_nWndHeight; i++)
  304. for( j=0; j<m_nWndWidth;  j++)
  305. {
  306. m_tResPixelArray[i][j].rgbBlue = m_tResPixelArray[i][j].rgbGreen =
  307. m_tResPixelArray[i][j].rgbRed  = (int)(method1->m_pBinaryArray[i][j]*255);
  309. }
  311. // MakeBitMap();
  313. SetCursor(LoadCursor(NULL,IDC_ARROW));
  314. //MyDraw();
  315. }
  316. ////////////////////////////////////////////////////////////////////////////////
  317. // 画图
  318. ////////////////////////////////////////////////////////////////////////////////
  319. /*
  320. void CFaceDetectDoc::MyDraw()
  321. {
  322. CRect rc;
  324. m_pResMap=new CBitmap();
  325. pDCShow->GetWindow()->GetClientRect(&rc);
  327. pDCShow->Rectangle(&rc);
  328. CDC dc;
  329. CBitmap *pOldBitmap;
  330. dc.CreateCompatibleDC(pDCShow);
  332. pOldBitmap=dc.SelectObject(m_pResMap);
  333. pDCShow->StretchBlt(0,0,m_nWndWidth,m_nWndHeight,&dc,0,0,m_nWndWidth,m_nWndHeight,SRCCOPY);
  334. dc.SelectObject(pOldBitmap);
  335. dc.DeleteDC();
  336. }
  337. */
  338. ////////////////////////////////////////////////////////////////////////////////
  339. // 垂直方向的直方图
  340. ////////////////////////////////////////////////////////////////////////////////
  341. void CFaceDetectDoc::HistogramV() 
  342. {
  344. SetCursor(LoadCursor(NULL,IDC_WAIT));
  345. for(int i=0; i<m_nWndHeight; i++)
  346. {
  347. int count = 0;
  348. for(int j=0; j<m_nWndWidth;  j++)
  349. {
  350. if(method1->m_pBinaryArray[i][j] == 1) count++;
  351. m_tResPixelArray[i][j].rgbBlue = m_tResPixelArray[i][j].rgbGreen =
  352. m_tResPixelArray[i][j].rgbRed  = 255;
  353. }
  354. for(j=0; j<count;  j++)
  355. {
  356. m_tResPixelArray[i][j].rgbBlue = m_tResPixelArray[i][j].rgbGreen =
  357. m_tResPixelArray[i][j].rgbRed  = 0;
  358. }
  359. }
  361. // MakeBitMap();
  362. SetCursor(LoadCursor(NULL,IDC_ARROW));
  363. }
  365. ////////////////////////////////////////////////////////////////////////////////
  366. // 水平方向直方图
  367. ////////////////////////////////////////////////////////////////////////////////
  368. void CFaceDetectDoc::OnBtnHistogramH() 
  369. {
  372. SetCursor(LoadCursor(NULL,IDC_WAIT));
  373. for(int j=0; j<m_nWndWidth;  j++)
  374. {
  375. int count = 0;
  376. for(int i=0; i<m_nWndHeight; i++)
  377. {
  378. if(method1->m_pBinaryArray[i][j] == 1) count++;
  379. m_tResPixelArray[i][j].rgbBlue = m_tResPixelArray[i][j].rgbGreen =
  380. m_tResPixelArray[i][j].rgbRed  = 255;
  381. }
  382. for(i=m_nWndHeight-1; i>=m_nWndHeight-count;i--)
  383. {
  384. m_tResPixelArray[i][j].rgbBlue = m_tResPixelArray[i][j].rgbGreen =
  385. m_tResPixelArray[i][j].rgbRed  = 0;
  386. }
  387. }
  389. // MakeBitMap();
  390. SetCursor(LoadCursor(NULL,IDC_ARROW));
  391. }
  392. ////////////////////////////////////////////////////////////////////////////////
  393. //标记脸部区域
  394. ////////////////////////////////////////////////////////////////////////////////
  395. void CFaceDetectDoc::OnBtnMarkFace() 
  396. {
  399. SetCursor(LoadCursor(NULL,IDC_WAIT));
  400. int *temp = new int[m_nWndWidth];
  401. int max = 0;
  402. int pos = -1;
  403. for(int j=0; j<m_nWndWidth;  j++)////////////////计算出每行占有的脸部像素点数
  404. {
  405. int count = 0;
  406. for(int i=0; i<m_nWndHeight; i++)
  407. {
  408. if(method1->m_pBinaryArray[i][j]== 1) count++;
  409. }
  410. temp[j] = count;
  411. if(count > max)////////////////////////获取哪一行具有的脸部像素点数最多，记下数目和第几行
  412. {
  413. max = count;
  414. pos = j;
  415. }
  416. }
  419. for(l=pos; l>=0; l--)////////////////////////向上获取具有的脸部像素点数最多的一半的行
  420. {
  421. if(temp[l]<max*0.2||l==0)
  422. {
  423. left = l;
  424. break;
  425. }
  426. }
  427. for(l=pos; l<m_nWndWidth; l++)////////////////////////向下获取具有的脸部像素点数最多的1/3的行
  428. {
  429. if(temp[l]<max*0.3||l==m_nWndWidth-1)
  430. {
  431. right = l;
  432. break;
  433. }
  434. }
  436. for(int i=0; i<m_nWndHeight; i++)
  437. {
  438. int count = 0;
  439. for(l = left;l<=right;l++)
  440. {
  441. if(method1->m_pBinaryArray[i][l] == 1) count++;
  442. }
  443. if(count>=(right-left)*0.5)
  444. {
  445. top = i;
  446. break;
  447. }
  448. }
  449. bottom = (int)(top+(right-left)*1.5)>=m_nWndHeight? m_nWndHeight-1:(int)(top+(right-left)*1.5);
  451. CopyBitMap(m_tResPixelArray,m_tOriPixelArray);
  452.  
  453. for(i=top;i<=bottom;i++)
  454. {
  455. m_tResPixelArray[i][left].rgbBlue=255;
  456. m_tResPixelArray[i][left].rgbGreen = m_tResPixelArray[i][left].rgbRed = 0;
  457. m_tResPixelArray[i][right].rgbBlue=255;
  458. m_tResPixelArray[i][right].rgbGreen = m_tResPixelArray[i][right].rgbRed = 0;
  459. }
  460. for(j=left;j<=right;j++)
  461. {
  462. m_tResPixelArray[top][j].rgbBlue=255;
  463. m_tResPixelArray[top][j].rgbGreen = m_tResPixelArray[top][j].rgbRed = 0;
  464. m_tResPixelArray[bottom][j].rgbBlue=255;
  465. m_tResPixelArray[bottom][j].rgbGreen = m_tResPixelArray[bottom][j].rgbRed = 0;
  466. }
  467. // MakeBitMap();
  469. SetCursor(LoadCursor(NULL,IDC_ARROW));
  470.    //////////////////////////////////////////////////////////////////////////////////////
  471.   
  472. for(int h=bottom;h<m_nWndHeight;h++)
  473.    {    int count=0;
  474.    for(int w=left;w<right;w++)
  475.    {
  476.    if(method1->m_pBinaryArray[h][w] == 1)
  477.    count++;
  478.    }
  479.               if(count>=(right-left)*0.5)
  480.   bottom++;
  481.    
  483.    
  484.    }
  485.    
  487. CFaceDetectView *Gd=new CFaceDetectView();
  488. Gd->DrawLine(left,top,bottom,right);
  490.   //////////////////////////////////////////////////////////////////////////////////////
  491.        
  493. if(m_bFaceOK)
  494. {
  495. //ReplaceDoc dlg;
  496. // if(dlg.DoModal()==IDOK)
  497. // {
  498. CopyBitMap(m_tResPixelArray,m_tOriPixelArray);
  499. CRect rect(left,top,right,bottom);
  500. m_rFaceRegion = rect;
  501. // MakeBitMap();
  502. //}
  503. }
  504. else
  505. {
  506. m_bFaceOK = true;
  507. CopyBitMap(m_tResPixelArray,m_tOriPixelArray);
  508. CRect rect(left,top,right,bottom);
  509. m_rFaceRegion = rect;
  510. // MakeBitMap();
  511. }
  512. }
  513. ////////////////////////////////////////////////////////////////////////////////
  514. // 拷贝位图
  515. // 参数：  dest－目标位图指针
  516. //         source－源位图指针
  517. ////////////////////////////////////////////////////////////////////////////////
  518. bool CFaceDetectDoc::CopyBitMap(RGBQUAD **dest, RGBQUAD **source)
  519. {
  520. if(source==NULL || dest==NULL) 
  521. return false;
  522. for(int i=0; i<m_nWndHeight; i++)
  523. for(int j=0; j<m_nWndWidth; j++)
  524. dest[i][j]=source[i][j];
  526. return true;
  527. }
  528. ////////////////////////////////////////////////////////////////////////////////
  529. // 生成新的位图
  530. ////////////////////////////////////////////////////////////////////////////////
  531. void CFaceDetectDoc::MakeBitMap()
  532. {
  533. CClientDC ClientDC(pDCShow->GetWindow());
  534. if(m_pResMap!=NULL) delete m_pResMap;
  536. m_pResMap=new CBitmap();
  537. m_pResMap->CreateCompatibleBitmap(&ClientDC,m_nWndWidth,m_nWndHeight);
  539. CDC  dc;
  540. dc.CreateCompatibleDC(&ClientDC);
  541. dc.SelectObject(m_pResMap);
  543. for(int i=0; i<m_nWndHeight; i++)
  544. for(int j=0; j<m_nWndWidth; j++)
  545. dc.SetPixelV(j,i,RGB(m_tResPixelArray[i][j].rgbRed,m_tResPixelArray[i][j].rgbGreen,m_tResPixelArray[i][j].rgbBlue));
  546.     
  547. if(m_bFaceOK )
  548. {
  549. CBrush Pen;
  550. Pen.CreateSolidBrush(RGB(255,0,0));
  551. dc.FrameRect(m_rFaceRegion,&Pen);
  552.   Pen.DeleteObject();
  553. }
  555. if(m_bLeftEyeOK) DrawCross(&dc,m_LeftEye,RGB(255,0,0));
  556. if(m_bRightEyeOK) DrawCross(&dc,m_RightEye,RGB(255,0,0));
  557. if(m_bLeftEyeLeftCornerOK) DrawCross(&dc,m_LeftEyeLeftCorner,RGB(255,0,255));
  558. if(m_bLeftEyeRightCornerOK) DrawCross(&dc,m_LeftEyeRightCorner,RGB(255,255,0));
  559. if(m_bRightEyeLeftCornerOK) DrawCross(&dc,m_RightEyeLeftCorner,RGB(255,0,255));
  560. if(m_bRightEyeRightCornerOK) DrawCross(&dc,m_RightEyeRightCorner,RGB(255,255,0));
  561. if(m_bLeftNostrilOK) DrawCross(&dc,m_LeftNostril,RGB(0,255,0));
  562. if(m_bRightNostrilOK) DrawCross(&dc,m_RightNostril,RGB(0,255,0));
  563. if(m_bMidNoseOK) DrawCross(&dc,m_MidNose,RGB(0,255,0));
  564. if(m_bLeftMouthCornerOK) DrawCross(&dc,m_LeftMouthCorner,RGB(0,0,255));
  565. if(m_bRightMouthCornerOK) DrawCross(&dc,m_RightMouthCorner,RGB(0,0,255));
  566. if(m_bMidMouthOK) DrawCross(&dc,m_MidMouth,RGB(0,0,255));
  568. dc.DeleteDC();
  570. // MyDraw();
  571. }
  572. ////////////////////////////////////////////////////////////////////////////////
  573. // 画十字形标记
  574. // 参数：  pDC－CDC指针
  575. //         point－要画的点的坐标
  576. //         crColor－标记得颜色
  577. ////////////////////////////////////////////////////////////////////////////////
  578. void CFaceDetectDoc::DrawCross(CDC *pDC, CPoint point, COLORREF crColor)
  579. {
  580. CPen pen,*oldPen;
  581. pen.CreatePen(PS_SOLID,1,crColor);
  582. oldPen = (CPen*)pDC->SelectObject(&pen);
  583. pDC->MoveTo(point.x-7,point.y);
  584. pDC->LineTo(point.x+7,point.y);
  585. pDC->MoveTo(point.x,point.y-7);
  586. pDC->LineTo(point.x,point.y+7);
  587. pDC->SelectObject(oldPen);
  588. pen.DeleteObject();
  589. }