Changeset 432

Timestamp:

Nov 30, 2011, 6:37:59 PM (13 years ago)

Author:

acolaitis

Message:

PYTHON. Modifs relative to movies. To make a movie, the user must now specify --rate 20 (or any other positive value). Fixed gcm plots because of last commit on LES files. Fixed movies with GCM files, also not tested (until mencoder is on the farm)

Location:

trunk/UTIL/PYTHON

Files:

• myplot.py (modified) (1 diff)
• myscript.py (modified) (1 diff)
• planetoplot.py (modified) (5 diffs)
• pp.py (modified) (1 diff)

trunk/UTIL/PYTHON/myplot.py

@@ -39 +39 @@
     if 'controle' in nc.variables:             typefile = 'gcm'
     elif 'phisinit' in nc.variables:           typefile = 'gcm'
-    elif '9999' in getattr(nc,'START_DATE') :  typefile = 'mesoideal'
+    elif hasattr(nc,'START_DATE'):
+      if '9999' in getattr(nc,'START_DATE') :  typefile = 'mesoideal'
     elif 'vert' in nc.variables:               typefile = 'mesoapi'
     elif 'U' in nc.variables:                  typefile = 'meso'

trunk/UTIL/PYTHON/myscript.py

@@ -61 +61 @@
     ### SPECIAL
     parser.add_option('--tsat',         action='store_true',dest='tsat',               default=False,help='convert temperature field T in Tsat-T using pressure')
+    parser.add_option('--rate',         action='store'     ,dest='rate', type="int",   default=None, help='Output is a movie along Time dimension. --time must not be specified. [None]')
 
     return parser

trunk/UTIL/PYTHON/planetoplot.py

@@ -51 +51 @@
            tsat=False,\
            flagnolow=False,\
-           movie_axis=0):
+           mrate=None):
 
 
@@ -306 +306 @@
        varname = all_varname[index_f]
        if varname:
-           what_I_plot, error = reducefield(all_var[index_f], d4=indextime, d1=indexlon, d2=indexlat , d3=indexvert , yint=yintegral, alt=vert, anomaly=anomaly)
+           zindtime=indextime
+           if mrate is not None:zindtime=None
+           what_I_plot, error = reducefield(all_var[index_f], d4=zindtime, d1=indexlon, d2=indexlat , d3=indexvert , yint=yintegral, alt=vert, anomaly=anomaly)
            what_I_plot = what_I_plot*mult
            if not error: 
@@ -334 +336 @@
                if len(what_I_plot.shape) >= 2:
                  istart=0
-                 if indextime is None:iend=len(time)-1
+                 if mrate is not None:iend=len(time)-1
                  else:iend=istart
                  imov=istart
@@ -340 +342 @@
                  while imov <= iend:
                     what_I_plot_frame=what_I_plot
-                    if len(what_I_plot.shape) is 3:
+                    if mrate is not None:
                        what_I_plot_frame=what_I_plot[imov,:,:]
                        print "-> frame ",imov+1
@@ -369 +371 @@
                         mframe=fig2img(mpl.pyplot.gcf())
 
-                        if ((len(what_I_plot.shape) is 3) and (imov is 0)):
+                        if ((mrate is not None) and (imov is 0)):
                             W,H = mpl.pyplot.gcf().canvas.get_width_height()
-                            rate = 10
-                            video = VideoSink((H,W), "test", rate=rate, byteorder="rgba")
-                        if len(what_I_plot.shape) is 3: 
+                            video = VideoSink((H,W), "test", rate=mrate, byteorder="rgba")
+                        if mrate is not None: 
                             video.run(mframe)
                             mpl.pyplot.close()
 
                         imov=imov+1
-                 if len(what_I_plot.shape) is 3: video.close
+                 if mrate is not None: video.close
                ##### 1D field
                elif len(what_I_plot.shape) is 1:

trunk/UTIL/PYTHON/pp.py

@@ -137 +137 @@
                 ope=opt.operat,fileref=reffile,minop=opt.vminope,maxop=opt.vmaxope,titleref=opt.titref,\
                 invert_y=opt.inverty,xaxis=zexaxis,yaxis=zeyaxis,ylog=opt.logy,yintegral=opt.column,\
-                blat=opt.blat,tsat=opt.tsat,flagnolow=opt.nolow)
+                blat=opt.blat,tsat=opt.tsat,flagnolow=opt.nolow,mrate=opt.rate)
         print 'DONE: '+name
         system("rm -f to_be_erased")