[英]Controlling a QThread in pyqt5?
我用 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_())
我通过使用 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()
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.