| #! /ufs/guido/bin/sgi/python |
| |
| # Edit CMIF movies interactively -- copy one or more files to an output file |
| |
| |
| # XXX To do: |
| # |
| # - convert between formats (grey, rgb, rgb8, ...) |
| # - change size |
| # - cut out a given area of the image |
| # - change time base (a la Vtime) |
| |
| |
| import sys |
| import os |
| import gl, GL, DEVICE |
| import fl, FL |
| import flp |
| import Viewer |
| import getopt |
| import string |
| import watchcursor |
| |
| ARROW = 0 |
| WATCH = 1 |
| watchcursor.defwatch(WATCH) |
| |
| |
| def main(): |
| qsize = 40 |
| opts, args = getopt.getopt(sys.argv[1:], 'q:') |
| for o, a in opts: |
| if o == '-q': |
| qsize = string.atoi(a) |
| ed = Editor(qsize) |
| if args[0:]: |
| ed.open_input(args[0]) |
| if args[1:]: |
| ed.open_output(args[1]) |
| while 1: |
| dummy = fl.do_forms() |
| |
| |
| class Editor: |
| |
| def __init__(self, qsize): |
| self.qsize = qsize |
| self.vin = None |
| self.vout = None |
| self.ifile = '' |
| self.ofile = '' |
| formdef = flp.parse_form('VeditForm', 'form') |
| flp.create_full_form(self, formdef) |
| self.form.show_form(FL.PLACE_SIZE, FL.TRUE, 'Vedit') |
| fl.set_event_call_back(self.do_event) |
| |
| def do_event(self, dev, val): |
| if dev == DEVICE.REDRAW: |
| if self.vin: |
| self.vin.redraw(val) |
| if self.vout: |
| self.vout.redraw(val) |
| |
| def busy(self): |
| gl.winset(self.form.window) |
| gl.setcursor(WATCH, 0, 0) |
| |
| def ready(self): |
| gl.winset(self.form.window) |
| gl.setcursor(ARROW, 0, 0) |
| |
| |
| def iocheck(self): |
| self.msg('') |
| if self.vin == None and self.vout == None: |
| self.err('Please open input and output files first') |
| return 0 |
| return self.icheck() and self.ocheck() |
| |
| def icheck(self): |
| self.msg('') |
| if self.vin == None: |
| self.err('Please open an input file first') |
| return 0 |
| return 1 |
| |
| def ocheck(self): |
| self.msg('') |
| if self.vout == None: |
| self.err('Please open an output file first') |
| return 0 |
| return 1 |
| |
| |
| def cb_in_new(self, *args): |
| self.msg('') |
| hd, tl = os.path.split(self.ifile) |
| filename = fl.file_selector('Input video file', hd, '', tl) |
| if not filename: return |
| self.open_input(filename) |
| |
| def cb_in_close(self, *args): |
| self.msg('') |
| self.close_input() |
| |
| def cb_in_skip(self, *args): |
| if not self.icheck(): return |
| if not self.vin.get(): self.err('End of input file') |
| self.ishow() |
| |
| def cb_in_back(self, *args): |
| if not self.icheck(): return |
| if not self.vin.backup(): self.err('Begin of input file') |
| self.ishow() |
| |
| def cb_in_slider(self, *args): |
| if not self.icheck(): return |
| left, pos, right = self.vin.qinfo() |
| i = int(self.in_slider.get_slider_value()) |
| i = max(i, left) |
| i = min(i, right) |
| if i == pos: return |
| if not self.vin.seek(i): |
| self.err('Input seek failed') |
| self.ishow() |
| |
| def cb_in_rewind(self, *args): |
| if not self.icheck(): return |
| self.vin.rewind() |
| self.ishow() |
| |
| |
| def cb_copy(self, *args): |
| if not self.iocheck(): return |
| data = self.vin.get() |
| if not data: |
| self.err('End of input file') |
| self.ishow() |
| return |
| if self.vout.getinfo() <> self.vin.getinfo(): |
| print 'Copying info...' |
| self.vout.setinfo(self.vin.getinfo()) |
| if self.vin.format == 'compress': |
| self.vout.setcompressheader(\ |
| self.vin.getcompressheader()) |
| self.vout.put(data) |
| self.oshow() |
| self.ishow() |
| |
| def cb_uncopy(self, *args): |
| if not self.iocheck(): return |
| if not self.vout.backup(): |
| self.err('Output buffer exhausted') |
| return |
| self.oshow() |
| if not self.vin.backup(): |
| self.err('Begin of input file') |
| return |
| self.ishow() |
| |
| |
| def cb_out_new(self, *args): |
| self.msg('') |
| hd, tl = os.path.split(self.ofile) |
| filename = fl.file_selector('Output video file', hd, '', tl) |
| if not filename: return |
| self.open_output(filename) |
| |
| def cb_out_close(self, *args): |
| self.msg('') |
| self.close_output() |
| |
| def cb_out_skip(self, *args): |
| if not self.ocheck(): return |
| if not self.vout.forward(): self.err('Output buffer exhausted') |
| self.oshow() |
| |
| def cb_out_back(self, *args): |
| if not self.ocheck(): return |
| if not self.vout.backup(): self.err('Output buffer exhausted') |
| self.oshow() |
| |
| def cb_out_slider(self, *args): |
| if not self.ocheck(): return |
| i = int(self.out_slider.get_slider_value()) |
| left, pos, right = self.vout.qinfo() |
| i = int(self.out_slider.get_slider_value()) |
| i = max(i, left) |
| i = min(i, right) |
| if i == pos: return |
| if not self.vout.seek(i): |
| self.err('Output seek failed') |
| self.oshow() |
| |
| def cb_out_trunc(self, *arcs): |
| if not self.ocheck(): return |
| self.vout.trunc() |
| self.oshow() |
| |
| def cb_out_rewind(self, *args): |
| if not self.ocheck(): return |
| self.vout.rewind() |
| self.oshow() |
| |
| |
| def cb_quit(self, *args): |
| self.close_input() |
| self.close_output() |
| sys.exit(0) |
| |
| |
| def open_input(self, filename): |
| self.ifile = filename |
| basename = os.path.split(filename)[1] |
| title = 'in: ' + basename |
| try: |
| vin = Viewer.InputViewer(filename, title) |
| except: |
| self.err('Can\'t open input file', filename) |
| return |
| self.close_input() |
| self.vin = vin |
| self.in_file.label = basename |
| self.ishow() |
| |
| def close_input(self): |
| if self.vin: |
| self.busy() |
| self.msg('Closing input file...') |
| self.vin.close() |
| self.ready() |
| self.msg('') |
| self.vin = None |
| self.in_file.label = '(none)' |
| self.format('in') |
| |
| def ishow(self): |
| self.vin.show() |
| self.format('in') |
| |
| def open_output(self, filename): |
| self.ofile = filename |
| basename = os.path.split(filename)[1] |
| title = 'out: ' + basename |
| try: |
| vout = Viewer.OutputViewer(filename, \ |
| title, self.qsize) |
| except: |
| self.err('Can\'t open output file', filename) |
| return |
| self.close_output() |
| self.vout = vout |
| self.out_file.label = basename |
| if self.vin: |
| self.vout.setinfo(self.vin.getinfo()) |
| self.oshow() |
| |
| def close_output(self): |
| if self.vout: |
| self.busy() |
| self.msg('Closing output file...') |
| self.vout.close() |
| self.ready() |
| self.msg('') |
| self.vout = None |
| self.out_file.label = '(none)' |
| self.format('out') |
| |
| def oshow(self): |
| self.vout.show() |
| self.format('out') |
| |
| |
| def msg(self, *args): |
| str = string.strip(string.join(args)) |
| self.msg_area.label = str |
| |
| def err(self, *args): |
| gl.ringbell() |
| apply(self.msg, args) |
| |
| def format(self, io): |
| v = getattr(self, 'v' + io) |
| if v == None: |
| left = right = pos = 0 |
| else: |
| left, pos, right = v.qinfo() |
| getattr(self, io + '_info1').label = `left` |
| getattr(self, io + '_info2').label = `pos` |
| getattr(self, io + '_info3').label = `right` |
| sl = getattr(self, io + '_slider') |
| sl.freeze_object() |
| sl.set_slider_bounds(left, right) |
| sl.set_slider_value(pos) |
| sl.unfreeze_object() |
| |
| |
| try: |
| main() |
| except KeyboardInterrupt: |
| print '[Interrupt]' |