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

相关推荐
Yan-英杰33 分钟前
百度搜索和文心智能体接入DeepSeek满血版——AI搜索的新纪元
图像处理·人工智能·python·深度学习·deepseek
weixin_307779132 小时前
Azure上基于OpenAI GPT-4模型验证行政区域数据的设计方案
数据仓库·python·云计算·aws
玩电脑的辣条哥3 小时前
Python如何播放本地音乐并在web页面播放
开发语言·前端·python
taoqick3 小时前
对PosWiseFFN的改进: MoE、PKM、UltraMem
人工智能·pytorch·深度学习
多想和从前一样5 小时前
Django 创建表时 “__str__ ”方法的使用
后端·python·django
charles_vaez6 小时前
开源模型应用落地-LangGraph101-探索 LangGraph 短期记忆
深度学习·语言模型·自然语言处理
WHATEVER_LEO7 小时前
【每日论文】Latent Radiance Fields with 3D-aware 2D Representations
人工智能·深度学习·神经网络·机器学习·计算机视觉·自然语言处理
小喵要摸鱼7 小时前
【Pytorch 库】自定义数据集相关的类
pytorch·python
bdawn7 小时前
深度集成DeepSeek大模型:WebSocket流式聊天实现
python·websocket·openai·api·实时聊天·deepseek大模型·流式输出
Jackson@ML7 小时前
Python数据可视化简介
开发语言·python·数据可视化