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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > OpenCV > 查看源码
mymfcDlg.cpp
资源名称：mymfc_v2.rar　[点击查看]
上传用户：wjt888999
上传日期：2022-08-11
资源大小：5225k
文件大小：16k
源码类别：

OpenCV

开发平台：

Visual C++

mymfcDlg.cpp：源码内容
  1. // mymfcDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "mymfc.h"
  6. #include "mymfcDlg.h"
  8. #ifdef _DEBUG
  9. #define new DEBUG_NEW
  10. #endif
  13. // CAboutDlg dialog used for App About
  15. class CAboutDlg : public CDialog
  16. {
  17. public:
  18. CAboutDlg();
  20. // Dialog Data
  21. enum { IDD = IDD_ABOUTBOX };
  23. protected:
  24. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  26. // Implementation
  27. protected:
  28. DECLARE_MESSAGE_MAP()
  29. };
  31. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  32. {
  33. }
  35. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  36. {
  37. CDialog::DoDataExchange(pDX);
  38. }
  40. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  41. END_MESSAGE_MAP()
  44. // CmymfcDlg dialog
  49. CmymfcDlg::CmymfcDlg(CWnd* pParent /*=NULL*/)
  50. : CDialog(CmymfcDlg::IDD, pParent)
  51. , TheImage(NULL)
  52. , vPath(_T(""))
  53. , pCapture(NULL)
  54. , m_run(0)
  55. , mCamIndex(0)
  56. {
  57. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  58. }
  60. void CmymfcDlg::DoDataExchange(CDataExchange* pDX)
  61. {
  62. CDialog::DoDataExchange(pDX);
  63. DDX_Control(pDX, IDC_CamList, m_camList);
  64. }
  66. BEGIN_MESSAGE_MAP(CmymfcDlg, CDialog)
  67. ON_WM_SYSCOMMAND()
  68. ON_WM_PAINT()
  69. ON_WM_QUERYDRAGICON()
  70. //}}AFX_MSG_MAP
  71. ON_BN_CLICKED(IDC_ReadImg, &CmymfcDlg::OnBnClickedReadimg)
  72. ON_BN_CLICKED(IDC_EdgeDetect, &CmymfcDlg::OnBnClickedEdgedetect)
  73. ON_BN_CLICKED(IDC_ReadVideo, &CmymfcDlg::OnBnClickedReadvideo)
  74. ON_BN_CLICKED(IDC_Play, &CmymfcDlg::OnBnClickedPlayVideo)
  75. ON_CBN_SELCHANGE(IDC_CamList, &CmymfcDlg::OnCbnSelchangeCamlist)
  76. ON_BN_CLICKED(IDC_Pause, &CmymfcDlg::OnBnClickedPause)
  77. ON_BN_CLICKED(IDOK, &CmymfcDlg::OnBnClickedOk)
  78. ON_BN_CLICKED(IDCANCEL, &CmymfcDlg::OnBnClickedCancel)
  79. ON_BN_CLICKED(IDC_StopCam, &CmymfcDlg::OnBnClickedStopcam)
  80. ON_BN_CLICKED(IDC_RunCam, &CmymfcDlg::OnBnClickedRuncam)
  81. END_MESSAGE_MAP()
  84. // CmymfcDlg message handlers
  86. BOOL CmymfcDlg::OnInitDialog()
  87. {
  88. CDialog::OnInitDialog();
  90. // Add "About..." menu item to system menu.
  92. // IDM_ABOUTBOX must be in the system command range.
  93. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  94. ASSERT(IDM_ABOUTBOX < 0xF000);
  96. CMenu* pSysMenu = GetSystemMenu(FALSE);
  97. if (pSysMenu != NULL)
  98. {
  99. CString strAboutMenu;
  100. strAboutMenu.LoadString(IDS_ABOUTBOX);
  101. if (!strAboutMenu.IsEmpty())
  102. {
  103. pSysMenu->AppendMenu(MF_SEPARATOR);
  104. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  105. }
  106. }
  108. // Set the icon for this dialog.  The framework does this automatically
  109. //  when the application's main window is not a dialog
  110. SetIcon(m_hIcon, TRUE); // Set big icon
  111. SetIcon(m_hIcon, FALSE); // Set small icon
  113. // TODO: Add extra initialization here
  114. // 初始化 创建 TheImage
  115. CvSize ImgSize;
  116. ImgSize.height = IMAGE_HEIGHT;
  117. ImgSize.width = IMAGE_WIDTH;
  118. TheImage = cvCreateImage( ImgSize, IPL_DEPTH_8U, IMAGE_CHANNELS );
  120. /*
  121. // 初始化 BMP 的显示缓存
  122. bmi = ( BITMAPINFO* )buffer;
  123. bmih = &( bmi->bmiHeader );
  124. memset( bmih, 0, sizeof( *bmih ) ); // 把bmih里的内容清零，返回bmih
  125. bmih->biSize = sizeof( BITMAPINFOHEADER );
  126. bmih->biWidth = IMAGE_WIDTH;
  127. bmih->biHeight = -IMAGE_HEIGHT;
  128. bmih->biPlanes = 1;
  129. bmih->biCompression = BI_RGB;
  130. bmih->biBitCount = 8 * TheImage->nChannels;
  131. palette = bmi->bmiColors;
  133. if( TheImage->nChannels == 1 )
  134. {
  135. for( int i = 0; i < 256; i++ )
  136. {
  137. palette[i].rgbBlue = palette[i].rgbGreen = palette[i].rgbRed = (BYTE)i;
  138. palette[i].rgbReserved = 0;
  139. }
  140. }
  141. */
  143. // 初始化视频、摄像头显示循环的执行和中断（m_run）
  144. m_run = 0;
  146. // 获取摄像头数目
  147. int cam_count = CCameraDS::CameraCount();
  148. // 在组合框CamList中添加摄像头名称的字符串
  149. for(int i=0; i < cam_count; i++)
  150. {
  151. char camera_name[1024];  
  152. int retval = CCameraDS::CameraName(i, camera_name, sizeof(camera_name) );
  154. char istr[25];
  155. sprintf_s(istr, " # %d", i);
  156. //strcat_s( camera_name, istr );
  157. CString camstr;
  158. camstr.Format("%s%s",camera_name, istr );
  159. if(retval >0)
  160. m_camList.AddString( camstr );
  161. else
  162. AfxMessageBox(_T("不能获取摄像头的名称"));
  163. }
  164. UpdateData( FALSE );
  166. // 使边缘检测按钮、视频播放和暂停按钮失效
  167. GetDlgItem( IDC_EdgeDetect )->EnableWindow( FALSE );
  168. GetDlgItem( IDC_Play )->EnableWindow( FALSE );
  169. GetDlgItem( IDC_Pause )->EnableWindow( FALSE );
  170. GetDlgItem( IDC_RunCam )->EnableWindow( FALSE );
  171. GetDlgItem( IDC_StopCam )->EnableWindow( FALSE );
  174. return TRUE;  // return TRUE  unless you set the focus to a control
  175. }
  177. void CmymfcDlg::OnSysCommand(UINT nID, LPARAM lParam)
  178. {
  179. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  180. {
  181. CAboutDlg dlgAbout;
  182. dlgAbout.DoModal();
  183. }
  184. else
  185. {
  186. CDialog::OnSysCommand(nID, lParam);
  187. }
  188. }
  190. // If you add a minimize button to your dialog, you will need the code below
  191. //  to draw the icon.  For MFC applications using the document/view model,
  192. //  this is automatically done for you by the framework.
  194. void CmymfcDlg::OnPaint()
  195. {
  196. if (IsIconic())
  197. {
  198. CPaintDC dc(this); // device context for painting
  200. SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
  202. // Center icon in client rectangle
  203. int cxIcon = GetSystemMetrics(SM_CXICON);
  204. int cyIcon = GetSystemMetrics(SM_CYICON);
  205. CRect rect;
  206. GetClientRect(&rect);
  207. int x = (rect.Width() - cxIcon + 1) / 2;
  208. int y = (rect.Height() - cyIcon + 1) / 2;
  210. // Draw the icon
  211. dc.DrawIcon(x, y, m_hIcon);
  212. }
  213. else
  214. {
  215. CDialog::OnPaint(); // 重绘对话框
  216. CDialog::UpdateWindow(); // 更新windows窗口，如果无这步调用，图片显示还会出现问题
  217. ShowImage( TheImage, IDC_ShowImg ); // 重绘图片函数
  218. }
  219. }
  221. // The system calls this function to obtain the cursor to display while the user drags
  222. //  the minimized window.
  223. HCURSOR CmymfcDlg::OnQueryDragIcon()
  224. {
  225. return static_cast<HCURSOR>(m_hIcon);
  226. }
  229. /*=======================================
  230. *************图片读取处理****************
  231. ========================================*/
  233. void CmymfcDlg::OnBnClickedReadimg()
  234. {
  235. // TODO: Add your control notification handler code here
  236. CFileDialog dlg(
  237. TRUE, _T("*.bmp"), NULL,
  238. OFN_FILEMUSTEXIST | OFN_PATHMUSTEXIST | OFN_HIDEREADONLY,
  239. _T("image files (*.bmp; *.jpg) |*.bmp; *.jpg; *.jpeg | All Files (*.*) |*.*||"), NULL
  240. ); // 选项图片的约定
  241. dlg.m_ofn.lpstrTitle = _T("Open Image"); // 打开文件对话框的标题名
  242. if( dlg.DoModal() != IDOK ) // 判断是否获得图片
  243. return;
  245. CString mPath = dlg.GetPathName(); // 获取图片路径
  247. IplImage* ipl = cvLoadImage( mPath, 1 ); // 读取图片、缓存到一个局部变量 ipl 中
  248. if( !ipl ) // 判断是否成功读取图片
  249. return;
  250. if( TheImage ) // 对上一幅显示的图片数据清零
  251. cvZero( TheImage );
  252. // 使边缘检测按钮生效
  253. GetDlgItem( IDC_EdgeDetect )->EnableWindow( TRUE );
  255. ResizeImage( ipl ); // 对读入的图片进行缩放，使其宽或高最大值者刚好等于 256，再复制到 TheImage 中
  256. ShowImage( TheImage, IDC_ShowImg ); // 调用显示图片函数
  257. cvReleaseImage( &ipl ); // 释放 ipl 占用的内存
  258. }
  260. void CmymfcDlg::OnBnClickedEdgedetect()
  261. {
  262. // TODO: Add your control notification handler code here
  264. IplImage *gray = 0, *edge = 0;
  265. gray = cvCreateImage( cvSize(IMAGE_WIDTH, IMAGE_HEIGHT), IPL_DEPTH_8U, 1 );
  266. edge = cvCreateImage( cvSize(IMAGE_WIDTH, IMAGE_HEIGHT), IPL_DEPTH_8U, 1 );
  267. cvCvtColor( TheImage, gray, CV_BGR2GRAY );
  268. cvCanny( gray, edge, 30, 100, 3 );
  269. cvCvtColor( edge, TheImage, CV_GRAY2BGR );
  270. ShowImage( TheImage, IDC_ShowImg ); // 调用显示图片函数
  272. // 边缘检测只执行一次，完成后则令按钮失效
  273. GetDlgItem( IDC_EdgeDetect )->EnableWindow( FALSE );
  275. cvReleaseImage( &gray );
  276. cvReleaseImage( &edge );
  277. }
  279. void CmymfcDlg::ShowImage( IplImage* img, UINT ID ) // ID 是Picture Control控件的ID号
  280. {
  281. CDC* pDC = GetDlgItem( ID ) ->GetDC(); // 获得显示控件的 DC
  282. HDC hDC = pDC ->GetSafeHdc(); // 获取 HDC(设备句柄) 来进行绘图操作
  284. CRect rect;
  285. GetDlgItem(ID) ->GetClientRect( &rect );
  286. int rw = rect.right - rect.left; // 求出图片控件的宽和高
  287. int rh = rect.bottom - rect.top;
  288. int iw = img->width; // 读取图片的宽和高
  289. int ih = img->height;
  290. int tx = (int)(rw - iw)/2; // 使图片的显示位置正好在控件的正中
  291. int ty = (int)(rh - ih)/2;
  292. SetRect( rect, tx, ty, tx+iw, ty+ih );
  294. CvvImage cimg;
  295. cimg.CopyOf( img ); // 复制图片
  296. cimg.DrawToHDC( hDC, &rect ); // 将图片绘制到显示控件的指定区域内
  298. ReleaseDC( pDC );
  299. }
  301. void CmymfcDlg::ResizeImage(IplImage* img)
  302. {
  303. // 读取图片的宽和高
  304.     int w = img->width;
  305. int h = img->height;
  307. // 找出宽和高中的较大值者
  308. int max = (w > h)? w: h;
  310. // 计算将图片缩放到TheImage区域所需的比例因子
  311. float scale = (float) ( (float) max / 256.0f );
  313. // 缩放后图片的宽和高
  314. int nw = (int)( w/scale );
  315. int nh = (int)( h/scale );
  317. // 为了将缩放后的图片存入 TheImage 的正中部位，需计算图片在 TheImage 左上角的期望坐标值
  318. int tlx = (nw > nh)? 0: (int)(256-nw)/2;
  319. int tly = (nw > nh)? (int)(256-nh)/2: 0;
  321. // 设置 TheImage 的 ROI 区域，用来存入图片 img
  322. cvSetImageROI( TheImage, cvRect( tlx, tly, nw, nh) );
  324. // 对图片 img 进行缩放，并存入到 TheImage 中
  325. cvResize( img, TheImage );
  327. // 重置 TheImage 的 ROI 准备读入下一幅图片
  328. cvResetImageROI( TheImage );
  329. }
  331. /*=======================================
  332. *************本地视频处理****************
  333. ========================================*/
  335. void CmymfcDlg::OnBnClickedReadvideo()
  336. {
  337. // TODO: Add your control notification handler code here
  338. CFileDialog dlg(
  339. TRUE, _T("*.avi"), NULL,
  340. OFN_FILEMUSTEXIST | OFN_PATHMUSTEXIST | OFN_HIDEREADONLY,
  341. _T("*.avi|*.avi|*.rmvb|*.rmvb|*.mp4|*.mp4|*.wmv|*.wmv| All Files (*.*) |*.*||"), NULL
  342. ); // 选项视频的约定
  343. dlg.m_ofn.lpstrTitle = _T("Open Video"); // 打开文件对话框的标题名
  344. if( dlg.DoModal() != IDOK ) // 判断是否获得视频
  345. return;
  347. vPath = dlg.GetPathName(); // 获取视频路径
  348. // 使能视频播放按钮
  349. GetDlgItem( IDC_Play )->EnableWindow( TRUE );
  350. GetDlgItem( IDC_Pause )->EnableWindow( TRUE );
  351. }
  353. void CmymfcDlg::OnBnClickedPlayVideo()
  354. {
  355. // TODO: Add your control notification handler code here
  356. CvCapture *pCapture = NULL;
  357.     IplImage *pFrame = NULL;
  358.         
  359. if( vPath == "" )  //判断文件路径是否为空
  360. {
  361. MessageBox("请先选择视频文件!");
  362. return;
  363. }
  364. else
  365. {
  366. if(!(pCapture = cvCreateFileCapture( vPath )))
  367. {
  368. MessageBox("打开视频文件失败!");
  369. return;
  370. }
  371. }
  374. // 播放视频时令读入图片和视频、以及播放的按钮失效
  375. GetDlgItem( IDC_ReadImg )->EnableWindow( FALSE );
  376. GetDlgItem( IDC_ReadVideo )->EnableWindow( FALSE );
  377. GetDlgItem( IDC_Play )->EnableWindow( FALSE );
  378. GetDlgItem( IDC_CamList )->EnableWindow( FALSE );
  379. GetDlgItem( IDC_RunCam )->EnableWindow( FALSE );
  380. GetDlgItem( IDC_StopCam )->EnableWindow( FALSE );
  383. cvNamedWindow("video");
  384. cvResizeWindow("video",1,1);
  385. HWND hWnd = (HWND) cvGetWindowHandle("video");
  386. HWND hParent = ::GetParent(hWnd);
  387. HWND hwnd1=::FindWindow("CmymfcDlg","mymfc");
  388. ::SetParent(hWnd, hwnd1);
  389. ::ShowWindow(hParent, SW_HIDE);
  391. // 读取视频文件的帧数
  392. int frames = (int) cvGetCaptureProperty( pCapture, CV_CAP_PROP_FRAME_COUNT );
  393. int numfrm = 0;
  395. // 对图像数据清零
  396. if( TheImage )
  397. cvZero( TheImage );
  399. m_run = 1;
  400.     while( (numfrm < frames) & (m_run == 1) )
  401.     { 
  402. pFrame = cvQueryFrame( pCapture );
  403. IplImage* newframe = cvCloneImage(pFrame);
  404. ResizeImage( newframe );
  405. ShowImage( TheImage, IDC_ShowImg );
  406. if (cvWaitKey(20) == 27) break;
  407. numfrm++;
  408. cvReleaseImage( &newframe );
  409. }
  410. m_run = 0;
  412. // 对图像数据清零
  413. if( TheImage )
  414. {
  415. cvZero( TheImage );
  416. ShowImage( TheImage, IDC_ShowImg );
  417. }
  418. // 视频结束后令读入图片和视频按钮生效
  419. GetDlgItem( IDC_ReadImg )->EnableWindow( TRUE );
  420. GetDlgItem( IDC_ReadVideo )->EnableWindow( TRUE );
  421. GetDlgItem( IDC_Play )->EnableWindow( TRUE );
  422. GetDlgItem( IDC_CamList )->EnableWindow( TRUE );
  424.     cvReleaseCapture(&pCapture);
  425. cvDestroyWindow("video");
  426. return;
  427. }
  429. void CmymfcDlg::OnBnClickedPause()
  430. {
  431. // TODO: Add your control notification handler code here
  432. m_run = 0;
  433. return;
  434. }
  436. /*=======================================
  437. 摄像头处理区
  438. ========================================*/
  440. void CmymfcDlg::OnCbnSelchangeCamlist()
  441. {
  442. // TODO: Add your control notification handler code here
  443. mCamIndex = m_camList.GetCurSel();
  444. GetDlgItem( IDC_RunCam )->EnableWindow( TRUE );
  445. return;
  446. }
  448. void CmymfcDlg::OnBnClickedStopcam()
  449. {
  450. // TODO: Add your control notification handler code here
  451. m_run = 0;
  452. return;
  453. }
  455. void CmymfcDlg::OnBnClickedRuncam()
  456. {
  457. // TODO: Add your control notification handler code here
  458. CString camName;
  459. camName.Format("%s%d","Camera # ", mCamIndex );
  460. UpdateData( FALSE );
  463. // 创建1个摄像头实例
  464. CCameraDS camera;
  466. //打开第一个摄像头
  467. //if(! camera.OpenCamera(0, true)) //弹出属性选择窗口
  468. if( (! camera.OpenCamera(mCamIndex, false, 320,240) ) ) //不弹出属性选择窗口，用代码制定图像宽和高
  469. {
  470. AfxMessageBox(_T("Can not open camera.n"));
  471. return;
  472. }
  474. // 读取摄像头数据时令读入图片和视频、播放、暂停的按钮失效
  475. GetDlgItem( IDC_ReadImg )->EnableWindow( FALSE );
  476. GetDlgItem( IDC_ReadVideo )->EnableWindow( FALSE );
  477. GetDlgItem( IDC_Play )->EnableWindow( FALSE );
  478. GetDlgItem( IDC_Pause )->EnableWindow( FALSE );
  479. // 使关闭摄像头按钮生效
  480. GetDlgItem( IDC_StopCam )->EnableWindow( TRUE );
  482. // 创建一个Ipl图像指针来读取摄像头的帧画面
  483. IplImage* cFrame;
  485. // 创建一个视频播放窗口
  486. cvNamedWindow("camera");
  487. // 将这个窗口缩至最小
  488. cvResizeWindow("camera",1,1);
  489. // 获取窗口句柄
  490. HWND hWnd = (HWND) cvGetWindowHandle("camera");
  491. // 获取该窗口的父窗口句柄
  492. HWND hParent = ::GetParent(hWnd);
  493. // 获取mymfc的GUI窗口句柄
  494. HWND hwnd1=::FindWindow("CmymfcDlg","mymfc");
  495. // 将GUI窗口设置为视频播放窗口的父窗口，使视频在 GUI 的指定区域播放
  496. ::SetParent(hWnd, hwnd1);
  497. // 隐藏所创建的视频播放窗口
  498. ::ShowWindow(hParent, SW_HIDE);
  500. // 对图像数据清零
  501. if( TheImage )
  502. cvZero( TheImage );
  504. // 初始化在图像中显示字符的字体格式
  505. CvFont tFont;
  506. cvInitFont(&tFont,  CV_FONT_HERSHEY_COMPLEX, 0.5f,0.7f,0,1,8);
  508. // 为读取系统时间信息分配内存
  509. char timestr[26];
  510. memset(timestr, 0, 26 * sizeof(char));
  512. m_run = 1;
  513. while( m_run )
  514. { 
  515. //获取一帧
  516. cFrame = camera.QueryFrame();
  517. IplImage* newframe = cvCloneImage(cFrame);
  519. // 获取系统时间信息
  520. __time64_t rawtime; 
  521. struct tm timeinfo ; 
  522. errno_t err;
  523. // Get time as 64-bit integer
  524. _time64( &rawtime );
  525. // Convert to local time
  526. err = _localtime64_s( &timeinfo, &rawtime );
  527. // Convert to an ASCII representation
  528. asctime_s( timestr, 26, &timeinfo );
  530. CString strTime;
  531. strTime.Format( "%s", timestr );
  532. strTime.Replace( "n", "" );
  535. /*rawtime = time( NULL ); 
  536. timeinfo = localtime( &rawtime ); 
  537. char* p = asctime( timeinfo );
  539. // 字符串 p 的第25个字符是换行符 'n'，但在图像中将乱码显示
  540. // 故仅读取 p 的前 24 个字符
  541. for (int i = 0; i < 24; i++)
  542. {
  543. timestr[i] = *p;
  544. p++;
  545. }
  546. p = NULL; */
  548. // 在图像上显示摄像头序号以及系统时间信息
  549. cvPutText( newframe, camName, cvPoint(95,15), &tFont,  CV_RGB(255,0,0) );
  550. cvPutText( newframe, strTime, cvPoint(5,225), &tFont,  CV_RGB(255,0,0) );
  552. ResizeImage( newframe );
  553. ShowImage( TheImage, IDC_ShowImg );
  555. if (cvWaitKey(20) == 27) break;
  556. cvReleaseImage( &newframe );
  557. }
  560. // 对图像数据清零
  561. if( TheImage )
  562. {
  563. cvZero( TheImage );
  564. ShowImage( TheImage, IDC_ShowImg );
  565. }
  567. camera.CloseCamera();
  568. cvDestroyWindow("camera");
  571. // 读取摄像头数据时令读入图片和视频、播放、暂停的按钮失效
  572. GetDlgItem( IDC_ReadImg )->EnableWindow( TRUE );
  573. GetDlgItem( IDC_ReadVideo )->EnableWindow( TRUE );
  574. GetDlgItem( IDC_Play )->EnableWindow( TRUE );
  575. GetDlgItem( IDC_Pause )->EnableWindow( TRUE );
  577. return;
  578. }
  580. /*=======================================
  581. *************退出GUI窗口****************
  582. ========================================*/
  584. void CmymfcDlg::OnBnClickedOk()
  585. {
  586. // TODO: Add your control notification handler code here
  587. m_run = 0;
  588. cvDestroyAllWindows();
  589. OnOK();
  590. }
  592. void CmymfcDlg::OnBnClickedCancel()
  593. {
  594. // TODO: Add your control notification handler code here
  595. m_run = 0;
  596. cvDestroyAllWindows();
  597. OnCancel();
  598. }