Changeset 240 for trunk/MESOSCALE_DEV/PLOT/PYTHON/mylib

Timestamp:

Jul 22, 2011, 9:46:21 AM (13 years ago)

Author:

aslmd

Message:

MESOSCALE: corrected a bug in the previous commit in makemeso, works now fine. added anomaly calculations in python interface + better treatment for output; default is now GUI where one can zoom etc... and further explore the plotted field

File:

• trunk/MESOSCALE_DEV/PLOT/PYTHON/mylib/myplot.py (modified) (6 diffs)

trunk/MESOSCALE_DEV/PLOT/PYTHON/mylib/myplot.py

@@ -100 +100 @@
                 wlon = [-60.,60.]
         elif area == "North_Pole":
-                wlat = [60.,90.]
+                wlat = [50.,90.]
                 wlon = [-180.,180.]
         elif area == "Close_North_Pole":
                 wlat = [75.,90.]
+                wlon = [-180.,180.]
+        elif area == "South_Pole":
+                wlat = [-90.,-50.]
+                wlon = [-180.,180.]
+        elif area == "Close_South_Pole":
+                wlat = [-90.,-75.]
                 wlon = [-180.,180.]
         return wlon,wlat
@@ -244 +250 @@
     return [wlon,wlat]
 
-def makeplotpngres (filename,res,pad_inches_value=0.25,folder='',disp=True):
+def makeplotres (filename,res=None,pad_inches_value=0.25,folder='',disp=True,ext='png',erase=False):
     import  matplotlib.pyplot as plt
-    res = int(res)
-    name = filename+"_"+str(res)+".png"
+    from os import system 
+    addstr = ""
+    if res is not None:
+        res = int(res)
+        addstr = "_"+str(res)
+    name = filename+addstr+"."+ext
     if folder != '':      name = folder+'/'+name
     plt.savefig(name,dpi=res,bbox_inches='tight',pad_inches=pad_inches_value)
-    if disp:              display(name)         
+    if disp:              display(name)
+    if ext in ['eps','ps','svg']:   system("tar czvf "+name+".tar.gz "+name+" ; rm -f "+name)
+    if erase:   system("mv "+name+" to_be_erased")              
     return
 
-def makeplotpng (filename,pad_inches_value=0.25,minres=100.,folder='',disp=True):
-    makeplotpngres(filename,minres,     pad_inches_value=pad_inches_value,folder=folder,disp=disp)
-    makeplotpngres(filename,minres+200.,pad_inches_value=pad_inches_value,folder=folder,disp=False)
-    return
-
-def dumpbdy (field,stag=None):
+def dumpbdy (field,n,stag=None):
     nx = len(field[0,:])-1
     ny = len(field[:,0])-1
     if stag == 'U': nx = nx-1
     if stag == 'V': ny = ny-1
-    return field[5:ny-5,5:nx-5]
+    return field[n:ny-n,n:nx-n]
 
 def getcoorddef ( nc ):   
@@ -270 +277 @@
     if typefile in ['mesoapi','meso']:
         [lon2d,lat2d] = getcoord2d(nc)
-        lon2d = dumpbdy(lon2d)
-        lat2d = dumpbdy(lat2d)
+        lon2d = dumpbdy(lon2d,6)
+        lat2d = dumpbdy(lat2d,6)
     elif typefile in ['gcm']:
         [lon2d,lat2d] = getcoord2d(nc,nlat="latitude",nlon="longitude",is1d=True)
@@ -376 +383 @@
     import  numpy                           as np
     import  matplotlib                      as mpl
+    from mymath import max
     meanlon = 0.5*(wlon[0]+wlon[1])
     meanlat = 0.5*(wlat[0]+wlat[1])
     if   wlat[0] >= 80.:   blat =  40. 
-    elif wlat[0] <= -80.:  blat = -40. 
-    else:                  blat = wlat[0]
+    elif wlat[1] <= -80.:  blat = -40. 
+    else:                  blat = max([wlat[0],wlat[1]])
+    print blat
     h = 50.  ## en km
     radius = 3397200.
@@ -482 +491 @@
              "ICETOT":       "%.1e",\
              "TAU_ICE":      "%.2f",\
+             "anomaly":      "%.1f",\
                     }
     if whichvar not in fmtvar:
@@ -499 +509 @@
              "ICETOT":       "YlGnBu",\
              "TAU_ICE":      "Blues",\
+             "anomaly":      "RdBu_r",\
                      }
+#spectral BrBG RdBu_r
     if whichone not in whichcolorb:
         whichone = "def"