Changeset 817 for trunk/UTIL/PYTHON/myplot.py

Timestamp:

Oct 25, 2012, 11:24:38 PM (12 years ago)

Author:

aslmd

Message:

UTIL PYTHON : improved projections. improved local time ticks. small bug fix with vertical coordinates. added a keyword nocolorb.

File:

• trunk/UTIL/PYTHON/myplot.py (modified) (9 diffs)

trunk/UTIL/PYTHON/myplot.py

@@ -65 +65 @@
 def check_localtime(time):
     a=-1
-    print time
     for i in range(len(time)-1):
        if (time[i] > time[i+1]): a=i
@@ -452 +451 @@
 
 ## Author: AS + TN
-def definesubplot ( numplot, fig ):
+def definesubplot ( numplot, fig, ipreferline=False):
     from matplotlib.pyplot import rcParams
     rcParams['font.size'] = 12. ## default (important for multiple calls)
@@ -470 +469 @@
         subh = 1
         fig.subplots_adjust(hspace = 0.5)
+        if ipreferline: subv = 1 ; subh = 3 ; fig.subplots_adjust(wspace = 0.35)
         rcParams['font.size'] = int( rcParams['font.size'] * 1. / 2. )
     elif numplot == 4:
@@ -883 +883 @@
     h = 50.  ## en km
     radius = 3397200.
-    #print meanlat, meanlon
     if   char == "cyl":     m = Basemap(rsphere=radius,projection='cyl',\
                               llcrnrlat=wlat[0],urcrnrlat=wlat[1],llcrnrlon=wlon[0],urcrnrlon=wlon[1])
@@ -900 +899 @@
     elif char == "merc":    m = Basemap(rsphere=radius,projection='merc',lat_ts=0.,\
                               llcrnrlat=wlat[0],urcrnrlat=wlat[1],llcrnrlon=wlon[0],urcrnrlon=wlon[1])
+    elif char == "geos":    m = Basemap(rsphere=radius,projection='geos',lon_0=meanlon)
+    elif char == "robin":   m = Basemap(rsphere=radius,projection='robin',lon_0=0)
+    elif char == "cass":    
+             if zewidth > 60000.:  ## approx. more than one degree
+                m = Basemap(rsphere=radius,projection='cass',\
+                              lon_0=meanlon,lat_0=meanlat,\
+                              width=zewidth,height=zeheight)
+             else:
+                m = Basemap(rsphere=radius,projection='cass',\
+                              lon_0=meanlon,lat_0=meanlat,\
+                              llcrnrlat=wlat[0],urcrnrlat=wlat[1],llcrnrlon=wlon[0],urcrnrlon=wlon[1])
+    else:                   errormess("projection not supported.")
     fontsizemer = int(mpl.rcParams['font.size']*3./4.)
-    if char in ["cyl","lcc","merc","nsper","laea"]:   step = findstep(wlon)
-    else:                                             step = 10.
-    steplon = step*2.
+    if zewidth > 60000.:
+        if char in ["cyl","lcc","merc","nsper","laea"]:   step = findstep(wlon)
+        else:                                             step = 10.
+        steplon = step*2.
+    else:
+        print "very small domain !"
+        steplon = 0.5
+        step = 0.5
     zecolor ='grey'
     zelinewidth = 1
-    zelatmax = 80
-    if meanlat > 75.: zelatmax = 90. ; step = step/2.
+    zelatmax = 80.
+    if meanlat > 75.:  zelatmax = 90. ; step = step/2. ; steplon = steplon*2.
     # to show gcm grid:
     #zecolor = 'r'
@@ -914 +930 @@
     #steplon = 360./64.
     #zelatmax = 90. - step/3
-    if char not in ["moll"]:
+    if char not in ["moll","robin"]:
         if wlon[1]-wlon[0] < 2.:  ## LOCAL MODE
             m.drawmeridians(np.r_[-1.:1.:0.05], labels=[0,0,0,1], color=zecolor, linewidth=zelinewidth, fontsize=fontsizemer, fmt='%5.2f')
@@ -1041 +1057 @@
              "TPOT":         "%.0f",\
              "TSURF":        "%.0f",\
+             "TSK":          "%.0f",\
              "U_OUT1":       "%.0f",\
              "T_OUT1":       "%.0f",\
@@ -1092 +1109 @@
              "TPOT":         "Paired",\
              "TSURF":        "RdBu_r",\
+             "TSK":          "RdBu_r",\
              "QH2O":         "PuBu",\
              "USTM":         "YlOrRd",\
@@ -1350 +1368 @@
       else: y = lat
       count = count+1
-   if indexvert is None and ((dim0 == 4) or (y is None)):
+   if indexvert is None and len(vert) > 1 and ((dim0 == 4) or (y is None)):
       print "AXIS is vert"
       if vertmode == 0: # vertical axis is as is (GCM grid)