Demo/sgi/video/video.py

   1 import getopt
   2 from gl import *
   3 from GL import *
   4 from DEVICE import *
   5 import time
   6 import sys
   7 import al
   8 import AL
   9
  10 sys.path.append('/ufs/guido/src/video') # Increase chance to find colorsys
  11 import colorsys
  12
  13 BUFFERSIZE = 32000
  14
  15 class Struct(): pass
  16 epoch = Struct()
  17 epoch.correcttiming = 1
  18 EndOfFile = 'End of file'
  19 bye = 'bye'
  20
  21 def openspkr():
  22     conf = al.newconfig()
  23     conf.setqueuesize(BUFFERSIZE)
  24     conf.setwidth(AL.SAMPLE_16)
  25     conf.setchannels(AL.MONO)
  26     return al.openport('spkr','w',conf)
  27
  28 def openvideo(name):
  29     try:
  30         f = open(name, 'r')
  31     except:
  32         sys.stderr.write(name + ': cannot open\n')
  33         sys.exit(1)
  34     line = f.readline()
  35     if not line: raise EndOfFile
  36     colorinfo = (8, 0, 0, 0)
  37     if line[:4] == 'CMIF':
  38         if line[:14] == 'CMIF video 2.0':
  39             line = f.readline()
  40             colorinfo = eval(line[:-1])
  41         line = f.readline()
  42     x = eval(line[:-1])
  43     if len(x) == 3: w, h, pf = x
  44     else: w, h = x; pf = 2
  45     if pf and w/pf % 4 <> 0:
  46         sys.stderr.write( \
  47         'warning: stride not a multiple of 4 -- may not work on Indigo XS\n')
  48     return f, w, h, pf, colorinfo
  49
  50 def loadframe(f,w,h,pf,af,spkr, (ybits,ibits,qbits,chrompack),mf):
  51     line = f.readline()
  52     if line == '':
  53         raise EndOfFile
  54     x = eval(line[:-1])
  55     if type(x) == type(0) or type(x) == type(0.0):
  56         tijd = x
  57         if pf == 0:
  58                 size = w*h*4
  59         else:
  60                 size = (w/pf) * (h/pf)
  61     else:
  62         tijd, size = x
  63     data = f.read(size)
  64     if len(data) <> size:
  65         raise EndOfFile
  66     if pf:
  67         w = w/pf
  68         h = h/pf
  69     if chrompack:
  70         cw = (w+chrompack-1)/chrompack
  71         ch = (h+chrompack-1)/chrompack
  72         chromdata = f.read(2*cw*ch)
  73         rectzoom(pf*chrompack*mf,pf*chrompack*mf)
  74         pixmode(PM_SIZE,16)
  75         writemask(0x7ff - ((1<<ybits)-1))
  76         lrectwrite(0,0,cw-1,ch-1,chromdata)
  77         writemask((1<<ybits)-1)
  78         pixmode(PM_SIZE,8)
  79     if pf:
  80         rectzoom(pf*mf, pf*mf)
  81     elif mf <> 1:
  82         rectzoom(mf,mf)
  83     lrectwrite(0,0,w-1,h-1,data)
  84     # This is ugly here, but the only way to get the two
  85     # channels started in sync
  86     #if af <> None:
  87     #   playsound(af,spkr)
  88     ct = time.millitimer() - epoch.epoch
  89     if epoch.correcttiming and tijd > 0 and ct < tijd:
  90         time.millisleep(tijd-ct)
  91     #swapbuffers()
  92     return tijd
  93
  94 def initcmap(ybits,ibits,qbits,chrompack):
  95     if ybits+ibits+qbits > 11:
  96         raise 'Sorry, 11 bits max'
  97     maxy = pow(2,ybits)
  98     maxi = pow(2,ibits)
  99     maxq = pow(2,qbits)
 100     for i in range(2048,4096-256):
 101         mapcolor(i, 0, 255, 0)
 102     for y in range(maxy):
 103       yv = float(y)/float(maxy-1)
 104       for i in range(maxi):
 105         if maxi == 1: iv = 0
 106         else: iv = (float(i)/float(maxi-1))-0.5
 107         for q in range(maxq):
 108           if maxq == 1: qv = 0
 109           else: qv = (float(q)/float(maxq-1))-0.5
 110           index = 2048 + y + (i << ybits) + (q << (ybits+ibits))
 111           rv,gv,bv = colorsys.yiq_to_rgb(yv,iv,qv)
 112           r,g,b = int(rv*255.0), int(gv*255.0), int(bv*255.0)
 113           if index < 4096 - 256:
 114               mapcolor(index, r,g,b)
 115
 116 def playsound(af, spkr):
 117     nsamp = spkr.getfillable()
 118     data = af.read(nsamp*2)
 119     spkr.writesamps(data)
 120
 121 def main():
 122         looping = 0
 123         packfactor = 0
 124         magfactor = 1
 125         try:
 126                 opts, args = getopt.getopt(sys.argv[1:], 'm:p:lF')
 127         except getopt.error:
 128                 sys.stderr.write('usage: video ' + \
 129         '[-l] [-p pf] [-m mag] [-F] [moviefile [soundfile [skipbytes]]]\n')
 130                 sys.exit(2)
 131         for opt, arg in opts:
 132                 if opt == '-m':
 133                         magfactor = int(eval(arg))
 134                 elif opt == '-p':
 135                         packfactor = int(eval(arg))
 136                 elif opt == '-l':
 137                         looping = 1
 138                 elif opt == '-F':
 139                         epoch.correcttiming = 0
 140         if args:
 141                 filename = args[0]
 142         else:
 143                 filename = 'film.video'
 144         f, w, h, pf, cinfo = openvideo(filename)
 145         if 0 < packfactor <> pf:
 146                 w = w/pf*packfactor
 147                 h = h/pf*packfactor
 148                 pf = packfactor
 149         if args[1:]:
 150                 audiofilename = args[1]
 151                 af = open(audiofilename, 'r')
 152                 spkr = openspkr()
 153                 afskip = 0
 154                 if args[2:]:
 155                         afskip = eval(args[2])
 156                 af.seek(afskip)
 157         else:
 158                 af, spkr = None, None
 159         foreground()
 160         prefsize(w*magfactor,h*magfactor)
 161         win = winopen(filename)
 162         if pf:
 163             #doublebuffer()
 164             cmode()
 165         else:
 166             RGBmode()
 167         #doublebuffer()
 168         gconfig()
 169         if pf:
 170             initcmap(cinfo)
 171             color(2048)
 172             clear()
 173             writemask(2047)
 174             pixmode(PM_SIZE,8)  # 8 bit pixels
 175         qdevice(ESCKEY)
 176         qdevice(WINSHUT)
 177         qdevice(WINQUIT)
 178         running = 1
 179         epoch.epoch = time.millitimer()
 180         nframe = 0
 181         tijd = 1
 182         if looping:
 183                 looping = f.tell()
 184         try:
 185             while 1:
 186                 if running:
 187                     try:
 188                         tijd = loadframe(f, w, h, pf, af, spkr, cinfo,magfactor)
 189                         nframe = nframe + 1
 190                     except EndOfFile:
 191                         running = 0
 192                         t = time.millitimer()
 193                         if tijd > 0:
 194                                 print 'Recorded at',
 195                                 print 0.1 * int(nframe * 10000.0 / tijd),
 196                                 print 'frames/sec'
 197                         print 'Played', nframe, 'frames at',
 198                         print 0.1 * int(nframe * 10000.0 / (t-epoch.epoch)),
 199                         print 'frames/sec'
 200                         if looping:
 201                                 f.seek(looping)
 202                                 epoch.epoch = time.millitimer()
 203                                 nframe = 0
 204                                 running = 1
 205                                 if af <> None:
 206                                         af.seek(afskip)
 207                 if af <> None:
 208                         playsound(af,spkr)
 209                 if not running or qtest():
 210                     dev, val = qread()
 211                     if dev in (ESCKEY, WINSHUT, WINQUIT):
 212                         raise bye
 213                     elif dev == REDRAW:
 214                         reshapeviewport()
 215         except bye:
 216             pass
 217
 218 main()