Demo/sgi/video/camcorder.py

from gl import *
from GL import *
from DEVICE import *
import time
import sys
import getopt
import socket
import posix
import vtime

# Preallocation parameter
PREALLOC = 4 # Megabyte

# Sync audio parameters
SYNCPORT = 10000
CTLPORT = 10001

from vpregs import *

class Struct(): pass
epoch = Struct()

def getvideosize():
    w = getvideo(VP_WIDTH)
    h = getvideo(VP_HEIGHT)
    print 'WIDTH,HEIGHT:', w, h
    print 'GB{X,Y}ORG:', getvideo(VP_GBXORG), getvideo(VP_GBYORG)
    print 'FB{X,Y}ORG:',  getvideo(VP_FBXORG), getvideo(VP_FBYORG)
    x = 0
    y = 0
    return x,y,w,h

framelist = []

def prealloc(w, h):
	nbytes = w*h*4
	limit = PREALLOC*1024*1024
	total = 0
	list = []
	print 'Prealloc to', PREALLOC, 'Megabytes...'
	while total+nbytes <=  limit:
		list.append('x'*nbytes)
		total = total + nbytes
	print 'Done.'

def grabframe(f,x,y,w,h,pf):
    readsource(SRC_FRONT)
    if pf:
    	w = w/pf*pf
    	h = h/pf*pf
    data = lrectread(x,y,x+w-1,y+h-1)
    t = time.millitimer()-epoch.epoch
    framelist.append(data, t)
    readsource(SRC_FRAMEGRABBER)

def saveframes(f, w, h, pf):
	for data, t in framelist:
		if pf:
		    	w = w/pf*pf
		    	h = h/pf*pf
			data = packrect(w,h,pf,data)
		f.write(`t` + ',' + `len(data)` + '\n')
		f.write(data)
	framelist[:] = []

def saveframe(f,x,y,w,h,pf, notime):
    readsource(SRC_FRONT)
    if pf:
    	w = w/pf*pf
    	h = h/pf*pf
    data = lrectread(x,y,x+w-1,y+h-1)
    if pf: data = packrect(w,h,pf,data)
    if notime: t = 0
    else: t = time.millitimer()-epoch.epoch
    f.write(`t` + ',' + `len(data)` + '\n')
    f.write(data)
    readsource(SRC_FRAMEGRABBER)

def drawframe(x,y,w,h,col):
    drawmode(OVERDRAW)
    color(col)
    bgnline()
    v2i(x-1,y-1) ; v2i(x+w,y-1); v2i(x+w,y+h); v2i(x-1,y+h); v2i(x-1,y-1)
    endline()
    drawmode(NORMALDRAW)

def usage():
    sys.stderr.write('Usage: camcorder ' + \
	'[-c] [-p packfactor] [-a audiomachine [-s]] [outputfile]\n')
    sys.exit(2)

def wrheader(f, w, h, pf):
    	f.write('CMIF video 1.0\n')
	f.write(`w,h,pf` + '\n')
	print 'width,height,pf:', w, h, pf,
	if pf == 0: pf = 4
	print '(i.e.,', w*h*pf, 'bytes/frame)'

