cappicture.cpp

使用Visual C++提供的AVICap窗口类进行视频捕获和播放

#include <windows.h>                   //说明嵌入的头文件
#include <vfw.h>
#include <commdlg.h>
#include "CapPicture.h"

LRESULT WINAPI MainWndProc( HWND, UINT, WPARAM, LPARAM );   //定义函数
LRESULT WINAPI SelCapDrvProc( HWND, UINT, WPARAM, LPARAM );
int EnumCapDrv();
VOID APIENTRY HandlePopupMenu(HWND, POINT);
DWORD WINAPI videoThreadProc(LPVOID lParam);

 //说明常量
HANDLE ghInstance;               
HWND hwndMain;
HWND hwndVideo;
HWND hwndSelCapDrvDlg;
HWND hwndSelCapDrvDlg_LBox;
HWND hwndExit;
HWND hwndMin;
HWND hwndHelp;
HWND hwndRecord;
HANDLE hVideoThread;
HRGN hRegion1;
CAPDRIVERCAPS CapDrvCaps;

//说明变量
bool isRecordFileOpen = false;    
char recordFile[260];
bool isPicFileOpen = false;
char pictureFile[260];
bool isRecording = false;
bool threadEnd = false;

//消息循环函数：每个C程序的第一个进入点

int PASCAL WinMain( HINSTANCE hInstance,
                    HINSTANCE hPrevInstance,
                    LPSTR lpszCmdLine,
                    int nCmdShow )
{
   WNDCLASS wc;
   MSG msg;

   if( !hPrevInstance )
   {
      wc.lpszClassName = "GenericAppClass";
      wc.lpfnWndProc = MainWndProc;                //主线程名称
      wc.style = CS_OWNDC | CS_VREDRAW | CS_HREDRAW;   //设置窗口风格
      wc.hInstance = hInstance;
      wc.hIcon = LoadIcon( NULL, IDI_APPLICATION );
      wc.hCursor = LoadCursor( NULL, IDC_ARROW );            //设置光标
      wc.hbrBackground = CreateSolidBrush (RGB(156, 150, 104)); //设置窗口背景颜色
      wc.lpszMenuName = "GenericAppMenu";

      wc.cbClsExtra = 0;
      wc.cbWndExtra = 0;

      RegisterClass( &wc );
   }

   ghInstance = hInstance;

   //创建主窗口

   hwndMain = CreateWindow( "GenericAppClass",
      "Super Video",
      WS_POPUP,
      100,
      100,
      500,
      500,
      NULL,
      NULL,
      hInstance,
      NULL
   );

   ShowWindow( hwndMain, nCmdShow );
   SetWindowRgn(hwndMain,hRegion1,1);//设置窗口位置和尺寸

   while( GetMessage( &msg, NULL, 0, 0 ) ) {
      TranslateMessage( &msg );

      DispatchMessage( &msg );
   }

   return msg.wParam;
}

