Changeset 981 in lmdz_wrf for trunk/tools

Timestamp:

Aug 8, 2016, 6:22:33 PM (8 years ago)

Author:

lfita

Message:

Fixing wrong 'None' attribution of values in `ColorsLinesPointsStyles?'
Adding 'ColorsLinesPointsStyles?' in `plot_lines'

File:

• trunk/tools/drawing_tools.py (modified) (14 diffs)

trunk/tools/drawing_tools.py

@@ -5647 +5647 @@
     return
 
-def plot_lines(vardv, varvv, vaxis, dtit, linesn, vtit, vunit, gtit, gloc, kfig):
+def plot_lines(vardv, varvv, vaxis, dtit, linesn, vtit, vunit, gtit, gloc, cs, ls,   \
+  ps, ws, ss, fv, fign, kfig):
     """ Function to plot a collection of lines
       vardv= list of set of dimension values
@@ -5661 +5662 @@
         5: 'right', 6: 'center left', 7: 'center right', 8: 'lower center',
         9: 'upper center', 10: 'center'
+      cs= list of color names
+      ls= list of style of lines
+      ps= list of style of points
+      ws= list of withs of lines
+      ss= list of size of points
+      fv= frequency of values
+      fign= name of the figure
       kfig= kind of figure
       plot_lines([np.arange(10)], [np.sin(np.arange(10)*np.pi/2.5)], 'y', 'time (s)',      \
-  ['2.5'], 'sin', '-', 'sinus frequency dependency', 'pdf')
+  ['2.5'], 'sin', '-', 'sinus frequency dependency', 'sinus', 'pdf')
     """
     fname = 'plot_lines'
@@ -5673 +5681 @@
 
 # Canging line kinds every 7 lines (end of standard colors)
-    linekinds=['.-','x-','o-']
-    pointkindsauto=['.', ',', 'x', 'o', '*', '+', '<', '|', '_', '>', '1', '8', 's', \
-      'p', 'h', 'D']
-
     Ntraj = len(vardv)
 
-    N7lines = 0
+    cols, lins, pts, lws, pss = ColorsLinesPointsStyles(Ntraj, cs, ls, ps, ws, ss, fv)
 
     xmin = 100000.
@@ -5702 +5706 @@
     if vaxis == 'x':
         for il in range(Ntraj):
-            plt.plot(varvv[il], vardv[il], linekinds[N7lines], label= linesn[il])
-            if il == 6: N7lines = N7lines + 1
+            plt.plot(varvv[il], vardv[il], lins[il], marker=pts[il],                 \
+              linewidth=lws[il], markersize=pss[il], label= linesn[il], color=cols[il])
 
         plt.xlabel(vtit + ' (' + vunit + ')')
@@ -5712 +5716 @@
     else:
         for il in range(Ntraj):
-            plt.plot(vardv[il], varvv[il], linekinds[N7lines], label= linesn[il])
-            if il == 6: N7lines = N7lines + 1
+            plt.plot(vardv[il], varvv[il], lins[il], marker=pts[il],                 \
+              linewidth=lws[il], markersize=pss[il], label= linesn[il], color=cols[il])
 
         plt.xlabel(dtit)
@@ -5721 +5725 @@
         plt.ylim(xmin,xmax)
 
-    figname = 'lines'
     graphtit = gtit.replace('_','\_').replace('&','\&')
 
@@ -5727 +5730 @@
     plt.legend(loc=gloc)
     
-    output_kind(kfig, figname, True)
+    output_kind(kfig, fign, True)
 
     return
@@ -5766 +5769 @@
         Ncols = len(colorsauto)
         for ic in range(Nstyles):
-            iic = np.mod(ic,Ncols) - 1
-            if iic == 0: iic = Ncols - 1
+            imc = int(ic/Ncols)
+            iic = ic - imc * Ncols
             usecolors.append(colorsauto[iic])
     else:
@@ -5786 +5789 @@
         Nklns = len(linekindsauto)
         for il in range(Nstyles):
-            iil = np.mod(il,Nklns) - 1
-            if iil == 0: iil = Nklns - 1
+            iml = int(il/Nklns)
+            iil = il - iml * Nklns
             uselines.append(linekindsauto[iil])
     else:
@@ -5806 +5809 @@
         Nkpts = len(pointkindsauto)
         for ip in range(Nstyles):
-            iip = np.mod(ip,Nkpts) - 1
-            if iip == 0: iip = Nkpts - 1
+            imc = int(ip/Nkpts)
+            iip = ip - imc * Nkpts
             usepoints.append(pointkindsauto[iip])
     else:
@@ -5826 +5829 @@
         Nwlns = len(linewidthsauto)
         for il in range(Nstyles):
-            iil = np.mod(il,Nwlns) - 1
-            if iil == 0: iil = Nwlns - 1
+            iml = int(il/Nwlns)
+            iil = il - iml * Nwlns
             usewlines.append(linewidthsauto[iil])
     else:
@@ -5846 +5849 @@
         Nspts = len(pointsizesauto)
         for ip in range(Nstyles):
-            iip = np.mod(ip,Nspts) - 1
-            if iip == 0: iip = Nspts - 1
+            ipc = int(ip/Nspts)
+            iip = ip - ipc * Nspts
             usespoints.append(pointsizesauto[iip])
     else:
@@ -5884 +5887 @@
 
 #Nstyles = 3
-#colors = ['blue']
+##colors = ['blue']
+#colors = None
 #lines = ['-']
 #points = ['x', '*', 'o']
@@ -5999 +6003 @@
     if vardv == 'h':
         print fname + '_____________________________________________________________'
-        print plot_lines.__doc__
+        print plot_lines_time.__doc__
         quit()