捕获屏幕有比PIL.ImageGrab.grab()更好的方法吗?

时间:2012-09-25 20:58:09

标签: python windows windows-xp python-imaging-library python-2.5

我正在用python制作一个截屏程序。我目前的问题是PIL.ImageGrab.grab()在2秒后给出了相同的输出。例如,因为我认为我不清楚,在下面的程序中,几乎所有图像都是相同的,具有相同的Image.tostring()值,即使我在PIL.ImageGrab期间移动我的屏幕。抓斗循环正在执行。

>>> from PIL.ImageGrab import grab
>>> l = []
>>> import time
>>> for a in l:
        l.append(grab())
        time.sleep(0.01)


    >>> for a in range(0, 30):
        l.append(grab())
        time.sleep(0.01)


>>> b = []
>>> for a in l:
        b.append(a.tostring())


>>> len(b)
30
>>> del l
>>> last = []
>>> a = 0
>>> a = -1
>>> last = ""
>>> same = -1
>>> for pic in b:
        if b == last:
            same = same + 1
        last = b


>>> same
28
>>> 

这是一个问题,因为所有图像都是相同的,但是1中有30个是不同的。这将成为一个绝对可怕的高质量视频。请告诉我是否有更好的替代品PIL.ImageGrab.grab()。我需要捕捉整个屏幕。谢谢!

编辑:

到目前为止,看起来最好的选择是使用pywin32。我现在正在使用它,但它很慢。我并不真正关心兼容性,因为现在这个项目是个人的。每次我计算出程序的帧速率时,pywin32都是如此之慢,它是负面的。请告诉我是否有更快的替代方案。谢谢!

2 个答案:

答案 0 :(得分:5)

PIL.ImageGrab有很多替代品。

如果您想要跨平台,几乎每个主要的窗口库都具有屏幕抓取功能。以下是使用PyQt4的示例:

import sys
from PyQt4.QtGui import QPixmap, QApplication
app = QApplication(sys.argv)
QPixmap.grabWindow(QApplication.desktop().winId()).save('screenshot.jpg', 'jpg')

如果您想要与Qt认为“桌面”的默认设置不同的东西,您需要深入了解PyQt或Qt文档。

缺点是这些跨平台窗口库在安装,分发方面通常非常重要,有时甚至是你如何构建程序。

另一种方法是使用Windows特定的代码。虽然可以直接使用Windows API,但更简单的解决方案是PyWin32

import win32gui, win32ui, win32con, win32api
hwin = win32gui.GetDesktopWindow()
width = win32api.GetSystemMetrics(win32con.SM_CXVIRTUALSCREEN)
height = win32api.GetSystemMetrics(win32con.SM_CYVIRTUALSCREEN)
left = win32api.GetSystemMetrics(win32con.SM_XVIRTUALSCREEN)
top = win32api.GetSystemMetrics(win32con.SM_YVIRTUALSCREEN)
hwindc = win32gui.GetWindowDC(hwin)
srcdc = win32ui.CreateDCFromHandle(hwindc)
memdc = srcdc.CreateCompatibleDC()
bmp = win32ui.CreateBitmap()
bmp.CreateCompatibleBitmap(srcdc, width, height)
memdc.SelectObject(bmp)
memdc.BitBlt((0, 0), (width, height), srcdc, (left, top), win32con.SRCCOPY)
bmp.SaveBitmapFile(memdc, 'screenshot.bmp')

与Win32一样,您必须准确指定“桌面”的含义;此版本指定当前会话的“虚拟屏幕”,如果您在控制台本地运行,则为所有监视器的组合,如果您在终端服务上运行,则为远程视图。如果您想要不同的东西,您必须阅读MSDN上的相关Win32文档。但在大多数情况下,从文档转换到PyWin32是微不足道的。

(顺便说一句,即使我说“Win32”,同样适用于Win64。)

答案 1 :(得分:2)

现代的跨平台解决方案是使用Python MSS

from mss import mss
from PIL import Image

def capture_screenshot():
    # Capture entire screen
    with mss() as sct:
        monitor = sct.monitors[1]
        sct_img = sct.grab(monitor)
        # Convert to PIL/Pillow Image
        return Image.frombytes('RGB', sct_img.size, sct_img.bgra, 'raw', 'BGRX')

img = capture_screenshot()
img.show()

在速度较慢的笔记本电脑上,此功能可以以27 fps的速度返回屏幕截图。