动态Qt内容捕获与视频导出：基于QPainter和Imageio的教程

· Python · 208554

在Qt应用程序中，实现动态图形内容的实时显示并将其导出为视频是一个常见的需求。开发者可能希望在窗口中展示动画、模拟或数据可视化，并能够将整个过程记录下来。然而，在实现这一功能时，尤其是在尝试将QPainter绘制的内容直接捕获并保存为视频帧时，可能会遇到一些挑战，例如QPainter上下文冲突或递归绘制的问题。本教程将提供一个清晰、专业的解决方案，利用PySide6/PyQt6和imageio库来解决这些问题。

1. 核心概念与技术栈

在深入实现之前，我们首先了解本方案所依赖的核心组件：

  • PySide6/PyQt6: Qt框架的Python绑定，用于构建图形用户界面和处理绘图事件。

    • QWidget: 应用程序窗口或用户界面元素的基础类。
    • QPainter: 用于在绘制设备（如QWidget、QImage或QPixmap）上进行低级绘图操作。
    • QTimer: 提供重复或单次触发的定时器事件，常用于驱动动画或周期性任务。
    • QPixmap.grab(): 捕获指定QWidget或其子部件的当前显示内容，返回一个QPixmap对象。
    • QImage: 独立于硬件的图像表示，可用于直接像素操作。
  • NumPy: Python的科学计算库，用于高效处理多维数组。在这里，它用于将QImage的像素数据转换为imageio所需的NumPy数组格式。
  • Imageio: 一个Python库，用于读取和写入各种图像和视频格式。它提供了简洁的API来创建视频文件。

2. 环境准备

在开始编码之前，请确保您的Python环境中安装了以下库：

pip install PySide6 numpy imageio imageio[ffmpeg]

imageio[ffmpeg]是必需的，因为它提供了ffmpeg后端，imageio通常依赖它来处理视频文件的编码和解码。

3. 理解常见问题与解决方案

初次尝试实现动态绘制和视频捕获时，开发者可能会尝试在QWidget的paintEvent中直接使用QPainter绘制到QImage，然后将QImage渲染到QWidget。这种方法通常会导致以下错误：

  • QPainter::begin: A paint device can only be painted by one painter at a time.：这表明在同一时间，一个绘制设备（例如QWidget或QImage）被多个QPainter实例尝试操作。paintEvent本身已经提供了一个QPainter来绘制到QWidget，如果在此事件中又创建另一个QPainter来绘制到QImage，并试图将QImage渲染回QWidget，就可能发生冲突。
  • QWidget::render: Cannot render with an inactive painter：当QPainter对象未正确激活或已结束时，尝试使用其进行渲染会导致此错误。
  • QWidget::repaint: Recursive repaint detected：在paintEvent内部调用self.render()或self.update()可能导致无限循环的重绘，从而触发此错误。

正确的解决方案是分离关注点：

  1. paintEvent的职责：paintEvent应专注于将图形内容直接绘制到QWidget上，使其在屏幕上可见。
  2. 动画与捕获的职责：动画逻辑（更新数据、触发重绘）和视频帧捕获应由一个独立的定时器回调函数来处理。在这个回调函数中，我们首先触发QWidget的重绘（self.update()），然后等待paintEvent完成绘制，最后通过self.grab()捕获已经绘制好的QWidget内容。

4. 详细实现步骤

我们将创建一个名为PlotWidget的QWidget子类，它将负责绘制动态点并将其保存为视频。

千面视频动捕

千面视频动捕是一个AI视频动捕解决方案，专注于将视频中的人体关节二维信息转化为三维模型动作。

27

查看详情

4.1 PlotWidget的初始化

在__init__方法中，我们设置窗口尺寸、初始化QTimer来驱动动画，并准备imageio视频写入器。

import imageio, numpy as np
from PySide6.QtWidgets import QApplication, QWidget, QVBoxLayout
from PySide6.QtCore import QPoint, QRect, QTimer, Qt
from PySide6.QtGui import QPainter, QPointList, QImage, QPixmap

WIDTH = 720
HEIGHT = 720

