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

Timestamp:

Dec 1, 2011, 3:58:28 PM (13 years ago)

Author:

acolaitis

Message:

PYTHON. Improved quality of movies for really small weight. Also added an option to force even better quality (option is --quality, weight is also reasonnable but rendering is slower).

File:

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

trunk/UTIL/PYTHON/planetoplot.py

@@ -52 +52 @@
            tsat=False,\
            flagnolow=False,\
-           mrate=None):
+           mrate=None,\
+           mquality=False):
 
 
@@ -73 +74 @@
     from numpy.core.defchararray import find
     from videosink import VideoSink
+    import subprocess
 
     ################################
@@ -153 +155 @@
           if varname in ['PHTOT','W']:    vertdim='BOTTOM-TOP_PATCH_END_STAG'
           else:                           vertdim='BOTTOM-TOP_PATCH_END_UNSTAG'
+          if (var2 is not None and var2 not in ['PHTOT','W']):
+               vertdim='BOTTOM-TOP_PATCH_END_UNSTAG'
+               dumped_vert_stag=True
           if varname in ['V']:  latdim='SOUTH-NORTH_PATCH_END_STAG'
           else:                 latdim='SOUTH-NORTH_PATCH_END_UNSTAG'
@@ -291 +296 @@
                                              yint=yintegral, alt=vert, anomaly=anomaly )
            what_I_plot = what_I_plot*mult
-           #if typefile in ['mesoideal']:    what_I_plot = dumpbdy(what_I_plot,0,stag='W')
+#           if typefile in ['mesoideal']:    what_I_plot = dumpbdy(what_I_plot,0,stag='W')
            if not error: 
                fvar = varname 
@@ -324 +329 @@
                     if mrate is not None:      
                         what_I_plot_frame = what_I_plot[imov,:,:] ; print "-> frame ",imov+1
-                        if mapmode == 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 typefile in ['mesoapi','meso'] and mapmode == 1:   what_I_plot_frame = dumpbdy(what_I_plot_frame,6,condition=True)
+                    if typefile in ['mesoideal']:    what_I_plot_frame = dumpbdy(what_I_plot_frame,0,stag='W',condition=dumped_vert_stag)
+
                     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)
@@ -358 +365 @@
                         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=fig2img(mpl.pyplot.gcf())
+                    figframe=mpl.pyplot.gcf()
+                    if mrate is not None:
+                       if mquality: figframe.set_dpi(600.)
+                       else: figframe.set_dpi(200.)
+                    mframe=fig2img(figframe)
                     if ((mrate is not None) and (imov is 0)):
-                        W,H = mpl.pyplot.gcf().canvas.get_width_height()
-                        video = VideoSink((H,W), "test", rate=mrate, byteorder="rgba")
+                        moviename='test'
+                        W,H = figframe.canvas.get_width_height()
+                        video = VideoSink((H,W), moviename, rate=mrate, byteorder="rgba")
                     if mrate is not None: 
                         video.run(mframe)
                         mpl.pyplot.close()
                     imov=imov+1
-                 if mrate is not None: video.close
+                 if mrate is not None: video.close()
                ##### 1D field
                elif len(what_I_plot.shape) is 1:
@@ -386 +398 @@
            if not error:
                if typefile in ['mesoapi','meso']:    
-                   [vecx,vecy] = [dumpbdy(vecx,6,stag=uchar), dumpbdy(vecy,6,stag=vchar)]
+                   [vecx,vecy] = [dumpbdy(vecx,6,stag=uchar,condition=True), dumpbdy(vecy,6,stag=vchar,condition=True)]
                    key = True
                elif typefile in ['gcm']:            
@@ -469 +481 @@
             show()
 
+    ##################################
+    ### Getting more out of this video
+  
+    if mrate is not None:
+        print "Re-encoding movie.. first pass"
+        video.first_pass(filename=moviename,quality=mquality)
+        print "Re-encoding movie.. second pass"
+        video.second_pass(filename=moviename,quality=mquality)   
+
     ###############
     ### Now the end