source: trunk/UTIL/PYTHON/planetoplot.py @ 430

Last change on this file since 430 was 430, checked in by acolaitis, 13 years ago


PYTHON

Added... movies !

Example :

pp.py -f wrfout_d01_9999-01-01_07:11:40 --var upward --lat 50 --time 5 -m 0 -M 10 -c Spectral_r --div 20 --ymax 200

will generate a 2D altitude-latitude plot of upward tracer concentration, whereas :

pp.py -f wrfout_d01_9999-01-01_07:11:40 --var upward --lat 50 -m 0 -M 10 -c Spectral_r --div 20 --ymax 200

will generate a movie of the previous plot, along the time axis (default axis for movies for now).

Requirements :

The way the movie maker is written is very light-weight and fast, as no intermediary pictures are saved. The data is buffered and streamed right into an encoder.
I could not make this work with ffmpeg... So you MUST have mencoder installed. It is not on the farm... It is on ciclad, and I know that Aymeric installed it on his machine (at least before getting Ulrich).

Known problems :

I spent litteraly all day making this work, but the option is still basic for now, i.e. it is not yea possible to use -w with movies, or 1D plot movies. (several other stuff are not possible yet, like title specification, axis limits (ylim,xlim), etc... I know how to make it work, I just need some time.

Default name for the output movie is "test.avi" for now.

Tested with LES data without projections. Not test on meso or gcm in mapmode 1.

Details :

A videosink.py
----------------- New class that contains some routines for video stuff. This could either be transformed in a function in an other .py, or other functions could be added to this class to make it more usefull.

M 427 mymath.py
----------------- New routines to convert figures to RGBA data (4-dimensions arrays of pixel data) and figures to PIL images.

M * 428 planetoplot.py
----------------- Added support for movies of 2D plots without overlines

M * 428 myplot.py
----------------- Minor stuff about LES and dumpbdy for W. This could be made in a much cleaner way... maybe later.
  • Property svn:executable set to *
File size: 25.3 KB
Line 
1#######################
2##### PLANETOPLOT #####
3#######################
4
5### A. Spiga     -- LMD -- 06~09/2011 -- General building and mapping capabilities
6### T. Navarro   -- LMD -- 10~11/2011 -- Improved use for GCM and added sections + 1Dplot capabilities
7### A. Colaitis  -- LMD --    11/2011 -- Mostly minor improvements and inter-plot operation capabilities + zrecast interpolation for gcm
8### A. Spiga     -- LMD --    11/2011 -- Extended multivar subplot capabilities + cosmetic changes + general cleaning and tests
9
10def planetoplot (namefiles,\
11           level=0,\
12           vertmode=0,\
13           proj=None,\
14           back=None,\
15           target=None,
16           stride=3,\
17           var=None,\
18           colorb="def",\
19           winds=False,\
20           addchar=None,\
21           interv=[0,1],\
22           vmin=None,\
23           vmax=None,\
24           tile=False,\
25           zoom=None,\
26           display=True,\
27           hole=False,\
28           save="gui",\
29           anomaly=False,\
30           var2=None,\
31           ndiv=10,\
32           mult=1.,\
33           zetitle="fill",\
34           slon=None,\
35           slat=None,\
36           svert=None,\
37           stime=None,\
38           outputname=None,\
39           resolution=200,\
40           ope=None,\
41           fileref=None,\
42           minop=0.,\
43           maxop=0.,\
44           titleref="fill",\
45           invert_y=False,\
46           xaxis=[None,None],\
47           yaxis=[None,None],\
48           ylog=False,\
49           yintegral=False,\
50           blat=None,\
51           tsat=False,\
52           flagnolow=False,\
53           movie_axis=0):
54
55
56    ####################################################################################################################
57    ### Colorbars http://www.scipy.org/Cookbook/Matplotlib/Show_colormaps?action=AttachFile&do=get&target=colormaps3.png
58
59    #################################
60    ### Load librairies and functions
61    from netCDF4 import Dataset
62    from myplot import getcoord2d,define_proj,makeplotres,simplinterv,vectorfield,ptitle,latinterv,getproj,wrfinterv,dumpbdy,\
63                       fmtvar,definecolorvec,defcolorb,getprefix,putpoints,calculate_bounds,errormess,definesubplot,\
64                       zoomset,getcoorddef,getwinddef,whatkindfile,reducefield,bounds,getstralt,getfield,smooth,nolow,\
65                       getname,localtime,polarinterv,getsindex,define_axis,determineplot,readslices,bidimfind,getlschar,hole_bounds
66    from mymath import deg,max,min,mean,get_tsat,writeascii,fig2data,fig2img
67    import matplotlib as mpl
68    from matplotlib.pyplot import contour,contourf, subplot, figure, rcParams, savefig, colorbar, pcolor, show, plot, clabel, title
69    from matplotlib.cm import get_cmap
70    import numpy as np
71    from numpy.core.defchararray import find
72    from videosink import VideoSink
73
74    ################################
75    ### Preliminary stuff
76    ################################
77    print "********************************************"
78    print "********** WELCOME TO PLANETOPLOT **********"
79    print "********************************************"
80    if not isinstance(namefiles, np.ndarray): namefiles = [namefiles]
81    if not isinstance(var, np.ndarray):       var = [var]
82
83    ################################
84    ### Which plot needs to be done?
85    ################################
86    nlon, nlat, nvert, ntime, mapmode, nslices = determineplot(slon, slat, svert, stime)
87    vlon = None ; vlat = None
88    if slon is not None: vlon = slon[0][0]
89    if slat is not None: vlat = slat[0][0]
90    if mapmode == 0:       winds=False
91    elif mapmode == 1:     
92        if svert is None:  svert = readslices(str(level)) ; nvert=1
93    zelen = len(namefiles)*len(var)
94    numplot = zelen*nslices
95    print "********** FILES, SLICES, VARS, TOTAL PLOTS: ", len(namefiles), nslices, len(var), numplot
96    if ope is not None:
97        if fileref is not None:       zelen = zelen + 2
98        elif "var" in ope:            zelen = zelen + 1
99    all_var  = [[]]*zelen ; all_var2  = [[]]*zelen ; all_title = [[]]*zelen ; all_varname = [[]]*zelen ; all_namefile = [[]]*zelen ; all_time = [[]]*zelen
100 
101    #################################################################################################
102    ### Loop over the files + vars initially separated by commas to be plotted on the same figure ###
103    #################################################################################################
104    k = 0 ; firstfile = True
105    for nnn in range(len(namefiles)):
106     for vvv in range(len(var)): 
107
108      print "********** LOOP..... THIS IS SUBPLOT NUMBER.....",k
109
110      ######################
111      ### Load NETCDF object
112      namefile = namefiles[nnn] ; print "********** THE NAMEFILE IS....", namefile
113      nc  = Dataset(namefile)
114
115      ##################################
116      ### Initial checks and definitions
117      ### ... TYPEFILE
118      typefile = whatkindfile(nc)                                 
119      if typefile in ['mesoideal']:   mapmode=0;winds=False
120      if firstfile: print "********** MAPMODE: ", mapmode
121      if firstfile:                 typefile0 = typefile
122      elif typefile != typefile0:   errormess("Not the same kind of files !", [typefile0, typefile])
123      ### ... VAR
124      varname=var[vvv]
125      print "********** THE VAR IS....",varname, var2
126      if varname not in nc.variables: varname = False
127      ### ... WINDS
128      if winds:                                                   
129         [uchar,vchar,metwind] = getwinddef(nc)             
130         if uchar == 'not found': winds = False
131      if not varname and not winds: errormess("please set at least winds or var",printvar=nc.variables)
132      ### ... COORDINATES, could be moved below
133      [lon2d,lat2d] = getcoorddef(nc)
134      ### ... PROJECTION
135      if ((proj == None) and (typefile not in ['mesoideal'])):   proj = getproj(nc)                 
136
137##########################################################
138      if typefile == "gcm":
139          lat = nc.variables["latitude"][:] ; lon = nc.variables["longitude"][:]
140          if "Time" in nc.variables:      time = nc.variables["Time"][:]
141          elif "time" in nc.variables:    time = nc.variables["time"][:]
142          else:                           errormess("no time axis found.")
143          vert = nc.variables["altitude"][:]
144      elif typefile in ['meso','mesoapi','geo','mesoideal']:
145          if vlon is not None or vlat is not None:   indices = bidimfind(lon2d,lat2d,vlon,vlat) ; print '********** INDICES: ', indices
146          if slon is not None: slon[0][0] = indices[0] ; slon[0][1] = indices[0]
147          if slat is not None: slat[0][0] = indices[1] ; slat[0][1] = indices[1]
148          if varname in ['PHTOT','W']:    vertdim='BOTTOM-TOP_PATCH_END_STAG'
149          else:                           vertdim='BOTTOM-TOP_PATCH_END_UNSTAG'
150          if varname in ['V']:  latdim='SOUTH-NORTH_PATCH_END_STAG'
151          else:                 latdim='SOUTH-NORTH_PATCH_END_UNSTAG'
152          if varname in ['U']:  londim='WEST-EAST_PATCH_END_STAG'
153          else:                 londim='WEST-EAST_PATCH_END_UNSTAG'
154          lon = np.arange(0,getattr(nc,londim),1) ; lat = np.arange(0,getattr(nc,latdim),1)
155          time = np.arange(0,len(nc.variables["Times"]),1)
156          if typefile in ['geo']:   vert = [0.] ; stime = readslices(str(0))
157          else:
158              if vertmode is None:  vertmode=0
159              if vertmode == 0:     vert = np.arange(0,getattr(nc,vertdim),1)
160              else:                 vert = nc.variables["vert"][:]
161       #if firstfile:
162       #   lat0 = lat
163       #elif len(lat0) != len(lat):
164       #   errormess("Not the same latitude lengths !", [len(lat0), len(lat)])
165       #elif sum((lat == lat0) == False) != 0:
166       #   errormess("Not the same latitudes !", [lat,lat0])
167       ## Faire d'autre checks sur les compatibilites entre fichiers!!
168##########################################################
169
170      if firstfile:
171         ##########################
172         ### Define plot boundaries
173         ### todo: possible areas in latinterv in argument (ex: "Far_South_Pole")
174         if proj in ["npstere","spstere"]: [wlon,wlat] = polarinterv(lon2d,lat2d)
175         elif proj in ["lcc","laea"]:      [wlon,wlat] = wrfinterv(lon2d,lat2d)
176         else:                             [wlon,wlat] = simplinterv(lon2d,lat2d)
177         if zoom:                          [wlon,wlat] = zoomset(wlon,wlat,zoom) 
178
179      all_varname[k] = varname
180      all_namefile[k] = namefile
181      all_time[k] = time
182      if var2: all_var2[k] = getfield(nc,var2)
183      ##### SPECIFIC
184      if varname in ["temp","t","T_nadir_nit","T_nadir_day","temp_day","temp_night"] and tsat:
185          tt=getfield(nc,varname) ; print "computing Tsat-T, I ASSUME Z-AXIS IS PRESSURE"
186          if type(tt).__name__=='MaskedArray':  tt.set_fill_value([np.NaN]) ; tinput=tt.filled()
187          else:                                 tinput=tt
188          all_var[k]=get_tsat(vert,tinput,zlon=lon,zlat=lat,zalt=vert,ztime=time)
189      else:
190      ##### GENERAL STUFF HERE
191          all_var[k] = getfield(nc,varname)
192      print "********** all_var[k].shape", all_var[k].shape
193      k += 1
194      firstfile = False
195      #### End of for namefile in namefiles
196
197    ##################################
198    ### Operation on files
199    if ope is not None:
200        print ope
201        if "var" not in ope:
202             if len(var) > 1: errormess("for this operation... please set only one var !")
203             if ope in ["-","+"]:
204                if fileref is not None:   all_var[k] = getfield(Dataset(fileref),all_varname[k-1]) ; all_varname[k] = all_varname[k-1] ; all_time[k] = all_time[k-1] ; all_namefile[k] = all_namefile[k-1]
205                else:                     errormess("fileref is missing!") 
206                if ope == "-":     all_var[k+1]= all_var[k-1] - all_var[k]
207                elif ope == "+":   all_var[k+1]= all_var[k-1] + all_var[k]
208                all_varname[k+1] = all_varname[k] ; all_time[k+1] = all_time[k] ; all_namefile[k+1] = all_namefile[k] ; numplot = numplot+2
209             elif ope in ["cat"]:
210                tab = all_var[0];k = 1
211                while k != len(namefiles)-1: 
212                    tab = np.append(tab,all_var[k],axis=0) ; k += 1
213                all_time[0] = np.arange(0,len(tab),1) ### AS: time reference is too simplistic, should be better
214                all_var[0] = np.array(tab) ; numplot = 1
215             else: errormess(ope+" : non-implemented operation. Check pp.py --help")
216        else:
217             if len(namefiles) > 1: errormess("for this operation... please set only one file !") 
218             if len(var) > 2:       errormess("not sure this works for more than 2 vars... please check.")
219             if   ope in ["div_var"]: all_var[k] = all_var[k-2] / all_var[k-1] ; insert = '_div_'
220             elif ope in ["mul_var"]: all_var[k] = all_var[k-2] * all_var[k-1] ; insert = '_mul_'
221             elif ope in ["add_var"]: all_var[k] = all_var[k-2] + all_var[k-1] ; insert = '_add_'
222             elif ope in ["sub_var"]: all_var[k] = all_var[k-2] - all_var[k-1] ; insert = '_sub_'
223             else:                    errormess(ope+" : non-implemented operation. Check pp.py --help")
224             numplot = numplot + 1 ; all_time[k] = all_time[k-1] ; all_namefile[k] = all_namefile[k-1]
225             all_varname[k] = all_varname[k-2] + insert + all_varname[k-1] 
226
227    ##################################
228    ### Open a figure and set subplots
229    fig = figure()
230    subv,subh = definesubplot( numplot, fig ) 
231 
232    #################################
233    ### Time loop for plotting device
234    nplot = 1;error = False
235    print "********************************************"
236    while error is False:
237       print "********** NPLOT", nplot
238     
239       ### General plot settings
240       if nplot > numplot: break
241       if numplot > 1:  subplot(subv,subh,nplot)
242
243       ### Map projection                   
244       if mapmode == 1:
245           m = define_proj(proj,wlon,wlat,back=back,blat=blat)
246           x, y = m(lon2d, lat2d)
247
248       ####################################################################
249       ## get all indexes to be taken into account for this subplot and then reduce field
250       ## We plot 1) all lon slices 2) all lat slices 3) all vert slices 4) all time slices and then go to the next slice
251       indexlon  = getsindex(slon,(nplot-1)%nlon,lon)
252       indexlat  = getsindex(slat,((nplot-1)//nlon)%nlat,lat)
253       indexvert = getsindex(svert,((nplot-1)//(nlon*nlat))%nvert,vert) 
254       if ope is not None:
255           if fileref is not None:      index_f = ((nplot-1)//(nlon*nlat*nvert*ntime))%(len(namefiles)+2)  ## OK only 1 var,  see test in the beginning
256           elif "var" in ope:           index_f = ((nplot-1)//(nlon*nlat*nvert*ntime))%(len(var)+1)        ## OK only 1 file, see test in the beginning
257       else:                            yeah = len(namefiles)*len(var) ; index_f = ((nplot-1)//(nlon*nlat*nvert*ntime))%yeah
258       time = all_time[index_f]
259       if stime is not None:
260           if stime[0][0] < 0:
261               if typefile in ['mesoapi','meso']:
262                   for i in range(len(time)):  time[i] = localtime ( interv[0]+time[i]*interv[1], 0.5*(wlon[0]+wlon[1]) )
263                   print "OK... WORKING WITH LOCAL TIMES"
264               else: errormess("local times not supported for GCM files. not too hard to modify the code though.")
265       if mapmode == 1 and stime is None:   indextime = 1
266       else:                                indextime = getsindex(stime,((nplot-1)//(nlon*nlat*nvert))%ntime,time)
267       ltst = None 
268       if typefile in ['mesoapi','meso'] and indextime is not None:  ltst = localtime ( interv[0]+indextime*interv[1], 0.5*(wlon[0]+wlon[1]) ) 
269       print "********** index lon, lat, vert, time ",indexlon,indexlat,indexvert,indextime
270       ####################################################################
271
272       ticks = ndiv + 1
273
274       #### Contour plot
275       if var2:
276           what_I_plot, error = reducefield(all_var2[index_f], d4=indextime, d1=indexlon, d2=indexlat , d3=indexvert, yint=yintegral, alt=vert)
277           #what_I_plot = what_I_plot*mult
278           if not error:
279              if mapmode == 1:
280                  if typefile in ['mesoapi','meso']:    what_I_plot = dumpbdy(what_I_plot,6)
281              zevmin, zevmax = calculate_bounds(what_I_plot)
282              zelevels = np.linspace(zevmin,zevmax,ticks) #20)
283              if var2 == 'HGT':  zelevels = np.arange(-10000.,30000.,2000.)
284              if mapmode == 0:
285                  if typefile in ['mesoideal']:    what_I_plot = dumpbdy(what_I_plot,0,stag='W')
286                  itime=indextime
287                  if len(what_I_plot.shape) is 3:itime=[0]
288                  what_I_plot, x, y = define_axis(lon,lat,vert,time,indexlon,indexlat,indexvert,\
289                        itime,what_I_plot, len(all_var2[index_f].shape),vertmode)
290              ### If we plot a 2-D field
291              if len(what_I_plot.shape) is 2:
292                  zelevels=[-10.,-8.,-6.,-4.,-2.,0.,2.,4.,6.,8.,10.]
293                  cs = contour(x,y,what_I_plot, zelevels, colors='k', linewidths = 1 ) #0.33 colors='w' )# , alpha=0.5)
294                  clabel(cs,zelevels,
295                     inline=3,
296                     fmt='%1.1f',
297                     fontsize=7)
298              ### If we plot a 1-D field
299              elif len(what_I_plot.shape) is 1:
300                  plot(what_I_plot,x) 
301           else:
302              errormess("There is an error in reducing field !")
303
304       #### Shaded plot
305       varname = all_varname[index_f]
306       if varname:
307           what_I_plot, error = reducefield(all_var[index_f], d4=indextime, d1=indexlon, d2=indexlat , d3=indexvert , yint=yintegral, alt=vert, anomaly=anomaly)
308           what_I_plot = what_I_plot*mult
309           if not error: 
310               fvar = varname
311               if anomaly: fvar = 'anomaly'
312               ##### MAPMODE-SPECIFIC SETTINGS #####
313               if mapmode == 1:
314                   if typefile in ['mesoapi','meso']:    what_I_plot = dumpbdy(what_I_plot,6)
315               elif mapmode == 0:
316                   itime=indextime
317                   if len(what_I_plot.shape) is 3:itime=[0]
318                   what_I_plot, x, y = define_axis(lon,lat,vert,time,indexlon,indexlat,indexvert,\
319                         itime,what_I_plot, len(all_var[index_f].shape),vertmode)
320                   zxmin, zxmax = xaxis ; zymin, zymax = yaxis
321                   if zxmin is not None: mpl.pyplot.xlim(xmin=zxmin)
322                   if zxmax is not None: mpl.pyplot.xlim(xmax=zxmax)
323                   if zymin is not None: mpl.pyplot.ylim(ymin=zymin) 
324                   if zymax is not None: mpl.pyplot.ylim(ymax=zymax)
325                   if invert_y:     lima,limb = mpl.pyplot.ylim() ; mpl.pyplot.ylim(limb,lima)
326                   if ylog:         mpl.pyplot.semilogy()
327               if (fileref is not None) and (index_f is numplot-1):    zevmin, zevmax = calculate_bounds(what_I_plot,vmin=minop,vmax=maxop)
328               else:                                                   zevmin, zevmax = calculate_bounds(what_I_plot,vmin=vmin,vmax=vmax)
329               if colorb in ["def","nobar"]:   palette = get_cmap(name=defcolorb(fvar.upper()))
330               elif (fileref is not None) and (index_f is numplot-1): palette = get_cmap(name="RdBu_r")
331               else:                           palette = get_cmap(name=colorb)
332               ##### simple 2D field and movies of 2D fields
333               if len(what_I_plot.shape) >= 2:
334                 istart=0
335                 if indextime is None:iend=len(time)-1
336                 else:iend=istart
337                 imov=istart
338
339                 while imov <= iend:
340                    what_I_plot_frame=what_I_plot
341                    if len(what_I_plot.shape) is 3:
342                       what_I_plot_frame=what_I_plot[imov,:,:]
343                       print "-> frame ",imov+1
344                    if hole: what_I_plot_frame = hole_bounds(what_I_plot_frame,zevmin,zevmax)
345                    else: what_I_plot_frame = bounds(what_I_plot_frame,zevmin,zevmax)
346                    if flagnolow: what_I_plot_frame = nolow(what_I_plot_frame)
347                    if not tile:
348                        #zelevels = np.linspace(zevmin*(1. + 1.e-7),zevmax*(1. - 1.e-7)) #,num=20)
349                        zelevels = np.linspace(zevmin,zevmax,num=ticks)
350                        if imov is 0:
351                           print np.array(x).shape
352                           print np.array(y).shape
353                           print np.array(what_I_plot_frame).shape
354   
355                        if mapmode == 1:       m.contourf( x, y, what_I_plot_frame, zelevels, cmap = palette)
356                        elif mapmode == 0:     contourf( x, y, what_I_plot_frame, zelevels, cmap = palette)
357                    else:
358                        if mapmode == 1:       m.pcolor( x, y, what_I_plot_frame, cmap = palette, vmin=zevmin, vmax=zevmax )
359                        elif mapmode == 0:     pcolor( x, y, what_I_plot_frame, cmap = palette, vmin=zevmin, vmax=zevmax )
360                    if colorb != 'nobar':       
361                        if (fileref is not None) and (index_f is numplot-1):   daformat = "%.3f"
362                        else:                                                  daformat = fmtvar(fvar.upper())
363                        colorbar( fraction=0.05,pad=0.03,format=daformat,\
364                                  ticks=np.linspace(zevmin,zevmax,num=min([ticks/2+1,20])),extend='neither',spacing='proportional' ) 
365                        #mframe=mpl.pyplot.gci().to_rgba(mpl.pyplot.gci().get_array())
366                        #mframe=mpl.pyplot.imshow()
367                        #mframe=fig2data(mpl.pyplot.gcf())
368                        mframe=fig2img(mpl.pyplot.gcf())
369
370                        if ((len(what_I_plot.shape) is 3) and (imov is 0)):
371                            W,H = mpl.pyplot.gcf().canvas.get_width_height()
372                            rate = 10
373                            video = VideoSink((H,W), "test", rate=rate, byteorder="rgba")
374                        if len(what_I_plot.shape) is 3: 
375                            video.run(mframe)
376                            mpl.pyplot.close()
377
378                        imov=imov+1
379                 if len(what_I_plot.shape) is 3: video.close
380               ##### 1D field
381               elif len(what_I_plot.shape) is 1:
382                 plot(x,what_I_plot)
383                 if save == 'txt':  writeascii(np.transpose(what_I_plot),'profile'+str(nplot)+'.txt')
384
385               #### Other cases: (maybe plot 3-D field one day or movie ??)
386               else:
387                 print "WARNING!!! ",len(what_I_plot.shape),"-D PLOT NOT SUPPORTED !!! dimensions: ",what_I_plot.shape
388                 errormess("Are you sure you did not forget to prescribe a dimension ?")
389           else:
390               errormess("There is an error in reducing field !")
391
392       ### Vector plot --- a adapter
393       if winds:
394           vecx, error = reducefield( getfield(nc,uchar), d4=indextime, d3=indexvert , yint=yintegral , alt=vert)
395           vecy, error = reducefield( getfield(nc,vchar), d4=indextime, d3=indexvert , yint=yintegral , alt=vert)
396           #what_I_plot, error = reducefield(all_var[index_f], d4=indextime, d1=indexlon, d2=indexlat , d3=indexvert )
397           if not error:
398               if typefile in ['mesoapi','meso']:   
399                   [vecx,vecy] = [dumpbdy(vecx,6,stag=uchar), dumpbdy(vecy,6,stag=vchar)]
400                   key = True
401               elif typefile in ['gcm']:           
402                   key = False
403               if metwind:  [vecx,vecy] = m.rotate_vector(vecx, vecy, lon2d, lat2d)
404               if varname == False:       colorvec = definecolorvec(back)
405               else:                  colorvec = definecolorvec(colorb)
406               vectorfield(vecx, vecy,\
407                          x, y, stride=stride, csmooth=2,\
408                          #scale=15., factor=300., color=colorvec, key=key)
409                          scale=20., factor=250., color=colorvec, key=key)
410                                            #200.         ## or csmooth=stride
411   
412       ### Next subplot
413       basename = getname(var=varname,winds=winds,anomaly=anomaly)
414       basename = basename + getstralt(nc,level) 
415       if typefile in ['mesoapi','meso']:
416            if slon is not None: basename = basename + "_lon_" + str(int(lon2d[indices[1],indices[0]]))
417            if slat is not None: basename = basename + "_lat_" + str(int(lat2d[indices[1],indices[0]]))
418            plottitle = basename
419            if addchar: 
420                [addchar,gogol,gogol2] = getlschar ( all_namefile[index_f] )
421                plottitle = plottitle + addchar + "_LT"
422            else:        plottitle = plottitle + "_LT"
423            if ltst is not None: plottitle = plottitle + str(ltst)
424       else:
425            if fileref is not None:
426                if index_f is numplot-1:     plottitle = basename+' '+"fig(1) "+ope+" fig(2)"
427                elif index_f is numplot-2:   plottitle = basename+' '+fileref
428                else:                        plottitle = basename+' '+namefiles[0]#index_f]
429            else:                            plottitle = basename+' '+namefiles[0]#index_f]
430       if mult != 1:                         plottitle = str(mult) + "*" + plottitle
431       if zetitle != "fill":                 
432          plottitle = zetitle
433          if titleref is "fill":             titleref=zetitle
434          if fileref is not None:
435             if index_f is numplot-2:        plottitle = titleref
436             if index_f is numplot-1:        plottitle = "fig(1) "+ope+" fig(2)"
437#       if indexlon is not None:      plottitle = plottitle + " lon: " + str(min(lon[indexlon])) +" "+ str(max(lon[indexlon]))
438#       if indexlat is not None:      plottitle = plottitle + " lat: " + str(min(lat[indexlat])) +" "+ str(max(lat[indexlat]))
439#       if indexvert is not None:     plottitle = plottitle + " vert: " + str(min(vert[indexvert])) +" "+ str(max(vert[indexvert]))
440#       if indextime is not None:     plottitle = plottitle + " time: " + str(min(time[indextime])) +" "+ str(max(time[indextime]))
441       title( plottitle )
442       if nplot >= numplot: error = True
443       nplot += 1
444
445 
446
447
448
449
450
451     
452    ##########################################################################
453    ### Save the figure in a file in the data folder or an user-defined folder
454    if outputname is None:
455       if typefile in ['meso','mesoapi']:   prefix = getprefix(nc)
456       elif typefile in ['gcm']:            prefix = 'LMD_GCM_'
457       else:                                prefix = ''
458    ###
459       zeplot = prefix + basename
460       if addchar:         zeplot = zeplot + addchar
461       if numplot <= 0:    zeplot = zeplot + "_LT"+str(abs(numplot))
462    ###
463       if not target:      zeplot = namefile[0:find(namefile,'wrfout')] + zeplot
464       else:               zeplot = target + "/" + zeplot 
465    ###
466    else:
467       zeplot=outputname
468
469    pad_inches_value = 0.35
470    print "********** SAVE ", save
471    if save == 'png': 
472        if display: makeplotres(zeplot,res=100.,pad_inches_value=pad_inches_value) #,erase=True)  ## a miniature
473        makeplotres(zeplot,res=resolution,pad_inches_value=pad_inches_value,disp=False)
474    elif save in ['eps','svg','pdf']:     makeplotres(zeplot,pad_inches_value=pad_inches_value,disp=False,ext=save)
475    elif save == 'gui':                   show()
476    elif save == 'txt':                   print "Saved results in txt file." 
477    else: 
478         print "INFO: save mode not supported. using gui instead."
479         show()
480
481    ###############
482    ### Now the end
483    return zeplot
Note: See TracBrowser for help on using the repository browser.