Demo/sgi/video/video2rgb.py

   1 #! /usr/bin/env python
   2
   3 # Convert CMIF movie file(s) to a sequence of rgb images
   4
   5
   6 # Help function
   7
   8 def help():
   9         print 'Usage: video2rgb [options] [file] ...'
  10         print
  11         print 'Options:'
  12         print '-q         : quiet, no informative messages'
  13         print '-m         : create monochrome (greyscale) image files'
  14         print '-f prefix  : create image files with names "prefix0000.rgb"'
  15         print 'file ...   : file(s) to convert; default film.video'
  16
  17
  18 # Imported modules
  19
  20 import sys
  21 sys.path.append('/ufs/jack/src/av/video') # Increase chance of finding VFile
  22 import VFile
  23 import time
  24 import getopt
  25 import string
  26 import imgfile
  27 import imgconv
  28
  29
  30 # Global options
  31
  32 quiet = 0
  33 prefix = 'film'
  34 seqno = 0
  35 mono = 0
  36
  37
  38 # Main program -- mostly command line parsing
  39
  40 def main():
  41         global quiet, prefix, mono
  42
  43         # Parse command line
  44         try:
  45                 opts, args = getopt.getopt(sys.argv[1:], 'qmf:')
  46         except getopt.error, msg:
  47                 sys.stdout = sys.stderr
  48                 print 'Error:', msg, '\n'
  49                 help()
  50                 sys.exit(2)
  51
  52         # Interpret options
  53         try:
  54                 for opt, arg in opts:
  55                         if opt == '-q': quiet = 1
  56                         if opt == '-f': prefix = arg
  57                         if opt == '-m': mono = 1
  58         except string.atoi_error:
  59                 sys.stdout = sys.stderr
  60                 print 'Option', opt, 'requires integer argument'
  61                 sys.exit(2)
  62
  63         # Process all files
  64         if not args: args = ['film.video']
  65         sts = 0
  66         for filename in args:
  67                 sts = (process(filename) or sts)
  68
  69         # Exit with proper exit status
  70         sys.exit(sts)
  71
  72
  73 # Process one movie file
  74
  75 def process(filename):
  76         try:
  77                 vin = VFile.VinFile(filename)
  78         except IOError, msg:
  79                 sys.stderr.write(filename + ': I/O error: ' + `msg` + '\n')
  80                 return 1
  81         except VFile.Error, msg:
  82                 sys.stderr.write(msg + '\n')
  83                 return 1
  84         except EOFError:
  85                 sys.stderr.write(filename + ': EOF in video header\n')
  86                 return 1
  87
  88         if not quiet:
  89                 vin.printinfo()
  90
  91         width, height = int(vin.width), int(vin.height)
  92
  93         try:
  94                 if mono:
  95                         cf = imgconv.getconverter(vin.format, 'grey')
  96                 else:
  97                         cf = imgconv.getconverter(vin.format, 'rgb')
  98         except imgconv.error:
  99                 print 'Sorry, no converter available for type',vin.format
 100                 return
 101
 102         if mono:
 103                 depth = 1
 104                 bpp = 1
 105         else:
 106                 depth = 3
 107                 bpp = 4
 108
 109         convert(vin, cf, width, height, depth, bpp, vin.packfactor)
 110
 111 def convert(vin, cf, width, height, depth, bpp, pf):
 112         global seqno
 113
 114         if type(pf) == type(()):
 115                 xpf, ypf = pf
 116         elif pf == 0:
 117                 xpf = ypf = 1
 118         else:
 119                 xpf = ypf = pf
 120         while 1:
 121                 try:
 122                         time, data, cdata = vin.getnextframe()
 123                 except EOFError:
 124                         return
 125                 if cdata:
 126                         print 'Film contains chromdata!'
 127                         return
 128                 data = cf(data, width/xpf, height/abs(ypf))
 129                 if pf:
 130                         data = applypackfactor(data, width, height, pf, bpp)
 131                 s = `seqno`
 132                 s = '0'*(4-len(s)) + s
 133                 fname = prefix + s + '.rgb'
 134                 seqno = seqno + 1
 135                 if not quiet:
 136                         print 'Writing',fname,'...'
 137                 imgfile.write(fname, data, width, height, depth)
 138
 139 def applypackfactor(image, w, h, pf, bpp):
 140         import imageop
 141         if type(pf) == type(()):
 142                 xpf, ypf = pf
 143         elif pf == 0:
 144                 xpf = ypf = 1
 145         else:
 146                 xpf = ypf = pf
 147         w1 = w/xpf
 148         h1 = h/abs(ypf)
 149         if ypf < 0:
 150                 ypf = -ypf
 151                 image = imageop.crop(image, bpp, w1, h1, 0, h1-1, w1-1, 0)
 152         return imageop.scale(image, bpp, w1, h1, w, h)
 153
 154 # Don't forget to call the main program
 155
 156 try:
 157         main()
 158 except KeyboardInterrupt:
 159         print '[Interrupt]'