程式師世界 >> 編程語言 >> C語言 >> VC >> 關於VC++ >> 抓圖程序開發實踐

抓圖程序開發實踐

編輯：關於VC++

前言

本程序是利用通用的熱鍵管理DLL實現的熱鍵功能，在收到熱鍵通知後截取屏幕的內容並保存到bmp文件中。本文提供了全部源代碼供大家參考。

程序運行界面

例如我們設置圖片保存路徑為c:\,熱鍵為F9 + Control,然後按Change按鈕設置好熱鍵，當我們在任意程序中按下CTRL+F9,當前界面將被保存在C:\下的BMP文件中。下面我們來看看程序的實現原理與基本結構。

一、熱鍵管理DLL

這個DLL實際上是一個鍵盤鉤子，由它來監視系統的鍵盤事件。如果有和程序登記符合的按鍵和按鍵組合就通知該程序的窗口。為了應用方便，我把它做成了一個標准的管理庫來為其它的程序通過熱鍵服務，它有兩個輸出函數：AddHotkey和DeleteHotkey，程序只需要調用這兩個函數就可以了，如果編譯之後不用改變熱鍵，則只需要AddHotkey就可以了。

1. 全局變量

所有的全局變量都放在一個共享段中，定義如下：

#pragma data_seg("shareddata") 　　HHOOK hHook =NULL; //鉤子句柄　　UINT nHookCount =0; //掛接的程序數目　　static UCHAR HotKey[MAX_KEY] = {0}; //熱鍵虛擬鍵碼　　static UCHAR HotKeyMask[MAX_KEY] = {0}; //組合掩碼, control=4,alt=2,shift=1 可以“或” 　　static HWND hCallWnd[MAX_KEY] = {0}; //window handle associated with hotkey 　　static int KeyCount =0; 　　static UCHAR MaskBits =0; //00000 Ctrl=4 & Alt=2 & Shift=1 #pragma data_seg()

關於共享段，有幾點重要的說明：

a. 必須在鏈接選項裡指定該段為共享，project->settings->link->object/library，加上/section:shareddata,rws

b. 第二種指定共享段的方法：在def文件的sections裡加上一句shareddata read write shared

c. 第三種指定共享段的方法：在程序裡加上一句#pragma comment(linker,"section:shareddata,rws")

d. 所有的變量必須初始化，否則鏈接程序會把它放到普通數據段

e. 如果不初始化變量，需要在段外用“__declspec(allocate("shareddata")) 變量類型變量名”的方式定義

2. 兩個輸出函數如下：

//添加熱鍵，如果cKey和cMask均為0則監視所有按鍵事件 BOOL __declspec(dllexport) __stdcall AddHotkey(HWND hWnd,UCHAR cKey,UCHAR cMask) { 　　BOOL bAdded=FALSE; 　　for(int index=0;index<MAX_KEY;index++){ 　　　　if(hCallWnd[index]==0){ 　　　　　hCallWnd[index]=hWnd; 　　　　　HotKey[index]=cKey; 　　　　　HotKeyMask[index]=cMask; 　　　　　bAdded=TRUE; 　　　　　KeyCount++; 　　　　　break; 　　　　} 　　} 　　return bAdded; } //刪除熱鍵 BOOL __declspec(dllexport) __stdcall DeleteHotkey(HWND hWnd,UCHAR cKey,UCHAR　cMask) { 　　BOOL bRemoved=FALSE; 　　for(int index=0;index<MAX_KEY;index++){ 　　　　if(hCallWnd[index]==hWnd){ 　　　　　if(HotKey[index]==cKey&&HotKeyMask[index]==cMask){ 　　　　　　hCallWnd[index]=NULL; 　　　　　　HotKey[index]=0; 　　　　　　HotKeyMask[index]=0; 　　　　　　bRemoved=TRUE; 　　　　　　KeyCount--; 　　　　　　break; 　　　　　} 　　　　} 　　} 　　return bRemoved; }

3. 鉤子函數如下：

