Ошибка PIL ImageGrab на втором виртуальном мониторе VirtualBox - PullRequest
Новая
       29

Ошибка PIL ImageGrab на втором виртуальном мониторе VirtualBox

2 голосов
/ 27 августа 2010

Я настроил VirtualBox для использования двух мониторов. Я попытался сделать скриншот окна на втором мониторе: 

import ImageGrab
im = ImageGrab.grab(windowrect)
im.save("img.png")

windowrect Я подтвердил, что являюсь правильным прямоугольником окна, в данном случае (1616, 2, 2594, 732). Однако img.png - это просто большой черный ящик. Любые идеи, как исправить это взаимодействие между захватом экрана и VirtualBox, чтобы я мог делать скриншоты 2-го виртуального монитора?

Ответы [ 2 ]

2 голосов
/ 10 ноября 2013

Я сталкивался с такой же проблемой. Если бы мне пришлось угадывать, я бы предположил, что ImageGrab получает свои координаты из флага SM_SCREEN, который дает координаты только основного монитора, а не SM_VIRTUALSCREEN, который дает весь виртуальный экран (я не смотрел хотя через источник, так что просто предположение).

Тем не менее, достаточно просто обойти это, взаимодействуя напрямую с Windows API, а затем преобразовав его растровый вывод в более удобный объект PIL.

Некоторые коды: 

def _get_screen_buffer(self, bounds=None):
                # Grabs a DC to the entire virtual screen, but only copies to 
                # the bitmap the the rect defined by the user. 

                SM_XVIRTUALSCREEN = 76  # coordinates for the left side of the virtual screen. 
                SM_YVIRTUALSCREEN = 77  # coordinates for the right side of the virtual screen.  
                SM_CXVIRTUALSCREEN = 78 # width of the virtual screen
                SM_CYVIRTUALSCREEN = 79 # height of the virtual screen

                hDesktopWnd = windll.user32.GetDesktopWindow() #Entire virtual Screen

                left = windll.user32.GetSystemMetrics(SM_XVIRTUALSCREEN)
                top = windll.user32.GetSystemMetrics(SM_YVIRTUALSCREEN)
                width = windll.user32.GetSystemMetrics(SM_CXVIRTUALSCREEN)
                height = windll.user32.GetSystemMetrics(SM_CYVIRTUALSCREEN)

                if bounds:
                        left, top, right, bottom = bounds
                        width = right - left 
                        height = bottom - top

                hDesktopDC = windll.user32.GetWindowDC(hDesktopWnd)
                if not hDesktopDC: print 'GetDC Failed'; sys.exit()

                hCaptureDC = windll.gdi32.CreateCompatibleDC(hDesktopDC)
                if not hCaptureDC: print 'CreateCompatibleBitmap Failed'; sys.exit()

                hCaptureBitmap = windll.gdi32.CreateCompatibleBitmap(hDesktopDC, width, height)
                if not hCaptureBitmap: print 'CreateCompatibleBitmap Failed'; sys.exit()

                windll.gdi32.SelectObject(hCaptureDC, hCaptureBitmap)

                SRCCOPY = 0x00CC0020
                windll.gdi32.BitBlt(
                        hCaptureDC, 
                        0, 0, 
                        width, height, 
                        hDesktopDC, 
                        left, top, 
                        0x00CC0020
                )
                return hCaptureBitmap

        def _make_image_from_buffer(self, hCaptureBitmap):
                import Image
                bmp_info = BITMAPINFO()
                bmp_header = BITMAPFILEHEADER()
                hdc = windll.user32.GetDC(None)

                bmp_info.bmiHeader.biSize = sizeof(BITMAPINFOHEADER)

                DIB_RGB_COLORS = 0
                windll.gdi32.GetDIBits(hdc, 
                        hCaptureBitmap, 
                        0,0, 
                        None, byref(bmp_info), 
                        DIB_RGB_COLORS
                )

                bmp_info.bmiHeader.biSizeImage = bmp_info.bmiHeader.biWidth *abs(bmp_info.bmiHeader.biHeight) * (bmp_info.bmiHeader.biBitCount+7)/8;
                size = (bmp_info.bmiHeader.biWidth, bmp_info.bmiHeader.biHeight )
                print size
                pBuf = (c_char * bmp_info.bmiHeader.biSizeImage)()

                windll.gdi32.GetBitmapBits(hCaptureBitmap, bmp_info.bmiHeader.biSizeImage, pBuf)

                return Image.frombuffer('RGB', size, pBuf, 'raw', 'BGRX', 0, 1)

Первая функция получает растровое изображение экрана, а второй преобразует его в объект PIL.

Если вы не хотите самостоятельно определять координаты других мониторов, у меня есть небольшой модуль под названием PyRobot , который имеет функции для нацеливания на конкретные мониторы и еще много чего. И это чистый Python, поэтому не нужно устанавливать PyWin32 :)

1 голос
/ 04 июня 2014

Audionautics / chriskiehl!

Я взял ваш пост и часть вашей библиотеки и переписал его для Python 3!Я надеюсь, что ты в порядке с этим.Это работает для меня, используя PILLOW (PIL) и движок python 3.4.

Я только что назвал файл duelMonitory.py, содержащий : 

# https://github.com/chriskiehl/pyrobot
# started from Audionautics code on /3827085/oshibka-pil-imagegrab-na-vtorom-virtualnom-monitore-virtualbox
# updated for PILLOW and Python 3 by Alan Baines (Kizrak)


from PIL import *
#from ctypes import windll, Structure, byref, c_uint
#import ctypes

import ctypes
from ctypes import *
from ctypes.wintypes import *

def get_screen_buffer(bounds=None):
    # Grabs a DC to the entire virtual screen, but only copies to 
    # the bitmap the the rect defined by the user. 

    SM_XVIRTUALSCREEN = 76  # coordinates for the left side of the virtual screen. 
    SM_YVIRTUALSCREEN = 77  # coordinates for the right side of the virtual screen.  
    SM_CXVIRTUALSCREEN = 78 # width of the virtual screen
    SM_CYVIRTUALSCREEN = 79 # height of the virtual screen

    hDesktopWnd = windll.user32.GetDesktopWindow() #Entire virtual Screen

    left = windll.user32.GetSystemMetrics(SM_XVIRTUALSCREEN)
    top = windll.user32.GetSystemMetrics(SM_YVIRTUALSCREEN)
    width = windll.user32.GetSystemMetrics(SM_CXVIRTUALSCREEN)
    height = windll.user32.GetSystemMetrics(SM_CYVIRTUALSCREEN)

    if bounds:
        left, top, right, bottom = bounds
        width = right - left 
        height = bottom - top

    hDesktopDC = windll.user32.GetWindowDC(hDesktopWnd)
    if not hDesktopDC:
        print ('GetDC Failed')
        sys.exit()

    hCaptureDC = windll.gdi32.CreateCompatibleDC(hDesktopDC)
    if not hCaptureDC:
        print ('CreateCompatibleBitmap Failed')
        sys.exit()

    hCaptureBitmap = windll.gdi32.CreateCompatibleBitmap(hDesktopDC, width, height)
    if not hCaptureBitmap:
        print ('CreateCompatibleBitmap Failed')
        sys.exit()

    windll.gdi32.SelectObject(hCaptureDC, hCaptureBitmap)

    SRCCOPY = 0x00CC0020
    windll.gdi32.BitBlt(
        hCaptureDC, 
        0, 0, 
        width, height, 
        hDesktopDC, 
        left, top, 
        0x00CC0020
    )
    return hCaptureBitmap

def make_image_from_buffer(hCaptureBitmap):
    from PIL import Image
    bmp_info = BITMAPINFO()
    bmp_header = BITMAPFILEHEADER()
    hdc = windll.user32.GetDC(None)

    bmp_info.bmiHeader.biSize = sizeof(BITMAPINFOHEADER)

    DIB_RGB_COLORS = 0
    windll.gdi32.GetDIBits(hdc, 
        hCaptureBitmap, 
        0,0, 
        None, byref(bmp_info), 
        DIB_RGB_COLORS
    )

    bmp_info.bmiHeader.biSizeImage = int( bmp_info.bmiHeader.biWidth *abs(bmp_info.bmiHeader.biHeight) * (bmp_info.bmiHeader.biBitCount+7)/8 );
    size = (bmp_info.bmiHeader.biWidth, bmp_info.bmiHeader.biHeight )
    print (size)
    pBuf = (c_char * bmp_info.bmiHeader.biSizeImage)()

    windll.gdi32.GetBitmapBits(hCaptureBitmap, bmp_info.bmiHeader.biSizeImage, pBuf)

    return Image.frombuffer('RGB', size, pBuf, 'raw', 'BGRX', 0, 1)


class BITMAPFILEHEADER(ctypes.Structure):
    _fields_ = [
        ('bfType', ctypes.c_short),
        ('bfSize', ctypes.c_uint32),
        ('bfReserved1', ctypes.c_short),
        ('bfReserved2', ctypes.c_short),
        ('bfOffBits', ctypes.c_uint32)
    ]


class BITMAPINFOHEADER(ctypes.Structure):
    _fields_ = [
        ('biSize', ctypes.c_uint32),
        ('biWidth', ctypes.c_int),
        ('biHeight', ctypes.c_int),
        ('biPlanes', ctypes.c_short),
        ('biBitCount', ctypes.c_short),
        ('biCompression', ctypes.c_uint32),
        ('biSizeImage', ctypes.c_uint32),
        ('biXPelsPerMeter', ctypes.c_long),
        ('biYPelsPerMeter', ctypes.c_long),
        ('biClrUsed', ctypes.c_uint32),
        ('biClrImportant', ctypes.c_uint32)
    ]


class BITMAPINFO(ctypes.Structure):
    _fields_ = [
        ('bmiHeader', BITMAPINFOHEADER),
        ('bmiColors', ctypes.c_ulong * 3)
    ]

И мой тестовый код был duelMonitor.test.py, содержащий : 

from PIL import *

from duelMonitor import *

hCaptureBitmap = get_screen_buffer()

pimage = make_image_from_buffer(hCaptureBitmap)

pimage.save("Hello.png","PNG")

По сути, просто сочетание поста Audionautics и его библиотеки с веб-страницы https://github.com/chriskiehl/pyrobot и конвертация в Python 3 (и PILLOW).

Веселитесь!

(PS. Я бы написал как комментарий, но мне не хватает очков / грусти)

...