//第一个消息处理函数
LRESULT CALLBACK MainWndProc( HWND hwndMain, UINT msg, WPARAM wParam,
   LPARAM lParam )
{
   HDC hDC = GetDC(hwndMain);
   RECT rc;           
   POINT pt;  

   switch( msg ) {
	  case WM_LBUTTONDBLCLK://调整窗口位置？这个消息是由发出的？
	  	    SetFocus(hwndMain);
	  	    break;
      case WM_RBUTTONDOWN: //弹出窗口？这个消息是由发出的？
            GetClientRect(hwndMain, (LPRECT) &rc); 
            pt.x = LOWORD(lParam); 
            pt.y = HIWORD(lParam); 
            if (PtInRect((LPRECT) &rc, pt)) 
                HandlePopupMenu(hwndMain, pt); 
            break; 
      case WM_PAINT: //调整窗口颜色
         FrameRgn(hDC,hRegion1,CreateSolidBrush(RGB(0,0,0)),2,2); 
  		 BringWindowToTop(hwndSelCapDrvDlg);//调整对话框位置
         return( DefWindowProc( hwndMain, msg, wParam, lParam ));
	  case WM_COMMAND:
	  	 CAPSTATUS CapStatus;
	  	 switch( wParam ) {
 		
		 	case SOURCE://这个消息是由发出的？答：右键电击主窗口边缘区域后弹出一个对话框中第一个按钮产生的消息
	  	 	   if(CapDrvCaps.fHasDlgVideoSource)
	  	 	   		capDlgVideoSource(hwndVideo);//设置视频采样参数
	  	 	   break;
	  	 	case FORMAT://这个消息是由发出的？答：右键电击主窗口边缘区域后弹出一个对话框中第二个按钮产生的消息
	  	 	   if(CapDrvCaps.fHasDlgVideoFormat)
	  	 	   {		
	  	 	   		capDlgVideoFormat(hwndMain);//这个函数的作用？答：设置摄像窗口参数
	  	 	   		capGetStatus(hwndVideo, &CapStatus, sizeof(CAPSTATUS));
	  	 	   		SetWindowPos(hwndVideo, NULL, 0, 0, CapStatus.uiImageWidth,
	  	 	   					CapStatus.uiImageHeight, SWP_NOZORDER | SWP_NOMOVE);
	  	 	   }
	  	 	   break;
	  	 	case DISPLAY://这个消息是由选择按钮发出的？答：右键电击主窗口边缘区域后弹出一个对话框中第三个按钮产生的消息
	  	 		if (CapDrvCaps.fHasDlgVideoDisplay)
	  	 			capDlgVideoDisplay(hwndVideo);//设置视频图像参数
	  	 		break;
            case EXIT:
               SendMessage(hwndMain, WM_SYSCOMMAND, SC_CLOSE, 0);//该消息接收者为hwndMain或OS
               break;
            case MINIMIZE:
               SendMessage(hwndMain, WM_SYSCOMMAND, SC_MINIMIZE, 0);//该消息接收者为hwndMain或OS
               break;
            case HELP:
               SendMessage(hwndMain, WM_SYSCOMMAND, SC_CONTEXTHELP, 0); //该消息接收者为hwndMain或OS            
               break;
            case RECORDVIDEO: //这个消息是由HWNDRecord按钮发出的 表示要进行视频捕捉
            if(HIWORD(wParam) == BN_CLICKED && (HWND) lParam == hwndRecord)
            {
			   if (!isRecordFileOpen) //设置录像存储文件并进行第一次视频捕捉
			   {
			   		OPENFILENAME ofn;
			   		ZeroMemory(&ofn, sizeof(OPENFILENAME));
					ofn.lStructSize = sizeof(OPENFILENAME);
					ofn.hwndOwner = hwndMain;
					ofn.lpstrFile = recordFile;
					ofn.nMaxFile = sizeof(recordFile);
					ofn.lpstrFilter = "Video\0*.avi";
					ofn.nFilterIndex = 1;
					ofn.lpstrFileTitle = NULL;
					ofn.nMaxFileTitle = 0;
					ofn.lpstrInitialDir = NULL;
					ofn.Flags = OFN_PATHMUSTEXIST | OFN_FILEMUSTEXIST;
					if(GetSaveFileName(&ofn) == TRUE)
					{
						strcpy(recordFile, ofn.lpstrFile);
						strcat(recordFile, ".avi");
						isRecordFileOpen = true; 
						DWORD id;
						SECURITY_ATTRIBUTES sa;
						sa.nLength = sizeof(SECURITY_ATTRIBUTES);
						sa.lpSecurityDescriptor = NULL;
						sa.bInheritHandle = TRUE;
						hVideoThread = CreateThread(&sa, (ULONG)0, 
											videoThreadProc, (LPVOID)(ULONG)0, (ULONG)0, &id);//第一次视频捕捉	
						if(hVideoThread == NULL)
							MessageBox(NULL, "Creation of Record Thread failed!", "Thread", MB_OK | MB_ICONEXCLAMATION);
						
						break;
					}
				}

				if (isRecordFileOpen) 
				{
					if(isRecording)
					{
						threadEnd = true;
						capFileSaveAs(hwndVideo, recordFile);
						SetWindowText(hwndRecord, "开始下一次录像");
						isRecording = false;
						break;
					}
					
					if(!isRecording )
					{
						int a = 0;
						
						MessageBox(hwndMain, "Do you want to write over the open file?",
									"File warning", MB_YESNO | MB_ICONWARNING);
						if (a != IDYES)
						{
							isRecordFileOpen = false;
							SendMessage(hwndMain, WM_COMMAND, MAKEWPARAM(RECORDVIDEO, BN_CLICKED), (LPARAM) hwndRecord);
						}
						
						if (a == IDYES)
						{
							capCaptureSequence(hwndVideo);//输出文件建立前的图像捕获
							capGrabFrame(hwndVideo);
							isRecording = true;//图像捕获完成
						}
						
					break;
					}
				}
			}	
         	break;
         }
      break;
     case WM_CREATE:  //需要在窗口上建立按钮，名称分别为 EXIT,MINIMIZE,HELP,RECORDVIDEO
		              // 这个消息由谁发出的？
      
      	RECT helpRect, minRect, exitRect;
      	HRGN helpRgn, minRgn, exitRgn;
        hRegion1 = CreateRoundRectRgn(10,100,500, 500,20,20);   
	  	hwndVideo = capCreateCaptureWindow(
	  					(LPSTR) "My Capture Window",
	  					WS_CHILD | WS_VISIBLE,
	  					150, 150, 300, 280,
	  					(HWND) hwndMain,
	  					(int) 1);
		hwndExit = CreateWindow (
				"button",
				"退出",
				WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON | BS_CENTER | BS_VCENTER,
				430, 110, 40, 20,
				hwndMain,
				(HMENU) EXIT,
				(HINSTANCE)ghInstance,
				(LPVOID)NULL);
		hwndMin = CreateWindow (
				"button",
				"隐藏",
				WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON | BS_CENTER | BS_VCENTER,
				380, 110, 40, 20,
				hwndMain,
				(HMENU) MINIMIZE,
				(HINSTANCE)ghInstance,
				(LPVOID)NULL); // 330, 110, BUTTONSIZE, BUTTONSIZE, 		
		hwndHelp = CreateWindow (
				"button",
				"?",
				WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON | BS_CENTER | BS_VCENTER,
				330, 110, 40, 20, 
				hwndMain, 
				(HMENU) HELP,
				(HINSTANCE)ghInstance,
				(LPVOID)NULL);
		hwndRecord = CreateWindow (
				"button", 
				"开始第一次录像", 
				WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON | BS_CENTER | BS_VCENTER, 
				250, 464, 120, 28,
				hwndMain, 
				(HMENU) RECORDVIDEO,
				(HINSTANCE)ghInstance,
				(LPVOID)NULL);				
				
	  	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( SELCAPDRVDLG  ), 
										0, (DLGPROC)SelCapDrvProc);
		hwndSelCapDrvDlg_LBox = GetDlgItem(hwndSelCapDrvDlg, 
   										   SELCAPDRVDLG_LSTBOX);
	    EnumCapDrv();
      break;
      case WM_DESTROY:     //谁发出该消息？本程序中有一个视频选择窗口几消息处理函数未建立
      	 capPreview(hwndVideo, FALSE);
      	 capDriverDisconnect(hwndVideo);
         PostQuitMessage( 0 );
         break;
      default:
         return( DefWindowProc( hwndMain, msg, wParam, lParam ));

   }

   return 0;
}


