PIL и растровое изображение из WinAPI - PullRequest
Новая
       21

PIL и растровое изображение из WinAPI

3 голосов
/ 05 августа 2011

У меня есть код, который делает скриншот окна, используя winapi.Затем я должен сохранить образ на диск и загрузить его снова с диска в память PIL.Есть ли способ сразу без сохранения на диск передать эту битовую карту в PIL. 

import win32gui, win32ui, win32con
import Image

win_name='Book'
bmpfilenamename='1.bmp'
hWnd = win32gui.FindWindow(None, win_name)
windowcor = win32gui.GetWindowRect(hWnd)
w=windowcor[2]-windowcor[0]
h=windowcor[3]-windowcor[1]
wDC = win32gui.GetWindowDC(hWnd)
dcObj=win32ui.CreateDCFromHandle(wDC)
cDC=dcObj.CreateCompatibleDC()
dataBitMap = win32ui.CreateBitmap()
dataBitMap.CreateCompatibleBitmap(dcObj, w, h)
cDC.SelectObject(dataBitMap)
cDC.BitBlt((0,0),(w, h) , dcObj, (0,0), win32con.SRCCOPY)
dataBitMap.SaveBitmapFile(cDC, bmpfilenamename)

#dcObj.DeleteDC()
#cDC.DeleteDC()
#win32gui.ReleaseDC(hWnd, wDC)

im=Image.open(bmpfilenamename)
im.load()

Ответы [ 2 ]

10 голосов
/ 05 августа 2011

Закомментируйте эту строку: 

dataBitMap.SaveBitmapFile(cDC, bmpfilenamename)

и добавьте это вместо: 

bmpinfo = dataBitMap.GetInfo()
bmpstr = dataBitMap.GetBitmapBits(True)
im = Image.frombuffer(
    'RGB',
    (bmpinfo['bmWidth'], bmpinfo['bmHeight']),
    bmpstr, 'raw', 'BGRX', 0, 1)

[из еще один вопрос SO ]

0 голосов
/ 05 августа 2011

Если вы используете PIL, почему бы не использовать его для скриншотов?
PIL содержит модуль ImageGrab, который можно использовать как: 

import win32gui, win32ui, win32con
import Image
import ImageGrab

win_name='Book'
hWnd = win32gui.FindWindow(None, win_name)
windowcor = win32gui.GetWindowRect(hWnd)

im = ImageGrab.grab(windowcor)

Однако он получит правильный скриншот, только если ваше приложение находится на переднем плане.

...