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

Timestamp:

Nov 30, 2011, 5:41:30 PM (13 years ago)

Author:

acolaitis

Message:

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.}

File:

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

trunk/UTIL/PYTHON/planetoplot.py

@@ -50 +50 @@
            blat=None,\
            tsat=False,\
-           flagnolow=False):
+           flagnolow=False,\
+           movie_axis=0):
 
 
@@ -63 +64 @@
                        zoomset,getcoorddef,getwinddef,whatkindfile,reducefield,bounds,getstralt,getfield,smooth,nolow,\
                        getname,localtime,polarinterv,getsindex,define_axis,determineplot,readslices,bidimfind,getlschar,hole_bounds
-    from mymath import deg,max,min,mean,get_tsat,writeascii
+    from mymath import deg,max,min,mean,get_tsat,writeascii,fig2data,fig2img
     import matplotlib as mpl
     from matplotlib.pyplot import contour,contourf, subplot, figure, rcParams, savefig, colorbar, pcolor, show, plot, clabel, title
@@ -69 +70 @@
     import numpy as np
     from numpy.core.defchararray import find
+    from videosink import VideoSink
 
     ################################
@@ -281 +283 @@
               if var2 == 'HGT':  zelevels = np.arange(-10000.,30000.,2000.)
               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]
                   what_I_plot, x, y = define_axis(lon,lat,vert,time,indexlon,indexlat,indexvert,\
-                        indextime,what_I_plot, len(all_var2[index_f].shape),vertmode)
+                        itime,what_I_plot, len(all_var2[index_f].shape),vertmode)
               ### If we plot a 2-D field
               if len(what_I_plot.shape) is 2:
+                  zelevels=[-10.,-8.,-6.,-4.,-2.,0.,2.,4.,6.,8.,10.]
                   cs = contour(x,y,what_I_plot, zelevels, colors='k', linewidths = 1 ) #0.33 colors='w' )# , alpha=0.5)
-#                  clabel(cs,#zelevels[::2],
-#                     inline=3,
-#                     fmt='%1.1e',
-#                     fontsize=7)
+                  clabel(cs,zelevels,
+                     inline=3,
+                     fmt='%1.1f',
+                     fontsize=7)
               ### If we plot a 1-D field
               elif len(what_I_plot.shape) is 1:
@@ -308 +314 @@
                    if typefile in ['mesoapi','meso']:    what_I_plot = dumpbdy(what_I_plot,6)
                elif mapmode == 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,\
-                         indextime,what_I_plot, len(all_var[index_f].shape),vertmode)
+                         itime,what_I_plot, len(all_var[index_f].shape),vertmode)
                    zxmin, zxmax = xaxis ; zymin, zymax = yaxis
                    if zxmin is not None: mpl.pyplot.xlim(xmin=zxmin)
@@ -322 +330 @@
                elif (fileref is not None) and (index_f is numplot-1): palette = get_cmap(name="RdBu_r")
                else:                           palette = get_cmap(name=colorb)
-               ##### 2D field
-               if len(what_I_plot.shape) is 2:
-                 if hole: what_I_plot = hole_bounds(what_I_plot,zevmin,zevmax)
-                 else: what_I_plot = bounds(what_I_plot,zevmin,zevmax)
-                 if flagnolow: what_I_plot = nolow(what_I_plot)
-                 if not tile:
-                     #zelevels = np.linspace(zevmin*(1. + 1.e-7),zevmax*(1. - 1.e-7)) #,num=20)
-                     zelevels = np.linspace(zevmin,zevmax,num=ticks)
-                     print np.array(x).shape
-                     print np.array(y).shape
-                     print np.array(what_I_plot).shape
-
-                     if mapmode == 1:       m.contourf( x, y, what_I_plot, zelevels, cmap = palette)
-                     elif mapmode == 0:     contourf( x, y, what_I_plot, zelevels, cmap = palette)
-                 else:
-                     if mapmode == 1:       m.pcolor( x, y, what_I_plot, cmap = palette, vmin=zevmin, vmax=zevmax )
-                     elif mapmode == 0:     pcolor( x, y, what_I_plot, cmap = palette, vmin=zevmin, vmax=zevmax )
-                 if colorb != 'nobar':        
-                     if (fileref is not None) and (index_f is numplot-1):   daformat = "%.3f"
-                     else:                                                  daformat = fmtvar(fvar.upper())
-                     colorbar( fraction=0.05,pad=0.03,format=daformat,\
-                               ticks=np.linspace(zevmin,zevmax,num=min([ticks/2+1,20])),extend='neither',spacing='proportional' ) 
+               ##### simple 2D field and movies of 2D fields
+               if len(what_I_plot.shape) >= 2:
+                 istart=0
+                 if indextime is None:iend=len(time)-1
+                 else:iend=istart
+                 imov=istart
+
+                 while imov <= iend:
+                    what_I_plot_frame=what_I_plot
+                    if len(what_I_plot.shape) is 3:
+                       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 not tile:
+                        #zelevels = np.linspace(zevmin*(1. + 1.e-7),zevmax*(1. - 1.e-7)) #,num=20)
+                        zelevels = np.linspace(zevmin,zevmax,num=ticks)
+                        if imov is 0:
+                           print np.array(x).shape
+                           print np.array(y).shape
+                           print np.array(what_I_plot_frame).shape
+   
+                        if mapmode == 1:       m.contourf( x, y, what_I_plot_frame, zelevels, cmap = palette)
+                        elif mapmode == 0:     contourf( x, y, what_I_plot_frame, zelevels, cmap = palette)
+                    else:
+                        if mapmode == 1:       m.pcolor( x, y, what_I_plot_frame, cmap = palette, vmin=zevmin, vmax=zevmax )
+                        elif mapmode == 0:     pcolor( x, y, what_I_plot_frame, cmap = palette, vmin=zevmin, vmax=zevmax )
+                    if colorb != 'nobar':        
+                        if (fileref is not None) and (index_f is numplot-1):   daformat = "%.3f"
+                        else:                                                  daformat = fmtvar(fvar.upper())
+                        colorbar( fraction=0.05,pad=0.03,format=daformat,\
+                                  ticks=np.linspace(zevmin,zevmax,num=min([ticks/2+1,20])),extend='neither',spacing='proportional' ) 
+                        #mframe=mpl.pyplot.gci().to_rgba(mpl.pyplot.gci().get_array())
+                        #mframe=mpl.pyplot.imshow()
+                        #mframe=fig2data(mpl.pyplot.gcf())
+                        mframe=fig2img(mpl.pyplot.gcf())
+
+                        if ((len(what_I_plot.shape) is 3) 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.run(mframe)
+                            mpl.pyplot.close()
+
+                        imov=imov+1
+                 if len(what_I_plot.shape) is 3: video.close
                ##### 1D field
                elif len(what_I_plot.shape) is 1: