Changeset 948 in lmdz_wrf for trunk/tools

Timestamp:

Jun 24, 2016, 1:39:50 PM (8 years ago)

Author:

lfita

Message:

Adding legend font size in 'plot_time_lines'

Location:

trunk/tools

Files:

• drawing.py (modified) (4 diffs)
• drawing_tools.py (modified) (3 diffs)

trunk/tools/drawing.py

@@ -1799 +1799 @@
                 'l': milisecond
            [tfmt]; desired format
-        [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'
+        [legvals]=[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)
         [graphk]: kind of the graphic
         [lines]: ',' list of type of lines, None for automatic, single value all the same
@@ -1841 +1843 @@
     title = values.split(';')[5].replace('|',' ')
     timevals = values.split(';')[6]
-    locleg = int(values.split(';')[7])
+    legvalues = values.split(';')[7]
     graphk = values.split(';')[8]
     lines0 = values.split(';')[9]
@@ -1926 +1928 @@
     timekind = timevals.split('|')[2]
     timefmt = timevals.split('|')[3]
+
+    locleg = int(legvalues.split('|')[0])
+    if legvalues.split('|')[1] == 'auto':
+        legfontsize = 12
+    else:
+        legfontsize = int(legvalues.split('|')[1])
 
 # Getting values
@@ -2093 +2101 @@
 
     drw.plot_lines_time(dimvalues, varvalues, valuesaxis, dimtit, legvals, vartit,   \
-      varunits, timepos, timelabels, title, locleg, graphk, lines, collines, points, \
-      linewidths, pointsizes, pointfreq)
+      varunits, timepos, timelabels, title, locleg, legfontsize, graphk, lines,      \
+      collines, points, linewidths, pointsizes, pointfreq)
 
     return

trunk/tools/drawing_tools.py

@@ -5731 +5731 @@
 
 def plot_lines_time(vardv, varvv, vaxis, dtit, linesn0, vtit, vunit, tpos, tlabs,    \
-  gtit, gloc, kfig, lsl, coll, ptl, lwidth, psize, ptf):
+  gtit, gloc, gsize, kfig, lsl, coll, ptl, lwidth, psize, ptf):
     """ Function to plot a collection of lines with a time axis
       vardv= list of set of dimension values
@@ -5747 +5747 @@
         5: 'right', 6: 'center left', 7: 'center right', 8: 'lower center',
         9: 'upper center', 10: 'center'
+      gsize= fontsize of the legend
       kfig= kind of figure
       lsl= ',' list of line styles
@@ -5866 +5867 @@
     if linesn0 is not None:
         if Ntraj < 10:
-            plt.legend(loc=gloc)
+            plt.legend(loc=gloc, prop={'size':gsize})
         elif 10 < Ntraj < 20:
-            plt.legend(loc=gloc, prop={'size':10})
+            plt.legend(loc=gloc, prop={'size':int(gsize*0.8)})
         else:
-            plt.legend(loc=gloc, prop={'size':8})
+            plt.legend(loc=gloc, prop={'size':int(gsize*0.6)})
 
     print plt.xlim(),':', plt.ylim()