Changeset 1231 in lmdz_wrf for trunk/tools/drawing.py

Timestamp:

Oct 25, 2016, 7:56:03 PM (8 years ago)

Author:

lfita

Message:

Generlizing, improving and adding to the web `2D_shadow_line'

File:

• trunk/tools/drawing.py (modified) (9 diffs)

trunk/tools/drawing.py

@@ -21 +21 @@
 ## e.g. # drawing.py -o draw_2D_shad_cont -f wrfout_d01_2001-11-11_00:00:00 -S 'huss,tas:west_east|-1,south_north|-1,Time|2:Time|2:XLONG:XLAT:auto:Blues,auto,auto:fixc,r:%3g:Srange,Srange:260,300,9:huss|&|tas|at|2001-11-11|02|UTC:png:None:cyl,c:True' -v Q2,T2 
 ## e.g. # drawing.py -o draw_2D_shad_cont_time -f wrfout_d01_2001-11-11_00:00:00 -S 'hfls,tas;west_east|-1,south_north|27,Time|-1;south_north|27;XLONG;WRFtime;auto;BuPu,auto,auto;fixc,y;%3g;Srange,Srange;260,300,9;huss|&|tas|evolution|at|south_north=27;png;None;WRFtime|hours!since!1949-12-01|exct,1,h|$%d^{%H}$|date!($[DD]^{[HH]}$);True' -v LH,T2
+## e.g. # drawing.py -o draw_2D_shad_line -f wrfout_d01_2001-11-11_00:00:00,wrfout_d01_2001-11-11_00:00:00 -S 'hus,hgt:west_east|-1,south_north|96,Time|2,bottom_top|-1:XLONG:ZNU:auto:rainbow,auto,horizontal:Srange,Srange:k,0.,4000.,auto,auto,auto,45.:vert.|sec.|hus|at|y=96|on|2001-11-11|02|UTC:png:flip@y:None:True' -v QVAPOR,HGT
 ## e.g. # drawing.py -f ~/etudes/domains/MEDCORDEX/geo_em.d01.nc -o draw_2D_shad_cont -S 'height,landmask:Time|0:Time|0:XLONG_M:XLAT_M:terrain:fixc,k:None:0.,3000.:0,1,10:MEDCORDEX height & landmask:pdf:False:lcc,i' -v HGT_M,LANDMASK
 ## e.g. # drawing.py -o draw_2D_shad_line -f 'mean_dtcon-pluc-pres_lat.nc,mean_dtcon-pluc-pres_lat.nc' -S 'dtcon,prc:bottom_top|-1,south_north|-1:latmean:presmean:seismic,k:-5.,5.:monthly|dtcon|&|prc:pdf:flip@y:None:True' -v 'dtconmean,prcmean'
@@ -774 +775 @@
     """ plotting two fields, one with shading and the other with contour lines being
     one of the dimensions of time characteristics
-    draw_2D_shad_cont(ncfile, values, varn)
+    draw_2D_shad_cont_time(ncfile, values, varn)
       ncfile= [ncfilevars],[ncfilevarc] files to use (one value, same file)
       values=[vnamefs];[dimvals];[dimvalc];[dimxvn];[dimyvn];[dimxyf];[colorbarvals];[ckind];[clabfmt];[sminv],[smaxv];[sminc],[smaxv],[Nlev];[figt];[kindfig];[reverse];[timevals];[close]
@@ -1061 +1062 @@
     draw_2D_shad_line(ncfile, values, varn)
       ncfile= [ncfiles],[ncfilel] file to use for the shading and for the line
-      values=[vnamefs],[vnamefl]:[dimvals]:[dimxvn]:[dimyvn]:[colorbar],[colline]:[sminv],[smaxv]:[figt]:
+      values=[vnamefs],[vnamefl]:[dimvals]:[dimxvn]:[dimyvn]:[dimxyfmt]:[colorbarvals]:[smin/axv]:[linevalues]:[figt]:
        [kindfig]:[reverse]:[mapv]:[close]
         [vnamefs]: Name in the figure of the variable to be shaded
@@ -1068 +1069 @@
           variable a given value is required (-1, all the length)
         [dimx/yvn]: name of the variables with the values of the final dimensions (x,y)
-        [colorbar]: name of the color bar
-        [colline]: name of the color for the line
+        [dimxyfmt]=[dxs],[dxf],[Ndx],[ordx],[dys],[dyf],[Ndy],[ordx]: format of the values at each axis (or 'auto')
+          [dxs]: style of x-axis ('auto' for 'pretty')
+            'Nfix', values computed at even 'Ndx'
+            'Vfix', values computed at even 'Ndx' increments
+            'pretty', values computed following aprox. 'Ndx' at 'pretty' intervals (2.,2.5,4,5,10)
+          [dxf]: format of the labels at the x-axis ('auto' for '%5g')
+          [Ndx]: Number of ticks at the x-axis ('auto' for 5)
+          [ordx]: angle of orientation of ticks at the x-axis ('auto' for horizontal)
+          [dys]: style of y-axis ('auto' for 'pretty')
+          [dyf]: format of the labels at the y-axis ('auto' for '%5g')
+          [Ndy]: Number of ticks at the y-axis ('auto' for 5)
+          [ordy]: angle of orientation of ticks at the y-axis ('auto' for horizontal)
+        [colorbarvals]=[colbarn],[fmtcolorbar],[orientation]
+          [colorbarn]: name of the color bar
+          [fmtcolorbar]: format of the numbers in the color bar 'C'-like ('auto' for %6g)
+          [orientation]: orientation of the colorbar ('vertical' (default, by 'auto'), 'horizontal')
+        [smin/axv]: minimum and maximum value for the shading or:
+          'Srange': for full range
+          'Saroundmean@val': for mean-xtrm,mean+xtrm where xtrm = np.min(mean-min@val,max@val-mean)
+          'Saroundminmax@val': for min*val,max*val
+          'Saroundpercentile@val': for median-xtrm,median+xtrm where xtrm = np.min(median-percentile_(val),
+            percentile_(100-val)-median)
+          'Smean@val': for -xtrm,xtrm where xtrm = np.min(mean-min*@val,max*@val-mean)
+          'Smedian@val': for -xtrm,xtrm where xtrm = np.min(median-min@val,max@val-median)
+          'Spercentile@val': for -xtrm,xtrm where xtrm = np.min(median-percentile_(val),
+             percentile_(100-val)-median)
+        [linevalues]=[colline],[sminl],[smaxl],[dls],[dlf],[Ndl],[ordl]
+          [colline]: name of the color for the line
+          [smin/axv]: minimum and maximum value for the line or:
+            'Srange': for full range
+             'Saroundmean@val': for mean-xtrm,mean+xtrm where xtrm = np.min(mean-min@val,max@val-mean)
+            'Saroundminmax@val': for min*val,max*val
+            'Saroundpercentile@val': for median-xtrm,median+xtrm where xtrm = np.min(median-percentile_(val),
+              percentile_(100-val)-median)
+            'Smean@val': for -xtrm,xtrm where xtrm = np.min(mean-min*@val,max*@val-mean)
+            'Smedian@val': for -xtrm,xtrm where xtrm = np.min(median-min@val,max@val-median)
+            'Spercentile@val': for -xtrm,xtrm where xtrm = np.min(median-percentile_(val),
+               percentile_(100-val)-median)
+          [dls]: style of line-axis ('auto' for 'pretty')
+          [dlf]: format of the labels at the line-axis ('auto' for '%5g')
+          [Ndl]: Number of ticks at the line-axis ('auto' for 5)
+          [ordl]: angle of orientation of ticks at the line-axis ('auto' for horizontal)
         [smin/axv]: minimum and maximum value for the shading
         [figt]: title of the figure ('|' for spaces)
@@ -1087 +1128 @@
             * 'h', high
             * 'f', full
+        [close]: Whether figure should be finished or not
       valules= 'rh:z|-1,x|-1:z|-1,x|-1:lat:pressure:BuPu:0.,100.:rh:pdf:flip@y:None'
       varn= [varsn],[varnl] name of the variable to plot with shading and with line
@@ -1098 +1140 @@
 
     farguments = '[vnamefs],[vnamefl]:[dimvals]:[dimxvn]:[dimyvn]:' +                \
-      '[colorbar],[colline]:[sminv],[smaxv]:[figt]:[kindfig]:[reverse]:' +           \
-      '[mapv]:[close]'
+      '[dimxyfmt]:[colorbarvals]:[smin/axv]:[linevalues]:[figt]:[kindfig]:' +        \
+      '[reverse]:[mapv]:[close]'
     drw.check_arguments(fname,values,farguments,':')
 
@@ -1106 +1148 @@
     vdimxn = values.split(':')[2]
     vdimyn = values.split(':')[3]
-    colbarn = values.split(':')[4].split(',')[0]
-    shadminmax = values.split(':')[5]
-    figtitle = values.split(':')[6].replace('|',' ')
-    figkind = values.split(':')[7]
-    revals = values.split(':')[8]
-    mapvalue = values.split(':')[9]
-#    varn = values.split(':')[10]
+    dimxyf = values.split(':')[4]
+    colorbarvals = values.split(':')[5]
+    shadminmax = values.split(':')[6]
+    linevalues = values.split(':')[7]
+    figtitle = values.split(':')[8].replace('|',' ')
+    figkind = values.split(':')[9]
+    revals = values.split(':')[10]
+    mapvalue = values.split(':')[11]
+    close = gen.Str_Bool(values.split(':')[12])
 
     ncfiles = ncfile.split(',')[0]
@@ -1137 +1181 @@
 
 # Dimensions names
-##    print fname + ' obj dimnames: ', objvars.dimensions, dimvals, len(dimvals.split(','))
-##    dimnamesv = []
-##    for idd in range(len(objvars.dimensions)):
-##        cutdim = False
-##        for idc in range(len(dimvals.split(','))):
-##            dimcutn = dimvals.split(',')[idc].split(':')[0]
-##            print objvars.dimensions[idd], dimcutn
-##            if objvars.dimensions[idd] == dimcutn: 
-##                cutdim = True
-##                break
-##        if not cutdim: dimnamesv.append(objvars.dimensions[idd])
     dimnamesv = [vdimxn, vdimyn]
 
@@ -1165 +1198 @@
     objdimx = objsf.variables[vdimxn]
     objdimy = objsf.variables[vdimyn]
-    odimxu = objdimx.getncattr('units')
-    odimyu = objdimy.getncattr('units')
-
-    if len(objdimx.shape) <= 2:
-#        odimxv = objdimx[valshad.shape]
-#        odimyv = objdimy[valshad.shape]
-        odimxv = objdimx[:]
-        odimyv = objdimy[:]
-
-    elif len(objdimx.shape) == 3:
-#        dimcut = [0, slice(0,valshad.shape[0]), slice(0,valshad.shape[1])]
-#        odimxv = objdimx[tuple(dimcut)]
-#        odimyv = objdimy[tuple(dimcut)]
-        odimxv = objdimx[0,:]
-        odimyv = objdimy[0,:]
-    else:
-        print errormsg
-        print '  ' + fname + ': shape of dimension variable:', objdimx.shape,        \
-          ' not ready!!'
-        quit(-1)
-
-    shading_nx = np.zeros((2), dtype=np.float)
-    shading_nx[0] = np.float(shadminmax.split(',')[0])
-    shading_nx[1] = np.float(shadminmax.split(',')[1])
-
-    if mapvalue == 'None': mapvalue = None
+    if drw.searchInlist(objdimx.ncattrs(),'units'):
+        odimxu = objdimx.getncattr('units')
+    else:
+        print warnmsg
+        print '  ' + fname + ": variable dimension '" + vdimxn + "' without units!!"
+        odimxu = '-'
+
+    if drw.searchInlist(objdimy.ncattrs(),'units'):
+        odimyu = objdimy.getncattr('units')
+    else:
+        print warnmsg
+        print '  ' + fname + ": variable dimension '" + vdimyn + "' without units!!"
+        odimyu = '-'
+
+    odimxv, odimyv = drw.dxdy_lonlatDIMS(objdimx[:], objdimy[:], objdimx.dimensions,     \
+      objdimy.dimensions, dimvals.replace(':','|').split(','))
+
+    shading_nx = []
+    if shadminmax.split(',')[0][0:1] != 'S':
+            shading_nx.append(np.float(shadminmax.split(',')[0]))
+    else:
+        shading_nx.append(shadminmax.split(',')[0])
+
+    if shadminmax.split(',')[1][0:1] != 'S':
+        shading_nx.append(np.float(shadminmax.split(',')[1]))
+    else:
+        shading_nx.append(shadminmax.split(',')[1])
 
 # line plot
 ##
+    linearg = linevalues.split(',')
+    if linevalues.split(',')[1][0:1] != 'S':
+        linearg[1] = np.float(linevalues.split(',')[1])
+    if linevalues.split(',')[2][0:1] != 'S':
+        linearg[2] = np.float(linevalues.split(',')[2])
+    if linearg[3] == 'auto': linearg[3] = 'pretty'
+    if linearg[4] == 'auto': linearg[4] = '5g'
+    if linearg[5] == 'auto': linearg[5] = 5
+    if linearg[6] == 'auto': linearg[6] = 0.
+
     ncfilel = ncfile.split(',')[1]
     vnamelfig = values.split(':')[0].split(',')[1]
     varnl = varn.split(',')[1]
-    colline = values.split(':')[4].split(',')[1]
-
+
+    if not os.path.isfile(ncfilel):
+        print errormsg
+        print '  ' + fname + ': file for line "' + ncfilel + '" does not exist !!'
+        quit(-1)
     objlf = NetCDFFile(ncfilel,'r')
+
+    if  not objlf.variables.has_key(varnl):
+        print errormsg
+        print '  ' + fname + ': line file "' + ncfilel +                            \
+          '" does not have variable "' +  varnl + '" !!'
+        quit(-1)
     objlvar = objlf.variables[varnl]
-
-    linevals = objlvar[:]
+    linevals, dimsline = drw.slice_variable(objlvar, dimvals.replace(',','|'))
     varlunits = objlvar.units
 
+    if mapvalue == 'None': mapvalue = None
+
+    colbarn, fmtcolbar, colbaror = drw.colorbar_vals(colorbarvals,',')
+    colormapv = [colbarn, fmtcolbar, colbaror]
+
+    xstyl, xaxf, Nxax, xaxor, ystyl, yaxf, Nyax, yaxor = drw.format_axes(dimxyf,',')
+    xaxis = [xstyl, xaxf, Nxax, xaxor]
+    yaxis = [ystyl, yaxf, Nyax, yaxor]
+
+    if revals == 'None':
+        revals = None
+
     drw.plot_2D_shadow_line(valshad, linevals, vnamesfig, vnamelfig, odimxv, odimyv, \
-      odimxu, odimyu, dimnamesv, colbarn, colline, shading_nx, varunits, varlunits,  \
-      figtitle, figkind, revals, mapvalue, True)
+      odimxu, odimyu, dimnamesv, xaxis, yaxis, colormapv, linearg, shading_nx,       \
+      varunits, varlunits, figtitle, figkind, revals, mapvalue, close)
 
     objsf.close()