class PlotWidget(QWidget):
    def __init__(self, parent=None):
        super().__init__(parent)
        # 初始化定时器，用于触发动画和帧捕获
        self._timer = QTimer(self)
        self._timer.setInterval(100) # 每100毫秒触发一次，即10 FPS
        self._timer.timeout.connect(self.frame)

        # 存储绘制点的数据
        self._points = QPointList()

        # 设置窗口固定大小，确保视频帧尺寸一致
        self.setFixedSize(WIDTH, HEIGHT)

        # 视频帧计数器和imageio写入器
        self._totalFrames = 100 # 假设我们要录制100帧
        self._vid_writer = imageio.get_writer('video.avi', fps=10) # 视频文件名为video.avi，帧率为10 FPS

        # 启动定时器
        self._timer.start()

4.2 处理窗口关闭事件

为了确保视频文件正确关闭并释放资源，我们需要重写closeEvent。

    def closeEvent(self, event):
        if not self._vid_writer.closed:
            self._vid_writer.close() # 关闭视频写入器
        self._timer.stop() # 停止定时器
        event.accept() # 接受关闭事件

4.3 绘制事件 paintEvent

paintEvent是Qt用于处理绘制请求的函数。在这里，我们只进行实际的绘制操作，不涉及任何视频捕获或QImage渲染到QWidget的逻辑。

    def paintEvent(self, event):
        # 使用QPainter(self)直接在QWidget上进行绘制
        with QPainter(self) as painter:
            rect = QRect(QPoint(0, 0), self.size())
            painter.fillRect(rect, Qt.white) # 填充背景为白色
            painter.drawPoints(self._points) # 绘制点

4.4 动画逻辑与帧捕获 frame 方法

frame方法由QTimer定时调用，负责更新动画数据、触发窗口重绘，并在绘制完成后捕获当前窗口内容作为视频帧。

    def frame(self):
        # 模拟动画数据更新
        self._points.clear()
        # 示例：每次都在(0,0)处绘制一个点。实际应用中可在此处更新复杂图形数据
        self._points.append(QPoint(0,0)) 
        # 可以添加一些动态变化的代码，例如：
        # self._points.append(QPoint(self._totalFrames % WIDTH, self._totalFrames % HEIGHT))

        if self._totalFrames > 0:
            self.update() # 触发paintEvent，使QWidget重新绘制

            # 捕获QWidget的当前显示内容为QPixmap
            pixmap = self.grab() 

            # 将QPixmap转换为QImage，并指定为RGB888格式，这对于imageio是兼容的
            qimg = pixmap.toImage().convertToFormat(QImage.Format_RGB888)

            # 将QImage的原始像素数据转换为NumPy数组
            # (height, width, 3)表示图像的尺寸和3个颜色通道 (RGB)
            # strides参数确保NumPy正确解释QImage的内存布局
            array = np.ndarray((qimg.height(), qimg.width(), 3), 
                               buffer=qimg.constBits(), 
                               strides=[qimg.bytesPerLine(), 3, 1], 
                               dtype=np.uint8)

            # 如果视频写入器未关闭，则将当前帧添加到视频
            if not self._vid_writer.closed:
                self._vid_writer.append_data(array)
        else:
            # 帧数用尽，停止定时器并关闭视频写入器
            self._timer.stop()
            if not self._vid_writer.closed:
                self._vid_writer.close()

        self._totalFrames -= 1 # 减少剩余帧数

4.5 完整代码示例

将以上所有部分组合起来，形成一个完整的可运行示例。

import imageio, numpy as np
from PySide6.QtWidgets import QApplication, QWidget, QVBoxLayout
from PySide6.QtCore import QPoint, QRect, QTimer, Qt
from PySide6.QtGui import QPainter, QPointList, QImage, QPixmap

WIDTH = 720
HEIGHT = 720

