C++ Получение RGB из hBitmap

Question

Работа с растровыми изображениями очень полезна для меня, поэтому я действительно борюсь с онлайн-учебниками и стратегиями, которые я прочитал. В основном моя цель - сканировать экран для определенного значения RGB. Я считаю, что шаги для этого - захватить экран в hBitmap, а затем создать из него массив значений RGB, которые я могу проверить.

Первоначально я начал с GetPixel, но это очень медленно. Решением было использование GetDIBits, которое создает массив значений RGB. Проблема в том, что он возвращает странные и, возможно, случайные значения RGB.

Я использую следующий код, который я нашел из другого учебника:

/* Globals */ 
int ScreenX = GetDeviceCaps(GetDC(0), HORZRES); 
int ScreenY = GetDeviceCaps(GetDC(0), VERTRES); 
BYTE* ScreenData = new BYTE[3*ScreenX*ScreenY]; 

void ScreenCap() { 
    HDC hdc = GetDC(GetDesktopWindow()); 
    HDC hdcMem = CreateCompatibleDC (hdc); 
    HBITMAP hBitmap = CreateCompatibleBitmap(hdc, ScreenX, ScreenY); 
    BITMAPINFOHEADER bmi = {0}; 
    bmi.biSize = sizeof(BITMAPINFOHEADER); 
    bmi.biPlanes = 1; 
    bmi.biBitCount = 24; 
    bmi.biWidth = ScreenX; 
    bmi.biHeight = -ScreenY; 
    bmi.biCompression = BI_RGB; 
    bmi.biSizeImage = ScreenX * ScreenY; 
    SelectObject(hdcMem, hBitmap); 
    BitBlt(hdcMem, 0, 0, ScreenX, ScreenY, hdc, 0, 0, SRCCOPY); 
    GetDIBits(hdc, hBitmap, 0, ScreenY, ScreenData, (BITMAPINFO*)&bmi, DIB_RGB_COLORS); 

    DeleteDC(hdcMem); 
    ReleaseDC(NULL, hdc); 
} 

inline int PosR(int x, int y) { 
    return ScreenData[3*((y*ScreenX)+x)+2]; 
} 

inline int PosG(int x, int y) { 
    return ScreenData[3*((y*ScreenX)+x)+1]; 
} 

inline int PosB(int x, int y) { 
    return ScreenData[3*((y*ScreenX)+x)]; 
} 

я проверить это с помощью следующего кода. Я нажал Shift, чтобы вызвать ScreenCap, а затем переместите курсор в нужное место и нажмите «Пробел», чтобы увидеть, какое значение RGB находится в этом месте. Я совершенно орехи?

int main() { 

while (true) { 

    if (GetAsyncKeyState(VK_SPACE)){ 

     // Print out current cursor position 
     GetCursorPos(&p); 
     printf("X:%d Y:%d \n",p.x,p.y); 
     // Print out RGB value at that position 
     int r = PosR(p.x, p.y); 
     int g = PosG(p.x, p.y); 
     int b = PosB(p.x, p.y); 
     printf("r:%d g:%d b:%d \n",r,g,b); 

    } else if (GetAsyncKeyState(VK_ESCAPE)){ 
     printf("Quit\n"); 
     break; 
    } else if (GetAsyncKeyState(VK_SHIFT)){ 
     ScreenCap(); 
     printf("Captured\n"); 
    } 
} 

system("PAUSE"); 
return 0; 
} 

Answer 1

Вопрос заключается в том, что экран на самом деле 32бит глубоко не 24. Код ниже даст вам результат вам нужно:

/* Globals */ 
int ScreenX = 0; 
int ScreenY = 0; 
BYTE* ScreenData = 0; 

void ScreenCap() 
{ 
    HDC hScreen = GetDC(GetDesktopWindow()); 
    ScreenX = GetDeviceCaps(hScreen, HORZRES); 
    ScreenY = GetDeviceCaps(hScreen, VERTRES); 

    HDC hdcMem = CreateCompatibleDC (hScreen); 
    HBITMAP hBitmap = CreateCompatibleBitmap(hScreen, ScreenX, ScreenY); 
    HGDIOBJ hOld = SelectObject(hdcMem, hBitmap); 
    BitBlt(hdcMem, 0, 0, ScreenX, ScreenY, hScreen, 0, 0, SRCCOPY); 
    SelectObject(hdcMem, hOld); 

    BITMAPINFOHEADER bmi = {0}; 
    bmi.biSize = sizeof(BITMAPINFOHEADER); 
    bmi.biPlanes = 1; 
    bmi.biBitCount = 32; 
    bmi.biWidth = ScreenX; 
    bmi.biHeight = -ScreenY; 
    bmi.biCompression = BI_RGB; 
    bmi.biSizeImage = 0;// 3 * ScreenX * ScreenY; 

    if(ScreenData) 
     free(ScreenData); 
    ScreenData = (BYTE*)malloc(4 * ScreenX * ScreenY); 

    GetDIBits(hdcMem, hBitmap, 0, ScreenY, ScreenData, (BITMAPINFO*)&bmi, DIB_RGB_COLORS); 

    ReleaseDC(GetDesktopWindow(),hScreen); 
    DeleteDC(hdcMem); 
    DeleteObject(hBitmap); 
} 

inline int PosB(int x, int y) 
{ 
    return ScreenData[4*((y*ScreenX)+x)]; 
} 

inline int PosG(int x, int y) 
{ 
    return ScreenData[4*((y*ScreenX)+x)+1]; 
} 

inline int PosR(int x, int y) 
{ 
    return ScreenData[4*((y*ScreenX)+x)+2]; 
} 

bool ButtonPress(int Key) 
{ 
    bool button_pressed = false; 

    while(GetAsyncKeyState(Key)) 
     button_pressed = true; 

    return button_pressed; 
} 

int main() 
{ 
    while (true) 
    { 
     if (ButtonPress(VK_SPACE)) 
     { 

      // Print out current cursor position 
      POINT p; 
      GetCursorPos(&p); 
      printf("X:%d Y:%d \n",p.x,p.y); 
      // Print out RGB value at that position 
      std::cout << "Bitmap: r: " << PosR(p.x, p.y) << " g: " << PosG(p.x, p.y) << " b: " << PosB(p.x, p.y) << "\n"; 

     } else if (ButtonPress(VK_ESCAPE)) 
     { 
      printf("Quit\n"); 
      break; 
     } else if (ButtonPress(VK_SHIFT)) 
     { 
      ScreenCap(); 
      printf("Captured\n"); 
     } 
    } 

    system("PAUSE"); 
    return 0; 
} 

Answer 2

Ваш размер указан в пикселей, он должен быть указан в байт

**bmi.biSizeImage = ScreenX * ScreenY;** 
**bmi.biBitCount = 24;** 
bmi.biWidth = ScreenX; 
bmi.biHeight = -ScreenY; 
**bmi.biCompression = BI_RGB;** 

biSizeImage его определенные единицы байт и вы задаете RGB 3 байта на пиксели.

http://msdn.microsoft.com/en-us/library/windows/desktop/dd183376(v=vs.85).aspx

biSizeImage размер, в байт, изображения. Для битовых карт BI_RGB это может быть установлено равным нулю.