cpython/Demo/sgi/video/Vrec.py

218 lines
4.9 KiB
Python
Raw Normal View History

1992-08-20 08:46:28 -03:00
#! /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()