class PlotWidget(QWidget):
    def __init__(self, parent=None):
        super().__init__(parent)
        self._timer = QTimer(self)
        self._timer.setInterval(100)
        self._timer.timeout.connect(self.frame)

        self._points = QPointList()

        self.setFixedSize(WIDTH, HEIGHT)

        self._totalFrames = 100 # 录制100帧
        self._vid_writer = imageio.get_writer('video.avi', fps=10) # 10 FPS

        self._timer.start() # 启动定时器

    def closeEvent(self, event):
        if not self._vid_writer.closed:
            self._vid_writer.close()
        self._timer.stop()
        event.accept()

    def frame(self):
        self._points.clear()
        # 示例：每次都在(0,0)处绘制一个点。可以修改此处实现动态内容
        self._points.append(QPoint(0,0)) 

        if self._totalFrames > 0:
            self.update() # 触发paintEvent

            pixmap = self.grab() # 捕获窗口内容
            qimg = pixmap.toImage().convertToFormat(QImage.Format_RGB888) # 转换为RGB888 QImage

            # 转换为NumPy数组
            array = np.ndarray((qimg.height(), qimg.width(), 3), 
                               buffer=qimg.constBits(), 
                               strides=[qimg.bytesPerLine(), 3, 1], 
                               dtype=np.uint8)

            if not self._vid_writer.closed:
                self._vid_writer.append_data(array) # 添加到视频
        else:
            self._timer.stop()
            if not self._vid_writer.closed:
                self._vid_writer.close()

        self._totalFrames -= 1

    def paintEvent(self, event):
        with QPainter(self) as painter:
            rect = QRect(QPoint(0, 0), self.size())
            painter.fillRect(rect, Qt.white)
            painter.drawPoints(self._points)

if __name__ == '__main__':
    app = QApplication([])
    window = PlotWidget()
    window.show()
    app.exec()

5. 注意事项与最佳实践

  • 性能考量：self.grab()操作会捕获整个QWidget的内容，对于非常大的窗口或极高的帧率，这可能会带来一定的性能开销。如果只需要捕获部分区域，可以考虑使用self.grab(QRect)。
  • QImage格式：imageio通常期望标准的RGB或RGBA格式的NumPy数组。QImage.Format_RGB888是一个很好的选择，因为它直接对应于3通道的8位RGB数据。
  • 资源管理：务必在应用程序关闭或不再需要时，调用_vid_writer.close()来确保视频文件被正确写入和完成。同时，停止QTimer以避免不必要的CPU周期。
  • 动态内容：在frame方法中，self._points.append(QPoint(0,0))是一个简化示例。在实际应用中，您将在此处实现更复杂的动画逻辑，更新图形数据以创建丰富的动态效果。
  • 错误处理：在生产环境中，应增加更多的错误处理机制，例如检查imageio.get_writer是否成功创建文件，以及文件路径是否存在等。

6. 总结

通过本教程，我们学习了一种在PySide6/PyQt6中实现动态图形显示并同时将其导出为视频的有效方法。关键在于将QPainter的绘制操作限定在paintEvent中，专注于在QWidget上渲染，而将视频帧的捕获和写入逻辑放在一个由QTimer驱动的独立方法中。这种分离关注点的方法不仅解决了常见的QPainter上下文冲突和递归绘制问题，而且提供了一个清晰、可维护的架构，便于开发各种需要实时动画和视频输出的Qt应用程序。

以上就是动态Qt内容捕获与视频导出：基于QPainter和Imageio的教程的详细内容，更多请关注昆居客【www.kunjuke.com】。

相关推荐：

《绍宋：还我河山》下载安装教程介绍

这是一款沉浸式真人互动历史权谋影视游戏！你将化身赵玖，魂穿南宋最具争议的皇帝赵构，每次抉择都将左右王朝气运、定夺天下兴亡！面对靖康之变后的山河破碎，你能否打破宿命再造山河？真人互动影视游戏《绍宋：还我河山》邀你于乱世绝境之中，一起喊出宋可亡，天下不可亡的怒吼！绍宋：还我河山怎么下载安装Steam平台需要先安装Steam平台，之后注册一个账号登录，然后搜索“绍宋：还我河山”进入商店购买页面。点击添加...

贝乐虎儿歌app投屏教程

