Made much faster, but for SHORT movies only, by saving lrectread
data unprocessed in memory. As long as memory doesn't run out this gets us 19 frames/sec! Also many cosmetic changes.
This commit is contained in:
Guido van Rossum
parent
96059b318c
commit
5395c067de
|
|
@ -1,4 +1,3 @@
|
|||
#!/ufs/guido/bin/sgi/python3.3
|
||||
from gl import *
|
||||
from GL import *
|
||||
from DEVICE import *
|
||||
|
|
@ -9,15 +8,14 @@ import socket
|
|||
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()
|
||||
|
|
@ -25,17 +23,52 @@ 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
|
||||
|
|
@ -43,6 +76,7 @@ def saveframe(f,x,y,w,h,pf, notime):
|
|||
f.write(`t` + ',' + `len(data)` + '\n')
|
||||
f.write(data)
|
||||
readsource(SRC_FRAMEGRABBER)
|
||||
|
||||
def drawframe(x,y,w,h,col):
|
||||
drawmode(OVERDRAW)
|
||||
color(col)
|
||||
|
|
@ -50,12 +84,25 @@ def drawframe(x,y,w,h,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
|
||||
|
|
@ -64,13 +111,21 @@ def main():
|
|||
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:
|
||||
|
|
@ -81,15 +136,14 @@ def main():
|
|||
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)
|
||||
|
|
@ -98,6 +152,8 @@ def main():
|
|||
qdevice(CKEY)
|
||||
qdevice(PKEY)
|
||||
qdevice(ESCKEY)
|
||||
qdevice(WINQUIT)
|
||||
qdevice(WINSHUT)
|
||||
inrunning = 1
|
||||
outrunning = 0
|
||||
stop = 'stop'
|
||||
|
|
@ -108,9 +164,10 @@ def main():
|
|||
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
|
||||
|
|
@ -125,7 +182,8 @@ def main():
|
|||
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)
|
||||
|
|
@ -134,12 +192,16 @@ def main():
|
|||
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:
|
||||
|
|
@ -155,14 +217,12 @@ def main():
|
|||
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:
|
||||
|
|
@ -174,24 +234,34 @@ def main():
|
|||
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()
|
||||
|
|
|
|||
Loading…
Reference in New Issue