//第二个消息处理函数
LRESULT CALLBACK SelCapDrvProc( HWND hWnd, UINT msg, 
								WPARAM wParam, LPARAM lParam )
{

	switch(msg)
	{
		case WM_INITDIALOG:
		    return TRUE;
		case WM_COMMAND:
			switch ( wParam )
			{
				case SELCAPDRVDLG_BUTTON://SELECTDRVDLG中的按钮发出的消息
					int sel = 0;
					SendMessage( hwndSelCapDrvDlg_LBox, LB_GETSELITEMS, 1, sel);
					SendMessage( hwndVideo, WM_CAP_DRIVER_CONNECT, sel, 0L); 
					//将摄像机与窗口hwndVideo连接，操作系统的线程接收该消息
					SendMessage( hwndSelCapDrvDlg, WM_CLOSE, 0, 0);
				  // 关闭该对话窗口
					SendMessage( hwndVideo, WM_CAP_DRIVER_GET_CAPS,
								 sizeof(CAPDRIVERCAPS), (LONG) (LPVOID) &CapDrvCaps);
			        //启动摄像机 
					capPreviewRate( hwndVideo, 66 );
					//capPreview( hwndVideo, TRUE );
                 //将摄像机图像显示到窗口hwndVideo
			}
			return TRUE;
		case WM_CLOSE:
			DestroyWindow(hwndSelCapDrvDlg);
			//销毁窗口hwndSelCapDrvDlg
			return TRUE;
	}

	return( 0L );
}

int EnumCapDrv()
{
	char szDeviceName[80];
	char szDeviceVersion[80];
	char item[161];
	int i;

	for (i=0; i<10; i++)
	{
		if ( capGetDriverDescription(i, szDeviceName, sizeof(szDeviceName),
									szDeviceVersion, sizeof(szDeviceVersion)) )
		{
				strcpy(item, szDeviceName);
				strcat(item, " ");
				strcat(item, szDeviceVersion);
				SendMessage(hwndSelCapDrvDlg_LBox, LB_ADDSTRING, 0, 
            			   (LPARAM) item);
            	SendMessage(hwndSelCapDrvDlg_LBox, LB_SETITEMDATA, i, (LPARAM) i);
        }
    }
    
    return 0;
}

VOID APIENTRY HandlePopupMenu(HWND hwnd, POINT pt) 
{ 
    HMENU hmenu;    
    HMENU hmenuTrackPopup;
	hmenu = LoadMenu((HINSTANCE)ghInstance, "PopupMenu"); 
    if (hmenu == NULL) 
        return; 
    hmenuTrackPopup = GetSubMenu(hmenu, 0); 
    ClientToScreen(hwnd, (LPPOINT) &pt); 
    TrackPopupMenu(hmenuTrackPopup, TPM_LEFTALIGN | TPM_LEFTBUTTON, 
        pt.x, pt.y, 0, hwnd, NULL); 
    DestroyMenu(hmenu); 
}

DWORD WINAPI videoThreadProc(LPVOID lParam)
{
	SetWindowText(hwndRecord, "停止本次录像");
	capCaptureSequence(hwndVideo);
	capGrabFrameNoStop(hwndVideo);
	//capCaptureSingleFrameOpen(hwndVideo);已通过受动打开输出文件
	capCaptureSingleFrame(hwndVideo);
	char *szName="H:/MyFrame.bmp";
	capFileSaveDIB(hwndVideo,szName);//将当前frame存储到位图文件H:/MyFrame.bmp中
	//capCaptureSingleFrameClose(hwndVideo);
	isRecording = true;
	while (!threadEnd)
		;
	MessageBox(NULL, "关闭本次录像线程", "thread", NULL);
	return 0;
}