diff scripts/play-video.py @ 385:1917db662aa7

added rescaling options to scripts play-video and display-trajectories
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Mon, 22 Jul 2013 18:33:47 -0400
parents 1d90e9080cb2
children 3058e00887bc
line wrap: on
line diff
--- a/scripts/play-video.py	Mon Jul 22 18:11:01 2013 -0400
+++ b/scripts/play-video.py	Mon Jul 22 18:33:47 2013 -0400
@@ -1,22 +1,23 @@
 #! /usr/bin/env python
 
-import sys,getopt
+import sys, argparse
 import cvutils
 
-options, args = getopt.getopt(sys.argv[1:], 'hi:f:',['help', 'fps=']) 
-options = dict(options)
-print options
 
-if '--help' in options.keys() or '-h' in options.keys() or len(sys.argv) == 1:
-    print('Usage: '+sys.argv[0]+' --help|-h -i video-filename [-f first_frame] [--fps frame_rate]')
-    sys.exit()
+parser = argparse.ArgumentParser(description='The program displays the video.')
+parser.add_argument('-i', dest = 'videoFilename', help = 'name of the video file', required = True)
+parser.add_argument('-f', dest = 'firstFrameNum', help = 'number of first frame number to display', default = 0, type = int)
+parser.add_argument('--fps', dest = 'frameRate', help = 'approximate frame rate to replay', type = float)
+parser.add_argument('-r', dest = 'rescale', help = 'rescaling factor for the displayed image', default = 1., type = float)
+
+args = parser.parse_args()
 
 firstFrameNum = 0
-if '-f' in options.keys():
-    firstFrameNum = int(options['-f'])
+if args.firstFrameNum != None:
+    firstFrameNum = args.firstFrameNum
 
 frameRate = -1
-if '--fps' in options.keys():
-    frameRate = int(options['--fps'])
+if args.frameRate != None:
+    frameRate = args.frameRate
 
-cvutils.playVideo(options['-i'], firstFrameNum, frameRate)
+cvutils.playVideo(args.videoFilename, firstFrameNum, frameRate, rescale = args.rescale)