PIL与WinAPI中的位图

Question

我有一段代码，它使用winapi来截取窗口的屏幕截图。然后，我需要把这个图片保存到硬盘上，再从硬盘加载到内存中的PIL（Python Imaging Library）。有没有办法可以直接把这个位图传递给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()

Answer 1

如果你在使用PIL（Python图像库），那为什么不利用它来截屏呢？
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)

不过，它只有在你的应用程序在最前面的时候才能正确截屏。

Answer 2

把这一行注释掉：

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问题]