使用Pygame和OpenCV记录窗口的内容为视频

python 复制代码
from PIL import ImageGrab
import numpy as np

    pg.init()

    gl_version = (3, 3) # GL Version number (Major, Minor)

    pg.display.gl_set_attribute(pg.GL_CONTEXT_MAJOR_VERSION, gl_version[0])

    pg.display.gl_set_attribute(pg.GL_CONTEXT_MINOR_VERSION, gl_version[1])

    pg.display.gl_set_attribute(pg.GL_CONTEXT_PROFILE_MASK, pg.GL_CONTEXT_PROFILE_CORE)

    display_size = (1440, 720)

    pg.display.set_mode(display_size, pg.OPENGL | pg.DOUBLEBUF | pg.RESIZABLE)

    screen_info = pg.display.Info()

    screen_width, screen_height = screen_info.current_w, screen_info.current_h

    window_x, window_y = (screen_width - display_size[0])/2, (screen_height - display_size[1])/2 #for recording screen

    record_screen = True#False#

    if record_screen:

        fourcc = cv2.VideoWriter_fourcc(*'XVID')

        window_out = cv2.VideoWriter('window_output.avi', fourcc, 30.0, display_size)


    if record_screen:

        if window_out is not None:
            image = ImageGrab.grab(bbox=(window_x, window_y, window_x+display_size[0],         window_y+display_size[1]))

            #image.save('screenshot.png')

            image = np.array(image)

            image_bgr = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)

            window_out.write(image_bgr)


    if record_screen:

        window_out.release()

这段代码使用Pygame和OpenCV在屏幕上创建一个OpenGL窗口,并记录窗口的内容为视频。

首先,它初始化了Pygame库:

pg.init() 

然后,设置OpenGL的版本和配置:

gl_version = (3, 3) # 设置OpenGL版本号为3.3 pg.display.gl_set_attribute(pg.GL_CONTEXT_MAJOR_VERSION, gl_version[0]) pg.display.gl_set_attribute(pg.GL_CONTEXT_MINOR_VERSION, gl_version[1]) pg.display.gl_set_attribute(pg.GL_CONTEXT_PROFILE_MASK, pg.GL_CONTEXT_PROFILE_CORE) 

接下来,设置窗口的大小和一些Pygame的窗口属性:

display_size = (1440, 720) # 设置窗口大小为1440x720 pg.display.set_mode(display_size, pg.OPENGL | pg.DOUBLEBUF | pg.RESIZABLE) # 设置窗口模式为OpenGL、双缓冲和可调整大小 

然后,获取屏幕信息,计算窗口的位置:

screen_info = pg.display.Info() screen_width, screen_height = screen_info.current_w, screen_info.current_h window_x, window_y = (screen_width - display_size[0])/2, (screen_height - display_size[1])/2 # 计算窗口的位置 

接下来,如果设置了record_screenTrue,则创建一个视频写入器以记录窗口的内容:

record_screen = True if record_screen: fourcc = cv2.VideoWriter_fourcc(*'XVID') # 使用XVID编解码器 window_out = cv2.VideoWriter('window_output.avi', fourcc, 30.0, display_size) # 创建视频写入器 

然后,如果需要记录屏幕,就获取窗口的截图,并将其转换为OpenCV可以处理的格式:

if record_screen: if window_out is not None: image = ImageGrab.grab(bbox=(window_x, window_y, window_x+display_size[0], window_y+display_size[1])) # 获取窗口的截图 image = np.array(image) # 将截图转换为NumPy数组 image_bgr = cv2.cvtColor(image, cv2.COLOR_RGB2BGR) # 将RGB格式转换为BGR格式(OpenCV默认的颜色顺序) window_out.write(image_bgr) # 将截图写入视频 

最后,如果需要记录屏幕,则释放视频写入器:

if record_screen: window_out.release() # 释放视频写入器 

这样,就完成了创建OpenGL窗口并记录窗口内容的整个过程。

相关推荐
鸡鸭扣1 小时前
Docker:3、在VSCode上安装并运行python程序或JavaScript程序
运维·vscode·python·docker·容器·js
paterWang1 小时前
基于 Python 和 OpenCV 的酒店客房入侵检测系统设计与实现
开发语言·python·opencv
东方佑2 小时前
使用Python和OpenCV实现图像像素压缩与解压
开发语言·python·opencv
神秘_博士2 小时前
自制AirTag,支持安卓/鸿蒙/PC/Home Assistant,无需拥有iPhone
arm开发·python·物联网·flutter·docker·gitee
Moutai码农3 小时前
机器学习-生命周期
人工智能·python·机器学习·数据挖掘
Jackilina_Stone4 小时前
【DL】浅谈深度学习中的知识蒸馏 | 输出层知识蒸馏
人工智能·深度学习·机器学习·蒸馏
小白教程4 小时前
python学习笔记,python处理 Excel、Word、PPT 以及邮件自动化办公
python·python学习·python安装
武陵悭臾5 小时前
网络爬虫学习:借助DeepSeek完善爬虫软件,实现模拟鼠标右键点击,将链接另存为本地文件
python·selenium·网络爬虫·pyautogui·deepseek·鼠标右键模拟·保存链接为htm
代码猪猪傻瓜coding5 小时前
关于 形状信息提取的说明
人工智能·python·深度学习
Kai HVZ6 小时前
《深度学习》——自然语言处理(NLP)
人工智能·深度学习·自然语言处理