使用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窗口并记录窗口内容的整个过程。

相关推荐
CoovallyAIHub20 小时前
英伟达ViPE重磅发布!解决3D感知难题,SLAM+深度学习完美融合(附带数据集下载地址)
深度学习·算法·计算机视觉
这里有鱼汤21 小时前
miniQMT下载历史行情数据太慢怎么办?一招提速10倍!
前端·python
databook1 天前
Manim实现脉冲闪烁特效
后端·python·动效
程序设计实验室1 天前
2025年了,在 Django 之外,Python Web 框架还能怎么选?
python
倔强青铜三1 天前
苦练Python第46天:文件写入与上下文管理器
人工智能·python·面试
用户2519162427111 天前
Python之语言特点
python
刘立军1 天前
使用pyHugeGraph查询HugeGraph图数据
python·graphql
数据智能老司机2 天前
精通 Python 设计模式——创建型设计模式
python·设计模式·架构
数据智能老司机2 天前
精通 Python 设计模式——SOLID 原则
python·设计模式·架构
c8i2 天前
django中的FBV 和 CBV
python·django