QFabDVR.py

from PyQt4 import QtGui
from PyQt4.QtCore import Qt, pyqtSignal
from fabdvr import fabdvr


class QFabDVR(fabdvr, QtGui.QFrame):

    recording = pyqtSignal(bool)

    def __init__(self, **kwargs):
        super(QFabDVR, self).__init__(**kwargs)
        self.initUI()

    def initUI(self):
        self.setFrameShape(QtGui.QFrame.Box)
        layout = QtGui.QGridLayout(self)
        layout.setMargin(1)
        layout.setHorizontalSpacing(6)
        layout.setVerticalSpacing(3)
        title = QtGui.QLabel('Video Recorder')
        self.brecord = QtGui.QPushButton('Record', self)
        self.bstop = QtGui.QPushButton('Stop', self)
        self.wframe = QtGui.QLCDNumber(self)
        self.wframe.setNumDigits(5)
        wfilelabel = QtGui.QLabel('file name')
        wfilelabel.setAlignment(Qt.AlignRight | Qt.AlignVCenter)
        self.wfilename = QtGui.QLineEdit()
        self.wfilename.setText(self.filename)
        layout.addWidget(title, 1, 1, 1, 3)
        layout.addWidget(self.brecord, 2, 1)
        layout.addWidget(self.bstop, 2, 2)
        layout.addWidget(self.wframe, 2, 3)
        layout.addWidget(wfilelabel, 3, 1)
        layout.addWidget(self.wfilename, 3, 2, 1, 2)
        self.setLayout(layout)
        self.brecord.clicked.connect(self.handleRecord)
        self.bstop.clicked.connect(self.handleStop)

    def write(self, frame):
        super(QFabDVR, self).write(frame)
        self.wframe.display(self.framenumber)

    def handleRecord(self):
        super(QFabDVR, self).record(1000)
        self.recording.emit(True)

    def handleStop(self):
        super(QFabDVR, self).stop()
        self.recording.emit(False)


def main():
    import sys
    from QCameraDevice import QCameraDevice
    from QVideoItem import QVideoItem, QVideoWidget

    app = QtGui.QApplication(sys.argv)
    device = QCameraDevice(size=(640, 480), gray=True)
    video = QVideoItem(device)
    widget = QVideoWidget(video, background='w')
    widget.show()
    dvr = QFabDVR(source=video)
    dvr.show()
    sys.exit(app.exec_())


if __name__ == '__main__':
    main()