贝乐虎儿歌app怎么投屏，首先用户需要将电视和手机连接同一个wifi，然后打开软件，虽然点击一个歌曲或者是视频，点击TV，然后就可以看到投电视，选择你想要投放的电视即可完成投屏操作。 贝乐虎儿歌app投屏教程： 1、首先用户需要将电视和手机连接同一个wifi； 2、然后打开软件，虽然点击一个歌曲或者是视频，点击TV； 3、然后就可以看到投电视，选择你想要投放的电视即可完成投屏操作。 以上就是小编整...

开源电子书管理工具 Calibre 9.11 发布：支持 HTML 格式导出注释

7 月 3 日消息，科技媒体 Linuxiac 今天（7 月 3 日）发布博文，报道称开源电子书管理工具 Calibre 更新至 9.11 版本，新增将电子书注释导出为独立 HTML 网页的功能。根据官方更新日志，用户能以独立 HTML 网页格式导出电子书中的注释，后续用户可以在任意浏览器中打开页面，并支持切换深色 / 浅色主题，并提供搜索筛选与按高亮样式过滤。在阅读器体验方面，Calibre 升...

《幻鸟之森》下载安装教程介绍

《幻鸟之森》是一款陪伴你工作、学习和休息的桌面放置养成游戏。你可以在屏幕一侧观察、照料并引导神秘小鸟成长，看它们从雏鸟逐渐进化为幻想中的奇妙生物。无需频繁切换窗口，也能在日常间隙收获一份轻松与陪伴~幻鸟之森怎么下载安装Steam平台需要先安装Steam平台，之后注册一个账号登录，然后搜索“幻鸟之森”进入商店购买页面。点击添加至购物车，然后付钱购买，最后再次进入商店购买页面点击马上开玩选择文件下载位...

《超级变色龙》下载安装教程介绍

把自己涂满颜色，融入背景吧！《Meccha Chameleon》是一款全新感觉的捉迷藏游戏，玩家需要在自己纯白的身体上绘图以拟态成关卡中的样子。躲藏地点、姿势，以及最重要的“画功”是生存的关键。用连变色龙都自愧不如的技术欺骗追捕者的眼睛，在时限内逃出生天吧！您还可以轻松享受随机匹配或观众参与型直播。超级变色龙怎么下载安装Steam平台需要先安装Steam平台，之后注册一个账号登录，然后搜索“超级变...

《喵咪争霸》下载安装教程介绍

《喵咪争霸》是一款可爱画风的即时战斗休闲卡牌游戏。能力各异、萌态万千的猫咪卡牌，每一张都可能是破局的关键。研究不同的卡组与道具搭配，突破重重关卡，成为这片大陆的最强大佬吧！喵咪争霸怎么下载安装Steam平台需要先安装Steam平台，之后注册一个账号登录，然后搜索“喵咪争霸”进入商店购买页面。点击添加至购物车，然后付钱购买，最后再次进入商店购买页面点击马上开玩选择文件下载位置即可开始下载。可以去St...

《小闹花园》下载安装教程介绍

《小闹花园》是一款在桌面挂机养花的游戏。你可以随缘地为种子浇水、施肥，随缘地收一收可爱的花朵和果实，最后爽爽地杀一杀讨厌的虫子，放心，它们不会弄脏你的桌面！小闹花园怎么下载安装Steam平台需要先安装Steam平台，之后注册一个账号登录，然后搜索“小闹花园”进入商店购买页面。点击添加至购物车，然后付钱购买，最后再次进入商店购买页面点击马上开玩选择文件下载位置即可开始下载。可以去Steam平台下载，...

三角洲行动S10体验服安装教程

《三角洲行动》是一款设定在2035年的第一人称特战干员战术射击游戏。就在最近三角洲行动终于更新了s10赛季，那么有不少小伙伴好奇S10体验服怎么安装，下面就来告诉大家。三角洲行动S10体验服安装教程。 三角洲行动S10体验服安装教程 一、安装前必备准备 1、 确认资格生效 安装前务必先确认账号已获得测试资格，否则即使下载客户端也无法登录： 官网查询：登录 df.qq.com，点击右上角「测试资格查...