-#!/ufs/guido/bin/sgi/python3.3
from gl import *
from GL import *
from DEVICE import *
import posix
import vtime
+# Preallocation parameter
+PREALLOC = 8 # Megabyte
+
+# Sync audio parameters
SYNCPORT = 10000
CTLPORT = 10001
-VP_GBXORG = 0x1000001
-VP_GBYORG = 0x1000002
-VP_FBXORG = 0x1000003
-VP_FBYORG = 0x1000004
-VP_WIDTH = 0x1000005
-VP_HEIGHT = 0x1000006
+from vpregs import *
class Struct(): pass
epoch = Struct()
def getvideosize():
w = getvideo(VP_WIDTH)
h = getvideo(VP_HEIGHT)
- print getvideo(VP_GBXORG), getvideo(VP_GBYORG)
- print getvideo(VP_FBXORG), getvideo(VP_FBYORG)
+ 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):
+#### saveframes(f, 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 = None
data = lrectread(x,y,x+w-1,y+h-1)
if pf: data = packrect(w,h,pf,data)
if notime: t = 0
f.write(`t` + ',' + `len(data)` + '\n')
f.write(data)
readsource(SRC_FRAMEGRABBER)
+
def drawframe(x,y,w,h,col):
drawmode(OVERDRAW)
color(col)
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:s')
+ optlist, args = getopt.getopt(sys.argv[1:],'ca:sp:')
for opt, arg in optlist:
if opt = '-c':
pf = 0
aumachine = arg
elif opt = '-s':
austart = 1
+ elif opt = '-p':
+ pf = int(eval(arg))
else:
- print 'Usage: camcorder [-c] [-a audiomachine [-s]]'
- sys.exit(1)
+ 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'
- sys.exit(1)
+ usage()
print 'Starting audio recorder...'
posix.system('rsh '+aumachine+' syncrecord '+socket.gethostname()+' &')
if ausync:
aua = (socket.gethostbyname(aumachine), CTLPORT)
print 'Done.'
vidx, vidy, w, h = getvideosize()
- prefsize(w,h)
- win = winopen('Camcorder')
- if len(args) > 1:
- f = open(args, 'w')
- else:
- f = open('film.video', 'w')
+ #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()
+ #doublebuffer()
RGBmode()
gconfig()
qdevice(LEFTMOUSE)
qdevice(CKEY)
qdevice(PKEY)
qdevice(ESCKEY)
+ qdevice(WINQUIT)
+ qdevice(WINSHUT)
inrunning = 1
outrunning = 0
stop = 'stop'
sizewritten = 0
x, y = realw/4, realh/4
w, h = w/2, h/2
- drawframe(x,y,w,h,1)
- nframe = 0
+ prealloc(w, h)
try:
+ drawframe(x,y,w,h,1)
+ nframe = 0
num = 0
while 1:
insingle = 0
w = getvaluator(MOUSEX)-x-ox
h = getvaluator(MOUSEY)-y-oy
drawframe(x,y,w,h,1)
- if qtest():
+ 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)
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:
- f.write('CMIF video 1.0\n')
- f.write(`w,h,pf` + '\n')
+ wrheader(f, w, h, pf)
sizewritten = 1
- if ev = RKEY and val = 1:
+ prealloc(w, h)
+ elif ev = RKEY and val = 1:
if not inrunning:
ringbell()
else:
stoptime = time.millitimer()
if ausync:
ctl.sendto(`(0,stoptime)`, aua)
- nf = nframe * 1000.0 / (time.millitimer()-starttime)
- drawmode(OVERDRAW)
- color(0)
- clear()
- color(1)
- cmov2i(5,5)
- charstr('Recorded ' + `nf` + ' frames/sec')
- drawmode(NORMALDRAW)
+ 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 = 0
else:
insingle = 1
- elif ev = ESCKEY:
+ 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:
- rectcopy(vidx,vidy,realw,realh,0,0)
- swapbuffers()
+ 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:
- f.write('CMIF video 1.0\n')
- f.write(`w,h,pf` + '\n')
+ wrheader(f, w, h, pf)
sizewritten = 1
- saveframe(f, x, y, w, h, pf, outsingle)
+ if outrunning:
+ grabframe(f, x, y, w, h, pf)
+ else:
+ saveframe(f, x, y, w, h, pf, outsingle)
except stop:
pass
- drawmode(OVERDRAW)
- color(0)
- clear()
-#
+ finally:
+ drawmode(OVERDRAW)
+ color(0)
+ clear()
+
main()
projects / thirdparty / Python / cpython.git / commitdiff
