add options to change the width and height

Author: Michael Zingale

plot.py

@@ -9,14 +9,14 @@
 
 # plot an output file using the solver's dovis script
 
-def makeplot(myd, solverName, outfile):
+def makeplot(myd, solverName, outfile, W, H):
 
     exec 'import ' + solverName + ' as solver'
 
     sim = solver.Simulation(None, None)
     sim.cc_data = myd
 
-    pylab.figure(num=1, figsize=(8,4.5), dpi=100, facecolor='w')
+    pylab.figure(num=1, figsize=(W,H), dpi=100, facecolor='w')
 
     sim.dovis()
     pylab.savefig(outfile)
@@ -35,23 +35,30 @@ def usage():
   -h, --help    show this help message and exit
   -o image.png  output image name. The extension .png will generate a PNG
                 file, .eps will generate an EPS file (default: plot.png).
+  -W width      width in inches
+  -H height     height in inches
 """
     print usage
     sys.exit()
 
 
 if __name__== "__main__":
 
-    try: opts, next = getopt.getopt(sys.argv[1:], "o:h")
+    try: opts, next = getopt.getopt(sys.argv[1:], "o:h:W:H:")
     except getopt.GetoptError:
         sys.exit("invalid calling sequence")
 
     outfile = "plot.png"
 
+    W = 8.0
+    H = 4.5
+
     for o, a in opts:
         if o == "-h": usage()
         if o == "-o": outfile = a
-                                    
+        if o == "-W": W = float(a)
+        if o == "-H": H = float(a)
+
     try: solver = next[0]
     except: usage()
 
@@ -60,7 +67,7 @@ def usage():
 
     myg, myd = patch.read(file)
 
-    makeplot(myd, solver, outfile)
+    makeplot(myd, solver, outfile, W, H)