Changeset 436 for trunk/UTIL/PYTHON/planetoplot.py

Timestamp:

Dec 1, 2011, 2:19:47 AM (13 years ago)

Author:

aslmd

Message:

GRAPHICS: make movies work with mapmode=1. very nice animated movies are now possible with cool map projections.

File:

• trunk/UTIL/PYTHON/planetoplot.py (modified) (8 diffs)

trunk/UTIL/PYTHON/planetoplot.py

@@ -89 +89 @@
     if slon is not None: vlon = slon[0][0]
     if slat is not None: vlat = slat[0][0]
-    if mrate is not None: mapmode = 0 ### THIS IS DIRTY. MAPMPODE=1 SHOULD WORK WITH MOVIES. WILL BE FIXED.
     if mapmode == 0:       winds=False
     elif mapmode == 1:     
         if svert is None:  svert = readslices(str(level)) ; nvert=1
+        if stime is None and mrate is None:
+           stime = readslices(str(0)) ; ntime=1 ## this is a default choice
+           print "WELL... nothing about time axis. I took default: first time reference stored in file."
     zelen = len(namefiles)*len(var)
     numplot = zelen*nslices
@@ -235 +237 @@
     #################################
     ### Time loop for plotting device
-    nplot = 1;error = False
+    nplot = 1 ; error = False
     print "********************************************"
     while error is False:
@@ -265 +267 @@
                    for i in range(len(time)):  time[i] = localtime ( interv[0]+time[i]*interv[1], 0.5*(wlon[0]+wlon[1]) )
                    print "OK... WORKING WITH LOCAL TIMES"
-               else: errormess("local times not supported for GCM files. not too hard to modify the code though.")
+               else: errormess("local times not supported. not too hard to modify the code though.")
        if mrate is not None:                 indextime = None 
-       elif mapmode == 1 and stime is None:  indextime = 1
        else:                                 indextime = getsindex(stime,((nplot-1)//(nlon*nlat*nvert))%ntime,time)
        ltst = None 
@@ -288 +289 @@
               if mapmode == 0:
                   if typefile in ['mesoideal']:    what_I_plot = dumpbdy(what_I_plot,0,stag='W')
-                  itime=indextime
-                  if len(what_I_plot.shape) is 3:itime=[0]
+                  itime=indextime 
+                  if len(what_I_plot.shape) is 3: itime=[0]
                   what_I_plot, x, y = define_axis(lon,lat,vert,time,indexlon,indexlat,indexvert,\
                         itime,what_I_plot, len(all_var2[index_f].shape),vertmode)
@@ -306 +307 @@
        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)
+           what_I_plot, error = reducefield( all_var[index_f], d4=indextime, d1=indexlon, d2=indexlat, d3=indexvert, \
+                                             yint=yintegral, alt=vert, anomaly=anomaly )
            what_I_plot = what_I_plot*mult
            if not error: 
                fvar = varname 
                if anomaly: fvar = 'anomaly'
-               ##### MAPMODE-SPECIFIC SETTINGS #####
-               if mapmode == 1:
-                   if typefile in ['mesoapi','meso']:    what_I_plot = dumpbdy(what_I_plot,6)
-               elif mapmode == 0:
+               if mapmode == 0:
                    itime=indextime
                    if len(what_I_plot.shape) is 3:itime=[0]
@@ -328 +327 @@
                if (fileref is not None) and (index_f is numplot-1):    zevmin, zevmax = calculate_bounds(what_I_plot,vmin=minop,vmax=maxop)
                else:                                                   zevmin, zevmax = calculate_bounds(what_I_plot,vmin=vmin,vmax=vmax)
-               if colorb in ["def","nobar"]:   palette = get_cmap(name=defcolorb(fvar.upper()))
-               elif (fileref is not None) and (index_f is numplot-1): palette = get_cmap(name="RdBu_r")
-               else:                           palette = get_cmap(name=colorb)
+               if colorb in ["def","nobar"]:                           palette = get_cmap(name=defcolorb(fvar.upper()))
+               elif (fileref is not None) and (index_f is numplot-1):  palette = get_cmap(name="RdBu_r")
+               else:                                                   palette = get_cmap(name=colorb)
                ##### simple 2D field and movies of 2D fields
                if len(what_I_plot.shape) >= 2:
-                 if (len(what_I_plot.shape) is 3 and mrate is None): 
-                     errormess("3D field in input but not rate specified for movie (use --rate RATE or specify --time TIME). Exit.")
+                 if (len(what_I_plot.shape) is 3 and mrate is None):  errormess("3D field. Use --rate RATE for movie or specify --time TIME. Exit.")
                  istart=0 
                  if mrate is not None: iend=len(time)-1
@@ -341 +339 @@
                  while imov <= iend:
                     what_I_plot_frame = what_I_plot
-                    if mrate is not None:      what_I_plot_frame = what_I_plot[imov,:,:] ; print "-> frame ",imov+1
-                    if hole:                   what_I_plot_frame = hole_bounds(what_I_plot_frame,zevmin,zevmax)
-                    else:                      what_I_plot_frame = bounds(what_I_plot_frame,zevmin,zevmax)
-                    if flagnolow:              what_I_plot_frame = nolow(what_I_plot_frame)
+                    if mrate is not None:      
+                        what_I_plot_frame = what_I_plot[imov,:,:] ; print "-> frame ",imov+1
+                        if mapmode == 1: 
+                            m = define_proj(proj,wlon,wlat,back=back,blat=blat)  ## this is dirty, defined above but out of imov loop
+                            x, y = m(lon2d, lat2d)                               ## this is dirty, defined above but out of imov loop
+                    if typefile in ['mesoapi','meso'] and mapmode == 1:   what_I_plot_frame = dumpbdy(what_I_plot_frame,6)
+                    if hole:         what_I_plot_frame = hole_bounds(what_I_plot_frame,zevmin,zevmax)
+                    else:            what_I_plot_frame = bounds(what_I_plot_frame,zevmin,zevmax)
+                    if flagnolow:    what_I_plot_frame = nolow(what_I_plot_frame)
                     if not tile:
                         #zelevels = np.linspace(zevmin*(1. + 1.e-7),zevmax*(1. - 1.e-7)) #,num=20)
@@ -460 +463 @@
        zeplot=outputname
 
-    pad_inches_value = 0.35
-    print "********** SAVE ", save
-    if save == 'png': 
-        if display: makeplotres(zeplot,res=100.,pad_inches_value=pad_inches_value) #,erase=True)  ## a miniature
-        makeplotres(zeplot,res=resolution,pad_inches_value=pad_inches_value,disp=False)
-    elif save in ['eps','svg','pdf']:     makeplotres(zeplot,pad_inches_value=pad_inches_value,disp=False,ext=save)
-    elif save == 'gui':                   show()
-    elif save == 'txt':                   print "Saved results in txt file." 
-    else: 
-         print "INFO: save mode not supported. using gui instead."
-         show()
+    if mrate is None:
+        pad_inches_value = 0.35
+        print "********** SAVE ", save
+        if save == 'png': 
+            if display: makeplotres(zeplot,res=100.,pad_inches_value=pad_inches_value) #,erase=True)  ## a miniature
+            makeplotres(zeplot,res=resolution,pad_inches_value=pad_inches_value,disp=False)
+        elif save in ['eps','svg','pdf']:     makeplotres(zeplot,pad_inches_value=pad_inches_value,disp=False,ext=save)
+        elif save == 'gui':                   show()
+        elif save == 'txt':                   print "Saved results in txt file." 
+        else: 
+            print "INFO: save mode not supported. using gui instead."
+            show()
 
     ###############