Python抓取指定windows窗口

 

import cv2
import numpy as np
import win32api
import win32gui
import win32ui
import win32con
import os
import pygetwindow as gw

target_win = None
mydemo_list = gw.getWindowsWithTitle('Game')
target_win = mydemo_list[0]
hwnd = target_win._hWnd

if None == target_win:
    print("Target Not found")
    exit(-1)

# activate the window
target_win.activate()

#  获取显示器尺寸
width = win32api.GetSystemMetrics(win32con.SM_CXSCREEN)
height = win32api.GetSystemMetrics(win32con.SM_CYSCREEN)
out = cv2.VideoWriter(outpath, fourcc, FPS, tuple([width, height]))


#  创建设备描述表
hDC = win32gui.GetDC(hwnd)
dcObj = win32ui.CreateDCFromHandle(hDC)
memDC = dcObj.CreateCompatibleDC()
screenshot = win32ui.CreateBitmap()
screenshot.CreateCompatibleBitmap(dcObj, width, height)
memDC.SelectObject(screenshot)

memDC.BitBlt((0, 0), (width, height), dcObj, (0, 0), win32con.SRCCOPY)

#  转换Bitmap到OpenCV格式
img = np.fromstring(screenshot.GetBitmapBits(True), dtype='uint8')
img.shape = (height, width, 4)
# convert colors from RGBA to RGB
frame = cv2.cvtColor(img, cv2.COLOR_RGBA2RGB)

cv2.destroyAllWindows()
memDC.DeleteDC()
win32gui.ReleaseDC(win32con.NULL, hDC)