- //win32 摄像头捕获系统vfw
- // VideoRecord.h
- /// 用于定义一些资源ID
- #include "resource.h"
- //#define EXIT 104
- #define HELP 105
- #define MINIMIZE 106
- //#define DISPLAY 107
- #define BUTTONSIZE 15
- #define PHOTO 108
- #define RECORDVIDEO 109
- // #define RESOURCE 110
- // #define FORMAT 111
- #define CONNECT 112
- #include <Windows.h>
- #include <STRING>
- #include <vfw.h>
- #include "VideoRecord.h"
- #pragma comment(lib, "vfw32.lib")
- LRESULT WINAPI MainWndProc( HWND, UINT, WPARAM, LPARAM );
- LRESULT WINAPI SelCapDrcProc( HWND, UINT, WPARAM, LPARAM );
- int EnumCapDrv();
- VOID APIENTRY HandlePopupMenu(HWND, POINT); // 右键弹出菜单回调函数
- WORD WINAPI VideoThreadProc( LPVOID lParam); // 视频捕捉线程函数
- ////////////////全局变量
- HANDLE ghInstance; // 应用程序实例
- HWND hwndMain; // 主窗口句柄
- HWND hwndVideo; // 视频捕捉窗口句柄
- HWND hwndSelCapDrvDlg; // 选择捕捉驱动对话框句柄
- HWND hwndSelCapDrvLBox; // 选择驱动对话框列举驱动名称列表框句柄
- HWND hwndExit;
- HWND hwndMin;
- HWND hwndHelp;
- HWND hwndPhoto;
- HWND hwndRecord;
- HWND hwndBtnExit;
- HWND hwndSource;
- HWND hwndFormat;
- HWND hwndConnect;
- HANDLE hVideoTread; // 视频采集线程
- HRGN hRoundRgn; // 窗口修正区域
- CAPDRIVERCAPS capDrvCaps; // 驱动性能
- bool bRecordFileOpen = false; // 录像开始标志
- bool bPhotoFileOpen = false; // 照相开始标志
- bool bRecording = false; // 判断是否正在录像
- bool bThreadEnd = false; // 判断视频线程是否终止
- char recordFile[260]; //保持录像标志
- char photoFile[260]; // 保持照相标志
- char gachBuffer[260]; // 保存错误信息
- //////////////////////////////////////////////////////////////////////////////
- // ErrorCallbackProc: 错误回调函数,过capSetCallbackOnError宏来注册回调
- // hWnd: 捕获窗口句柄
- // nErrID: 错误代码
- // lpErrorText: 关于错误的文本信息
- ///////////////////////////////////////////////////////////////////////////////
- LRESULT CALLBACK ErrorCallbackProc(HWND hWnd,int nErrID,LPSTR lpErrorText)
- {
- if(!hwndMain)
- return FALSE;
- if(nErrID==0)
- return TRUE;//清除旧的错误
- wsprintf(gachBuffer,"Error# %d",nErrID);//显示错误标识和文本
- MessageBox(hWnd, lpErrorText, gachBuffer,MB_OK | MB_ICONEXCLAMATION);
- return (LRESULT) TRUE;
- }
- /////////////////////////////////////
- int PASCAL WinMain( IN HINSTANCE hInstance,
- IN HINSTANCE hPrevInstance,
- IN LPSTR lpCmdLine,
- IN int nShowCmd )
- {
- WNDCLASS wndClass;
- MSG msg;
- char className[] = "VideoClass";
- // 如果不是前实例
- if (!hPrevInstance)
- {
- wndClass.lpszClassName = className;
- wndClass.lpfnWndProc = MainWndProc;
- wndClass.style = CS_HREDRAW|CS_VREDRAW|CS_OWNDC;
- wndClass.hInstance = hInstance;
- wndClass.hCursor = LoadCursor(NULL, IDC_ARROW);
- wndClass.hIcon = LoadIcon(NULL, IDI_APPLICATION);
- wndClass.hbrBackground = CreateSolidBrush( RGB(200,200,100) );
- wndClass.lpszMenuName = NULL;
- wndClass.cbClsExtra = 0;
- wndClass.cbWndExtra = 0;
- RegisterClass( &wndClass );
- }
- ghInstance = hInstance;
- // 创建主窗口
- hwndMain = CreateWindow(className,
- "视频监控系统",
- WS_POPUP|WS_OVERLAPPED,
- GetSystemMetrics(SM_CXFULLSCREEN)/2 - 600/2,
- GetSystemMetrics(SM_CYFULLSCREEN)/2-600/2,
- 600, 600,
- NULL,
- NULL,
- hInstance,
- NULL);
- ShowWindow(hwndMain, nShowCmd);
- // 设定主窗口主要区域
- SetWindowRgn(hwndMain, hRoundRgn, 1);
- while( GetMessage(&msg, NULL, 0, 0))
- {
- TranslateMessage(&msg);//该函数将虚拟键消息转换为字符消息。
- //字符消息被寄送到调用线程的消息队列里,当下一次线程调用函数GetMessage或PeekMessage时被读出
- DispatchMessage(&msg);//该函数调度一个消息给窗口程序。
- //通常调度从GetMessage取得的消息。消息被调度到的窗口程序即是MainProc()函数
- }
- return msg.wParam;
- }
- LRESULT CALLBACK MainWndProc( HWND hMain, UINT msg, WPARAM wParam, LPARAM lParam )
- {
- HDC hdc = GetDC(hMain);
- RECT rc; // 客户区
- POINT pt; // 鼠标位置
- HFONT hFont = (HFONT)GetStockObject(DEFAULT_GUI_FONT);
- switch(msg)
- {
- case WM_LBUTTONDBLCLK:
- SetFocus(hMain);
- break;
- case WM_RBUTTONDOWN:
- GetClientRect(hMain, (LPRECT)&rc);
- pt.y = HIWORD(lParam);
- pt.x = LOWORD(lParam);
- if ( PtInRect(&rc, pt) )
- {
- HandlePopupMenu(hMain, pt);
- }
- break;
- case WM_PAINT:
- RECT helpRect, minRect, exitRect;
- HRGN helpRgn, minRgn, exitRgn;
- FrameRgn(hdc, hRoundRgn, CreateSolidBrush(RGB(0,0,0)), 2, 2);
- BringWindowToTop(hwndSelCapDrvDlg);
- return DefWindowProc(hMain, msg, wParam, lParam);
- case WM_CREATE:
- // 注册错误提示回调函数
- capSetCallbackOnError(hwndVideo, (FARPROC)ErrorCallbackProc);
- // 主窗口主要圆角矩形区域
- hRoundRgn = CreateRoundRectRgn(100, 70, 500, 460, 20, 20);
- // 捕捉视频区域s
- hwndVideo = capCreateCaptureWindow((LPSTR)"My Capture Window",
- WS_CHILD|WS_VISIBLE,
- 140, 100, 320, 280,
- (HWND)hMain,
- (int)1);
- hwndExit = CreateWindow("button", // 建立一个按钮,更多如BUTTON,COMBOBOX,LISTBOX
- "x", // button text
- WS_VISIBLE|WS_CHILD|BS_PUSHBUTTON|BS_CENTER|BS_VCENTER,
- 470, 75, BUTTONSIZE, BUTTONSIZE,
- hMain,
- (HMENU)EXIT,
- (HINSTANCE)ghInstance,
- (LPVOID)NULL );
- hwndMin = CreateWindow("button", // 建立一个按钮,更多如BUTTON,COMBOBOX,LISTBOX
- "-", // button text
- WS_VISIBLE|WS_CHILD|BS_PUSHBUTTON|BS_CENTER|BS_VCENTER,
- 450, 75, BUTTONSIZE, BUTTONSIZE,
- hMain,
- (HMENU)MINIMIZE,
- (HINSTANCE)ghInstance,
- (LPVOID)NULL );
- hwndHelp = CreateWindow("button", // 建立一个按钮,更多如BUTTON,COMBOBOX,LISTBOX
- "?", // button text
- WS_VISIBLE|WS_CHILD|BS_PUSHBUTTON/*|BS_CENTER|BS_VCENTER*/,
- 430, 75, BUTTONSIZE, BUTTONSIZE,
- hMain,
- (HMENU)HELP,
- (HINSTANCE)ghInstance,
- (LPVOID)NULL );
- hwndPhoto = CreateWindow("button",
- "照 相",
- WS_CHILD|WS_VISIBLE|BS_PUSHBUTTON,
- 150, 410, 50, 20,
- hMain,
- (HMENU)PHOTO,
- (HINSTANCE)ghInstance,
- (LPVOID)NULL );
- SendMessage(hwndPhoto, WM_SETFONT, (WPARAM)hFont, 1); // 设置按钮字体为系统默认字体
- hwndRecord = CreateWindow("button",
- "录 像",
- WS_CHILD|WS_VISIBLE|BS_PUSHBUTTON|BS_VCENTER|BS_CENTER,
- 220, 410, 50, 20,
- hMain,
- (HMENU)RECORDVIDEO,
- (HINSTANCE)ghInstance,
- (LPVOID)NULL );
- SendMessage(hwndRecord, WM_SETFONT, (WPARAM)hFont, 1);// 设置按钮字体为系统默认字体
- hwndBtnExit = CreateWindow("button", // 退出按钮
- "退 出",
- WS_CHILD|WS_VISIBLE|BS_PUSHBUTTON|BS_VCENTER|BS_CENTER,
- 400, 410, 50, 20,
- hMain,
- (HMENU)EXIT,
- (HINSTANCE)ghInstance,
- (LPVOID)NULL );
- SendMessage(hwndBtnExit, WM_SETFONT, (WPARAM)hFont, 1);
- // 设置按钮区域
- GetClientRect(hwndHelp, &helpRect);
- GetClientRect(hwndMin, &minRect);
- GetClientRect(hwndExit, &exitRect);
- helpRgn = CreateEllipticRgnIndirect(&helpRect);
- minRgn = CreateEllipticRgnIndirect(&minRect);
- exitRgn = CreateEllipticRgnIndirect(&exitRect);
- SetWindowRgn(hwndExit,exitRgn,1);
- SetWindowRgn(hwndMin,minRgn,1);
- SetWindowRgn(hwndHelp,helpRgn,1);
- // 创建选择驱动对话框
- hwndSelCapDrvDlg = CreateDialog((HINSTANCE)ghInstance,
- MAKEINTRESOURCE(DLG_SELCAPDRIVER),
- 0, (DLGPROC)SelCapDrcProc);
- hwndSelCapDrvLBox = GetDlgItem(hwndSelCapDrvDlg, LISTBOX_SELCAPDRIVER);
- EnumCapDrv();
- break;
- case WM_COMMAND:
- CAPSTATUS capStatus;
- switch(wParam)
- {
- case EXIT: // 退出系统
- SendMessage(hMain, WM_SYSCOMMAND, SC_CLOSE, 0);
- break;
- case MINIMIZE: // 最小化
- SendMessage(hMain, WM_SYSCOMMAND, SC_MINIMIZE, 0);
- break;
- case HELP: // 帮助按钮
- SendMessage(hMain, WM_SYSCOMMAND, SC_CONTEXTHELP, 0);
- break;
- case RESOURCE: // 视频源选项
- if (capDrvCaps.fHasDlgVideoSource)
- {
- capDlgVideoSource(hwndVideo);
- }
- break;
- case FORMAT: // 视频格式选项
- if (capDrvCaps.fHasDlgVideoFormat)
- {
- capDlgVideoFormat(hwndVideo);
- }
- break;
- case CONNECT:
- break;
- case DISPLAY: // 视频显示选项
- if (capDrvCaps.fHasDlgVideoDisplay)
- {
- capDlgVideoDisplay(hMain);
- capGetStatus(hwndVideo, &capStatus, sizeof(CAPSTATUS));// 得到当前捕获窗口的状态
- SetWindowPos(hwndVideo, NULL, 0, 0, capStatus.uiImageWidth, capStatus.uiImageHeight, SWP_NOZORDER|SWP_NOMOVE);
- }
- break;
- case PHOTO: // 抓取图片
- if ( capFileSaveDIB(hwndVideo, "test.bmp") ){
- MessageBox(hMain, "Save Bmp File Succeeded!", "photo save", MB_OK|MB_ICONINFORMATION);
- }
- else break;
- if (!bPhotoFileOpen)
- {
- OPENFILENAME ofnp; // open file name structure
- ZeroMemory(&ofnp, sizeof(OPENFILENAME));
- ofnp.lStructSize = sizeof(OPENFILENAME);
- ofnp.hwndOwner = hMain;
- ofnp.lpstrFile = photoFile;
- ofnp.lpstrFilter = "*.bmp";
- ofnp.nMaxFile = sizeof(photoFile);
- ofnp.nFilterIndex = 1;
- ofnp.nMaxFileTitle = 0;
- ofnp.lpstrInitialDir = NULL;
- ofnp.Flags = OFN_PATHMUSTEXIST|OFN_FILEMUSTEXIST;
- // 显示保存文件对话框
- if (GetSaveFileName(&ofnp) == TRUE)
- {
- bPhotoFileOpen = TRUE;
- strcpy(photoFile, ofnp.lpstrFile);
- strcat(photoFile, ".bmp");
- capDriverConnect(hwndPhoto, 0); // 与驱动连接
- capOverlay(hwndPhoto, TRUE); // //使用 hardware overlay
- //使 video input directlyinto framebuffer,使不用 CPU
- capPreview(hwndVideo, TRUE); // 抓取图像显示在视窗
- capPreviewScale(hwndVideo, TRUE); // 使图像延伸到所设定的视窗大小
- capPreviewRate(hwndVideo, 1000/30); // 每秒图像变化张数
- capPreview(hwndVideo, TRUE); // 预览图像显示
- capGrabFrameNoStop(hwndVideo);
- capEditCopy(hwndVideo);
- if (capFileSaveDIB(hwndVideo, photoFile)){
- MessageBox(hMain, "Save Bmp File Succeeded!", "photo save", MB_OK|MB_ICONINFORMATION);
- }
- break;
- }
- }
- break;
- case RECORDVIDEO: // 录制视频
- if (HIWORD(wParam) == BN_CLICKED && (HWND)lParam == hwndRecord)
- {
- OPENFILENAME ofnp;
- if (!bRecordFileOpen)
- {
- ZeroMemory(&ofnp, sizeof(OPENFILENAME));
- ofnp.lStructSize = sizeof(OPENFILENAME);
- ofnp.nMaxFile = sizeof(recordFile);
- ofnp.hwndOwner = hMain;
- ofnp.lpstrFile = recordFile;
- ofnp.lpstrFilter = "*.avi";
- ofnp.lpstrInitialDir = NULL;
- ofnp.nFilterIndex = 1;
- ofnp.lpstrFileTitle = NULL;
- ofnp.nMaxFileTitle = 0;
- ofnp.Flags = OFN_FILEMUSTEXIST|OFN_PATHMUSTEXIST;
- // 显示保存文件对话框
- if (GetSaveFileName(&ofnp) == TRUE)
- {
- strcpy(recordFile, ofnp.lpstrFile);
- strcat(recordFile, ".avi");
- bRecordFileOpen = TRUE;
- // 创建视频采集线程
- DWORD id;
- SECURITY_ATTRIBUTES sa;
- sa.nLength = sizeof(SECURITY_ATTRIBUTES);
- sa.lpSecurityDescriptor = NULL;
- sa.bInheritHandle = TRUE;
- hVideoTread = CreateThread(&sa,
- (ULONG)0,
- (LPTHREAD_START_ROUTINE)VideoThreadProc,
- (LPVOID)(ULONG)0,
- 0,
- &id);
- // 如果创建线程失败,弹出提示对话框
- if (hVideoTread == NULL)
- {
- MessageBox(NULL, "Video Thread Create Failure!", "Thread", MB_OK|MB_ICONINFORMATION);
- }
- }
- break;
- }
- if (bRecordFileOpen) // 已经打开一个选择对话框
- {
- if (bRecording) // 已经在录制视频中
- {
- bThreadEnd = true;
- // 结束录制,并保存
- if ( capFileSaveAs(hwndVideo, recordFile) ){
- MessageBox(NULL, "Save Avi File Succeeded!", "RecordVideo", MB_OK|MB_ICONINFORMATION);
- }
- SetWindowText(hwndRecord, "录 像");
- bRecording = false;
- break;
- }
- else // 没有录像但是选择了一个文件
- {
- int rt = 0;
- rt = MessageBox(hMain, "Do you want write over that file?", "系统提示", MB_YESNO|MB_ICONINFORMATION);
- if (rt == IDYES) // 继续录制,覆盖当前文件
- {
- if (capCaptureSequence(hwndRecord))
- {
- bRecording = true;
- SetWindowText(hwndRecord, "停 止");
- }
- }
- else
- {
- bRecordFileOpen = false;
- SendMessage(hMain, WM_COMMAND, MAKEWPARAM(RECORDVIDEO, BN_CLICKED), (LPARAM)hwndRecord);
- }
- break;
- }
- break;
- }
- }
- }
- break;
- case WM_DESTROY:
- capSetCallbackOnError(hwndVideo, NULL);
- capPreview(hwndVideo, FALSE); // 停止预览
- capCaptureAbort(hwndVideo); // 停止捕获
- capDriverDisconnect(hwndVideo); // 断开驱动连接
- PostQuitMessage(0);
- break;
- default:
- return DefWindowProc(hMain, msg, wParam, lParam);
- }
- return 0;
- }
- LRESULT WINAPI SelCapDrcProc( HWND hMian, UINT msg, WPARAM wParam, LPARAM lParam )
- {
- // ShowWindow(hwndSelCapDrvDlg, SW_SHOW);
- switch(msg)
- {
- case WM_INITDIALOG:
- return TRUE;
- case WM_COMMAND:
- switch(wParam)
- {
- case BTN_SEL:
- int sel = 0;
- SendMessage(hwndSelCapDrvLBox, LB_GETSELITEMS, 1, sel ); // 获取选择的驱动
- SendMessage(hwndVideo, WM_CAP_DRIVER_CONNECT, sel, 0L); // 连接驱动
- SendMessage(hwndSelCapDrvDlg, WM_CLOSE, 0, 0); // 关闭对话框
- SendMessage(hwndVideo, WM_CAP_DRIVER_GET_CAPS, sizeof(CAPDRIVERCAPS), (long)(LPVOID)&capDrvCaps); // 更新驱动的能力
- if (capDrvCaps.fHasOverlay)
- {
- capOverlay(hwndVideo, TRUE);
- }
- capPreviewRate(hwndVideo, 66); //设置视频捕获速率提高到66 miliseconds
- if ( !capPreview(hwndVideo, TRUE)) //用CapPreview启动预览功能
- {
- MessageBox(NULL, "预览启动失败", "提示", MB_OK|MB_ICONINFORMATION);
- SendMessage(hMian, WM_COMMAND, EXIT, 0);
- return FALSE;
- }
- }
- return TRUE;
- case WM_CLOSE:
- DestroyWindow(hwndSelCapDrvDlg);
- return TRUE;
- }
- return (0L);
- }
- int EnumCapDrv()
- {
- char drvName[100];
- char drvVesion[100];
- char drvStr[200];
- for (int i = 0; i < 5; i++) // 扫面获取驱动信息
- {
- if ( capGetDriverDescription(i, drvName, sizeof(drvName), drvVesion, sizeof(drvVesion)) )
- {
- strcpy(drvStr, drvName);
- strcat(drvStr, " ");
- strcat(drvStr, drvVesion);
- SendMessage(hwndSelCapDrvLBox, LB_ADDSTRING, 0, (LPARAM)drvStr); // 向列表框增加字符串
- SendMessage(hwndSelCapDrvLBox, LB_SETITEMDATA, i, (LPARAM)i);
- }
- }
- return 0;
- }
- VOID APIENTRY HandlePopupMenu( HWND hwnd, POINT pt )
- {
- HMENU hmenu;
- HMENU hPopupMenu;
- hmenu = LoadMenu((HINSTANCE)ghInstance, MAKEINTRESOURCE(IDR_POPUPMENU));
- if(hmenu == NULL) return;
- // 获取子菜单
- hPopupMenu = GetSubMenu(hmenu, 0);
- // 弹出菜单使用屏幕坐标,所以转换鼠标点击坐标到屏幕坐标
- ClientToScreen(hwnd, &pt);
- TrackPopupMenu(hPopupMenu, TPM_LEFTALIGN|TPM_LEFTBUTTON, pt.x, pt.y, 0, hwnd, NULL);
- DestroyMenu(hmenu); // 销毁指定菜单,并释放此菜单占用的内存
- }
- WORD WINAPI VideoThreadProc( LPVOID lParam )
- {
- // 捕获视频
- if( capCaptureSequence(hwndVideo) ) //该函数调控启动流视频和音频捕获到一个文件,你能使用这个功能发送消息
- {
- SetWindowText(hwndRecord, "停 止");
- MessageBox(NULL, "Start to capture video.", "提示", MB_OK|MB_ICONINFORMATION);
- bRecording = true;
- }
- while(!bThreadEnd);
- MessageBox(NULL, "Leaving Record Thread", "Thread",NULL);
- return 0;
- }