PySide6中QLabel显示QMovie并保持宽高比的教程

from PySide6.QtWidgets import QLabel, QApplication, QWidget, QVBoxLayout
from PySide6.QtGui import QMovie, QPainter, QPixmap
from PySide6.QtCore import Qt, QSize, QRect, QTimer
import sys

class ScaledLabel(QLabel):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        # 存储电影的原始尺寸（通过计算得出）
        self._movieSize = QSize()
        # 存储标签的最小尺寸提示（保持宽高比）
        self._minSize = QSize()
        # 设置默认的对齐方式，通常动画会居中显示
        self.setAlignment(Qt.AlignCenter)

    def minimumSizeHint(self):
        """
        提供一个最小尺寸提示，确保QLabel不会无限缩小，并保持宽高比。
        """
        if self._minSize.isValid():
            return self._minSize
        return super().minimumSizeHint()

    def setMovie(self, movie):
        """
        设置要显示的QMovie对象。
        此方法会计算动画的真实尺寸和最小尺寸提示。
        """
        # 如果设置的是同一个电影，则直接返回
        if self.movie() == movie:
            return

        # 调用父类的setMovie方法
        super().setMovie(movie)

        # 如果电影无效或不是QMovie类型，重置尺寸信息并更新几何形状
        if not isinstance(movie, QMovie) or not movie.isValid():
            self._movieSize = QSize()
            self._minSize = QSize()
            self.updateGeometry()
            return

        # 记录电影的当前帧和播放状态，以便后续恢复
        cf = movie.currentFrameNumber()
        state = movie.state()

        # 跳转到第一帧，并遍历所有帧以获取动画的真实边界框
        movie.jumpToFrame(0)
        rect = QRect()
        for i in range(movie.frameCount()):
            movie.jumpToNextFrame()
            # 使用逻辑或运算符来扩展矩形，以包含所有帧的区域
            rect |= movie.frameRect()

        # 计算动画的实际宽度和高度
        width = rect.x() + rect.width()
        height = rect.y() + rect.height()

        # 存储计算出的原始电影尺寸
        self._movieSize = QSize(width, height)

        # 计算基于原始宽高比的最小尺寸提示
        # 确保最小尺寸不会过小，且保持原始宽高比
        minimum_dim = min(width, height)
        maximum_dim = max(width, height)

        if minimum_dim > 0: # 避免除以零
            ratio = maximum_dim / minimum_dim
            base = min(4, minimum_dim) # 最小尺寸的基准，至少为4像素
            self._minSize = QSize(base, round(base * ratio))

            # 如果原始电影的高度是较小维度，则需要转置最小尺寸
            # 因为QSize(base, round(base * ratio))默认base是宽度
            if minimum_dim == height:
                self._minSize.transpose()
        else:
            self._minSize = QSize(4, 4) # 兜底，防止尺寸为0

        # 恢复电影的原始帧和播放状态
        movie.jumpToFrame(cf)
        if state == movie.MovieState.Running:
            movie.setPaused(False)

        # 更新控件的几何形状，触发重新布局和绘图
        self.updateGeometry()

    def paintEvent(self, event):
        """
        重写绘图事件，以在QLabel尺寸变化时正确缩放QMovie。
        """
        movie = self.movie()
        # 如果没有电影或电影无效，则调用父类的绘图事件
        if not isinstance(movie, QMovie) or not movie.isValid():
            super().paintEvent(event)
            return

        qp = QPainter(self)
        self.drawFrame(qp) # 绘制QLabel的边框

        # 获取QLabel的有效内容区域（排除边距）
        cr = self.contentsRect()
        margin = self.margin()
        cr.adjust(margin, margin, -margin, -margin)

        style = self.style()
        # 获取QLabel的对齐方式
        alignment = style.visualAlignment(self.layoutDirection(), self.alignment())

        # 根据内容区域和电影的原始尺寸，计算出保持宽高比的最大缩放尺寸
        maybeSize = self._movieSize.scaled(cr.size(), Qt.KeepAspectRatio)

        # 优化：如果计算出的尺寸与QMovie内部已设置的缩放尺寸不同
        # 则更新QMovie的缩放尺寸，并直接绘制当前帧的缩放版本
        if maybeSize != movie.scaledSize():
            movie.setScaledSize(maybeSize)
            style.drawItemPixmap(
                qp, cr, alignment, 
                movie.currentPixmap().scaled(cr.size(), Qt.KeepAspectRatio)
            )
        # 否则，如果QMovie内部尺寸已经匹配，直接绘制当前帧（QMovie已内部缩放）
        else:
            style.drawItemPixmap(
                qp, cr, alignment, 
                movie.currentPixmap()
            )

if __name__ == '__main__':
    app = QApplication(sys.argv)

    window = QWidget()
    window.setWindowTitle("PySide6 QMovie 保持宽高比示例")
    layout = QVBoxLayout(window)

    # 创建一个ScaledLabel实例
    label = ScaledLabel()
    # 创建一个QMovie对象，替换为你的GIF文件路径
    # 注意：请确保你的GIF文件存在且路径正确
    movie = QMovie("path/to/your/animation.gif") 

    if movie.isValid():
        label.setMovie(movie)
        movie.start() # 开始播放动画
    else:
        label.setText("无法加载动画文件或文件无效。")
        print("Error: Could not load movie or movie is invalid.")
        print(f"Movie status: {movie.isValid()}")
        print(f"Movie format: {movie.format()}")

    layout.addWidget(label)
    window.resize(400, 300) # 设置初始窗口大小
    window.show()

    sys.exit(app.exec())