在 pyqt5 中控制 QThread？

Question

我用 pyqt5 和 opencv 編寫了一個網絡攝像頭應用程序，它可以保存實時視頻。 截圖： https : //i.hizliresim.com/kM8GOD.png

我使用 QThread 進行流式傳輸而不會阻塞主應用程序。 但問題是，我無法使用按鈕控制流媒體和保存視頻。 它在我運行主程序時開始錄制，但我想用 RUN 按鈕進行錄制並用 STOP 按鈕保存。

我試過用錄音參數來做，但有點奇怪。 程序最初開始錄制，我第一次按下停止鍵時它完美地保存了視頻。 但是當我嘗試使用開始按鈕保存第二個視頻時。 它沒有開始在標簽上流式傳輸，但是當我關閉應用程序時，它正在保存視頻。

class Thread(QThread):
    recording = True
    changePixmap = pyqtSignal(QImage)
    cap = cv2.VideoCapture(0)
    frame_width = int(cap.get(3))
    frame_height = int(cap.get(4))

    out = cv2.VideoWriter('input.avi', cv2.VideoWriter_fourcc(
        'M', 'J', 'P', 'G'), 30, (frame_width, frame_height))

    def run(self):
        while self.recording:
            ret, frame = self.cap.read()
            if ret:
                self.out.write(frame)
                rgbImage = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
                h, w, ch = rgbImage.shape
                bytesPerLine = ch * w
                convertToQtFormat = QtGui.QImage(
                    rgbImage.data, w, h, bytesPerLine, QtGui.QImage.Format_RGB888)
                p = convertToQtFormat.scaled(
                    350, 350, PyQt5.QtCore.Qt.KeepAspectRatio)
                self.changePixmap.emit(p)
        # self.cap.release()
        # self.out.release()
        self.changePixmap.emit(p)


class Ui_MainWindow(QtWidgets.QWidget):
    def setupUi(self, MainWindow):
        MainWindow.setObjectName("MainWindow")
        MainWindow.resize(901, 593)
        self.centralwidget = QtWidgets.QWidget(MainWindow)
        self.centralwidget.setObjectName("centralwidget")
        self.label = QtWidgets.QLabel(self.centralwidget)
        self.label.setGeometry(QtCore.QRect(20, 20, 350, 350))
        self.label.setObjectName("label")
        self.th = Thread(self)
        self.th.changePixmap.connect(self.setImage)
        self.th.start()
        self.pushButton = QtWidgets.QPushButton(self.centralwidget)
        self.pushButton.setGeometry(QtCore.QRect(20, 320, 261, 31))
        self.pushButton.setObjectName("pushButton")
        self.pushButton.clicked.connect(self.func)
        self.label_2 = QtWidgets.QLabel(self.centralwidget)
        self.label_2.setGeometry(QtCore.QRect(470, 20, 261, 271))
        self.label_2.setText("")
        self.label_2.setObjectName("label_2")
        self.pushButton_2 = QtWidgets.QPushButton(self.centralwidget)
        self.pushButton_2.setGeometry(QtCore.QRect(20, 390, 261, 31))
        self.pushButton_2.setObjectName("pushButton_2")
        self.pushButton_2.clicked.connect(self.func2)
        MainWindow.setCentralWidget(self.centralwidget)
        self.menubar = QtWidgets.QMenuBar(MainWindow)
        self.menubar.setGeometry(QtCore.QRect(0, 0, 901, 22))
        self.menubar.setObjectName("menubar")
        MainWindow.setMenuBar(self.menubar)
        self.statusbar = QtWidgets.QStatusBar(MainWindow)
        self.statusbar.setObjectName("statusbar")
        MainWindow.setStatusBar(self.statusbar)

        self.retranslateUi(MainWindow)
        QtCore.QMetaObject.connectSlotsByName(MainWindow)

    def retranslateUi(self, MainWindow):
        _translate = QtCore.QCoreApplication.translate
        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
        self.pushButton.setText(_translate("MainWindow", "record"))
        self.pushButton_2.setText(_translate("MainWindow", "stop"))

    def setImage(self, image):
        self.label.setPixmap(QPixmap.fromImage(image))

    def func(self):
        self.th.recording = True

    def func2(self):
        self.th.recording = False


if __name__ == "__main__":
    import sys
    app = QtWidgets.QApplication(sys.argv)
    MainWindow = QtWidgets.QMainWindow()
    ui = Ui_MainWindow()
    ui.setupUi(MainWindow)
    MainWindow.show()
    sys.exit(app.exec_())

Answer 1

我通過使用 python 標准庫events解決了這個問題。

你的while self.recording:循環在run()只運行一次，完成后你不能再次輸入它。 在我的代碼中，將self.playing設置為 false 后，我等待再次設置 actionTaken 事件以繼續循環。

from threading import Event

class Controltool(QThread):
    changePixmap = pyqtSignal(QImage)

    def __init__(self, videopath):
        super(Controltool, self).__init__()

        self.running = Event()
        self.running.set()
        self.actionTaken = Event()
        self.playing = False

    def run(self):
        # We only leave this loop once we finish
        while self.running.is_set():
            self.actionTaken.clear()
            while self.playing:
                self.readFrame()
            self.actionTaken.wait()
            
    def readFrame(self):
        ret, frame = self.vidcap.read()
        if ret:
            [... blah blah ...]
            self.changePixmap.emit(p)
            
    def stopPlaying(self):
        self.playing = False

    def startPlaying(self):
        self.playing = True
        self.actionTaken.set()