PyQt+OpenCV 录制保存、播放视频

准备这一两个月看看以前的书，另外学习下视频方面的东西（CV方面）。之前学过几天QT/PyQt，于是用PyQt结合OpenCV，做了一个小程序。具有播放并保存摄像头捕获组成的视频，以及播放本地视频文件两个小功能。界面很简单，没图没真相。代码就100多行，很容易看懂。最后感叹下OpenCV和python的强大，噢，还有QT/PyQt。。#！ /usr/bin/env python
#coding=utf-8
from PyQt4.QtCore import *
from PyQt4.QtGui import *
import sys
import opencv
from PIL import Image
#this is important for capturing/displaying images
from opencv import highgui
from opencv import cv
import time
MPEG1VIDEO = 0x314D4950
camera = highgui.cvCreateCameraCapture（0）#找摄像头，一般填0-99都可以
class MainWindow（QWidget）:
def __init__（self, parent = None）:
QWidget.__init__（self）
self.resize（550, 550）
self.setWindowTitle（"vedio control"）
self.status = 0 #0 is init status;1 is play video; 2 is capture video
self.image = QImage（）

#录制的视频保存位置、格式等参数设定
self.videowriter = highgui.cvCreateVideoWriter（"test.mpg", highgui.CV_FOURCC（"m","p","g","1"）, 25, cv.cvSize（200,200）, 1）
#播放的视频位置
self.playcapture = highgui.cvCreateFileCapture（"test.avi"）

#初始化按钮
self.capturebtn = QPushButton（"capture"）
self.playbtn = QPushButton（"play"）
exitbtn = QPushButton（"exit"）

# 界面布局
vbox = QVBoxLayout（）
vbox.addWidget（self.capturebtn）
vbox.addWidget（self.playbtn）
vbox.addWidget（exitbtn）

self.piclabel = QLabel（"pic"）
hbox = QHBoxLayout（）
hbox.addLayout（vbox）
hbox.addStretch（1）
hbox.addWidget（self.piclabel）

self.setLayout（hbox）

#加载初始页面
if self.image.load（"1.jpg"）:
self.piclabel.setPixmap（QPixmap.fromImage（self.image））

#设定定时器
self.timer = Timer（） #录制视频
self.playtimer = Timer（"updatePlay（）"）#播放视频

#信号--槽
self.connect（self.timer, SIGNAL（"updateTime（）"）,
self.CaptureVGA）
self.connect（self.capturebtn, SIGNAL（"clicked（）"）,
self.PauseBegin）
self.connect（self.playtimer, SIGNAL（"updatePlay（）"）,
self.PlayVideo）
self.connect（self.playbtn, SIGNAL（"clicked（）"）,
self.VideoPlayPause）
self.connect（exitbtn, SIGNAL（"clicked（）"）,
app, SLOT（"quit（）"））

def PlayVideo（self）:
im = highgui.cvQueryFrame（self.playcapture）
im = opencv.adaptors.Ipl2PIL（im）
im = im.convert（"RGB"）.tostring（"jpeg", "RGB"）
self.image.loadFromData（QByteArray（im））
self.piclabel.setPixmap（QPixmap.fromImage（self.image））

def VideoPlayPause（self）:
self.status, playstr, capturestr = （（1, "pause", "capture"）, （0, "play", "capture"）, （1, "pause", "capture"））[self.status]#三种状态分别对应的显示、处理
self.playbtn.setText（playstr）
self.capturebtn.setText（capturestr）
if self.status is 1:#状态1，播放视频
self.timer.stop（）
self.playtimer.start（）
else:
self.playtimer.stop（）

def PauseBegin（self）:
self.status, playstr, capturestr = （（2, "play", "pause"）, （2, "play", "pause"）, （0, "play", "capture"））[self.status]
self.capturebtn.setText（capturestr）
self.playbtn.setText（playstr）
print self.status, playstr, capturestr
if self.status is 2:#状态2，录制并显示视频
self.timer.start（）
self.playtimer.stop（）
else:
self.timer.stop（）

def CaptureVGA（self）:
im = highgui.cvQueryFrame（camera）
highgui.cvWriteFrame（self.videowriter, im）#录制视频，写入文件
#convert Ipl image to PIL image
im = opencv.adaptors.Ipl2PIL（im）
im = im.convert（"RGB"）.tostring（"jpeg", "RGB"）# 转换格式，jpeg
self.image.loadFromData（QByteArray（im））#格式支持QT，直接加载
#im.save（"3.jpg"）#opencv 返回的是Ipl 格式，QT无法直接显示。不知道如何转换格式，采用保存、读取的方式。
#pic.load（"3.jpg"）
self.piclabel.setPixmap（QPixmap.fromImage（self.image）） #一帧一帧的显示

class Timer（QThread）:

def __init__（self, signal = "updateTime（）", parent=None）:
super（Timer, self）.__init__（parent）
self.stoped = False
self.signal = signal
self.mutex = QMutex（）
def run（self）:
with QMutexLocker（self.mutex）:
self.stoped = False
while True:
if self.stoped:
return
self.emit（SIGNAL（self.signal））
time.sleep（0.04） #40毫秒发送一次信号，每秒25帧

def stop（self）:
with QMutexLocker（self.mutex）:
self.stoped = True

def isStoped（self）:
with QMutexLocker（sellf.mutex）:
return self.stoped

if __name__ == "__main__" :
app = QApplication（sys.argv）
main = MainWindow（）
main.show（）
sys.exit（app.exec_（））