Использование OpenCV для распознавания рисунков - C++ OpenCV - Ответ 10767924

/*
 * ImageProcessor.cpp
 *
 */
 
#include "ImageProcessor.h"
 
ImageProcessor::ImageProcessor() {
    // TODO Auto-generated constructor stub
 
}
 
ImageProcessor::~ImageProcessor() {
    // TODO Auto-generated destructor stub
}
 
 
cv::Mat* ImageProcessor::getScreenshot(){
    return getScreenshot(0, 0, this->windowsize.right, this->windowsize.bottom);
}
cv::Mat* ImageProcessor::getScreenshot(int x, int y, int width, int height){
    HDC hwindowDC,hwindowCompatibleDC;
    HBITMAP hbwindow;
    cv::Mat* src = new cv::Mat();
    BITMAPINFOHEADER  bi;
 
    hwindowDC=GetDC(this->hwnd);
    hwindowCompatibleDC=CreateCompatibleDC(hwindowDC);
    SetStretchBltMode(hwindowCompatibleDC,COLORONCOLOR);
    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, x, y, width, height, 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
 
    DeleteObject (hbwindow);
    DeleteDC(hwindowCompatibleDC);
    ReleaseDC(this->hwnd, hwindowDC);
    cv::cvtColor(*src, *src, cv::COLOR_RGBA2RGB, 0);
    return src;
}
 
void ImageProcessor::mt                     (cv::Mat* src, cv::Mat* templ, cv::Mat* result, int mtmode){
    result->create(cvSize(src->cols - templ->cols + 1, src->rows - templ->rows + 1), CV_32FC1);
    cv::matchTemplate(*src, *templ, *result, mtmode);
}
void ImageProcessor::mt                     (cv::Mat* src, cv::Mat* templ, cv::Mat* result, cv::Mat* mask, int mtmode){
    result->create(cvSize(src->cols - templ->cols + 1, src->rows - templ->rows + 1), CV_32FC1);
    cv::matchTemplate(*src, *templ, *result, mtmode, *mask);
}
 
void ImageProcessor::mtNorm                 (cv::Mat* src, cv::Mat* templ, cv::Mat* result, int mtmode){
    mt(src, templ, result, mtmode);
    cv::normalize(*result, *result, 1, 0, cv::NORM_MINMAX);
}
void ImageProcessor::mtNorm                 (cv::Mat* src, cv::Mat* templ, cv::Mat* result, cv::Mat* mask, int mtmode){
    mt(src, templ, result, mask, mtmode);
    cv::normalize(*result, *result, 1, 0, cv::NORM_MINMAX);
}
 
void ImageProcessor::mtNormMM               (cv::Mat* src, cv::Mat* templ, cv::Mat* result, cv::Mat* mask, double &minval, double &maxval, cv::Point &minLoc, cv::Point &maxLoc, int mtmode){
    mtNorm(src, templ, result, mask, mtmode);
    cv::minMaxLoc(*result, &minval, &maxval, &minLoc, &maxLoc);
}
void ImageProcessor::mtNormMM               (cv::Mat* src, cv::Mat* templ, cv::Mat* result, double &minval, double &maxval, cv::Point &minLoc, cv::Point &maxLoc, int mtmode){
    mtNorm(src, templ, result, mtmode);
    cv::minMaxLoc(*result, &minval, &maxval, &minLoc, &maxLoc);
}
 
void ImageProcessor::mtMM                   (cv::Mat* src, cv::Mat* templ, cv::Mat* result, cv::Mat* mask, double &minval, double &maxval, cv::Point &minLoc, cv::Point &maxLoc, int mtmode){
    mt(src, templ, result, mask, mtmode);
    cv::minMaxLoc(*result, &minval, &maxval, &minLoc, &maxLoc);
}
void ImageProcessor::mtMM                   (cv::Mat* src, cv::Mat* templ, cv::Mat* result, double &minval, double &maxval, cv::Point &minLoc, cv::Point &maxLoc, int mtmode){
    mt(src, templ, result, mtmode);
    cv::minMaxLoc(*result, &minval, &maxval, &minLoc, &maxLoc);
}
 
