OPENCV桌面捕获第二部分

时间:2016-04-19 05:37:20

标签: c++ opencv visual-studio-2013 bitmap video-capture

继续上一篇文章OPENCV Destop Capture和此hwnd2mat函数,在opencv中捕获桌面作为源的方法是使用此hwnd2mat函数从屏幕图像创建位图。完成。 这个功能已经从屏幕图像创建了一个位图。但它产生了奇怪的效果,就像在视频内部拖尾一样,我只想要一个更像this one的普通视频源。我已经尝试找到导致它的原因,但我不知道了。

#include "opencv2/imgproc.hpp"
#include "opencv2/highgui.hpp"
#include <Windows.h>
#include <iostream>

using namespace std;
using namespace cv;

Mat hwnd2mat(HWND hwnd)
{
    HDC hwindowDC, hwindowCompatibleDC;

    int height, width, srcheight, srcwidth;
    HBITMAP hbwindow;
    Mat src;
    BITMAPINFOHEADER  bi;

    hwindowDC = GetDC(hwnd);
    hwindowCompatibleDC = CreateCompatibleDC(hwindowDC);
    SetStretchBltMode(hwindowCompatibleDC, COLORONCOLOR);

    RECT windowsize;    // get the height and width of the screen
    GetClientRect(hwnd, &windowsize);

    srcheight = windowsize.bottom;
    srcwidth = windowsize.right;
    height = windowsize.bottom / 1;  //change this to whatever size you want to resize to
    width = windowsize.right / 1;

    src.create(height, width, CV_8UC4);

    // create a bitmap
    hbwindow = CreateCompatibleBitmap(hwindowDC, width, height);
    bi.biSize = sizeof(BITMAPINFOHEADER);    //http://msdn.microsoft.com/en-us/library/windows/window/dd183402%28v=vs.85%29.aspx
    bi.biWidth = width;
    bi.biHeight = -height;  //this is the line that makes it draw upside down or not
    bi.biPlanes = 1;
    bi.biBitCount = 32;
    bi.biCompression = BI_RGB;
    bi.biSizeImage = 0;
    bi.biXPelsPerMeter = 0;
    bi.biYPelsPerMeter = 0;
    bi.biClrUsed = 0;
    bi.biClrImportant = 0;

    // use the previously created device context with the bitmap
    SelectObject(hwindowCompatibleDC, hbwindow);
    // copy from the window device context to the bitmap device context
    StretchBlt(hwindowCompatibleDC, 0, 0, width, height, hwindowDC, 0, 0, srcwidth, srcheight, SRCCOPY); //change SRCCOPY to NOTSRCCOPY for wacky colors !
    GetDIBits(hwindowCompatibleDC, hbwindow, 0, height, src.data, (BITMAPINFO *)&bi, DIB_RGB_COLORS);  //copy from hwindowCompatibleDC to hbwindow

    // avoid memory leak
    DeleteObject(hbwindow);
    DeleteDC(hwindowCompatibleDC);
    ReleaseDC(hwnd, hwindowDC);

    return src;
}

int main(int argc, char **argv)
{
    HWND hwndDesktop = GetDesktopWindow();
    namedWindow("output", CV_WINDOW_NORMAL);
    int key = 0;

    while (key != 30 )
    {
        Mat src = hwnd2mat(hwndDesktop);        
        // you can do some image processing here
        imshow("output", src);
        key = waitKey(30); // you can change wait time
    }
    ReleaseCapture();
    destroyAllWindows();
    return 0;
}

1 个答案:

答案 0 :(得分:0)

使用GetDesktopWindow()将产生无限镜像效果,而不是使用您想要捕获的窗口,而我,我使用FindWindowEx(),并使用Spy ++查找要捕获的窗口,应该有用。但当然会转到下一个bug :)。