Opencv截取屏幕(BitBlt API)

截取整个屏幕

直接获取整个屏幕当前显示画面，便于后期计算处理。

#ifdef _MSC_VER
#pragma comment( linker, "/subsystem:\"windows\" /entry:\"mainCRTStartup\"" )
#endif
#define _AFXDLL//为了方便是用mfc类  
#include<afxwin.h> 
#include <opencv2/opencv.hpp>
#include <opencv2/highgui.hpp>

LPVOID  screenCaptureData = NULL;
HBITMAP hBitmap;
HDC hDDC;
HDC hCDC;
int nWidth;
int nHeight;

void initGDI()
{
	nWidth = GetSystemMetrics(SM_CXSCREEN);//得到屏幕的分辨率的x  
	nHeight = GetSystemMetrics(SM_CYSCREEN);//得到屏幕分辨率的y  
	screenCaptureData = new char[nWidth*nHeight * 4];
	memset(screenCaptureData, 0, nWidth);
	// Get desktop DC, create a compatible dc, create a comaptible bitmap and select into compatible dc.  
	hDDC = GetDC(GetDesktopWindow());//得到屏幕的dc  
	hCDC = CreateCompatibleDC(hDDC);//  
	hBitmap = CreateCompatibleBitmap(hDDC, nWidth, nHeight);//得到位图  
	SelectObject(hCDC, hBitmap); //好像总得这么写。  
}

void gdiScreenCapture(LPVOID screenCaptureData)
{
	BitBlt(hCDC, 0, 0, nWidth, nHeight, hDDC, 0, 0, SRCCOPY);
	GetBitmapBits(hBitmap, nWidth*nHeight * 4, screenCaptureData);//得到位图的数据，并存到screenCaptureData数组中。  
	IplImage *img_tmp = cvCreateImage(cvSize(nWidth, nHeight), 8, 4);//创建一个rgba格式的IplImage,内容为空  
	memcpy(img_tmp->imageData, screenCaptureData, nWidth*nHeight * 4);//这样比较浪费时间，但写的方便,这里必须得是*4。  
	IplImage *img2_tmp = cvCreateImage(cvSize(nWidth, nHeight), 8, 3);//创建一个bgr格式的IplImage，可以没有这个Img2这个变量。 

	cv::Mat img = cv::cvarrToMat(img_tmp);
	cv::Mat img2 = cv::cvarrToMat(img2_tmp);

	cvtColor(img, img2, CV_BGRA2BGR);
	imwrite("rgba.jpg", img);
	cv::waitKey();
}

int main()
{
	initGDI();
	gdiScreenCapture(screenCaptureData);
	return 0;
}