Changeset 2105 in lmdz_wrf

Timestamp:

Aug 27, 2018, 5:49:19 PM (7 years ago)

Author:

lfita

Message:

Adding example for `draw_2Dshad_map'

Location:

trunk/tools

Files:

• drawing.py (modified) (9 diffs)
• drawing_tools.py (modified) (9 diffs)

trunk/tools/drawing.py

@@ -66 +66 @@
 ## e.g. # drawing.py -o draw_stations_map -f stations.inf -S 'snd|x|#FFAAAA|2|8@sfc|x|#AAFFAA|2|8:cyl,l:1.,20.,35.,49.:FPS!Alps!stations:pdf:true'
 ## e.g. # drawing.py -o draw_WRFeta_levels -f 120lev/simin_vars.nc,80lev/simin_vars.nc,50lev/simin_vars.nc,50lev_assigned/simin_vars.nc,38lev/simin_vars.nc -f '120lev,80lev,50lev,50leva,38lev:auto:auto:0|6|2:FPS!Alps!vert!levels:pdf:no:true'
-
+## e.g. # drawing.py -o draw_2Dshad_map -S 'orog;$conv^{ini}$;lat|-1,lon|-1;lon;lat;auto;rainbow#auto#auto#time@minutes!since!1949-12-01!00:00:00|exct,6,h|%d$^{%H}$|date!([DD]$^{[HH]}$);Srange,Srange;convini|Case1|1a;pdf;binary|shadow,NW,auto|-70.,-36,-62.,-30.;cyl,c;yes' -v convini -f UBA_ERA-I_1a_2D.nc
 
 #######
@@ -10632 +10632 @@
         [figt]: title of the figure ('|' for spaces)
         [kindfig]: kind of file for the figure output (ps, eps, png, pdf, ...)
-        [mapfmt] = [cbarmap]|[mapkind] values for the map 
+        [mapfmt] = [cbarmap]|[mapkind]|[lonlatbox] values for the map 
           cbarmap: name of the colorbar for the map
           mapkind: kind of map to use in the plot
             'direct': values are they come
-            'shadow',[pos]: pseudo-shadding from a given location of the sun
+            'shadow',[pos],[enhance]: pseudo-shadding from a given location of the sun
               [pos]: 'N', 'NW'
+              [enhance]: enhance factor for the shading ('auto' for 1./5.)
+          lonlatbox: [lonSW],[latSW],[lonNE],[latNE] to plot only a lon,lat box
+            'full': for the whole domain
         [mapv]: map characteristics: [proj],[res]
           see full documentation: http://matplotlib.org/basemap/
@@ -10712 +10715 @@
 
     mapkind = mapfmt[1]
+    if mapfmt[2] == 'full':
+        lonlatbox = None
+    elif len(mapfmt[2].split(',')) == 4:
+        lonlatbox = []
+        for lL in mapfmt[2].split(','):
+            lonlatbox.append(np.float(lL))
+    else:
+        print errormsg
+        print '  ' + fname + ": lonlatbox value '" + mapfmt[2] + "' not ready !!"
+        print '    for a lonlatbox 4 values are required. Values passed:',           \
+          mapfmt[2].splot(',')
+        quit(-1)
+    
 
     if mapkind == 'direct':
@@ -10718 +10734 @@
         mapv = np.zeros((dy,dx), dtype=np.float)
         sunpos = mapkind.split(',')[1]
+        enhance = gen.auto_val(mapkind.split(',')[2], 1./5.)
         if sunpos == 'N':
             ## Linear shadow from N
@@ -10733 +10750 @@
             print '    available ones:', availsunpos
             quit(-1)
+        xmapv = np.max(mapv)*enhance
+        mapv = np.where(mapv > xmapv, xmapv, mapv)
+        mapv = ma.masked_less(mapv, 0.)
     else:
         print errormsg
@@ -10824 +10844 @@
             cbarlab.append('{:6g}'.format(cpos))
     elif colorbarv[3][0:4] == 'time':
-        print 'Lluis colorbarv', colorbarv
         timevals = colorbarv[3].split('@')[1]
         timeunit = timevals.split('|')[0].replace('!',' ')
@@ -10832 +10851 @@
         cbarpos, cbarlab = drw.CFtimes_plot([nbarv, xbarv], timeunit, timekind, timefmt)
         vnamesfig = timelabel
-        varunits = '-'
+        varunits = None
         fmtcolbar = '%s'
 
@@ -10842 +10861 @@
 
     drw.plot_2Dshad_map(mapv, valshad, mapfmt[0], colormapv, xaxis, yaxis, odimxv,   \
-      odimyv, mapvalue, shading_nx, figtitle, figkind, close)
+      odimyv, lonlatbox, mapvalue, shading_nx, figtitle, figkind, close)
 
     return
@@ -10851 +10870 @@
 #values='orog;$conv^{ini}$;lat|-1,lon|-1;lon;lat;auto;rainbow#auto#auto#' +     \
 #  'time@minutes!since!1949-12-01!00:00:00|exct,12,h|%d$^{%H}$|date!'+          \
-#  '([DD]$^{[HH]}$);Srange,Srange;Case1|1a;pdf;binary|shadow,NW;cyl,c;yes'
+#  '([DD]$^{[HH]}$);Srange,Srange;Case1|1a;pdf;binary|shadow,NW,auto|' +        \
+#  '-70.,-36,-62.,-30.;cyl,c;yes'
 #draw_2Dshad_map(ncfile, values, 'convini', axfig=None, fig=None)
 

trunk/tools/drawing_tools.py

@@ -148 +148 @@
 # date_CFtime:
 # pot_values: Function to modify a seies of values by their potency of 10
-# CFtimes_plot:
+# CFtimes_plot: Function to provide a list of string values from a CF time values in order 
+#   to use them in a plot, according to the series of characteristics. 
 # color_lines:
 # output_kind:
@@ -2416 +2417 @@
         if trefu == 'd': timev = timev - secs0001_1901/(24.*3600.)
         if trefu == 'h': timev = timev - secs0001_1901/(3600.)
-        if trefu == 'm': timev = timev - secs0001_1901/(60.)
+        if trefu == 'i': timev = timev - secs0001_1901/(60.)
         if trefu == 's': timev = timev - secs0001_1901
         if trefu == 'l': timev = timev - secs0001_1901*1000.
@@ -2618 +2619 @@
                 timeout.append(datev)
                 it = it + 1                 
+
         elif tu == 'i':
             datev=firstTdt
@@ -2643 +2645 @@
                 datev = firstTdt + dt.timedelta(seconds=Nunits*60*it)
                 timeout.append(datev)
-                it = it + 1                 
+                it = it + 1        
         elif tu == 's':
             datev=firstTdt
@@ -13465 +13467 @@
 #prop={'size':legvs[1]}, ncol=legvs[2])prop={'size':8})prop={'size':8})prop={'size':8})
 
-def plot_2Dshad_map(mapvals, values, cbarmap, cbarv, xaxv, yaxv, dimxv, dimyv, mapv, \
-  vs, figtitle, figk, close):
+def plot_2Dshad_map(mapvals, values, cbarmap, cbarv, xaxv, yaxv, dimxv, dimyv, lLbox,\
+  mapv, vs, figtitle, figk, close):
     """ Function to plot a map of mask values on top of a map in shadow
       mapvals: 2D values to use for the map
@@ -13483 +13485 @@
       yaxv= list with the y-axis paramteres [style, format, number and orientation]
       dim[x/y]v: values along x/y axes
+      lLbox: whether only a lonlatbox for the map should be used or not
       mapv: values of the map
       vs: range of values in the figure
@@ -13544 +13547 @@
         dy = lat0.shape[0]
 
-        nlon = np.min(lon0)
-        xlon = np.max(lon0)
-        nlat = np.min(lat0)
-        xlat = np.max(lat0)
+        if lLbox is None:
+            nlon = np.min(lon0)
+            xlon = np.max(lon0)
+            nlat = np.min(lat0)
+            xlat = np.max(lat0)
+        else:
+            nlon = lLbox[0]
+            xlon = lLbox[2]
+            nlat = lLbox[1]
+            xlat = lLbox[3]
 
         lon2 = lon0[dy/2,dx/2]
@@ -13580 +13589 @@
     plt.pcolormesh(x, y, values, cmap=plt.get_cmap(cbarv[2]), vmin=vsend[0], vmax=vsend[1])
 
-    print 'Lluis cbarv:', cbarv
-
-    cbar = plt.colorbar(format=cbarv[3],orientation=cbarv[4], ticks=cbarv[5])
+    cbar = plt.colorbar(orientation=cbarv[4], ticks=cbarv[5])
     if cbarv[4] == 'horizontal':
         # From: http://stackoverflow.com/questions/32050030/rotation-of-colorbar-tick-labels-in-matplotlib
         cbar.ax.set_xticklabels(cbarv[6],rotation=90)
     else:
-        cbar.ax.set_xticklabels(cbarv[6])
+        cbar.ax.set_yticklabels(cbarv[6])
 
     if not mapv is None:
@@ -13668 +13675 @@
 
 # units labels
-    cbar.set_label(gen.latex_text(cbarv[0]) + ' (' + units_lunits(cbarv[1]) + ')')
+    if cbarv[1] is not None:
+        cbar.set_label(gen.latex_text(cbarv[0]) + ' (' + units_lunits(cbarv[1]) + ')')
+    else:
+        cbar.set_label(gen.latex_text(cbarv[0]))
     
     plt.title(gen.latex_text(figtitle))