void ImageProcessor::mtNormMMGiveScr        (cv::Mat& src, cv::Mat* templ, cv::Mat* result, cv::Mat* mask, double &minval, double &maxval, cv::Point &minLoc, cv::Point &maxLoc, int x, int y, int width, int height, int mtmode){
    src = *getScreenshot(x, y, width, height);
    mtNormMM(&src, templ, result, mask, minval, maxval, minLoc, maxLoc, mtmode);
}
void ImageProcessor::mtNormMMMakeScr        (cv::Mat* templ, cv::Mat* result, cv::Mat* mask, double &minval, double &maxval, cv::Point &minLoc, cv::Point &maxLoc, int x, int y, int width, int height, int mtmode){
    cv::Mat src = *getScreenshot(x, y, width, height);
    mtNormMM(&src, templ, result, mask, minval, maxval, minLoc, maxLoc, mtmode);
}
void ImageProcessor::mtNormMMGiveScr        (cv::Mat& src, cv::Mat* templ, cv::Mat* result, double &minval, double &maxval, cv::Point &minLoc, cv::Point &maxLoc, int x, int y, int width, int height, int mtmode){
    src = *getScreenshot(x, y, width, height);
    mtNormMM(&src, templ, result, minval, maxval, minLoc, maxLoc, mtmode);
}
void ImageProcessor::mtNormMMMakeScr        (cv::Mat* templ, cv::Mat* result, double &minval, double &maxval, cv::Point &minLoc, cv::Point &maxLoc, int x, int y, int width, int height, int mtmode){
    cv::Mat src = *getScreenshot(x, y, width, height);
    mtNormMM(&src, templ, result, minval, maxval, minLoc, maxLoc, mtmode);
}
 
void ImageProcessor::mtMMGiveScr            (cv::Mat& src, cv::Mat* templ, cv::Mat* result, cv::Mat* mask, double &minval, double &maxval, cv::Point &minLoc, cv::Point &maxLoc, int x, int y, int width, int height, int mtmode){
    src = *getScreenshot(x, y, width, height);
    mtMM(&src, templ, result, mask, minval, maxval, minLoc, maxLoc, mtmode);
}
/** Поиск картинки по скриншоту с маской. Скриншот создаёт сам. Ищет экстремумы. */
void ImageProcessor::mtMMMakeScr            (cv::Mat* templ, cv::Mat* result, cv::Mat* mask, double &minval, double &maxval, cv::Point &minLoc, cv::Point &maxLoc, int x, int y, int width, int height, int mtmode){
    cv::Mat src = *getScreenshot(x, y, width, height);
    mtMM(&src, templ, result, mask, minval, maxval, minLoc, maxLoc, mtmode);
}
/** Поиск картинки по скриншоту. Делает скриншот и отдаёт в виде аргумента. Ищет экстремумы. */
void ImageProcessor::mtMMGiveScr            (cv::Mat& src, cv::Mat* templ, cv::Mat* result, double &minval, double &maxval, cv::Point &minLoc, cv::Point &maxLoc, int x, int y, int width, int height, int mtmode){
    src = *getScreenshot(x, y, width, height);
    mtMM(&src, templ, result, minval, maxval, minLoc, maxLoc, mtmode);
}
/** Поиск картинки по скриншоту. Скриншот создаёт сам. Ищет экстремумы. */
void ImageProcessor::mtMMMakeScr            (cv::Mat* templ, cv::Mat* result, double &minval, double &maxval, cv::Point &minLoc, cv::Point &maxLoc, int x, int y, int width, int height, int mtmode){
    cv::Mat src = *getScreenshot(x, y, width, height);
    mtMM(&src, templ, result, minval, maxval, minLoc, maxLoc, mtmode);
}