正文

win32 摄像头捕获系统vfw

IP云
IP云V管理员/1 阅读
09/24
提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示
  1. //win32 摄像头捕获系统vfw
  1. //  VideoRecord.h
  2. ///  用于定义一些资源ID
  1. #include "resource.h"
  2. //#define EXIT  104
  3. #define HELP  105
  4. #define MINIMIZE 106
  5. //#define DISPLAY  107
  6. #define BUTTONSIZE  15
  7. #define PHOTO   108
  8. #define RECORDVIDEO 109
  9. // #define RESOURCE 110
  10. // #define FORMAT   111
  11. #define CONNECT 112
  1. #include <Windows.h>
  2. #include <STRING>
  3. #include <vfw.h>
  4. #include "VideoRecord.h"
  5. #pragma comment(lib, "vfw32.lib")
  6. LRESULT WINAPI MainWndProc( HWND, UINT, WPARAM, LPARAM );
  7. LRESULT WINAPI SelCapDrcProc( HWND, UINT, WPARAM, LPARAM );
  8. int EnumCapDrv();
  9. VOID APIENTRY HandlePopupMenu(HWND, POINT);  // 右键弹出菜单回调函数
  10. WORD WINAPI VideoThreadProc( LPVOID lParam); // 视频捕捉线程函数
  11. ////////////////全局变量
  12. HANDLE ghInstance;  // 应用程序实例
  13. HWND hwndMain;  // 主窗口句柄
  14. HWND hwndVideo;  // 视频捕捉窗口句柄
  15. HWND hwndSelCapDrvDlg; // 选择捕捉驱动对话框句柄
  16. HWND hwndSelCapDrvLBox; // 选择驱动对话框列举驱动名称列表框句柄
  17. HWND hwndExit;
  18. HWND hwndMin;
  19. HWND hwndHelp;
  20. HWND hwndPhoto;
  21. HWND hwndRecord;
  22. HWND hwndBtnExit;
  23. HWND hwndSource;
  24. HWND hwndFormat;
  25. HWND hwndConnect;
  26. HANDLE hVideoTread; // 视频采集线程
  27. HRGN hRoundRgn; // 窗口修正区域
  28. CAPDRIVERCAPS capDrvCaps; // 驱动性能
  29. bool bRecordFileOpen = false;  // 录像开始标志
  30. bool bPhotoFileOpen = false;  // 照相开始标志
  31. bool bRecording = false;  // 判断是否正在录像
  32. bool bThreadEnd = false; // 判断视频线程是否终止
  33. char recordFile[260]; //保持录像标志
  34. char photoFile[260]; // 保持照相标志
  35. char gachBuffer[260]; // 保存错误信息
  36. //////////////////////////////////////////////////////////////////////////////
  37. // ErrorCallbackProc:    错误回调函数,过capSetCallbackOnError宏来注册回调
  38. // hWnd:              捕获窗口句柄
  39. // nErrID:              错误代码
  40. // lpErrorText:          关于错误的文本信息
  41. ///////////////////////////////////////////////////////////////////////////////
  42. LRESULT CALLBACK ErrorCallbackProc(HWND hWnd,int nErrID,LPSTR lpErrorText)
  43. {
  44. if(!hwndMain)
  45. return FALSE;
  46. if(nErrID==0)
  47. return TRUE;//清除旧的错误
  48. wsprintf(gachBuffer,"Error# %d",nErrID);//显示错误标识和文本
  49. MessageBox(hWnd, lpErrorText, gachBuffer,MB_OK | MB_ICONEXCLAMATION);
  50. return (LRESULT) TRUE;
  51. }
  52. /////////////////////////////////////
  53. int PASCAL WinMain( IN HINSTANCE hInstance,
  54. IN HINSTANCE hPrevInstance,
  55. IN LPSTR lpCmdLine,
  56. IN int nShowCmd )
  57. {
  58. WNDCLASS wndClass;
  59. MSG msg;
  60. char className[] = "VideoClass";
  61. // 如果不是前实例
  62. if (!hPrevInstance)
  63. {
  64. wndClass.lpszClassName = className;
  65. wndClass.lpfnWndProc = MainWndProc;
  66. wndClass.style = CS_HREDRAW|CS_VREDRAW|CS_OWNDC;
  67. wndClass.hInstance = hInstance;
  68. wndClass.hCursor = LoadCursor(NULL, IDC_ARROW);
  69. wndClass.hIcon = LoadIcon(NULL, IDI_APPLICATION);
  70. wndClass.hbrBackground = CreateSolidBrush( RGB(200,200,100) );
  71. wndClass.lpszMenuName = NULL;
  72. wndClass.cbClsExtra = 0;
  73. wndClass.cbWndExtra = 0;
  74. RegisterClass( &wndClass );
  75. }
  76. ghInstance = hInstance;
  77. // 创建主窗口
  78. hwndMain = CreateWindow(className,
  79. "视频监控系统",
  80. WS_POPUP|WS_OVERLAPPED,
  81. GetSystemMetrics(SM_CXFULLSCREEN)/2 - 600/2,
  82. GetSystemMetrics(SM_CYFULLSCREEN)/2-600/2,
  83. 600, 600,
  84. NULL,
  85. NULL,
  86. hInstance,
  87. NULL);
  88. ShowWindow(hwndMain, nShowCmd);
  89. // 设定主窗口主要区域
  90. SetWindowRgn(hwndMain, hRoundRgn, 1);
  91. while( GetMessage(&msg, NULL, 0, 0))
  92. {
  93. TranslateMessage(&msg);//该函数将虚拟键消息转换为字符消息。
  94. //字符消息被寄送到调用线程的消息队列里，当下一次线程调用函数GetMessage或PeekMessage时被读出
  95. DispatchMessage(&msg);//该函数调度一个消息给窗口程序。
  96. //通常调度从GetMessage取得的消息。消息被调度到的窗口程序即是MainProc()函数
  97. }
  98. return msg.wParam;
  99. }
  100. LRESULT CALLBACK MainWndProc( HWND hMain, UINT msg, WPARAM wParam, LPARAM lParam )
  101. {
  102. HDC hdc = GetDC(hMain);
  103. RECT rc;   // 客户区
  104. POINT pt;  // 鼠标位置
  105. HFONT hFont = (HFONT)GetStockObject(DEFAULT_GUI_FONT);
  106. switch(msg)
  107. {
  108. case WM_LBUTTONDBLCLK:
  109. SetFocus(hMain);
  110. break;
  111. case WM_RBUTTONDOWN:
  112. GetClientRect(hMain, (LPRECT)&rc);
  113. pt.y = HIWORD(lParam);
  114. pt.x = LOWORD(lParam);
  115. if ( PtInRect(&rc, pt) )
  116. {
  117. HandlePopupMenu(hMain, pt);
  118. }
  119. break;
  120. case WM_PAINT:
  121. RECT helpRect, minRect, exitRect;
  122. HRGN helpRgn, minRgn, exitRgn;
  123. FrameRgn(hdc, hRoundRgn, CreateSolidBrush(RGB(0,0,0)), 2, 2);
  124. BringWindowToTop(hwndSelCapDrvDlg);
  125. return DefWindowProc(hMain, msg, wParam, lParam);
  126. case WM_CREATE:
  127. // 注册错误提示回调函数
  128. capSetCallbackOnError(hwndVideo, (FARPROC)ErrorCallbackProc);
  129. // 主窗口主要圆角矩形区域
  130. hRoundRgn = CreateRoundRectRgn(100, 70, 500, 460, 20, 20);
  131. // 捕捉视频区域s
  132. hwndVideo = capCreateCaptureWindow((LPSTR)"My Capture Window",
  133. WS_CHILD|WS_VISIBLE,
  134. 140, 100, 320, 280,
  135. (HWND)hMain,
  136. (int)1);
  137. hwndExit = CreateWindow("button",    // 建立一个按钮,更多如BUTTON,COMBOBOX,LISTBOX
  138. "x",    // button text
  139. WS_VISIBLE|WS_CHILD|BS_PUSHBUTTON|BS_CENTER|BS_VCENTER,
  140. 470, 75, BUTTONSIZE, BUTTONSIZE,
  141. hMain,
  142. (HMENU)EXIT,
  143. (HINSTANCE)ghInstance,
  144. (LPVOID)NULL );
  145. hwndMin = CreateWindow("button",    // 建立一个按钮,更多如BUTTON,COMBOBOX,LISTBOX
  146. "-",    // button text
  147. WS_VISIBLE|WS_CHILD|BS_PUSHBUTTON|BS_CENTER|BS_VCENTER,
  148. 450, 75, BUTTONSIZE, BUTTONSIZE,
  149. hMain,
  150. (HMENU)MINIMIZE,
  151. (HINSTANCE)ghInstance,
  152. (LPVOID)NULL );
  153. hwndHelp = CreateWindow("button",    // 建立一个按钮,更多如BUTTON,COMBOBOX,LISTBOX
  154. "?",    // button text
  155. WS_VISIBLE|WS_CHILD|BS_PUSHBUTTON/*|BS_CENTER|BS_VCENTER*/,
  156. 430, 75, BUTTONSIZE, BUTTONSIZE,
  157. hMain,
  158. (HMENU)HELP,
  159. (HINSTANCE)ghInstance,
  160. (LPVOID)NULL );
  161. hwndPhoto = CreateWindow("button",
  162. "照 相",
  163. WS_CHILD|WS_VISIBLE|BS_PUSHBUTTON,
  164. 150, 410, 50, 20,
  165. hMain,
  166. (HMENU)PHOTO,
  167. (HINSTANCE)ghInstance,
  168. (LPVOID)NULL );
  169. SendMessage(hwndPhoto, WM_SETFONT, (WPARAM)hFont, 1);  // 设置按钮字体为系统默认字体
  170. hwndRecord = CreateWindow("button",
  171. "录 像",
  172. WS_CHILD|WS_VISIBLE|BS_PUSHBUTTON|BS_VCENTER|BS_CENTER,
  173. 220, 410, 50, 20,
  174. hMain,
  175. (HMENU)RECORDVIDEO,
  176. (HINSTANCE)ghInstance,
  177. (LPVOID)NULL );
  178. SendMessage(hwndRecord, WM_SETFONT, (WPARAM)hFont, 1);// 设置按钮字体为系统默认字体
  179. hwndBtnExit = CreateWindow("button",   // 退出按钮
  180. "退 出",
  181. WS_CHILD|WS_VISIBLE|BS_PUSHBUTTON|BS_VCENTER|BS_CENTER,
  182. 400, 410, 50, 20,
  183. hMain,
  184. (HMENU)EXIT,
  185. (HINSTANCE)ghInstance,
  186. (LPVOID)NULL );
  187. SendMessage(hwndBtnExit, WM_SETFONT, (WPARAM)hFont, 1);
  188. // 设置按钮区域
  189. GetClientRect(hwndHelp, &helpRect);
  190. GetClientRect(hwndMin, &minRect);
  191. GetClientRect(hwndExit, &exitRect);
  192. helpRgn = CreateEllipticRgnIndirect(&helpRect);
  193. minRgn = CreateEllipticRgnIndirect(&minRect);
  194. exitRgn = CreateEllipticRgnIndirect(&exitRect);
  195. SetWindowRgn(hwndExit,exitRgn,1);
  196. SetWindowRgn(hwndMin,minRgn,1);
  197. SetWindowRgn(hwndHelp,helpRgn,1);
  198. // 创建选择驱动对话框
  199. hwndSelCapDrvDlg = CreateDialog((HINSTANCE)ghInstance,
  200. MAKEINTRESOURCE(DLG_SELCAPDRIVER),
  201. 0, (DLGPROC)SelCapDrcProc);
  202. hwndSelCapDrvLBox = GetDlgItem(hwndSelCapDrvDlg, LISTBOX_SELCAPDRIVER);
  203. EnumCapDrv();
  204. break;
  205. case WM_COMMAND:
  206. CAPSTATUS capStatus;
  207. switch(wParam)
  208. {
  209. case EXIT:  // 退出系统
  210. SendMessage(hMain, WM_SYSCOMMAND, SC_CLOSE, 0);
  211. break;
  212. case MINIMIZE: // 最小化
  213. SendMessage(hMain, WM_SYSCOMMAND, SC_MINIMIZE, 0);
  214. break;
  215. case HELP:  // 帮助按钮
  216. SendMessage(hMain, WM_SYSCOMMAND, SC_CONTEXTHELP, 0);
  217. break;
  218. case RESOURCE: // 视频源选项
  219. if (capDrvCaps.fHasDlgVideoSource)
  220. {
  221. capDlgVideoSource(hwndVideo);
  222. }
  223. break;
  224. case FORMAT:  // 视频格式选项
  225. if (capDrvCaps.fHasDlgVideoFormat)
  226. {
  227. capDlgVideoFormat(hwndVideo);
  228. }
  229. break;
  230. case CONNECT:
  231. break;
  232. case DISPLAY: // 视频显示选项
  233. if (capDrvCaps.fHasDlgVideoDisplay)
  234. {
  235. capDlgVideoDisplay(hMain);
  236. capGetStatus(hwndVideo, &capStatus, sizeof(CAPSTATUS));// 得到当前捕获窗口的状态
  237. SetWindowPos(hwndVideo, NULL, 0, 0, capStatus.uiImageWidth, capStatus.uiImageHeight, SWP_NOZORDER|SWP_NOMOVE);
  238. }
  239. break;
  240. case PHOTO:  // 抓取图片
  241. if ( capFileSaveDIB(hwndVideo, "test.bmp") ){
  242. MessageBox(hMain, "Save Bmp File Succeeded!", "photo save", MB_OK|MB_ICONINFORMATION);
  243. }
  244. else break;
  245. if (!bPhotoFileOpen)
  246. {
  247. OPENFILENAME ofnp;  // open file name structure
  248. ZeroMemory(&ofnp, sizeof(OPENFILENAME));
  249. ofnp.lStructSize = sizeof(OPENFILENAME);
  250. ofnp.hwndOwner = hMain;
  251. ofnp.lpstrFile = photoFile;
  252. ofnp.lpstrFilter = "*.bmp";
  253. ofnp.nMaxFile = sizeof(photoFile);
  254. ofnp.nFilterIndex = 1;
  255. ofnp.nMaxFileTitle = 0;
  256. ofnp.lpstrInitialDir = NULL;
  257. ofnp.Flags = OFN_PATHMUSTEXIST|OFN_FILEMUSTEXIST;
  258. // 显示保存文件对话框
  259. if (GetSaveFileName(&ofnp) == TRUE)
  260. {
  261. bPhotoFileOpen = TRUE;
  262. strcpy(photoFile, ofnp.lpstrFile);
  263. strcat(photoFile, ".bmp");
  264. capDriverConnect(hwndPhoto, 0);  // 与驱动连接
  265. capOverlay(hwndPhoto, TRUE); // //使用 hardware overlay
  266. //使 video input directlyinto framebuffer，使不用 CPU
  267. capPreview(hwndVideo, TRUE);  // 抓取图像显示在视窗
  268. capPreviewScale(hwndVideo, TRUE); // 使图像延伸到所设定的视窗大小
  269. capPreviewRate(hwndVideo, 1000/30); // 每秒图像变化张数
  270. capPreview(hwndVideo, TRUE); // 预览图像显示
  271. capGrabFrameNoStop(hwndVideo);
  272. capEditCopy(hwndVideo);
  273. if (capFileSaveDIB(hwndVideo, photoFile)){
  274. MessageBox(hMain, "Save Bmp File Succeeded!", "photo save", MB_OK|MB_ICONINFORMATION);
  275. }
  276. break;
  277. }
  278. }
  279. break;
  280. case RECORDVIDEO:  // 录制视频
  281. if (HIWORD(wParam) == BN_CLICKED && (HWND)lParam == hwndRecord)
  282. {
  283. OPENFILENAME ofnp;
  284. if (!bRecordFileOpen)
  285. {
  286. ZeroMemory(&ofnp, sizeof(OPENFILENAME));
  287. ofnp.lStructSize = sizeof(OPENFILENAME);
  288. ofnp.nMaxFile = sizeof(recordFile);
  289. ofnp.hwndOwner = hMain;
  290. ofnp.lpstrFile = recordFile;
  291. ofnp.lpstrFilter = "*.avi";
  292. ofnp.lpstrInitialDir = NULL;
  293. ofnp.nFilterIndex = 1;
  294. ofnp.lpstrFileTitle = NULL;
  295. ofnp.nMaxFileTitle = 0;
  296. ofnp.Flags = OFN_FILEMUSTEXIST|OFN_PATHMUSTEXIST;
  297. // 显示保存文件对话框
  298. if (GetSaveFileName(&ofnp) == TRUE)
  299. {
  300. strcpy(recordFile, ofnp.lpstrFile);
  301. strcat(recordFile, ".avi");
  302. bRecordFileOpen = TRUE;
  303. // 创建视频采集线程
  304. DWORD id;
  305. SECURITY_ATTRIBUTES sa;
  306. sa.nLength = sizeof(SECURITY_ATTRIBUTES);
  307. sa.lpSecurityDescriptor = NULL;
  308. sa.bInheritHandle = TRUE;
  309. hVideoTread = CreateThread(&sa,
  310. (ULONG)0,
  311. (LPTHREAD_START_ROUTINE)VideoThreadProc,
  312. (LPVOID)(ULONG)0,
  313. 0,
  314. &id);
  315. // 如果创建线程失败，弹出提示对话框
  316. if (hVideoTread == NULL)
  317. {
  318. MessageBox(NULL, "Video Thread Create Failure!", "Thread", MB_OK|MB_ICONINFORMATION);
  319. }
  320. }
  321. break;
  322. }
  323. if (bRecordFileOpen)  // 已经打开一个选择对话框
  324. {
  325. if (bRecording)  // 已经在录制视频中
  326. {
  327. bThreadEnd = true;
  328. // 结束录制，并保存
  329. if ( capFileSaveAs(hwndVideo, recordFile) ){
  330. MessageBox(NULL, "Save Avi File Succeeded!", "RecordVideo", MB_OK|MB_ICONINFORMATION);
  331. }
  332. SetWindowText(hwndRecord, "录 像");
  333. bRecording = false;
  334. break;
  335. }
  336. else  // 没有录像但是选择了一个文件
  337. {
  338. int rt = 0;
  339. rt = MessageBox(hMain, "Do you want write over that file?", "系统提示", MB_YESNO|MB_ICONINFORMATION);
  340. if (rt == IDYES)  // 继续录制，覆盖当前文件
  341. {
  342. if (capCaptureSequence(hwndRecord))
  343. {
  344. bRecording = true;
  345. SetWindowText(hwndRecord, "停 止");
  346. }
  347. }
  348. else
  349. {
  350. bRecordFileOpen = false;
  351. SendMessage(hMain, WM_COMMAND, MAKEWPARAM(RECORDVIDEO, BN_CLICKED), (LPARAM)hwndRecord);
  352. }
  353. break;
  354. }
  355. break;
  356. }
  357. }
  358. }
  359. break;
  360. case WM_DESTROY:
  361. capSetCallbackOnError(hwndVideo, NULL);
  362. capPreview(hwndVideo, FALSE); // 停止预览
  363. capCaptureAbort(hwndVideo);  // 停止捕获
  364. capDriverDisconnect(hwndVideo); // 断开驱动连接
  365. PostQuitMessage(0);
  366. break;
  367. default:
  368. return DefWindowProc(hMain, msg, wParam, lParam);
  369. }
  370. return 0;
  371. }
  372. LRESULT WINAPI SelCapDrcProc( HWND hMian, UINT msg, WPARAM wParam, LPARAM lParam )
  373. {
  374. //  ShowWindow(hwndSelCapDrvDlg, SW_SHOW);
  375. switch(msg)
  376. {
  377. case WM_INITDIALOG:
  378. return TRUE;
  379. case WM_COMMAND:
  380. switch(wParam)
  381. {
  382. case BTN_SEL:
  383. int sel = 0;
  384. SendMessage(hwndSelCapDrvLBox, LB_GETSELITEMS, 1, sel );  // 获取选择的驱动
  385. SendMessage(hwndVideo, WM_CAP_DRIVER_CONNECT, sel, 0L);   // 连接驱动
  386. SendMessage(hwndSelCapDrvDlg, WM_CLOSE, 0, 0);   // 关闭对话框
  387. SendMessage(hwndVideo, WM_CAP_DRIVER_GET_CAPS, sizeof(CAPDRIVERCAPS), (long)(LPVOID)&capDrvCaps); // 更新驱动的能力
  388. if (capDrvCaps.fHasOverlay)
  389. {
  390. capOverlay(hwndVideo, TRUE);
  391. }
  392. capPreviewRate(hwndVideo, 66);   //设置视频捕获速率提高到66 miliseconds
  393. if ( !capPreview(hwndVideo, TRUE))   //用CapPreview启动预览功能
  394. {
  395. MessageBox(NULL, "预览启动失败", "提示", MB_OK|MB_ICONINFORMATION);
  396. SendMessage(hMian, WM_COMMAND, EXIT, 0);
  397. return FALSE;
  398. }
  399. }
  400. return TRUE;
  401. case WM_CLOSE:
  402. DestroyWindow(hwndSelCapDrvDlg);
  403. return TRUE;
  404. }
  405. return (0L);
  406. }
  407. int EnumCapDrv()
  408. {
  409. char drvName[100];
  410. char drvVesion[100];
  411. char drvStr[200];
  412. for (int i = 0; i < 5; i++)  // 扫面获取驱动信息
  413. {
  414. if ( capGetDriverDescription(i, drvName, sizeof(drvName), drvVesion, sizeof(drvVesion)) )
  415. {
  416. strcpy(drvStr, drvName);
  417. strcat(drvStr, "  ");
  418. strcat(drvStr, drvVesion);
  419. SendMessage(hwndSelCapDrvLBox, LB_ADDSTRING, 0, (LPARAM)drvStr); // 向列表框增加字符串
  420. SendMessage(hwndSelCapDrvLBox, LB_SETITEMDATA, i, (LPARAM)i);
  421. }
  422. }
  423. return 0;
  424. }
  425. VOID APIENTRY HandlePopupMenu( HWND hwnd, POINT pt )
  426. {
  427. HMENU hmenu;
  428. HMENU hPopupMenu;
  429. hmenu = LoadMenu((HINSTANCE)ghInstance, MAKEINTRESOURCE(IDR_POPUPMENU));
  430. if(hmenu == NULL) return;
  431. // 获取子菜单
  432. hPopupMenu = GetSubMenu(hmenu, 0);
  433. // 弹出菜单使用屏幕坐标，所以转换鼠标点击坐标到屏幕坐标
  434. ClientToScreen(hwnd, &pt);
  435. TrackPopupMenu(hPopupMenu, TPM_LEFTALIGN|TPM_LEFTBUTTON, pt.x, pt.y, 0, hwnd, NULL);
  436. DestroyMenu(hmenu); // 销毁指定菜单，并释放此菜单占用的内存
  437. }
  438. WORD WINAPI VideoThreadProc( LPVOID lParam )
  439. {
  440. // 捕获视频
  441. if( capCaptureSequence(hwndVideo) ) //该函数调控启动流视频和音频捕获到一个文件,你能使用这个功能发送消息
  442. {
  443. SetWindowText(hwndRecord, "停 止");
  444. MessageBox(NULL, "Start to capture video.", "提示", MB_OK|MB_ICONINFORMATION);
  445. bRecording = true;
  446. }
  447. while(!bThreadEnd);
  448. MessageBox(NULL, "Leaving Record Thread", "Thread",NULL);
  449. return 0;
  450. }