动态Qt内容捕获与视频导出：基于QPainter和Imageio的教程-Python教程-PHP中文网

动态qt内容捕获与视频导出：基于qpainter和imageio的教程

在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()可能导致无限循环的重绘，从而触发此错误。

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

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

4. 详细实现步骤

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

千面视频动捕

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

查看详情

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的教程的详细内容，更多请关注php中文网其它相关文章！