LRESULT CALLBACK KeyboardProc(int nCode,WPARAM wParam,LPARAM lParam) { 　　BOOL bProcessed=FALSE; 　　if(HC_ACTION==nCode) 　　{ 　　　　if((lParam&0xc0000000)==0xc0000000){// 有鍵松開　　　　　switch(wParam) 　　　　　{ 　　　　　case VK_MENU: 　　　　　MaskBits&=~ALTBIT; 　　　　　break; 　　　　　case VK_CONTROL: 　　　　　MaskBits&=~CTRLBIT; 　　　　　break; 　　　　　case VK_SHIFT: 　　　　　MaskBits&=~SHIFTBIT; 　　　　　break; 　　　　　default: //judge the key and send message 　　　　　break; 　　　　　} 　　　　　for(int index=0;index<MAX_KEY;index++){ 　　　　　if(hCallWnd[index]==NULL) 　　　　　continue; 　　　　　if(IsWindow(hCallWnd[index])&&(HotKey[index]==wParam)&&(HotKeyMask[index]==MaskBits)){ 　　　　　SendMessage(hCallWnd[index],WM_HOTKEY,wParam,WM_KEYUP); 　　　　　bProcessed=TRUE; 　　　　　} 　　　　　} 　　　　} 　　　　else if((lParam&0xc000ffff)==1){ //有鍵按下　　　　　switch(wParam) 　　　　　{ 　　　　　case VK_MENU: 　　　　　MaskBits|=ALTBIT; 　　　　　break; 　　　　　case VK_CONTROL: 　　　　　MaskBits|=CTRLBIT; 　　　　　break; 　　　　　case VK_SHIFT: 　　　　　MaskBits|=SHIFTBIT; 　　　　　break; 　　　　　default: //judge the key and send message 　　　　　break; 　　　　　} 　　　　　for(int index=0;index<MAX_KEY;index++){ 　　　　　if(hCallWnd[index]==NULL) 　　　　　　continue; 　　　　　if(IsWindow(hCallWnd[index])&&(HotKey[index]==wParam)&&(HotKeyMask[index]==MaskBits)){ 　　　　　SendMessage(hCallWnd[index],WM_HOTKEY,wParam,WM_KEYDOWN); 　　　　　bProcessed=TRUE; 　　　　　} 　　　　　} 　　　　} 　　　　if(!bProcessed){ //一般按鍵事件，為監視鍵盤的程序留出余地　　　　for(int index=0;index<MAX_KEY;index++){ 　　　　if(hCallWnd[index]==NULL) 　　　　continue; 　　　　if(IsWindow(hCallWnd[index])&&(HotKey[index]==0)&&(HotKeyMask[index]==0)) 　　　　SendMessage(hCallWnd[index],WM_HOTKEY,WM_HOTKEY,lParam); //lParam的意義可看MSDN中WM_KEYDOWN部分　　　　} 　　　　} 　　} 　　return CallNextHookEx( hHook, nCode, wParam, lParam ); }

這樣，熱鍵管理的DLL就做好了，我們就可以在程序裡調用AddHotkey了

二、抓圖程序

抓圖程序是一個基於對話框的程序，它在建立對話框的時候調用前面的DLL，登記熱鍵

1. 添加hook.lib到工程裡，在程序裡給出兩個DLL函數的定義，也可以寫個頭文件，再包含進來

BOOL __declspec(dllexport)__stdcall AddHotkey(HWND,UCHAR key,UCHAR mask); BOOL __declspec(dllexport)__stdcall DeleteHotkey(HWND,UCHAR key,UCHAR mask);

也可以用動態載入的辦法調用

2. 添加對熱鍵通知事件的響應，程序中用的辦法是重載WindowProc

LRESULT CCaptureDlg::WindowProc(UINT message, WPARAM wParam, LPARAM lParam)　 { 　　if(message==WM_HOTKEY&&lParam==WM_KEYDOWN){ //lParam表示是按下還是松開，如果有多個熱鍵，由wParam來區分　　　SaveBmp(); 　　　return FALSE; 　　} 　　…… //其它處理及默認處理 }

3. 保存屏幕

保存屏幕由函數SaveBmp完成，如下：

void CCaptureDlg::SaveBmp() { 　　CDC dc; 　　dc.CreateDC("DISPLAY",NULL,NULL,NULL); 　　CBitmap bm; 　　int Width=GetSystemMetrics(SM_CXSCREEN); 　　int Height=GetSystemMetrics(SM_CYSCREEN); 　　bm.CreateCompatibleBitmap(&dc,Width,Height); 　　CDC tdc; 　　tdc.CreateCompatibleDC(&dc); 　　CBitmap*pOld=tdc.SelectObject(&bm); 　　tdc.BitBlt(0,0,Width,Height,&dc,0,0,SRCCOPY); 　　tdc.SelectObject(pOld); 　　BITMAP btm; 　　bm.GetBitmap(&btm); 　　DWORD size=btm.bmWidthBytes*btm.bmHeight; 　　LPSTR lpData=(LPSTR)GlobalAllocPtr(GPTR,size); 　　///////////////////////////////////////////// 　　BITMAPINFOHEADER bih; 　　bih.biBitCount=btm.bmBitsPixel; 　　bih.biClrImportant=0; 　　bih.biClrUsed=0; 　　bih.biCompression=0; 　　bih.biHeight=btm.bmHeight; 　　bih.biPlanes=1; 　　bih.biSize=sizeof(BITMAPINFOHEADER); 　　bih.biSizeImage=size; 　　bih.biWidth=btm.bmWidth; 　　bih.biXPelsPerMeter=0; 　　bih.biYPelsPerMeter=0; 　　GetDIBits(dc,bm,0,bih.biHeight,lpData,(BITMAPINFO*)&bih,DIB_RGB_COLORS); 　　//bm.GetBitmapBits(size,lpData);//此函數在處理5-5-5模式的16位色下會出現顏色混亂　　static int filecount=0; 　　CString name; 　　name.Format("pict%04d.bmp",filecount++); 　　name=m_Path+name; 　　BITMAPFILEHEADER bfh; 　　bfh.bfReserved1=bfh.bfReserved2=0; 　　bfh.bfType=((WORD)('M'<< 8)|'B'); 　　bfh.bfSize=54+size; 　　bfh.bfOffBits=54; 　　CFile bf; 　　if(bf.Open(name,CFile::modeCreate|CFile::modeWrite)){ 　　　　bf.WriteHuge(&bfh,sizeof(BITMAPFILEHEADER)); 　　　　bf.WriteHuge(&bih,sizeof(BITMAPINFOHEADER)); 　　　　bf.WriteHuge(lpData,size); 　　　　bf.Close(); 　　　　nCount++; 　　} 　　GlobalFreePtr(lpData); 　　if(nCount==1) 　　　　m_Number.Format("%d picture captured.",nCount); 　　else 　　　　m_Number.Format("%d pictures captured.",nCount); 　　UpdateData(FALSE); }