1 import os, glob, subprocess, platform
\r
5 #Try to find the OpenCV library for video capture.
\r
6 from opencv import cv
\r
7 from opencv import highgui
\r
12 #Use the vidcap library directly from the VideoCapture package. (Windows only)
\r
13 # http://videocapture.sourceforge.net/
\r
14 # We're using the binary interface, not the python interface, so we don't depend on PIL
\r
15 import vidcap as win32vidcap
\r
19 def hasWebcamSupport():
\r
20 if cv == None and win32vidcap == None:
\r
22 if not os.path.exists(getFFMPEGpath()):
\r
26 def getFFMPEGpath():
\r
27 if platform.system() == "Windows":
\r
28 return os.path.normpath(os.path.join(os.path.split(__file__)[0], "../ffmpeg.exe"))
\r
29 elif os.path.exists('/usr/bin/ffmpeg'):
\r
30 return '/usr/bin/ffmpeg'
\r
31 return os.path.normpath(os.path.join(os.path.split(__file__)[0], "../ffmpeg"))
\r
33 class webcam(object):
\r
37 self._cam = highgui.cvCreateCameraCapture(-1)
\r
38 elif win32vidcap != None:
\r
40 self._cam = win32vidcap.new_Dev(0, False)
\r
44 self._doTimelaps = False
\r
47 def hasCamera(self):
\r
48 return self._cam != None
\r
50 def propertyPages(self):
\r
51 if self._cam == None:
\r
53 if win32vidcap != None:
\r
54 return ['Image properties', 'Format properties']
\r
56 #TODO Make an OpenCV property page
\r
59 def openPropertyPage(self, pageType = 0):
\r
60 if self._cam == None:
\r
62 if win32vidcap != None:
\r
64 self._cam.displaycapturefilterproperties()
\r
66 self._cam.displaycapturepinproperties()
\r
68 def takeNewImage(self):
\r
69 if self._cam == None:
\r
72 frame = cv.QueryFrame(self._cam)
\r
73 cv.CvtColor(frame, frame, cv.CV_BGR2RGB)
\r
74 self._bitmap = wx.BitmapFromBuffer(frame.width, frame.height, frame.imageData)
\r
75 elif win32vidcap != None:
\r
76 buffer, width, height = self._cam.getbuffer()
\r
77 wxImage = wx.EmptyImage(width, height)
\r
78 wxImage.SetData(buffer[::-1])
\r
79 if self._bitmap != None:
\r
81 self._bitmap = wxImage.ConvertToBitmap()
\r
85 if self._doTimelaps:
\r
86 filename = os.path.normpath(os.path.join(os.path.split(__file__)[0], "../__tmp_snap", "__tmp_snap_%04d.jpg" % (self._snapshotCount)))
\r
87 self._snapshotCount += 1
\r
88 self._bitmap.SaveFile(filename, wx.BITMAP_TYPE_JPEG)
\r
92 def getLastImage(self):
\r
95 def startTimelaps(self, filename):
\r
96 if self._cam == None:
\r
98 self._cleanTempDir()
\r
99 self._timelapsFilename = filename
\r
100 self._snapshotCount = 0
\r
101 self._doTimelaps = True
\r
103 def endTimelaps(self):
\r
104 if self._doTimelaps:
\r
105 ffmpeg = getFFMPEGpath()
\r
106 basePath = os.path.normpath(os.path.join(os.path.split(__file__)[0], "../__tmp_snap", "__tmp_snap_%04d.jpg"))
\r
107 subprocess.call([ffmpeg, '-r', '12.5', '-i', basePath, '-vcodec', 'mpeg2video', '-pix_fmt', 'yuv420p', '-r', '25', '-y', '-b:v', '1500k', '-f', 'vob', self._timelapsFilename])
\r
108 self._doTimelaps = False
\r
110 def _cleanTempDir(self):
\r
111 basePath = os.path.normpath(os.path.join(os.path.split(__file__)[0], "../__tmp_snap"))
\r
113 os.makedirs(basePath)
\r
116 for filename in glob.iglob(basePath + "/*.jpg"):
\r
117 os.remove(filename)
\r