如何限制QRubberBand在PyQt5的QLabel/图像范围内绘制？

我用PyQt5开发图像查看器，目前功能正常，需要添加裁剪功能——允许用户在图像上绘制矩形，后续用该矩形进行裁剪。当前QRubberBand可正常绘制，但会超出图像范围，我曾尝试以下代码：

if eventQMouseEvent.pos().x() < self.imageLabel.size().width() and eventQMouseEvent.pos().y() < self.imageLabel.size().height():                
    self.currentQRubberBand.setGeometry(QRect(self.originQPoint, eventQMouseEvent.pos()).normalized())

但效果卡顿且不符合预期。

修改后的完整代码 #

from PyQt5 import QtCore
from PyQt5.QtCore import Qt, QRect
from PyQt5.QtGui import QImage, QPixmap, QPalette, QPainter
from PyQt5.QtWidgets import QLabel, QSizePolicy, QScrollArea, QMessageBox, QMainWindow, QMenu, QAction, qApp, QFileDialog, QApplication, QRubberBand
import sys

class QImageViewer(QMainWindow):
    def __init__(self):
        super().__init__()

        self.scaleFactor = 0.0
        
        self.opened = False

        self.imageLabel = QLabel()
        self.imageLabel.setBackgroundRole(QPalette.Base)
        self.imageLabel.setSizePolicy(QSizePolicy.Ignored, QSizePolicy.Ignored)
        self.imageLabel.setScaledContents(True)

        self.scrollArea = QScrollArea()
        self.scrollArea.setBackgroundRole(QPalette.Dark)
        self.scrollArea.setWidget(self.imageLabel)
        self.scrollArea.setVisible(False)

        self.setCentralWidget(self.scrollArea)

        self.createActions()
        self.createMenus()

        self.setWindowTitle("Image Viewer")
        self.resize(800, 600)

    def open(self):
        options = QFileDialog.Options()
        fileName, _ = QFileDialog.getOpenFileName(self, 'QFileDialog.getOpenFileName()', '',
                                                  'Images (*.png *.jpeg *.jpg *.bmp *.gif)', options=options)
        if fileName:
            image = QImage(fileName)
            if image.isNull():
                QMessageBox.information(self, "Image Viewer", "Cannot load %s." % fileName)
                return
            
            self.opened = True
            
            self.imageLabel.setPixmap(QPixmap.fromImage(image))
            self.scaleFactor = 1.0

            self.scrollArea.setVisible(True)
            self.fitToWindowAct.setEnabled(True)
            self.updateActions()

            if not self.fitToWindowAct.isChecked():
                self.imageLabel.adjustSize()

    def zoomIn(self):
        self.scaleImage(1.25)

    def zoomOut(self):
        self.scaleImage(0.8)

    def normalSize(self):
        self.imageLabel.adjustSize()
        self.scaleFactor = 1.0

    def fitToWindow(self):
        fitToWindow = self.fitToWindowAct.isChecked()
        self.scrollArea.setWidgetResizable(fitToWindow)
        if not fitToWindow:
            self.normalSize()

        self.updateActions()

    def createActions(self):
        self.openAct = QAction("&Open...", self, shortcut="Ctrl+O", triggered=self.open)
        self.exitAct = QAction("E&xit", self, shortcut="Ctrl+Q", triggered=self.close)
        self.zoomInAct = QAction("Zoom &In (25%)", self, shortcut="Ctrl++", enabled=False, triggered=self.zoomIn)
        self.zoomOutAct = QAction("Zoom &Out (25%)", self, shortcut="Ctrl+-", enabled=False, triggered=self.zoomOut)
        self.normalSizeAct = QAction("&Normal Size", self, shortcut="Ctrl+S", enabled=False, triggered=self.normalSize)
        self.fitToWindowAct = QAction("&Fit to Window", self, enabled=False, checkable=True, shortcut="Ctrl+F",
                                      triggered=self.fitToWindow)

    def createMenus(self):
        self.fileMenu = QMenu("&File", self)
        self.fileMenu.addAction(self.openAct)
        self.fileMenu.addSeparator()
        self.fileMenu.addAction(self.exitAct)

        self.viewMenu = QMenu("&View", self)
        self.viewMenu.addAction(self.zoomInAct)
        self.viewMenu.addAction(self.zoomOutAct)
        self.viewMenu.addAction(self.normalSizeAct)
        self.viewMenu.addSeparator()
        self.viewMenu.addAction(self.fitToWindowAct)

        self.menuBar().addMenu(self.fileMenu)
        self.menuBar().addMenu(self.viewMenu)

    def updateActions(self):
        self.zoomInAct.setEnabled(not self.fitToWindowAct.isChecked())
        self.zoomOutAct.setEnabled(not self.fitToWindowAct.isChecked())
        self.normalSizeAct.setEnabled(not self.fitToWindowAct.isChecked())

    def scaleImage(self, factor):
        self.scaleFactor *= factor
        self.imageLabel.resize(self.scaleFactor * self.imageLabel.pixmap().size())

        self.adjustScrollBar(self.scrollArea.horizontalScrollBar(), factor)
        self.adjustScrollBar(self.scrollArea.verticalScrollBar(), factor)

        self.zoomInAct.setEnabled(self.scaleFactor < 3.0)
        self.zoomOutAct.setEnabled(self.scaleFactor > 0.333)

    def adjustScrollBar(self, scrollBar, factor):
        scrollBar.setValue(int(factor * scrollBar.value()
                               + ((factor - 1) * scrollBar.pageStep() / 2)))
    
    def mousePressEvent (self, eventQMouseEvent):
        if self.opened:
            # 将全局坐标转换为imageLabel的局部坐标
            self.originQPoint = self.imageLabel.mapFromGlobal(eventQMouseEvent.globalPos())
            # 限制初始点在图像范围内
            self.originQPoint = self.imageLabel.rect().contains(self.originQPoint) and self.originQPoint or self.imageLabel.rect().topLeft()
            # 创建橡皮筋，父控件设为imageLabel，确保坐标对应图像区域
            self.currentQRubberBand = QRubberBand(QRubberBand.Rectangle, self.imageLabel)
            self.currentQRubberBand.setGeometry(QRect(self.originQPoint, QtCore.QSize()))
            self.currentQRubberBand.show()

    def mouseMoveEvent (self, eventQMouseEvent):
        if self.opened and hasattr(self, 'currentQRubberBand'):
            # 转换为imageLabel局部坐标
            currentPos = self.imageLabel.mapFromGlobal(eventQMouseEvent.globalPos())
            # 生成规范化的矩形，再和图像标签的区域取交集，限制范围
            rubberRect = QRect(self.originQPoint, currentPos).normalized()
            rubberRect = rubberRect.intersected(self.imageLabel.rect())
            self.currentQRubberBand.setGeometry(rubberRect)
            
    def mouseReleaseEvent (self, eventQMouseEvent):
        if self.opened and hasattr(self, 'currentQRubberBand'):
            self.currentQRubberBand.hide()
            currentQRect = self.currentQRubberBand.geometry()
            # 这里可以添加裁剪逻辑，注意要转换为原始图像的坐标（考虑缩放比例）
            # 原始图像坐标 = currentQRect / self.scaleFactor
            print("裁剪区域（图像标签坐标）：", currentQRect)
            print("原始图像对应区域：", QRect(currentQRect.topLeft()/self.scaleFactor, currentQRect.size()/self.scaleFactor))
            self.currentQRubberBand.deleteLater()
            delattr(self, 'currentQRubberBand')

if __name__ == '__main__':
    app = QApplication(sys.argv)
    imageViewer = QImageViewer()
    imageViewer.show()
    sys.exit(app.exec_()) 

关键改动说明 #

1. 调整QRubberBand的父控件：将橡皮筋的父控件从主窗口改为self.imageLabel，这样橡皮筋的坐标系统直接对应图像标签的区域，无需额外计算窗口偏移。
2. 坐标转换：鼠标事件中使用mapFromGlobal将全局坐标转换为图像标签的局部坐标，确保坐标对应正确。
3. 范围限制：使用QRect.intersected()方法，让橡皮筋矩形始终和图像标签的区域取交集，强制限制在图像范围内绘制，避免超出。
4. 初始点校验：鼠标按下时，确保初始点在图像范围内，防止从图像外开始绘制。

内容的提问来源于stack exchange，提问作者Mhmd Admn