def main():
    foreground()
    pf = 2
    ausync = 0
    austart = 0
    optlist, args = getopt.getopt(sys.argv[1:],'ca:sp:')
    for opt, arg in optlist:
	if opt == '-c':
	    pf = 0
	elif opt == '-a':
	    ausync = 1
	    aumachine = arg
	elif opt == '-s':
	    austart = 1
	elif opt == '-p':
	    pf = int(eval(arg))
	else:
	    usage()
    if args:
    	if len(args) > 1:
    	    print 'Too many arguments'
    	    usage()
    	filename = args[0]
    else:
    	filename = 'film.video'
    if austart:
	if not ausync:
	    print 'Cannot use -s without -a'
	    usage()
	print 'Starting audio recorder...'
	posix.system('rsh '+aumachine+' syncrecord '+socket.gethostname()+' &')
    if ausync:
	print 'Syncing to audio recorder...'
	globtime = vtime.VTime().init(1,aumachine,SYNCPORT)
	ctl = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
	ctl.bind((socket.gethostname(),CTLPORT))
	aua = (socket.gethostbyname(aumachine), CTLPORT)
	print 'Done.'
    vidx, vidy, w, h = getvideosize()
    #prefsize(w,h)
    winx, winy = 1280-w-10, 1024-h-30
    prefposition(winx,winx+w-1,winy,winy+h-1)
    win = winopen(filename)
    f = open(filename, 'w')
    w, h = getsize()
    realw, realh = w, h
    ####doublebuffer()
    RGBmode()
    gconfig()
    qdevice(LEFTMOUSE)
    qdevice(RKEY)
    qdevice(SKEY)
    qdevice(CKEY)
    qdevice(PKEY)
    qdevice(ESCKEY)
    qdevice(WINQUIT)
    qdevice(WINSHUT)
    inrunning = 1
    outrunning = 0
    stop = 'stop'
    readsource(SRC_FRAMEGRABBER)
    mousing = 0
    epoch.epoch = time.millitimer()
    stoptime = epoch.epoch
    sizewritten = 0
    x, y = realw/4, realh/4
    w, h = w/2, h/2
    prealloc(w, h)
    try:
	drawframe(x,y,w,h,1)
	nframe = 0
	num = 0
	while 1:
	    insingle = 0
	    outsingle = 0
	    if mousing:
		drawframe(x,y,w,h,0)
		ox, oy = getorigin()
		if sizewritten:
		    x = getvaluator(MOUSEX)-ox
		    y = getvaluator(MOUSEY)-oy
		else:
		    w = getvaluator(MOUSEX)-x-ox
		    h = getvaluator(MOUSEY)-y-oy
		drawframe(x,y,w,h,1)
	    if qtest() or \
	    not (mousing or inrunning or insingle or outrunning or outsingle):
		ev, val = qread()
		if ev == LEFTMOUSE and val == 1:
		    drawframe(x,y,w,h,0)
		    mousing = 1
		    ox, oy = getorigin()
		    x = getvaluator(MOUSEX)-ox
		    y = getvaluator(MOUSEY)-oy
		elif ev == LEFTMOUSE and val == 0:
		    if h < 0:
		    	y, h = y+h, -h
		    if w < 0:
		    	x, w = x+w, -w
		    mousing = 0
		    if not sizewritten:
		    	wrheader(f, w, h, pf)
			sizewritten = 1
			prealloc(w, h)
		elif ev == RKEY and val == 1:
		    if not inrunning:
			ringbell()
		    else:
			outrunning = 1
			wasstopped = time.millitimer() - stoptime
			epoch.epoch = epoch.epoch + wasstopped
			nframe = 0
			starttime = time.millitimer()
			if ausync:
			    ctl.sendto(`(1,starttime)`, aua)
		elif ev == PKEY and val == 1 and outrunning:
		    outrunning = 0
		    stoptime = time.millitimer()
		    if ausync:
			ctl.sendto(`(0,stoptime)`, aua)
		    fps =  nframe * 1000.0 / (time.millitimer()-starttime)
		    print 'Recorded', nframe,
		    print 'frames at', 0.1*int(fps*10),'frames/sec'
		    print 'Saving...'
		    saveframes(f, w, h, pf)
		    print 'Done.'
		elif ev == PKEY and val == 1 and not outrunning:
			outsingle = 1
		elif ev == CKEY and val == 1:
			inrunning = 1
		elif ev == SKEY and val == 1:
			if outrunning:
			    ringbell()
			elif inrunning:
			    inrunning = 0
			else:
			    insingle = 1
		elif ev in (ESCKEY, WINQUIT, WINSHUT):
		    if ausync:
			ctl.sendto(`(2,time.millitimer())`, aua)
		    raise stop
		elif ev == REDRAW:
			drawframe(x,y,w,h,0)
			reshapeviewport()
			drawframe(x,y,w,h,1)
	    if inrunning or insingle:
	    	if outrunning:
			rectcopy(vidx+x,vidy+y,vidx+x+w-1,vidy+y+h-1,x,y)
	    	else:
			rectcopy(vidx,vidy,vidx+realw-1,vidx+realh-1,0,0)
	        ####swapbuffers()
	    if outrunning or outsingle:
		nframe = nframe + 1
		if not sizewritten:
		    wrheader(f, w, h, pf)
		    sizewritten = 1
		if outrunning:
			grabframe(f, x, y, w, h, pf)
		else:
			saveframe(f, x, y, w, h, pf, outsingle)
    except stop:
	pass
    finally:
        drawmode(OVERDRAW)
        color(0)
        clear()

main()