Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 1 | # VFile -- two classes for Video Files. |
| 2 | # VinFile -- for video input. |
| 3 | # VoutFile -- for video output. |
| 4 | |
| 5 | import sys |
| 6 | import gl |
| 7 | import GL |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 8 | import colorsys |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 9 | |
| 10 | Error = 'VFile.Error' # Exception |
| 11 | |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 12 | MAXMAP = 4096 - 256 |
| 13 | |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 14 | def conv_grey(l,x,y): return colorsys.yiq_to_rgb(l,0,0) |
| 15 | def conv_yiq (y,i,q): return colorsys.yiq_to_rgb(y, (i-0.5)*1.2, q-0.5) |
| 16 | def conv_hls (l,h,s): return colorsys.hls_to_rgb(h,l,s) |
| 17 | def conv_hsv (v,h,s): return colorsys.hsv_to_rgb(h,s,v) |
| 18 | def conv_rgb (r,g,b): |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 19 | raise Error, 'Attempt to make RGB colormap' |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 20 | |
| 21 | # Class VinFile represents a video file used for input. |
| 22 | # |
| 23 | # It has the following methods: |
| 24 | # init(filename) |
| 25 | # initfp(fp, filename) |
| 26 | # rewind() |
| 27 | # getframe() |
| 28 | # skipframe() |
| 29 | # |
| 30 | # The following read-only data members provide public information: |
| 31 | # version |
| 32 | # filename |
| 33 | # width, height |
| 34 | # packfactor |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 35 | # c0bits, c1bits, c2bits, chrompack |
| 36 | # offset |
| 37 | # format |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 38 | # |
| 39 | # These writable data members provide additional parametrization: |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 40 | # magnify |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 41 | # xorigin, yorigin |
| 42 | |
| 43 | class VinFile(): |
| 44 | |
| 45 | # init() and initfp() raise Error if the header is bad. |
| 46 | # init() raises whatever open() raises if the file can't be opened. |
| 47 | |
| 48 | def init(self, filename): |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 49 | if filename == '-': |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 50 | return self.initfp(sys.stdin, filename) |
| 51 | return self.initfp(open(filename, 'r'), filename) |
| 52 | |
| 53 | def initfp(self, (fp, filename)): |
| 54 | self.colormapinited = 0 |
| 55 | self.magnify = 1.0 |
| 56 | self.xorigin = self.yorigin = 0 |
| 57 | self.fp = fp |
| 58 | self.filename = filename |
| 59 | # |
| 60 | line = self.fp.readline() |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 61 | if line == 'CMIF video 1.0\n': |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 62 | self.version = 1.0 |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 63 | elif line == 'CMIF video 2.0\n': |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 64 | self.version = 2.0 |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 65 | elif line == 'CMIF video 3.0\n': |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 66 | self.version = 3.0 |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 67 | else: |
| 68 | raise Error, self.filename + ': bad video format' |
| 69 | # |
| 70 | if self.version < 2.0: |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 71 | self.c0bits, self.c1bits, self.c2bits = 8, 0, 0 |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 72 | self.chrompack = 0 |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 73 | self.offset = 0 |
| 74 | self.format = 'grey' |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 75 | elif self.version == 2.0: |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 76 | line = self.fp.readline() |
| 77 | try: |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 78 | self.c0bits, self.c1bits, self.c2bits, \ |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 79 | self.chrompack = eval(line[:-1]) |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 80 | if self.c1bits or self.c2bits: |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 81 | self.format = 'yiq' |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 82 | else: |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 83 | self.format = 'grey' |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 84 | self.offset = 0 |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 85 | except: |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 86 | raise Error, \ |
| 87 | self.filename + ': bad 2.0 color info' |
| 88 | elif self.version == 3.0: |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 89 | line = self.fp.readline() |
| 90 | try: |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 91 | self.format, rest = eval(line[:-1]) |
| 92 | if self.format == 'rgb': |
| 93 | pass |
| 94 | elif self.format == 'grey': |
| 95 | self.offset = 0 |
| 96 | self.c0bits = rest |
| 97 | self.c1bits = self.c2bits = \ |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 98 | self.chrompack = 0 |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 99 | else: |
| 100 | self.c0bits,self.c1bits,self.c2bits,\ |
| 101 | self.chrompack,self.offset = rest |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 102 | except: |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 103 | raise Error, \ |
| 104 | self.filename + ': bad 3.0 color info' |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 105 | |
| 106 | try: |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 107 | self.convcolor = eval('conv_'+self.format) |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 108 | except: |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 109 | raise Error, \ |
| 110 | self.filename + ': unknown colorsys ' + self.format |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 111 | # |
| 112 | line = self.fp.readline() |
| 113 | try: |
| 114 | x = eval(line[:-1]) |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 115 | if self.version > 1.0 or len(x) == 3: |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 116 | self.width, self.height, self.packfactor = x |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 117 | if self.packfactor == 0: |
| 118 | self.format = 'rgb' |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 119 | else: |
| 120 | sef.width, self.height = x |
| 121 | self.packfactor = 2 |
| 122 | except: |
| 123 | raise Error, self.filename + ': bad (w,h,pf) info' |
| 124 | # |
| 125 | return self |
| 126 | |
| 127 | # rewind() raises Error if the header is bad (which can only |
| 128 | # happen if the file was written to since opened). |
| 129 | |
| 130 | def rewind(self): |
| 131 | self.fp.seek(0) |
| 132 | x = self.initfp(self.fp, self.filename) |
| 133 | |
| 134 | # getnextframe() raises EOFError (built-in) if there is no next frame, |
| 135 | # or if the next frame is broken. |
| 136 | # So to getnextframeheader(), getnextframedata() and skipnextframe(). |
| 137 | |
| 138 | def getnextframe(self): |
| 139 | time, size, chromsize = self.getnextframeheader() |
| 140 | data, chromdata = self.getnextframedata(size, chromsize) |
| 141 | return time, data, chromdata |
| 142 | |
| 143 | def getnextframedata(self, (size, chromsize)): |
| 144 | data = self.fp.read(size) |
| 145 | if len(data) <> size: raise EOFError |
| 146 | if chromsize: |
| 147 | chromdata = self.fp.read(chromsize) |
| 148 | if len(chromdata) <> chromsize: raise EOFError |
| 149 | else: |
| 150 | chromdata = None |
| 151 | # |
| 152 | return data, chromdata |
| 153 | |
| 154 | def skipnextframe(self): |
| 155 | time, size, chromsize = self.getnextframeheader() |
| 156 | self.skipnextframedata(size, chromsize) |
| 157 | return time |
| 158 | |
| 159 | def skipnextframedata(self, (size, chromsize)): |
| 160 | # Note that this won't raise EOFError for a partial frame. |
| 161 | try: |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 162 | self.fp.seek(size + chromsize, 1) # Relative seek |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 163 | except: |
| 164 | # Assume it's a pipe -- read the data to discard it |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 165 | dummy = self.fp.read(size + chromsize) |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 166 | |
| 167 | def getnextframeheader(self): |
| 168 | line = self.fp.readline() |
| 169 | if not line: |
| 170 | raise EOFError |
| 171 | # |
| 172 | w, h, pf = self.width, self.height, self.packfactor |
| 173 | try: |
| 174 | x = eval(line[:-1]) |
| 175 | if type(x) in (type(0), type(0.0)): |
| 176 | time = x |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 177 | if pf == 0: |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 178 | size = w * h * 4 |
| 179 | else: |
| 180 | size = (w/pf) * (h/pf) |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 181 | elif len(x) == 2: |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 182 | time, size = x |
| 183 | cp = self.chrompack |
| 184 | if cp: |
| 185 | cw = (w + cp - 1) / cp |
| 186 | ch = (h + cp - 1) / cp |
| 187 | chromsize = 2 * cw * ch |
| 188 | else: |
| 189 | chromsize = 0 |
| 190 | else: |
| 191 | time, size, chromsize = x |
| 192 | except: |
| 193 | raise Error, self.filename + ': bad frame header' |
| 194 | return time, size, chromsize |
| 195 | |
| 196 | def shownextframe(self): |
| 197 | time, data, chromdata = self.getnextframe() |
| 198 | self.showframe(data, chromdata) |
| 199 | return time |
| 200 | |
| 201 | def showframe(self, (data, chromdata)): |
| 202 | w, h, pf = self.width, self.height, self.packfactor |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 203 | if not self.colormapinited: |
| 204 | self.initcolormap() |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 205 | factor = self.magnify |
| 206 | if pf: factor = factor * pf |
| 207 | if chromdata: |
| 208 | cp = self.chrompack |
| 209 | cw = (w+cp-1)/cp |
| 210 | ch = (h+cp-1)/cp |
| 211 | gl.rectzoom(factor*cp, factor*cp) |
| 212 | gl.pixmode(GL.PM_SIZE, 16) |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 213 | gl.writemask(self.mask - ((1 << self.c0bits) - 1)) |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 214 | gl.lrectwrite(self.xorigin, self.yorigin, \ |
| 215 | self.xorigin + cw - 1, self.yorigin + ch - 1, \ |
| 216 | chromdata) |
| 217 | # |
| 218 | if pf: |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 219 | gl.writemask((1 << self.c0bits) - 1) |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 220 | gl.pixmode(GL.PM_SIZE, 8) |
| 221 | gl.rectzoom(factor, factor) |
| 222 | w = w/pf |
| 223 | h = h/pf |
| 224 | gl.lrectwrite(self.xorigin, self.yorigin, \ |
| 225 | self.xorigin + w - 1, self.yorigin + h - 1, data) |
| 226 | |
| 227 | def initcolormap(self): |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 228 | self.colormapinited = 1 |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 229 | if self.format == 'rgb': |
| 230 | gl.RGBmode() |
| 231 | gl.gconfig() |
| 232 | return |
| 233 | gl.cmode() |
| 234 | gl.gconfig() |
| 235 | sys.stderr.write('Initializing color map...') |
| 236 | initcmap(self.convcolor, self.c0bits, self.c1bits, \ |
| 237 | self.c2bits, self.chrompack, self.offset) |
| 238 | sys.stderr.write(' Done.\n') |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 239 | if self.offset == 0: |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 240 | gl.color(0x800) |
| 241 | self.mask = 0x7ff |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 242 | else: |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 243 | self.mask = 0xfff |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 244 | gl.clear() |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 245 | |
| 246 | |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 247 | def initcmap(convcolor, c0bits, c1bits, c2bits, chrompack, offset): |
| 248 | if c0bits+c1bits+c2bits > 11: |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 249 | raise Error, 'Sorry, 11 bits max' |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 250 | maxc0 = 1 << c0bits |
| 251 | maxc1 = 1 << c1bits |
| 252 | maxc2 = 1 << c2bits |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 253 | if offset == 0: |
| 254 | offset = 2048 |
| 255 | for i in range(offset, MAXMAP): |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 256 | gl.mapcolor(i, 0, 255, 0) |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 257 | for c0 in range(maxc0): |
| 258 | c0v = c0/float(maxc0-1) |
| 259 | for c1 in range(maxc1): |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 260 | if maxc1 == 1: |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 261 | c1v = 0 |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 262 | else: |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 263 | c1v = c1/float(maxc1-1) |
| 264 | for c2 in range(maxc2): |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 265 | if maxc2 == 1: |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 266 | c2v = 0 |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 267 | else: |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 268 | c2v = c2/float(maxc2-1) |
| 269 | index = offset + c0 + \ |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 270 | (c1<<c0bits) + (c2 << (c0bits+c1bits)) |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 271 | rv, gv, bv = convcolor(c0v, c1v, c2v) |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 272 | r, g, b = \ |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 273 | int(rv*255.0), int(gv*255.0), int(bv*255.0) |
| 274 | if index < MAXMAP: |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 275 | gl.mapcolor(index, r, g, b) |
| 276 | |
| 277 | |
| 278 | def test(): |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 279 | import sys, time |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 280 | filename = 'film.video' |
| 281 | if sys.argv[1:]: filename = sys.argv[1] |
| 282 | vin = VinFile().init(filename) |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 283 | print 'File: ', filename |
Guido van Rossum | 444339d | 1991-12-03 16:52:40 +0000 | [diff] [blame] | 284 | print 'Version: ', vin.version |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 285 | print 'Size: ', vin.width, 'x', vin.height |
| 286 | print 'Pack: ', vin.packfactor, '; chrom:', vin.chrompack |
| 287 | print 'Bits: ', vin.c0bits, vin.c1bits, vin.c2bits |
| 288 | print 'Format: ', vin.format |
| 289 | print 'Offset: ', vin.offset |
Guido van Rossum | 094a9de | 1991-12-03 16:50:00 +0000 | [diff] [blame] | 290 | gl.foreground() |
| 291 | gl.prefsize(vin.width, vin.height) |
Guido van Rossum | 58b38cc | 1992-02-11 14:45:43 +0000 | [diff] [blame^] | 292 | wid = gl.winopen(filename) |
| 293 | vin.initcolormap() |
| 294 | t0 = time.millitimer() |
| 295 | while 1: |
| 296 | try: |
| 297 | t, data, chromdata = vin.getnextframe() |
| 298 | except EOFError: |
| 299 | break |
| 300 | dt = t + t0 - time.millitimer() |
| 301 | if dt > 0: |
| 302 | time.millisleep(dt) |
| 303 | vin.showframe(data, chromdata) |
| 304 | print 'Done.' |
| 305 | time.sleep(2) |