1 #ifndef _CAPTURESCREEN_H 2 #define _CAPTURESCREEN_H 3 4 #include <windows.h> 5 class CaptureScreen 6 { 7 public: 8 CaptureScreen() 9 {}; 10 ~CaptureScreen(void) 11 {}; 12 13 void ScreenInit(); 14 void ScreenFree(); 15 bool ScreenSetWindow(LPRECT lpRect); 16 bool ScreenCopy(BYTE *dstBuf); 17 18 int rHeight; //要获得的图像尺寸 19 int rWidth; 20 private: 21 int nScreenWidth; 22 int nScreenHeight; 23 PRGBTRIPLE scan0; 24 BITMAPINFO pbi; 25 int stride; 26 int top_x; 27 int top_y; 28 int bot_x; 29 int bot_y; 30 }; 31 32 #endif
1 #include "CaptureScreen.h" 2 3 void CaptureScreen::ScreenInit() 4 { 5 nScreenWidth = GetSystemMetrics(SM_CXSCREEN); 6 nScreenHeight = GetSystemMetrics(SM_CYSCREEN); 7 8 pbi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER); 9 pbi.bmiHeader.biWidth = nScreenWidth; 10 pbi.bmiHeader.biHeight = nScreenHeight; 11 pbi.bmiHeader.biPlanes = 1; 12 pbi.bmiHeader.biBitCount = 24; 13 pbi.bmiHeader.biCompression = BI_RGB; 14 stride = ((nScreenWidth * 24 + 31) & 0xffffffe0) / 8; // 24位图像扫描线宽度 15 scan0 = (PRGBTRIPLE)malloc(stride * nScreenHeight); // 图像数据缓冲区，应释放 16 } 17 18 void CaptureScreen::ScreenFree() 19 { 20 free(scan0); 21 } 22 23 bool CaptureScreen::ScreenSetWindow(LPRECT lpRect) 24 { 25 top_x = lpRect->left; 26 top_y = lpRect->top; 27 bot_x = lpRect->right; 28 bot_y = lpRect->bottom; 29 if(top_x >= bot_x) 30 return false; 31 if(top_y >= bot_y) 32 return false; 33 if(top_x<0 || bot_x>nScreenWidth) 34 return false; 35 if(top_y<0 || bot_y>nScreenHeight) 36 return false; 37 38 rHeight = bot_y-top_y; 39 rWidth = bot_x-top_x; 40 return true; 41 } 42 43 bool CaptureScreen::ScreenCopy(BYTE *dstBuf) 44 { 45 HWND hDesktopWnd = GetDesktopWindow(); 46 HDC hDesktopDC = GetDC(hDesktopWnd); 47 HDC hCaptureDC = CreateCompatibleDC(hDesktopDC); 48 HBITMAP hCaptureBitmap =CreateCompatibleBitmap(hDesktopDC,nScreenWidth, nScreenHeight); 49 SelectObject(hCaptureDC,hCaptureBitmap); 50 BitBlt(hCaptureDC,0,0,nScreenWidth,nScreenHeight,hDesktopDC,0,0,SRCCOPY|NOMIRRORBITMAP); 51 DeleteDC(hCaptureDC); 52 53 GetDIBits(hDesktopDC, hCaptureBitmap, 0, nScreenHeight, scan0, &pbi, DIB_RGB_COLORS); 54 ReleaseDC(hDesktopWnd,hDesktopDC); 55 DeleteObject(hCaptureBitmap); 56 57 BYTE *srcBuf = (BYTE*) scan0; 58 59 int j; 60 srcBuf += (nScreenHeight-1-top_y)*stride+top_x*3;// scan0倒序 61 for(j=0;j<rHeight;j++) 62 { 63 memcpy(dstBuf,srcBuf,rWidth*3); 64 dstBuf += rWidth*3; 65 srcBuf -= stride; 66 } 67 return true; 68 }
1 #include "CaptureScreen.h" 2 #include <stdio.h> 3 #include <Mmsystem.h> 4 #pragma comment( lib,"winmm.lib" ) 5 void SaveBmp2(BYTE *bgr32,int w,int h,int mode) 6 { 7 8 ///////////////////////////////////////////// 9 DWORD size=w*h*3; 10 LPSTR lpData=(LPSTR)GlobalAlloc(GPTR,size); 11 ///////////////////////////////////////////// 12 BITMAPINFOHEADER bih; 13 bih.biSize=sizeof(BITMAPINFOHEADER); 14 bih.biWidth=w; 15 bih.biHeight=h; 16 bih.biPlanes=1; 17 bih.biBitCount=24; 18 bih.biCompression=0; 19 bih.biSizeImage=w*h*3;//size; 20 bih.biXPelsPerMeter=0; 21 bih.biYPelsPerMeter=0; 22 bih.biClrUsed=0; 23 bih.biClrImportant=0; 24 25 BITMAPFILEHEADER bfh; 26 bfh.bfType=0x4D42; // "BM" // 设置位图文件头 27 bfh.bfReserved1=bfh.bfReserved2=0; 28 bfh.bfSize=54+size; 29 bfh.bfOffBits=54; 30 31 if(mode == 0) 32 { 33 for(int i=0;i<h;i++) 34 { 35 memcpy(lpData+(h-1-i)*w*3,bgr32+i*w*3,w*3); 36 } 37 } 38 else 39 memcpy(lpData,bgr32,w*h*3); 40 41 static int filecount=0; 42 char filename[255]; 43 sprintf(filename,"C:\%d.bmp",filecount); 44 FILE *fp=fopen(filename,"wb"); 45 46 if(fp) 47 { 48 49 fwrite(&bfh,1,sizeof(BITMAPFILEHEADER),fp); 50 fwrite(&bih,1,sizeof(BITMAPINFOHEADER),fp); 51 fwrite(lpData,1,size,fp); 52 fclose(fp); 53 filecount++; 54 } 55 GlobalFree(lpData); 56 57 } 58 59 int main() 60 { 61 62 CaptureScreen myCapture; 63 myCapture.ScreenInit(); 64 RECT dstRect={0,0,1920,1080}; 65 myCapture.ScreenSetWindow(&dstRect); 66 BYTE *dstBuff = (BYTE*)malloc(myCapture.rWidth*myCapture.rHeight*3); 67 int lastTime = timeGetTime(); 68 myCapture.ScreenCopy(dstBuff); 69 printf("time:%dms",timeGetTime()-lastTime); 70 SaveBmp2((BYTE*)dstBuff,myCapture.rWidth,myCapture.rHeight,0); 71 myCapture.ScreenFree(); 72 73 74 return 0; 75 }