Guido van Rossum | f06ee5f | 1996-11-27 19:52:01 +0000 | [diff] [blame] | 1 | #! /usr/bin/env python |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 2 | |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 3 | # Video bag of tricks: record video(+audio) in various formats and modes |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 4 | |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 5 | # XXX To do: |
| 6 | # - audio |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 7 | # - improve user interface |
| 8 | # - help button? |
| 9 | # - command line options to set initial settings |
| 10 | # - save settings in a file |
| 11 | # - ...? |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 12 | |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 13 | import sys |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 14 | import time |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 15 | import getopt |
| 16 | import string |
| 17 | import os |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 18 | sts = os.system('makemap') # Must be before "import fl" to work |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 19 | import sgi |
| 20 | import gl |
| 21 | import GL |
| 22 | import DEVICE |
| 23 | import fl |
| 24 | import FL |
| 25 | import flp |
| 26 | import watchcursor |
| 27 | import sv |
| 28 | import SV |
| 29 | import VFile |
| 30 | import VGrabber |
| 31 | import imageop |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 32 | sys.path.append('/ufs/jack/src/av/vcr') |
Guido van Rossum | f6d8032 | 1993-06-10 13:40:51 +0000 | [diff] [blame] | 33 | import VCR |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 34 | try: |
| 35 | import cl |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 36 | except ImportError: |
| 37 | cl = None |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 38 | |
| 39 | ARROW = 0 |
| 40 | WATCH = 1 |
| 41 | watchcursor.defwatch(WATCH) |
| 42 | |
| 43 | def main(): |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 44 | ## fl.set_graphics_mode(0, 1) |
Guido van Rossum | 21a3ff9 | 1993-12-17 15:11:41 +0000 | [diff] [blame] | 45 | vb = VideoBagOfTricks() |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 46 | while 1: |
| 47 | dummy = fl.do_forms() |
| 48 | [dummy] |
| 49 | |
| 50 | StopCapture = 'StopCapture' |
| 51 | |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 52 | VideoFormatLabels = ['Video off', 'rgb8', 'grey8', 'grey4', 'grey2', \ |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 53 | 'grey2 dith', 'mono dith', 'mono thresh', 'rgb24', 'rgb24-jpeg', \ |
| 54 | 'compress'] |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 55 | VideoFormats = ['', 'rgb8', 'grey', 'grey4', 'grey2', \ |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 56 | 'grey2', 'mono', 'mono', 'rgb', 'jpeg', 'compress'] |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 57 | |
| 58 | VideoModeLabels = ['Continuous', 'Burst', 'Single frame', 'VCR sync'] |
| 59 | [VM_CONT, VM_BURST, VM_SINGLE, VM_VCR] = range(1, 5) |
| 60 | |
| 61 | AudioFormatLabels = ['Audio off', \ |
| 62 | '16 bit mono', '16 bit stereo', '8 bit mono', '8 bit stereo'] |
| 63 | [A_OFF, A_16_MONO, A_16_STEREO, A_8_MONO, A_8_STEREO] = range(1, 6) |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 64 | |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 65 | VcrSpeedLabels = ['normal', '1/3', '1/5', '1/10', '1/30', 'single-step'] |
| 66 | VcrSpeeds = [None, 5, 4, 3, 2, 1, 0] |
| 67 | |
| 68 | RgbSizeLabels = ['full', 'quarter', 'sixteenth'] |
| 69 | |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 70 | # init file stuff: |
| 71 | if os.environ.has_key('HOME'): |
| 72 | HOME=os.environ['HOME'] |
| 73 | else: |
| 74 | HOME='.' |
| 75 | VB_INIT_FILE=HOME + '/.Vb_init' |
| 76 | |
| 77 | VB_INIT_KEYS=['vfile', 'vmode', 'mono_thresh', 'vformat', 'comp_scheme', \ |
| 78 | 'rgb24_size', 'afile', 'aformat'] |
| 79 | |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 80 | class VideoBagOfTricks: |
| 81 | |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 82 | # Init/close stuff |
| 83 | |
Guido van Rossum | 21a3ff9 | 1993-12-17 15:11:41 +0000 | [diff] [blame] | 84 | def __init__(self): |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 85 | self.window = None |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 86 | formdef = flp.parse_form('VbForm', 'form') |
| 87 | flp.create_full_form(self, formdef) |
| 88 | self.setdefaults() |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 89 | if self.vmode <> VM_CONT: |
| 90 | self.g_cont.hide_object() |
| 91 | if self.vmode <> VM_BURST: |
| 92 | self.g_burst.hide_object() |
| 93 | if self.vmode <> VM_SINGLE: |
| 94 | self.g_single.hide_object() |
| 95 | if self.vmode <> VM_VCR: |
| 96 | self.g_vcr.hide_object() |
| 97 | if self.vformat <> 'compress': |
| 98 | self.g_compress.hide_object() |
| 99 | |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 100 | self.openvideo() |
| 101 | self.makewindow() |
| 102 | self.bindvideo() |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 103 | if self.use_24: |
| 104 | self.optfullsizewindow() |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 105 | self.showform() |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 106 | fl.set_event_call_back(self.do_event) |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 107 | |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 108 | def close(self): |
| 109 | self.close_video() |
| 110 | self.close_audio() |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 111 | self.savedefaults() |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 112 | raise SystemExit, 0 |
| 113 | |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 114 | def showform(self): |
| 115 | # Get position of video window |
| 116 | gl.winset(self.window) |
| 117 | x, y = gl.getorigin() |
| 118 | width, height = gl.getsize() |
| 119 | # Calculate position of form window |
| 120 | x1 = x + width + 10 |
| 121 | x2 = x1 + int(self.form.w) - 1 |
| 122 | y2 = y + height - 1 |
| 123 | y1 = y2 - int(self.form.h) + 1 |
| 124 | # Position and show form window |
| 125 | gl.prefposition(x1, x2, y1, y2) |
Guido van Rossum | 43df862 | 1993-06-11 15:48:39 +0000 | [diff] [blame] | 126 | self.form.show_form(FL.PLACE_FREE, FL.TRUE, 'Vb Control') |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 127 | |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 128 | def getdefaultdefaults(self): |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 129 | # Video defaults |
| 130 | self.vfile = 'film.video' |
| 131 | self.vmode = VM_CONT |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 132 | self.mono_thresh = 128 |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 133 | self.vformat = 'rgb8' |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 134 | self.comp_scheme = 'Uncompressed' |
| 135 | self.rgb24_size = 1 |
| 136 | # Missing: drop, rate, maxmem, nframes, rate, vcrspeed |
| 137 | # Audio defaults: |
| 138 | self.afile = 'film.aiff' |
| 139 | self.aformat = A_OFF |
| 140 | |
| 141 | def getdefaults(self): |
| 142 | self.getdefaultdefaults() |
| 143 | # XXXX Read defaults file and override. |
| 144 | try: |
| 145 | fp = open(VB_INIT_FILE, 'r') |
| 146 | except IOError: |
| 147 | print 'Vb: no init file' |
| 148 | self.initcont = {} |
| 149 | return |
| 150 | data = fp.read(1000000) |
| 151 | try: |
| 152 | self.initcont = eval(data) |
| 153 | except: |
| 154 | print 'Vb: Ill-formatted init file' |
| 155 | self.initcont = {} |
| 156 | for k in self.initcont.keys(): |
| 157 | if hasattr(self, k): |
| 158 | setattr(self, k, self.initcont[k]) |
| 159 | |
| 160 | def savedefaults(self): |
| 161 | newdb = {} |
| 162 | for k in VB_INIT_KEYS: |
| 163 | newdb[k] = getattr(self, k) |
| 164 | if newdb <> self.initcont: |
| 165 | try: |
| 166 | fp = open(VB_INIT_FILE, 'w') |
| 167 | except IOError: |
| 168 | print 'Vb: Cannot create', VB_INIT_FILE |
| 169 | return |
| 170 | fp.write(`newdb`) |
| 171 | fp.close() |
| 172 | |
| 173 | def setdefaults(self): |
| 174 | self.getdefaults() |
| 175 | self.vcr = None |
| 176 | self.vout = None |
| 177 | self.capturing = 0 |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 178 | self.c_vformat.clear_choice() |
| 179 | for label in VideoFormatLabels: |
| 180 | self.c_vformat.addto_choice(label) |
| 181 | self.c_vformat.set_choice(1 + VideoFormats.index(self.vformat)) |
| 182 | self.c_vmode.clear_choice() |
| 183 | for label in VideoModeLabels: |
| 184 | self.c_vmode.addto_choice(label) |
| 185 | self.c_vmode.set_choice(self.vmode) |
| 186 | self.get_vformat() |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 187 | self.b_drop.set_button(1) |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 188 | self.in_rate.set_input('2') |
| 189 | self.in_maxmem.set_input('1.0') |
| 190 | self.in_nframes.set_input('0') |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 191 | self.in_nframes_vcr.set_input('1') |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 192 | self.in_rate_vcr.set_input('1') |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 193 | self.c_vcrspeed.clear_choice() |
| 194 | for label in VcrSpeedLabels: |
| 195 | self.c_vcrspeed.addto_choice(label) |
| 196 | self.c_vcrspeed.set_choice(4) |
| 197 | self.c_rgb24_size.clear_choice() |
| 198 | for label in RgbSizeLabels: |
| 199 | self.c_rgb24_size.addto_choice(label) |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 200 | self.c_rgb24_size.set_choice(self.rgb24_size) |
| 201 | if cl: |
Guido van Rossum | 1554b7b | 1998-10-09 13:26:29 +0000 | [diff] [blame] | 202 | algs = cl.QueryAlgorithms(cl.VIDEO) |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 203 | self.all_comp_schemes = [] |
| 204 | for i in range(0, len(algs), 2): |
Guido van Rossum | 1554b7b | 1998-10-09 13:26:29 +0000 | [diff] [blame] | 205 | if algs[i+1] in (cl.COMPRESSOR, cl.CODEC): |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 206 | self.all_comp_schemes.append(algs[i]) |
| 207 | self.c_cformat.clear_choice() |
| 208 | for label in self.all_comp_schemes: |
| 209 | self.c_cformat.addto_choice(label) |
| 210 | i = self.all_comp_schemes.index(self.comp_scheme) |
| 211 | self.c_cformat.set_choice(i+1) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 212 | # Audio defaults |
| 213 | self.aout = None |
| 214 | self.aport = None |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 215 | self.c_aformat.clear_choice() |
| 216 | for label in AudioFormatLabels: |
| 217 | self.c_aformat.addto_choice(label) |
| 218 | self.c_aformat.set_choice(self.aformat) |
| 219 | self.get_aformat() |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 220 | |
| 221 | def openvideo(self): |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 222 | try: |
| 223 | self.video = sv.OpenVideo() |
| 224 | except sv.error, msg: |
| 225 | print 'Error opening video:', msg |
| 226 | self.video = None |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 227 | param = [SV.BROADCAST, SV.PAL] |
| 228 | if self.video: self.video.GetParam(param) |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 229 | if param[1] == SV.PAL: |
| 230 | x = SV.PAL_XMAX |
| 231 | y = SV.PAL_YMAX |
| 232 | elif param[1] == SV.NTSC: |
| 233 | x = SV.NTSC_XMAX |
| 234 | y = SV.NTSC_YMAX |
| 235 | else: |
| 236 | print 'Unknown video standard:', param[1] |
| 237 | sys.exit(1) |
| 238 | self.maxx, self.maxy = x, y |
Guido van Rossum | 43df862 | 1993-06-11 15:48:39 +0000 | [diff] [blame] | 239 | self.curx = 256 |
| 240 | self.cury = 256*3/4 |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 241 | |
| 242 | def makewindow(self): |
| 243 | x, y = self.maxx, self.maxy |
| 244 | gl.foreground() |
| 245 | gl.maxsize(x, y) |
| 246 | gl.keepaspect(x, y) |
| 247 | gl.stepunit(8, 6) |
Guido van Rossum | 43df862 | 1993-06-11 15:48:39 +0000 | [diff] [blame] | 248 | width = self.curx |
| 249 | height = self.cury |
| 250 | if width and height: |
| 251 | # Place the window at (150, 150) from top left |
| 252 | # (the video board likes this location...) |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 253 | x1 = 150 |
Guido van Rossum | 43df862 | 1993-06-11 15:48:39 +0000 | [diff] [blame] | 254 | x2 = x1+width-1 |
| 255 | SCRHEIGHT = 768 |
| 256 | y2 = SCRHEIGHT-1-150 |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 257 | y1 = y2-height+1 |
| 258 | gl.prefposition(x1, x2, y1, y2) |
Guido van Rossum | 43df862 | 1993-06-11 15:48:39 +0000 | [diff] [blame] | 259 | self.window = gl.winopen('Vb video') |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 260 | self.settitle() |
| 261 | if width: |
| 262 | gl.maxsize(x, y) |
| 263 | gl.keepaspect(x, y) |
| 264 | gl.stepunit(8, 6) |
| 265 | gl.winconstraints() |
| 266 | gl.qdevice(DEVICE.LEFTMOUSE) |
| 267 | gl.qdevice(DEVICE.WINQUIT) |
| 268 | gl.qdevice(DEVICE.WINSHUT) |
| 269 | |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 270 | def optfullsizewindow(self): |
| 271 | if not self.window: |
| 272 | return |
| 273 | gl.winset(self.window) |
Guido van Rossum | 43df862 | 1993-06-11 15:48:39 +0000 | [diff] [blame] | 274 | if self.use_24: |
| 275 | x, y = self.maxx, self.maxy |
| 276 | else: |
| 277 | x, y = self.curx, self.cury |
Guido van Rossum | 2055ee8 | 1993-06-11 14:13:13 +0000 | [diff] [blame] | 278 | left, bottom = gl.getorigin() |
| 279 | width, height = gl.getsize() |
| 280 | bottom = bottom+height-y |
| 281 | gl.prefposition(left, left+x-1, bottom, bottom+y-1) |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 282 | gl.winconstraints() |
Guido van Rossum | 43df862 | 1993-06-11 15:48:39 +0000 | [diff] [blame] | 283 | if not self.use_24: |
| 284 | gl.keepaspect(x, y) |
| 285 | gl.stepunit(8, 6) |
| 286 | gl.maxsize(self.maxx, self.maxy) |
| 287 | gl.winconstraints() |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 288 | self.bindvideo() |
| 289 | |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 290 | def bindvideo(self): |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 291 | if not self.video: return |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 292 | x, y = gl.getsize() |
Guido van Rossum | 43df862 | 1993-06-11 15:48:39 +0000 | [diff] [blame] | 293 | if not self.use_24: |
| 294 | self.curx, self.cury = x, y |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 295 | self.video.SetSize(x, y) |
| 296 | drop = self.b_drop.get_button() |
| 297 | if drop: |
| 298 | param = [SV.FIELDDROP, 1, SV.GENLOCK, SV.GENLOCK_OFF] |
| 299 | else: |
| 300 | param = [SV.FIELDDROP, 0, SV.GENLOCK, SV.GENLOCK_ON] |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 301 | if self.rgb: |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 302 | param = param+[SV.COLOR, SV.DEFAULT_COLOR, \ |
| 303 | SV.DITHER, 1, \ |
| 304 | SV.INPUT_BYPASS, 0] |
| 305 | else: |
| 306 | param = param+[SV.COLOR, SV.MONO, SV.DITHER, 0, \ |
| 307 | SV.INPUT_BYPASS, 1] |
| 308 | self.video.BindGLWindow(self.window, SV.IN_REPLACE) |
| 309 | self.video.SetParam(param) |
| 310 | |
| 311 | def rebindvideo(self): |
| 312 | gl.winset(self.window) |
| 313 | self.bindvideo() |
| 314 | |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 315 | def reset(self): |
| 316 | self.close_video() |
| 317 | self.close_audio() |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 318 | if self.vcr: |
| 319 | try: |
| 320 | ok = self.vcr.still() |
| 321 | except VCR.error: |
| 322 | pass |
| 323 | self.vcr = None |
| 324 | self.b_capture.set_button(0) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 325 | |
| 326 | # Event handler (catches resize of video window) |
| 327 | |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 328 | def do_event(self, dev, val): |
| 329 | #print 'Event:', dev, val |
| 330 | if dev in (DEVICE.WINSHUT, DEVICE.WINQUIT): |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 331 | self.close() |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 332 | if dev == DEVICE.REDRAW and val == self.window: |
| 333 | self.rebindvideo() |
| 334 | self.settitle() |
| 335 | |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 336 | # Video controls: format, mode, file |
| 337 | |
| 338 | def cb_vformat(self, *args): |
| 339 | self.reset() |
| 340 | self.get_vformat() |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 341 | if self.mono_use_thresh: |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 342 | s = `self.mono_thresh` |
| 343 | s = fl.show_input('Please enter mono threshold', s) |
| 344 | if s: |
| 345 | try: |
| 346 | self.mono_thresh = string.atoi(s) |
| 347 | except string.atoi_error: |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 348 | fl.show_message('Bad input, using', \ |
| 349 | `self.mono_thresh`, '') |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 350 | self.rebindvideo() |
| 351 | |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 352 | def cb_cformat(self, *args): |
| 353 | i = self.c_cformat.get_choice() |
| 354 | self.comp_scheme = self.all_comp_schemes[i-1] |
| 355 | |
| 356 | |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 357 | def cb_vmode(self, *args): |
Guido van Rossum | c17c84f | 1993-05-10 15:45:49 +0000 | [diff] [blame] | 358 | if self.vcr: |
| 359 | self.vcr = None |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 360 | self.vmode = self.c_vmode.get_choice() |
| 361 | self.form.freeze_form() |
| 362 | self.g_cont.hide_object() |
| 363 | self.g_burst.hide_object() |
| 364 | self.g_single.hide_object() |
| 365 | self.g_vcr.hide_object() |
| 366 | if self.vmode == VM_CONT: |
| 367 | self.g_cont.show_object() |
| 368 | elif self.vmode == VM_BURST: |
| 369 | self.g_burst.show_object() |
| 370 | elif self.vmode == VM_SINGLE: |
| 371 | self.g_single.show_object() |
| 372 | elif self.vmode == VM_VCR: |
| 373 | self.g_vcr.show_object() |
| 374 | self.form.unfreeze_form() |
| 375 | |
| 376 | def cb_vfile(self, *args): |
| 377 | filename = self.vfile |
| 378 | hd, tl = os.path.split(filename) |
| 379 | filename = fl.file_selector('Video save file:', hd, '', tl) |
| 380 | if filename: |
| 381 | self.reset() |
| 382 | hd, tl = os.path.split(filename) |
| 383 | if hd == os.getcwd(): |
| 384 | filename = tl |
| 385 | self.vfile = filename |
| 386 | |
| 387 | # Video mode specific video controls |
| 388 | |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 389 | def cb_rate(self, *args): |
| 390 | pass |
| 391 | |
| 392 | def cb_drop(self, *args): |
| 393 | self.rebindvideo() |
| 394 | |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 395 | def cb_maxmem(self, *args): |
| 396 | pass |
| 397 | |
| 398 | def cb_nframes(self, *args): |
| 399 | pass |
| 400 | |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 401 | def cb_fps(self, *args): |
| 402 | pass |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 403 | |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 404 | def cb_nframes_vcr(self, *args): |
| 405 | pass |
| 406 | |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 407 | def cb_rate_vcr(self, *args): |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 408 | pass |
| 409 | |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 410 | def cb_vcrspeed(self, *args): |
| 411 | pass |
| 412 | |
| 413 | def cb_rgb24_size(self, *args): |
| 414 | i = self.c_rgb24_size.get_choice() |
| 415 | if i: |
| 416 | self.rgb24_size = i |
| 417 | |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 418 | # Audio controls: format, file |
| 419 | |
| 420 | def cb_aformat(self, *args): |
| 421 | self.get_aformat() |
| 422 | |
| 423 | def cb_afile(self, *args): |
| 424 | filename = self.afile |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 425 | hd, tl = os.path.split(filename) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 426 | filename = fl.file_selector('Audio save file:', hd, '', tl) |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 427 | if filename: |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 428 | self.reset() |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 429 | hd, tl = os.path.split(filename) |
| 430 | if hd == os.getcwd(): |
| 431 | filename = tl |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 432 | self.afile = filename |
| 433 | |
| 434 | # General controls: capture, reset, play, quit |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 435 | |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 436 | def cb_capture(self, *args): |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 437 | if self.capturing: |
| 438 | raise StopCapture |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 439 | if not self.b_capture.get_button(): |
| 440 | return |
| 441 | if not self.video or not self.vformat: |
| 442 | gl.ringbell() |
| 443 | return |
| 444 | if self.vmode == VM_CONT: |
| 445 | self.cont_capture() |
| 446 | elif self.vmode == VM_BURST: |
| 447 | self.burst_capture() |
| 448 | elif self.vmode == VM_SINGLE: |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 449 | self.single_capture(None, None) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 450 | elif self.vmode == VM_VCR: |
| 451 | self.vcr_capture() |
| 452 | |
| 453 | def cb_reset(self, *args): |
| 454 | self.reset() |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 455 | |
| 456 | def cb_play(self, *args): |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 457 | self.reset() |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 458 | sts = os.system('Vplay -q ' + self.vfile + ' &') |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 459 | |
| 460 | def cb_quit(self, *args): |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 461 | self.close() |
| 462 | |
| 463 | # Capture routines |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 464 | |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 465 | def burst_capture(self): |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 466 | self.setwatch() |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 467 | gl.winset(self.window) |
| 468 | x, y = gl.getsize() |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 469 | if self.use_24: |
| 470 | fl.show_message('Sorry, no 24 bit continuous capture yet', '', '') |
| 471 | return |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 472 | vformat = SV.RGB8_FRAMES |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 473 | nframes = self.getint(self.in_nframes, 0) |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 474 | if nframes == 0: |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 475 | maxmem = self.getint(self.in_maxmem, 1.0) |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 476 | memsize = int(maxmem * 1024 * 1024) |
Guido van Rossum | 5461850 | 1993-10-26 10:23:14 +0000 | [diff] [blame] | 477 | nframes = self.calcnframes(memsize) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 478 | info = (vformat, x, y, nframes, 1) |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 479 | try: |
| 480 | info2, data, bitvec = self.video.CaptureBurst(info) |
| 481 | except sv.error, msg: |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 482 | self.b_capture.set_button(0) |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 483 | self.setarrow() |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 484 | fl.show_message('Capture error:', str(msg), '') |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 485 | return |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 486 | if info <> info2: print info, '<>', info2 |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 487 | self.save_burst(info2, data, bitvec) |
| 488 | self.setarrow() |
| 489 | |
Guido van Rossum | 5461850 | 1993-10-26 10:23:14 +0000 | [diff] [blame] | 490 | def calcnframes(self, memsize): |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 491 | gl.winset(self.window) |
| 492 | x, y = gl.getsize() |
| 493 | pixels = x*y |
| 494 | pixels = pixels/2 # XXX always assume fields |
| 495 | if self.mono or self.grey: |
| 496 | n = memsize/pixels |
| 497 | else: |
| 498 | n = memsize/(4*pixels) |
| 499 | return max(1, n) |
| 500 | |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 501 | def save_burst(self, info, data, bitvec): |
| 502 | (vformat, x, y, nframes, rate) = info |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 503 | self.open_if_closed() |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 504 | fieldsize = x*y/2 |
| 505 | nskipped = 0 |
| 506 | realframeno = 0 |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 507 | tpf = 1000 / 50.0 # XXX |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 508 | for frameno in range(0, nframes*2): |
| 509 | if frameno <> 0 and \ |
| 510 | bitvec[frameno] == bitvec[frameno-1]: |
| 511 | nskipped = nskipped + 1 |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 512 | continue |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 513 | # |
| 514 | # Save field. |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 515 | # XXX Works only for fields and top-to-bottom |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 516 | # |
| 517 | start = frameno*fieldsize |
| 518 | field = data[start:start+fieldsize] |
| 519 | realframeno = realframeno + 1 |
| 520 | fn = int(realframeno*tpf) |
| 521 | if not self.write_frame(fn, field): |
| 522 | break |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 523 | |
| 524 | def cont_capture(self): |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 525 | saved_label = self.b_capture.label |
| 526 | self.b_capture.label = 'Stop\n' + saved_label |
| 527 | self.open_if_closed() |
| 528 | self.init_cont() |
| 529 | fps = 59.64 # Fields per second |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 530 | # XXX (fps of Indigo monitor, not of PAL or NTSC!) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 531 | tpf = 1000.0 / fps # Time per field in msec |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 532 | self.capturing = 1 |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 533 | self.start_audio() |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 534 | while 1: |
| 535 | try: |
| 536 | void = fl.check_forms() |
| 537 | except StopCapture: |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 538 | break |
| 539 | try: |
| 540 | cd, id = self.video.GetCaptureData() |
| 541 | except sv.error: |
| 542 | sgi.nap(1) |
| 543 | continue |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 544 | id = id + 2*self.rate |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 545 | data = cd.InterleaveFields(1) |
| 546 | cd.UnlockCaptureData() |
| 547 | t = id*tpf |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 548 | if not self.write_frame(t, data): |
Guido van Rossum | e17c6c3 | 1993-05-06 16:27:25 +0000 | [diff] [blame] | 549 | break |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 550 | self.stop_audio() |
Guido van Rossum | e17c6c3 | 1993-05-06 16:27:25 +0000 | [diff] [blame] | 551 | self.capturing = 0 |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 552 | self.end_cont() |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 553 | if self.aout: |
| 554 | # If recording audio, can't capture multiple sequences |
| 555 | self.reset() |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 556 | self.b_capture.label = saved_label |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 557 | |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 558 | def single_capture(self, stepfunc, timecode): |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 559 | self.open_if_closed() |
| 560 | self.init_cont() |
| 561 | while 1: |
| 562 | try: |
| 563 | cd, id = self.video.GetCaptureData() |
| 564 | break |
| 565 | except sv.error: |
| 566 | pass |
| 567 | sgi.nap(1) |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 568 | if stepfunc: # This might step the video |
| 569 | d=stepfunc() # to the next frame |
| 570 | if not self.use_24: |
| 571 | data = cd.InterleaveFields(1) |
| 572 | else: |
| 573 | x, y = self.vout.getsize() |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 574 | if self.use_compress: |
| 575 | if self.rgb24_size == 1: |
| 576 | data = cd.YUVtoYUV422DC(0) |
| 577 | elif self.rgb24_size == 2: |
| 578 | data = cd.YUVtoYUV422DC_quarter(1) |
| 579 | x = x/2 |
| 580 | y = y/2 |
| 581 | elif self.rgb24_size == 3: |
| 582 | data = cd.YUVtoYUV422DC_sixteenth(1) |
| 583 | x = x/4 |
| 584 | y = y/4 |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 585 | else: |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 586 | data = cd.YUVtoRGB(1) |
| 587 | if self.maxx*self.maxy*4 <> len(data): |
| 588 | print 'maxx,maxy,exp,got=', self.maxx, |
| 589 | print self.maxy,self.maxx*self.maxy*4, |
| 590 | print len(data) |
| 591 | fl.showmessage('Wrong sized data') |
| 592 | return 0 |
| 593 | if self.rgb24_size <> 1: |
| 594 | data = imageop.scale(data, 4, \ |
| 595 | self.maxx, self.maxy, x, y) |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 596 | if self.use_jpeg: |
| 597 | import jpeg |
| 598 | data = jpeg.compress(data, x, y, 4) |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 599 | if self.use_compress: |
| 600 | data = self.compressor.Compress(1, data) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 601 | cd.UnlockCaptureData() |
| 602 | self.end_cont() |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 603 | if timecode == None: |
| 604 | timecode = (self.nframes+1) * (1000/25) |
| 605 | return self.write_frame(timecode, data) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 606 | |
| 607 | def vcr_capture(self): |
| 608 | if not self.vcr: |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 609 | try: |
Guido van Rossum | 2055ee8 | 1993-06-11 14:13:13 +0000 | [diff] [blame] | 610 | print 'Connecting to VCR ...' |
Guido van Rossum | 21a3ff9 | 1993-12-17 15:11:41 +0000 | [diff] [blame] | 611 | self.vcr = VCR.VCR() |
Guido van Rossum | 2055ee8 | 1993-06-11 14:13:13 +0000 | [diff] [blame] | 612 | print 'Waiting for VCR to come online ...' |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 613 | self.vcr.initvcr() |
Guido van Rossum | 2055ee8 | 1993-06-11 14:13:13 +0000 | [diff] [blame] | 614 | print 'Preparing VCR ...' |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 615 | if not (self.vcr.fmmode('dnr') and \ |
| 616 | self.vcr.dmcontrol('digital slow')): |
| 617 | self.vcr_error('digital slow failed') |
| 618 | return |
Guido van Rossum | 2055ee8 | 1993-06-11 14:13:13 +0000 | [diff] [blame] | 619 | print 'VCR OK.' |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 620 | except VCR.error, msg: |
Guido van Rossum | c17c84f | 1993-05-10 15:45:49 +0000 | [diff] [blame] | 621 | self.vcr = None |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 622 | self.vcr_error(msg) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 623 | return |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 624 | if not self.vcr.still(): |
| 625 | self.vcr_error('still failed') |
| 626 | return |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 627 | self.open_if_closed() |
| 628 | rate = self.getint(self.in_rate_vcr, 1) |
| 629 | rate = max(rate, 1) |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 630 | vcrspeed = self.c_vcrspeed.get_choice() |
| 631 | vcrspeed = VcrSpeeds[vcrspeed] |
| 632 | if vcrspeed == 0: |
| 633 | stepfunc = self.vcr.step |
| 634 | else: |
| 635 | stepfunc = None |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 636 | self.speed_factor = rate |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 637 | addr = start_addr = self.vcr.sense() |
| 638 | if not self.single_capture(None, 0): |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 639 | return |
| 640 | print 'captured %02d:%02d:%02d:%02d' % self.vcr.addr2tc(addr) |
| 641 | count = self.getint(self.in_nframes_vcr, 1) - 1 |
| 642 | if count <= 0: |
| 643 | while rate > 0: |
| 644 | if not self.vcr.step(): |
| 645 | self.vcr_error('step failed') |
| 646 | here = self.vcr.sense() |
| 647 | if here > addr: |
| 648 | rate = rate - (here - addr) |
| 649 | addr = here |
| 650 | return |
Guido van Rossum | 2055ee8 | 1993-06-11 14:13:13 +0000 | [diff] [blame] | 651 | if not self.vcr.fwdshuttle(vcrspeed): |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 652 | self.vcr_error('fwd shuttle failed') |
| 653 | return |
| 654 | cycle = 0 |
| 655 | while count > 0: |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 656 | try: |
| 657 | here = self.vcr.sense() |
| 658 | except VCR.error, msg: |
| 659 | self.vcr_error(msg) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 660 | break |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 661 | if here <> addr: |
| 662 | if here <> addr+1: |
| 663 | print 'Missed', here-addr-1, |
| 664 | print 'frame' + 's'*(here-addr-1 <> 1) |
| 665 | cycle = (cycle+1) % rate |
| 666 | if cycle == 0: |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 667 | tc = (here-start_addr)*40 |
| 668 | if not self.single_capture(stepfunc, \ |
| 669 | tc): |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 670 | break |
| 671 | print 'captured %02d:%02d:%02d:%02d' \ |
| 672 | % self.vcr.addr2tc(here) |
| 673 | count = count -1 |
| 674 | addr = here |
| 675 | if self.vcr and not self.vcr.still(): |
| 676 | self.vcr_error('still failed') |
| 677 | |
| 678 | def vcr_error(self, msg): |
| 679 | self.reset() |
| 680 | fl.show_message('VCR error:', str(msg), '') |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 681 | |
| 682 | # Init/end continuous capture mode |
| 683 | |
| 684 | def init_cont(self): |
| 685 | qsize = 1 |
| 686 | if self.vmode == VM_CONT: |
| 687 | self.rate = self.getint(self.in_rate, 2) |
| 688 | else: |
| 689 | self.rate = 2 |
| 690 | x, y = self.vout.getsize() |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 691 | if self.use_24: |
| 692 | info = (SV.YUV411_FRAMES, x, y, qsize, self.rate) |
| 693 | else: |
| 694 | info = (SV.RGB8_FRAMES, x, y, qsize, self.rate) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 695 | info2 = self.video.InitContinuousCapture(info) |
| 696 | if info2 <> info: |
| 697 | # XXX This is really only debug info |
| 698 | print 'Info mismatch: requested', info, 'got', info2 |
| 699 | |
| 700 | def end_cont(self): |
| 701 | self.video.EndContinuousCapture() |
| 702 | |
| 703 | # Misc stuff |
| 704 | |
| 705 | def settitle(self): |
| 706 | gl.winset(self.window) |
| 707 | x, y = gl.getsize() |
Guido van Rossum | 43df862 | 1993-06-11 15:48:39 +0000 | [diff] [blame] | 708 | title = 'Vb ' + self.vfile + ' (%dx%d)' % (x, y) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 709 | gl.wintitle(title) |
| 710 | |
| 711 | def get_vformat(self): |
| 712 | i = self.c_vformat.get_choice() |
| 713 | label = VideoFormatLabels[i-1] |
| 714 | format = VideoFormats[i-1] |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 715 | if format == 'compress' and cl == None: |
| 716 | fl.show_message('Sorry, no compression library support') |
| 717 | format = '' |
| 718 | label = 'Video off' |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 719 | self.vformat = format |
| 720 | if self.vformat == '': |
| 721 | self.form.freeze_form() |
| 722 | self.g_video.hide_object() |
| 723 | self.g_cont.hide_object() |
| 724 | self.g_burst.hide_object() |
| 725 | self.g_single.hide_object() |
| 726 | self.form.unfreeze_form() |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 727 | else: |
| 728 | self.g_video.show_object() |
| 729 | if self.vmode == VM_CONT: |
| 730 | self.g_cont.show_object() |
| 731 | elif self.vmode == VM_BURST: |
| 732 | self.g_burst.show_object() |
| 733 | elif self.vmode == VM_SINGLE: |
| 734 | self.g_single.show_object() |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 735 | # |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 736 | self.rgb = (format[:3] == 'rgb' or format == 'compress') |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 737 | self.mono = (format == 'mono') |
| 738 | self.grey = (format[:4] == 'grey') |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 739 | self.use_24 = (format in ('rgb', 'jpeg', 'compress')) |
| 740 | if self.use_24: |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 741 | self.g_rgb24.show_object() |
| 742 | else: |
| 743 | self.g_rgb24.hide_object() |
| 744 | self.use_jpeg = (format == 'jpeg') |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 745 | self.mono_use_thresh = (label == 'mono thresh') |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 746 | self.use_compress = (format == 'compress') |
| 747 | if self.use_compress: |
| 748 | self.g_compress.show_object() |
| 749 | else: |
| 750 | self.g_compress.hide_object() |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 751 | s = format[4:] |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 752 | if self.grey and s: |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 753 | self.greybits = string.atoi(s) |
| 754 | else: |
| 755 | self.greybits = 8 |
| 756 | if label == 'grey2 dith': |
| 757 | self.greybits = -2 |
| 758 | # |
| 759 | convertor = None |
| 760 | if self.grey: |
| 761 | if self.greybits == 2: |
| 762 | convertor = imageop.grey2grey2 |
| 763 | elif self.greybits == 4: |
| 764 | convertor = imageop.grey2grey4 |
| 765 | elif self.greybits == -2: |
| 766 | convertor = imageop.dither2grey2 |
| 767 | self.convertor = convertor |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 768 | self.optfullsizewindow() |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 769 | |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 770 | def get_aformat(self): |
| 771 | self.reset() |
| 772 | self.aformat = self.c_aformat.get_choice() |
| 773 | if self.aformat == A_OFF: |
| 774 | self.g_audio.hide_object() |
| 775 | else: |
| 776 | self.g_audio.show_object() |
| 777 | |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 778 | def init_compressor(self, w, h): |
| 779 | self.compressor = None |
Guido van Rossum | 1554b7b | 1998-10-09 13:26:29 +0000 | [diff] [blame] | 780 | scheme = cl.QuerySchemeFromName(cl.VIDEO, self.comp_scheme) |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 781 | self.compressor = cl.OpenCompressor(scheme) |
Guido van Rossum | 1554b7b | 1998-10-09 13:26:29 +0000 | [diff] [blame] | 782 | parambuf = [cl.IMAGE_WIDTH, w, \ |
| 783 | cl.IMAGE_HEIGHT, h, \ |
| 784 | cl.ORIGINAL_FORMAT, cl.YUV422DC] |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 785 | self.compressor.SetParams(parambuf) |
| 786 | return self.compressor.Compress(0, '') |
| 787 | |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 788 | def open_if_closed(self): |
| 789 | if not self.vout: |
| 790 | self.open_video() |
| 791 | if not self.aout: |
| 792 | self.open_audio() |
| 793 | |
| 794 | # File I/O handling |
| 795 | |
| 796 | def open_video(self): |
| 797 | self.close_video() |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 798 | gl.winset(self.window) |
| 799 | x, y = gl.getsize() |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 800 | if self.use_24: |
| 801 | if self.rgb24_size == 2: |
| 802 | x, y = x/2, y/2 |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 803 | elif self.rgb24_size == 3: |
Jack Jansen | 672754a | 1993-06-08 12:52:41 +0000 | [diff] [blame] | 804 | x, y = x/4, y/4 |
Guido van Rossum | 21a3ff9 | 1993-12-17 15:11:41 +0000 | [diff] [blame] | 805 | vout = VFile.VoutFile(self.vfile) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 806 | vout.setformat(self.vformat) |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 807 | if self.vformat == 'compress': |
| 808 | cheader = self.init_compressor(x, y) |
| 809 | vout.setcompressheader(cheader) |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 810 | vout.setsize(x, y) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 811 | if self.vmode == VM_BURST: |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 812 | vout.setpf((1, -2)) |
| 813 | vout.writeheader() |
| 814 | self.vout = vout |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 815 | self.nframes = 0 |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 816 | self.speed_factor = 1 |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 817 | self.t_nframes.label = `self.nframes` |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 818 | |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 819 | def write_frame(self, t, data): |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 820 | t = t * self.speed_factor |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 821 | if not self.vout: |
| 822 | gl.ringbell() |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 823 | return 0 |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 824 | if self.convertor: |
| 825 | data = self.convertor(data, len(data), 1) |
| 826 | elif self.mono: |
| 827 | if self.mono_use_thresh: |
| 828 | data = imageop.grey2mono(data, \ |
| 829 | len(data), 1,\ |
| 830 | self.mono_thresh) |
| 831 | else: |
| 832 | data = imageop.dither2mono(data, \ |
| 833 | len(data), 1) |
| 834 | try: |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 835 | self.vout.writeframe(int(t), data, None) |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 836 | except IOError, msg: |
Guido van Rossum | ad4fcd4 | 1993-05-11 18:36:54 +0000 | [diff] [blame] | 837 | self.reset() |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 838 | if msg == (0, 'Error 0'): |
| 839 | msg = 'disk full??' |
| 840 | fl.show_message('IOError', str(msg), '') |
| 841 | return 0 |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 842 | self.nframes = self.nframes + 1 |
| 843 | self.t_nframes.label = `self.nframes` |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 844 | return 1 |
| 845 | |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 846 | def close_video(self): |
| 847 | if not self.vout: |
| 848 | return |
| 849 | self.nframes = 0 |
| 850 | self.t_nframes.label = '' |
Guido van Rossum | c5a1433 | 1993-05-07 11:20:07 +0000 | [diff] [blame] | 851 | try: |
| 852 | self.vout.close() |
| 853 | except IOError, msg: |
| 854 | if msg == (0, 'Error 0'): |
| 855 | msg = 'disk full??' |
| 856 | fl.show_message('IOError', str(msg), '') |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 857 | self.vout = None |
Jack Jansen | 78991fd | 1993-07-23 11:59:25 +0000 | [diff] [blame] | 858 | self.compressor = None |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 859 | |
| 860 | # Watch cursor handling |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 861 | |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 862 | def setwatch(self): |
| 863 | gl.winset(self.form.window) |
| 864 | gl.setcursor(WATCH, 0, 0) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 865 | gl.winset(self.window) |
| 866 | gl.setcursor(WATCH, 0, 0) |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 867 | |
| 868 | def setarrow(self): |
| 869 | gl.winset(self.form.window) |
| 870 | gl.setcursor(ARROW, 0, 0) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 871 | gl.winset(self.window) |
| 872 | gl.setcursor(ARROW, 0, 0) |
Guido van Rossum | bc6d3c3 | 1993-05-07 09:37:42 +0000 | [diff] [blame] | 873 | |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 874 | # Numeric field handling |
| 875 | |
| 876 | def getint(self, field, default): |
| 877 | try: |
| 878 | value = string.atoi(field.get_input()) |
| 879 | except string.atoi_error: |
| 880 | value = default |
| 881 | field.set_input(`value`) |
| 882 | return value |
| 883 | |
| 884 | def getfloat(self, field, default): |
| 885 | try: |
| 886 | value = float(eval(field.get_input())) |
| 887 | except: |
| 888 | value = float(default) |
Guido van Rossum | c17c84f | 1993-05-10 15:45:49 +0000 | [diff] [blame] | 889 | field.set_input(`value`) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 890 | return value |
| 891 | |
| 892 | # Audio stuff |
| 893 | |
| 894 | def open_audio(self): |
| 895 | if self.aformat == A_OFF: |
| 896 | return |
| 897 | import aifc |
| 898 | import al |
| 899 | import AL |
| 900 | import thread |
| 901 | self.close_audio() |
| 902 | params = [AL.INPUT_RATE, 0] |
| 903 | al.getparams(AL.DEFAULT_DEVICE, params) |
| 904 | rate = params[1] |
| 905 | self.aout = aifc.open(self.afile, 'w') |
| 906 | if self.aformat in (A_16_STEREO, A_8_STEREO): |
| 907 | nch = AL.STEREO |
| 908 | else: |
| 909 | nch = AL.MONO |
| 910 | if self.aformat in (A_16_STEREO, A_16_MONO): |
| 911 | width = AL.SAMPLE_16 |
| 912 | else: |
| 913 | width = AL.SAMPLE_8 |
| 914 | self.aout.setnchannels(nch) |
| 915 | self.aout.setsampwidth(width) |
| 916 | self.aout.setframerate(rate) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 917 | c = al.newconfig() |
| 918 | c.setqueuesize(8000) |
| 919 | c.setchannels(nch) |
| 920 | c.setwidth(width) |
| 921 | self.aport = al.openport('Vb audio record', 'r', c) |
| 922 | self.audio_stop = 0 |
| 923 | self.audio_ok = 0 |
| 924 | self.audio_busy = 1 |
| 925 | thread.start_new_thread(self.record_audio, ()) |
| 926 | |
| 927 | def start_audio(self): |
| 928 | if self.aformat == A_OFF: |
| 929 | return |
| 930 | self.audio_ok = 1 |
| 931 | |
| 932 | def record_audio(self, *args): |
| 933 | # This function runs in a separate thread |
| 934 | # Currently no semaphores are used |
| 935 | while not self.audio_stop: |
| 936 | data = self.aport.readsamps(4000) |
| 937 | if self.audio_ok: |
Guido van Rossum | f6d8032 | 1993-06-10 13:40:51 +0000 | [diff] [blame] | 938 | self.aout.writeframes(data) |
Guido van Rossum | 9f42f4f | 1993-05-10 15:07:20 +0000 | [diff] [blame] | 939 | data = None |
| 940 | self.audio_busy = 0 |
| 941 | |
| 942 | def stop_audio(self): |
| 943 | self.audio_ok = 0 |
| 944 | |
| 945 | def close_audio(self): |
| 946 | if self.aout: |
| 947 | self.audio_ok = 0 |
| 948 | self.audio_stop = 1 |
| 949 | while self.audio_busy: |
| 950 | time.sleep(0.1) |
| 951 | self.aout.close() |
| 952 | self.aout = None |
| 953 | if self.aport: |
| 954 | self.aport.closeport() |
| 955 | self.aport = None |
| 956 | |
Guido van Rossum | b7e3cc1 | 1993-05-06 16:06:44 +0000 | [diff] [blame] | 957 | |
| 958 | try: |
| 959 | main() |
| 960 | except KeyboardInterrupt: |
| 961 | print '[Interrupt]' |
| 962 | sys.exit(1) |