Changeset 1540 in lmdz_wrf for trunk

Timestamp:

Apr 25, 2017, 9:28:24 PM (8 years ago)

Author:

lfita

Message:

Adding `draw_bar': Plotting a bar char plot with multiple sources

Location:

trunk/tools

Files:

• documentation/plotting/draw_bar.html (added)
• documentation/plotting/gallery.html (modified) (1 diff)
• documentation/plotting/plotting.html (modified) (1 diff)
• drawing.py (modified) (6 diffs)

trunk/tools/documentation/plotting/gallery.html

@@ -18 +18 @@
       <A HREF="2Dshad_line.html"><IMG SRC="figs/hus-hgt_vert-sec_shad-line.png" ALT="2D shadow line" STYLE="float:left;width:200px;height:200px" TARGET="value"></IMG></A>
       <A HREF="2Dshad_line_time.html"><IMG SRC="figs/hus-ps_vert-sec_shad-line_time.png" ALT="2D shadow line time" STYLE="float:left;width:200px;height:200px" TARGET="value"></IMG></A>
+      <A HREF="draw_bar.html"><IMG SRC="figs/bar_rain_CRU-NCEP_norm.png" ALT="bar char plot" STYLE="float:left;width:200px;height:200px" TARGET="value"></IMG></A>
       <A HREF="barbs.html"><IMG SRC="figs/wind_barbs_2001111003_wss.png" ALT="wind-barbs" STYLE="float:left;width:200px;height:200px" TARGET="value"></IMG></A>
       <A HREF="barbs.html"><IMG SRC="figs/wind_barbs_2001111003_uw.png" ALT="wind-barbs" STYLE="float:left;width:200px;height:200px" TARGET="value"></IMG></A>

trunk/tools/documentation/plotting/plotting.html

@@ -14 +14 @@
       <A CLASS="lc" HREF="2Dshad_line.html" TARGET="value">2D shadow line</A><BR>
       <A CLASS="lc" HREF="2Dshad_line_time.html" TARGET="value">2D shadow line time</A><BR>
+      <A CLASS="lc" HREF="draw_bar.html" TARGET="value">bar</A><BR>
       <A CLASS="lc" HREF="barbs.html" TARGET="value">barbs</A><BR>
       <A CLASS="lc" HREF="basins.html" TARGET="value">ORCHIDEE's basins</A><BR>

trunk/tools/drawing.py

@@ -45 +45 @@
 ## e.g. # drawing.py -o draw_river_desc -f river_desc.nc -S 'Y|lat|lat|-1,X|lon|lon|-1:red,green:Blues:cyl,l:ORCDHIEE|rivers:png:0:or_rivers:True' -v Amazon,Parana
 ## e.g. #drawing.py -f forcing_Amazonia.nc,forcing_Amazonia.nc -o draw_2lines_time -S 'tstep:Excts:Excts:exct,5,y;%Y:x:tas,wss:r,b:2.:-,-:None:,;,:WFD!forcing!evolution!at!Amazonia:year:0|12:../LaTeX/figs/forcing_Amazonia_tas-wss:png:yes' -v Tair,Wind
+## e.g. #drawing.py -o draw_bar -f 'Forcing_rainstats_OKstomate_CRUNCEP_spinup_SESA_norm_meanstd_stat2.nc;time_counter,0@12@1;time_counter;rainnorm:Forcing_rainstats_OKstomate_CRUNCEP_spinup_SACZ_norm_meanstd_stat2.nc;time_counter,0@12@1;time_counter;rainnorm' -S 'Stringlist,%s,January@Feburary@March@Arpil@May@June@July@August@September@October@November@December,25,pretty,auto,auto,auto:time!since!1958-01-01:pr:y:seconds:mmd-1:SESA,SACZ:#AA0000,#0000AA:1000000.:normalized!evolution!of!precipitaion!with!CRU-NCEP!forcing:0|10:png:bar_rain_CRU-NCEP_norm:yes'
 
 
@@ -79 +80 @@
 # draw_2lines: Fucntion to plot two lines in different axes (x/x2 or y/y2)
 # draw_2lines_time: Function to plot two time-lines in different axes (x/x2 or y/y2)
+# draw_bar: Function to plot a bar char figure with multiple sources
 # draw_barbs: Function to plot wind barbs
 # draw_basins: Function to plot river basins with their discharge vector and basins id (from 'routing.nc')
@@ -110 +112 @@
   'draw_2D_shad_cont', 'draw_2D_shad_2cont', 'draw_2D_shad_cont_time',               \
   'draw_2D_shad_line',                                                               \
-  'draw_2D_shad_line_time', 'draw_barbs', 'draw_basins',                             \
+  'draw_2D_shad_line_time', 'draw_bar', 'draw_barbs', 'draw_basins',                 \
   'draw_2lines', 'draw_2lines_time', 'draw_lines', 'draw_lines_time',                \
   'draw_Neighbourghood_evol',                                                        \
@@ -6429 +6431 @@
     return
 
+def draw_bar(ncfile, values):
+    """ Function to plot a bar char figure with multiple sources
+      ncfile = ':' list of [filen];[dimslice];[xvarn];[yvarn]
+        [filen]: name of the file
+        [dimslice]: '|' list of [dimn],[dimrange]
+          [dimn]: name of the dimension
+          [dimrange]: range for the slice
+            * [integer]: which value of the dimension
+            * -1: all along the dimension
+            * -9: last value of the dimension
+            * [beg]@[end]@[inc] slice from [beg] to [end] every [inc]
+            * NOTE, no dim name all the dimension size
+        [xvarn]: name of the variable to use for the x-positions
+        [yvarn]: name of the variable to use for the values
+      values = [dimxyfmt]:[xaxislab]:[yaxislab]:[xunits]:[yunits]:[aval]:[labels]:[colors]:[wdth]:[graphtit]:
+          [legvalues]:[kindfig]:[fign]:[close]
+        [dimxyfmt]=[dxs],[dxf],[Ndx],[ordx],[dys],[dyf],[Ndy],[ordy]: format of the values at each 
+            axis (or single 'auto' for 'auto','auto')
+          [dxs]: style of x-axis ('auto' for 'pretty')
+            'Nfix', values computed at even 'Ndx'
+            'Stringlist', a given list of strings
+            '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
+          [Ndx]: Number of ticks at the x-axis (for 'Stringlist', '@' list of labels)
+          [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
+          [Ndy]: Number of ticks at the y-axis
+          [ordy]: angle of orientation of ticks at the y-axis ('auto' for horizontal)
+        [xaxislab]: Label for the x-axis ('!' for spaces)
+        [yaxislab]: Label for the y-axis ('!' for spaces)
+        [xunits]: units of the x-label ('None' for no units)
+        [yunits]: units of the y-label
+        [aval]: which acis should be used for the values ('x' or 'y')
+        [labels]: ',' list of labels for the legend ('None' for no-labels)
+        [colors]: ',' list of colors (single value same color, 'auto' automatic)
+        [wdth]: width of the bars
+        [graphtit]: title of the figure '!' for spaces
+        [legvalues]=[locleg]|[fontsize]: 
+          [locleg]: location of the legend (0, autmoatic)
+            1: 'upper right', 2: 'upper left', 3: 'lower left', 4: 'lower right',
+            5: 'right', 6: 'center left', 7: 'center right', 8: 'lower center',
+            9: 'upper center', 10: 'center'
+          [fontsize]: font size for the legend (auto for 12)
+        [kindfig]: kind of output figure (png, pdf, ps)
+        [fign]: name of the figure (without extension)
+        [close]: whether figure should be close or not
+    """
+    fname = 'draw_bar'
+
+    if values == 'h':
+        print fname + '____________________________________________________________'
+        print draw_bar.__doc__
+        quit()
+
+    expectargs = '[dimxyfmt]:[xaxislab]:[yaxislab]:[aval]:[xunits]:[yunits]:' +      \
+      '[labels]:[colors]:[wdth]:[graphtit]:[legvalues]:[kindfig]:[fign]:[close]'
+    drw.check_arguments(fname,values,expectargs,':')
+
+    dimxyfmt = values.split(':')[0]
+    xaxislab = values.split(':')[1].replace('!',' ')
+    yaxislab = values.split(':')[2].replace('!',' ')
+    aval = values.split(':')[3]
+    xunits = values.split(':')[4]
+    yunits = values.split(':')[5]
+    labels = gen.str_list(values.split(':')[6], ',')
+    colors = gen.str_list(values.split(':')[7], ',')
+    wdth = np.float(values.split(':')[8])
+    graphtit = values.split(':')[9].replace('!',' ')
+    legvalues = values.split(':')[10]
+    kindfig = values.split(':')[11]
+    fign = values.split(':')[12]
+    close = gen.Str_Bool(values.split(':')[13])
+
+    # files and values
+    filedvs = ncfile.split(':')
+    xvals = []
+    yvals = []
+
+    for fdv in filedvs:
+        filen = fdv.split(';')[0]
+        dimrange = fdv.split(';')[1]
+        xvarn = fdv.split(';')[2]
+        yvarn = fdv.split(';')[3]
+
+        if not os.path.isfile(filen):
+            print errormsg
+            print '  ' + fname + ': file "' + filen + '" does not exist !!'
+            quit(-1)
+
+        onc = NetCDFFile(filen, 'r')
+        if not onc.variables.has_key(xvarn):
+            print errormsg
+            print '  ' + fname + ': file "' + filen + "' does not have variable '" + \
+              xvarn + "' for the x-values !!"
+            print '    available ones:', onc.variables.keys()
+            onc.close()
+            quit(-1)
+        if not onc.variables.has_key(yvarn):
+            print errormsg
+            print '  ' + fname + ': file "' + filen + "' does not have variable '" + \
+              yvarn + "' for the y-values !!"
+            print '    available ones:', onc.variables.keys()
+            onc.close()
+            quit(-1)
+
+        oxvar = onc.variables[xvarn]
+        oyvar = onc.variables[yvarn]
+        xvs, xdims = ncvar.slice_variable(oxvar, dimrange.replace(',',':'))
+        yvs, ydims = ncvar.slice_variable(oyvar, dimrange.replace(',',':'))
+        if len(xvs.shape) != 1:
+            print errormsg
+            print '  ' + fname + ': provided slice:', dimrange, "for variable '" +   \
+              xvarn + "' does not provide a rank-1 value !!"
+            print '    resultant shape:', xvs.shape, 'dimensions:', xdims, 'fix it!!'
+            quit(-1)
+        if len(yvs.shape) != 1:
+            print errormsg
+            print '  ' + fname + ': provided slice:', dimrange, "for variable '" +   \
+              yvarn + "' does not provide a rank-1 value !!"
+            print '    resultant shape:', yvs.shape, 'dimensions:', ydims, 'fix it!!'
+            quit(-1)
+        xvals.append(list(xvs))
+        yvals.append(list(yvs))
+
+        onc.close()
+
+    xstyl, xaxf, Nxax, xaxor, ystyl, yaxf, Nyax, yaxor = drw.format_axes(dimxyfmt,',')
+    xaxis = [xstyl, xaxf, Nxax, xaxor]
+    yaxis = [ystyl, yaxf, Nyax, yaxor]
+
+    # Legend 
+    legloc, legsiz = drw.legend_values(legvalues,'|')
+
+    # units
+    if xunits == 'None': dxu = None
+    else: dxu = xunits
+    dyu = yunits
+
+    drw.plot_bars(xvals, yvals, xaxis, yaxis, dxu, dyu, aval, xaxislab, yaxislab,    \
+     labels, colors, wdth, graphtit, legloc, legsiz, kindfig, fign, close)
+
+    return
 
 #quit()
@@ -6450 +6596 @@
 # Not checking file operation
 Notcheckingfile = ['draw_2D_shad_cont', 'draw_2D_shad_2cont',                        \
-   'draw_2D_shad_cont_time',                                                         \
+  'draw_2D_shad_cont_time',                                                          \
   'draw_2D_shad_line', 'draw_2D_shad_line_time', 'draw_2lines', 'draw_2lines_time',  \
-  'draw_lines',                                                                      \
+  'draw_bar', 'draw_lines',                                                          \
   'draw_lines_time', 'draw_points', 'draw_Taylor', 'draw_topo_geogrid_boxes',        \
   'draw_trajectories', 'draw_vals_trajectories', 'variable_values']
@@ -6508 +6654 @@
     elif oper == 'draw_2D_shad_line_time':
         draw_2D_shad_line_time(opts.ncfile, opts.values, opts.varname)
+    elif oper == 'draw_bar':
+        draw_bar(opts.ncfile, opts.values)
     elif oper == 'draw_barbs':
         draw_barbs(opts.ncfile, opts.values, opts.varname)