standardized lay-out; new syntax.

This commit is contained in:
Guido van Rossum 1992-02-11 14:45:43 +00:00
parent 5778949295
commit 58b38ccc9d
1 changed files with 78 additions and 63 deletions

View File

@ -9,12 +9,14 @@ import colorsys
Error = 'VFile.Error' # Exception
MAXMAP = 4096 - 256
def conv_grey(l,x,y): return colorsys.yiq_to_rgb(l,0,0)
def conv_yiq (y,i,q): return colorsys.yiq_to_rgb(y, (i-0.5)*1.2, q-0.5)
def conv_hls (l,h,s): return colorsys.hls_to_rgb(h,l,s)
def conv_hsv (v,h,s): return colorsys.hsv_to_rgb(h,s,v)
def conv_rgb (r,g,b):
raise Error, 'Attempt to make RGB colormap'
raise Error, 'Attempt to make RGB colormap'
# Class VinFile represents a video file used for input.
#
@ -35,6 +37,7 @@ def conv_rgb (r,g,b):
# format
#
# These writable data members provide additional parametrization:
# magnify
# xorigin, yorigin
class VinFile():
@ -43,7 +46,7 @@ class VinFile():
# init() raises whatever open() raises if the file can't be opened.
def init(self, filename):
if filename = '-':
if filename == '-':
return self.initfp(sys.stdin, filename)
return self.initfp(open(filename, 'r'), filename)
@ -55,11 +58,11 @@ class VinFile():
self.filename = filename
#
line = self.fp.readline()
if line = 'CMIF video 1.0\n':
if line == 'CMIF video 1.0\n':
self.version = 1.0
elif line = 'CMIF video 2.0\n':
elif line == 'CMIF video 2.0\n':
self.version = 2.0
elif line = 'CMIF video 3.0\n':
elif line == 'CMIF video 3.0\n':
self.version = 3.0
else:
raise Error, self.filename + ': bad video format'
@ -69,47 +72,50 @@ class VinFile():
self.chrompack = 0
self.offset = 0
self.format = 'grey'
elif self.version = 2.0:
elif self.version == 2.0:
line = self.fp.readline()
try:
self.c0bits, self.c1bits, self.c2bits, \
self.chrompack = eval(line[:-1])
if self.c1bits or self.c2bits:
self.format = 'yiq'
self.format = 'yiq'
else:
self.format = 'grey'
self.format = 'grey'
self.offset = 0
except:
raise Error, self.filename + ': bad 2.0 color info'
elif self.version = 3.0:
raise Error, \
self.filename + ': bad 2.0 color info'
elif self.version == 3.0:
line = self.fp.readline()
try:
self.format, rest = eval(line[:-1])
if self.format = 'rgb':
pass
elif self.format = 'grey':
self.offset = 0
self.c0bits = rest
self.c1bits = self.c2bits = \
self.format, rest = eval(line[:-1])
if self.format == 'rgb':
pass
elif self.format == 'grey':
self.offset = 0
self.c0bits = rest
self.c1bits = self.c2bits = \
self.chrompack = 0
else:
self.c0bits,self.c1bits,self.c2bits,\
self.chrompack,self.offset = rest
else:
self.c0bits,self.c1bits,self.c2bits,\
self.chrompack,self.offset = rest
except:
raise Error, self.filename + ': bad 3.0 color info'
raise Error, \
self.filename + ': bad 3.0 color info'
try:
self.convcolor = eval('conv_'+self.format)
self.convcolor = eval('conv_'+self.format)
except:
raise Error, self.filename + ': unknown colorsys ' + self.format
raise Error, \
self.filename + ': unknown colorsys ' + self.format
#
line = self.fp.readline()
try:
x = eval(line[:-1])
if self.version > 1.0 or len(x) = 3:
if self.version > 1.0 or len(x) == 3:
self.width, self.height, self.packfactor = x
if self.packfactor = 0:
self.format = 'rgb'
if self.packfactor == 0:
self.format = 'rgb'
else:
sef.width, self.height = x
self.packfactor = 2
@ -153,11 +159,10 @@ class VinFile():
def skipnextframedata(self, (size, chromsize)):
# Note that this won't raise EOFError for a partial frame.
try:
self.fp.seek(size + chromsize, 1) # Relatc1ve seek
self.fp.seek(size + chromsize, 1) # Relative seek
except:
# Assume it's a pipe -- read the data to discard it
dummy = self.fp.read(size)
dummy = self.fp.read(chromsize)
dummy = self.fp.read(size + chromsize)
def getnextframeheader(self):
line = self.fp.readline()
@ -169,11 +174,11 @@ class VinFile():
x = eval(line[:-1])
if type(x) in (type(0), type(0.0)):
time = x
if pf = 0:
if pf == 0:
size = w * h * 4
else:
size = (w/pf) * (h/pf)
elif len(x) = 2:
elif len(x) == 2:
time, size = x
cp = self.chrompack
if cp:
@ -221,70 +226,80 @@ class VinFile():
def initcolormap(self):
self.colormapinited = 1
if self.format = 'rgb':
gl.RGBmode()
gl.gconfig()
return
initcmap(self.convcolor, self.c0bits, self.c1bits, self.c2bits, self.chrompack, self.offset)
if self.format == 'rgb':
gl.RGBmode()
gl.gconfig()
return
gl.cmode()
gl.gconfig()
sys.stderr.write('Initializing color map...')
initcmap(self.convcolor, self.c0bits, self.c1bits, \
self.c2bits, self.chrompack, self.offset)
sys.stderr.write(' Done.\n')
if self.offset == 0:
gl.color(0x800)
self.mask = 0x7ff
gl.color(0x800)
self.mask = 0x7ff
else:
self.mask = 0xfff
self.mask = 0xfff
gl.clear()
def initcmap(convcolor, c0bits, c1bits, c2bits, chrompack, offset):
if c0bits+c1bits+c2bits > 11:
raise Error, 'Sorry, 11 bits max'
import colorsys
maxc0 = 1 << c0bits
maxc1 = 1 << c1bits
maxc2 = 1 << c2bits
if offset = 0:
offset = 2048
rng = (offset, 4192-256)
for i in range(rng):
if offset == 0:
offset = 2048
for i in range(offset, MAXMAP):
gl.mapcolor(i, 0, 255, 0)
for c0 in range(maxc0):
c0v = c0/float(maxc0-1)
for c1 in range(maxc1):
if maxc1 = 1:
if maxc1 == 1:
c1v = 0
else:
c1v = c1/float(maxc1-1)
for c2 in range(maxc2):
if maxc2 = 1:
if maxc2 == 1:
c2v = 0
else:
c2v = c2/float(maxc2-1)
index = offset + c0 + \
(c1 << c0bits) + (c2 << (c0bits+c1bits))
(c1<<c0bits) + (c2 << (c0bits+c1bits))
rv, gv, bv = convcolor(c0v, c1v, c2v)
r, g, b = \
int(rv*255.0), int(gv*255.0), int(bv*255.0)
if index < 4096 - 256:
int(rv*255.0), int(gv*255.0), int(bv*255.0)
if index < MAXMAP:
gl.mapcolor(index, r, g, b)
def test():
import sys
import sys, time
filename = 'film.video'
if sys.argv[1:]: filename = sys.argv[1]
vin = VinFile().init(filename)
print 'File: ', filename
print 'Version: ', vin.version
print 'Size: ', vin.width, 'x', vin.height
print 'Pack: ', vin.packfactor, vin.chrompack
print 'Bits: ', vin.c0bits, vin.c1bits, vin.c2bits
print 'Format: ', vin.format
print 'Offset: ', vin.offset
print 'Size: ', vin.width, 'x', vin.height
print 'Pack: ', vin.packfactor, '; chrom:', vin.chrompack
print 'Bits: ', vin.c0bits, vin.c1bits, vin.c2bits
print 'Format: ', vin.format
print 'Offset: ', vin.offset
gl.foreground()
gl.prefsize(vin.width, vin.height)
wid = gl.winopen('VFile.test: ' + filename)
try:
while 1:
t = vin.shownextframe()
except EOFError:
pass
import time
time.sleep(5)
wid = gl.winopen(filename)
vin.initcolormap()
t0 = time.millitimer()
while 1:
try:
t, data, chromdata = vin.getnextframe()
except EOFError:
break
dt = t + t0 - time.millitimer()
if dt > 0:
time.millisleep(dt)
vin.showframe(data, chromdata)
print 'Done.'
time.sleep(2)