Initial revision

Demo/sgi/video/Vrec.py

+#! /ufs/guido/bin/sgi/python-405
+#! /ufs/guido/bin/sgi/python
+
+# Capture a CMIF movie using the Indigo video library and board
+
+
+# Usage:
+#
+# makemovie [-a] [-q queuesize] [-r n/d] [moviefile [audiofile]]
+
+
+# Options:
+#
+# -a            : record audio as well
+# -q queuesize  : set the capture queue size (default and max 16)
+# -r n/d        : capture n out of every d frames (default 1/1)
+#                 XXX doesn't work yet
+# 
+# moviefile     : here goes the movie data (default film.video);
+#                 the format is documented in cmif-film.ms
+# audiofile     : with -a, here goes the audio data (default film.aiff);
+#                 audio data is recorded in AIFF format, using the
+#                 input sampling rate, source and volume set by the
+#                 audio panel, in mono, 8 bits/sample
+
+
+# User interface:
+#
+# Start the application.  Resize the window to the desired movie size.
+# Press the left mouse button to start recording, release it to end
+# recording.  XXX For now, you must restart the program to record again.
+# You can't resize the window once you have made a recording.
+#
+# Press ESC or select the window manager Quit or Close window option
+# to quit.  (You can do this without recording -- then the output
+# files are untouched.)  XXX Don't press ESC before the program has
+# finished writing the output file -- it prints "Done writing" when it
+# is done.
+
+
+# XXX To do:
+#
+# add audio
+
+
+import sys
+sys.path.append('/ufs/guido/src/video')
+import sv, SV
+import VFile
+import gl, GL, DEVICE
+import al, AL
+import time
+import posix
+import getopt
+import string
+
+
+# Main program
+
+def main():
+	QSIZE = 16
+	TIME = 5
+	audio = 0
+	num, den = 1, 1
+
+	opts, args = getopt.getopt(sys.argv[1:], 'aq:r:t:')
+	for opt, arg in opts:
+		if opt == '-a':
+			audio = 1
+		elif opt == '-q':
+			QSIZE = string.atoi(arg)
+		elif opt == '-r':
+			[nstr, dstr] = string.splitfields(arg, '/')
+			num, den = string.atoi(nstr), string.atoi(dstr)
+		elif opt == '-t':
+			TIME = string.atoi(arg)
+
+	if args:
+		filename = args[0]
+	else:
+		filename = 'film.video'
+
+	if audio:
+		if args[1:]:
+			audiofilename = args[1]
+		else:
+			audiofilename = 'film.aiff'
+	else:
+		audiofilename = None
+
+	gl.foreground()
+
+	x, y = SV.PAL_XMAX / 4, SV.PAL_YMAX / 4
+	print x, 'x', y
+
+	gl.minsize(40, 30)
+	gl.stepunit(8, 6)
+	gl.maxsize(SV.PAL_XMAX, SV.PAL_YMAX)
+	gl.keepaspect(SV.PAL_XMAX, SV.PAL_YMAX)
+	win = gl.winopen(filename)
+	x, y = gl.getsize()
+	print x, 'x', y
+
+	v = sv.OpenVideo()
+	v.BindGLWindow(win, SV.IN_REPLACE)
+	v.SetSize(x, y)
+	v.BindGLWindow(win, SV.IN_REPLACE)
+
+	v.SetCaptureFormat(SV.RGB_FRAMES)
+	v.SetCaptureMode(SV.BLOCKING_CAPTURE)
+	v.SetQueueSize(QSIZE)
+
+	v.InitCapture()
+	if v.GetQueueSize() != QSIZE:
+		QSIZE = v.GetQueueSize()
+		print 'Warning: QSIZE reduced to', QSIZE
+
+###	v.SetSamplingRate(num, den) # XXX dumps core
+
+	gl.qdevice(DEVICE.LEFTMOUSE)
+	gl.qdevice(DEVICE.WINQUIT)
+	gl.qdevice(DEVICE.WINSHUT)
+	gl.qdevice(DEVICE.ESCKEY)
+
+	print 'Press left mouse to start recording, release it to stop'
+
+	recorded = 0
+
+	while 1:
+		dev, val = gl.qread()
+		if dev == DEVICE.LEFTMOUSE:
+			if val == 1:
+				if recorded:
+					# XXX This would dump core
+					gl.ringbell()
+					continue
+				# Fix the window's size now
+				gl.prefsize(x, y)
+				gl.winconstraints()
+				record(v, filename, audiofilename)
+				recorded = 1
+				print 'Wait until "Done writing" is printed!'
+		elif dev == DEVICE.REDRAW:
+			# Window resize (or move)
+			if not recorded:
+				x, y = gl.getsize()
+				print x, 'x', y
+				v.SetSize(x, y)
+				v.BindGLWindow(win, SV.IN_REPLACE)
+		elif dev in (DEVICE.ESCKEY, DEVICE.WINQUIT, DEVICE.WINSHUT):
+			# Quit
+			if not recorded:
+				# XXX Avoid core dump in EndCapture
+				posix._exit(0)
+			v.EndCapture()
+			v.CloseVideo()
+			gl.winclose(win)
+			break
+
+
+# Record until the mouse is released (or any other GL event)
+# XXX audio not yet supported
+
+def record(v, filename, audiofilename):
+	import thread
+	x, y = gl.getsize()
+	vout = VFile.VoutFile().init(filename)
+	vout.format = 'rgb8'
+	vout.width = x
+	vout.height = y
+	vout.writeheader()
+	buffer = []
+	thread.start_new_thread(saveframes, (vout, buffer))
+	gl.wintitle('(rec) ' + filename)
+	v.StartCapture()
+	t0 = time.millitimer()
+	while not gl.qtest():
+		if v.GetCaptured() > 2:
+			t = time.millitimer() - t0
+			cd, st = v.GetCaptureData()
+			data = cd.interleave(x, y)
+			cd.UnlockCaptureData()
+			buffer.append(data, t)
+		else:
+			time.millisleep(10)
+	v.StopCapture()
+	while v.GetCaptured() > 0:
+		t = time.millitimer() - t0
+		cd, st = v.GetCaptureData()
+		data = cd.interleave(x, y)
+		cd.UnlockCaptureData()
+		buffer.append(data, t)
+	buffer.append(None) # Sentinel
+	gl.wintitle('(done) ' + filename)
+
+
+# Thread to save the frames to the file
+
+def saveframes(vout, buffer):
+	while 1:
+		if not buffer:
+			time.millisleep(10)
+		else:
+			x = buffer[0]
+			del buffer[0]
+			if not x:
+				break
+			data, t = x
+			vout.writeframe(t, data, None)
+			del data
+	sys.stderr.write('Done writing\n')
+	vout.close()
+
+
+# Don't forget to call the main program
+
+main()