grabscreen.py Python win32api

时间:2018-05-10 17:49:01

标签: linux win32gui

任何Linux或Mac OS等效的Win32gui库或此代码? 在外部项目上工作,这个Windows代码将帮助我抓住屏幕。还没有找到任何类似的库。谢谢

def grab_screen(region=None):

hwin = win32gui.GetDesktopWindow()

if region:
        left,top,x2,y2 = region
        width = x2 - left + 1
        height = y2 - top + 1
else:
    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)

signedIntsArray = bmp.GetBitmapBits(True)
img = np.fromstring(signedIntsArray, dtype='uint8')
img.shape = (height,width,4)

srcdc.DeleteDC()
memdc.DeleteDC()
win32gui.ReleaseDC(hwin, hwindc)
win32gui.DeleteObject(bmp.GetHandle())

return cv2.cvtColor(img, cv2.COLOR_BGRA2RGB)

2 个答案:

答案 0 :(得分:0)

您可以使用pyautogui来抓取屏幕:

import pyautogui
image = pyautogui.screenshot('filename.png')

答案 1 :(得分:0)

你可以这样做:) 我认为 Mac OS 不能使用那些 WiIn32gui 库。 相反,您可以使用枕头来抓取屏幕。 屏幕大小可以更改,具体取决于您想要的大小。

import cv2
import numpy as np
import pyautogui
from PIL import ImageGrab

screen_w = 1920
screen_h = 1080

while True:
    rgb = ImageGrab.grab(bbox=(0, 0, screen_w, screen_h)) 
    rgb = np.array(rgb)

    cv2.imshow('window_frame', rgb)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break