實時截圖轉為opencv格式處理
實時截圖 轉為 openCV 格式處理
(2011-09-21 14:11:08) 轉載▼標籤: it |
分類: 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;
}