實時截圖 轉為 openCV 格式處理

轉載▼

// abs_image.cpp : 定義控制檯應用程式的入口點。
//

#include "stdafx.h"
#include "cv.h"
#include "highgui.h"

//截圖圖片引數
int image_width;
int image_height;
int image_depth;
int image_nchannels;

IplImage* screemImage;
int flag=0;

//截圖函式
void CopyScreenToBitmap()
{
int right=GetSystemMetrics(SM_CXSCREEN),left=0,top=0,bottom=GetSystemMetrics(SM_CYSCREEN);//定義截圖範圍此處設為全屏

int nWidth, nHeight;

HDC hSrcDC = NULL, hMemDC = NULL;
HBITMAP hBitmap = NULL, hOldBitmap = NULL;

hSrcDC = CreateDC(L"DISPLAY", NULL, NULL, NULL);
hMemDC = CreateCompatibleDC(hSrcDC);

nWidth = right - left;
nHeight = bottom - top;

hBitmap = CreateCompatibleBitmap(hSrcDC, nWidth, nHeight);
hOldBitmap = (HBITMAP)SelectObject(hMemDC, hBitmap);

BitBlt(hMemDC, 0, 0, nWidth, nHeight, hSrcDC, left, top , SRCCOPY);
hBitmap = (HBITMAP)SelectObject(hMemDC, hOldBitmap);

BITMAP bmp;
int nChannels,depth;
BYTE *pBuffer;

GetObject(hBitmap,sizeof(BITMAP),&bmp);

image_nchannels = bmp.bmBitsPixel == 1 ? 1 : bmp.bmBitsPixel/8 ;
image_depth = bmp.bmBitsPixel == 1 ? IPL_DEPTH_1U : IPL_DEPTH_8U;
image_width=bmp.bmWidth;
image_height=bmp.bmHeight;

if(flag==0)
{
screemImage=cvCreateImage(cvSize(image_width,image_height),image_depth,image_nchannels);
flag=1;
}

pBuffer = new BYTE[image_width*image_height*image_nchannels];
GetBitmapBits(hBitmap,image_height*image_width*image_nchannels,pBuffer);
memcpy(screemImage->imageData,pBuffer,image_height*image_width*image_nchannels);

delete pBuffer;

SelectObject(hMemDC,hOldBitmap);
DeleteObject(hOldBitmap);
DeleteDC(hMemDC);

SelectObject(hSrcDC,hBitmap);
DeleteDC(hMemDC);
DeleteObject(hBitmap);
}

int _tmain(int argc, _TCHAR* argv[])
{

IplImage* screenRGB=0;
IplImage* screen_resize=0;

while(1)
{

CopyScreenToBitmap(); //得到的圖片為RGBA格式,即4通道。

if(!screen_resize)screen_resize=cvCreateImage(cvSize(320,240),image_depth,image_nchannels);
cvResize(screemImage,screen_resize,CV_INTER_LINEAR);

if(!screenRGB)screenRGB=cvCreateImage(cvSize(320,240),IPL_DEPTH_8U,3);
cvCvtColor(screen_resize,screenRGB,CV_RGBA2RGB);

cvShowImage("s_laplace",screenRGB);

cvWaitKey(10);
}
cvDestroyAllWindows();
return 0;
}