Changeset 620 in lmdz_wrf for trunk/tools/drawing_tools.py

Timestamp:

Aug 21, 2015, 3:26:11 PM (10 years ago)

Author:

lfita

Message:

Adding 'points' as paramter on 'draw_lines_time'

File:

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

trunk/tools/drawing_tools.py

@@ -5205 +5205 @@
 
 def plot_lines_time(vardv, varvv, vaxis, dtit, linesn0, vtit, vunit, tpos, tlabs,    \
-  gtit, gloc, kfig, coll):
+  gtit, gloc, kfig, coll, ptl):
     """ Function to plot a collection of lines with a time axis
       vardv= list of set of dimension values
@@ -5223 +5223 @@
       kfig= kind of figure
       coll= ',' list of colors for the lines or None for automatic
+      coll= ',' list of colors for the lines, None for automatic, single
+          value all the same
+      ptl= ',' list of type of points for the lines, None for automatic, single
+          value all the same
+
       plot_lines([np.arange(10)], [np.sin(np.arange(10)*np.pi/2.5)], 'y', 'time (s)',      \
   ['2.5'], 'sin', '-', 'sinus frequency dependency', 'pdf')
@@ -5234 +5239 @@
 
 # Canging line kinds every 7 lines (end of standard colors)
-    linekinds=['.-','x-','o-']
+    linekinds = []
+    if ptl is None:
+        linekindsauto=['.-','x-','o-']
+        for ptype in range(4):
+            for ip in range(7):
+                linekinds.append(linekindsauto[ptype])
+    else:
+        linekinds = ptl
 
     Ntraj = len(vardv)
@@ -5256 +5268 @@
         for il in range(Ntraj):
             if coll is None:
-                plt.plot(varvv[il], vardv[il], linekinds[N7lines], label= linesn[il])
+                plt.plot(varvv[il], vardv[il], linekinds[il], label= linesn[il])
             else:
-                plt.plot(varvv[il], vardv[il], linekinds[N7lines], label= linesn[il],\
+                plt.plot(varvv[il], vardv[il], linekinds[il], label= linesn[il],\
                   color=coll[il])
 
@@ -5270 +5282 @@
             if mindv < xtrmdv[0]: xtrmdv[0] = mindv
             if maxdv > xtrmdv[1]: xtrmdv[1] = maxdv
-
-            if il == 6: N7lines = N7lines + 1
 
         plt.xlabel(vtit + ' (' + vunit + ')')
@@ -5284 +5294 @@
         for il in range(Ntraj):
             if coll is None:
-                plt.plot(vardv[il], varvv[il], linekinds[N7lines], label= linesn[il])
+                plt.plot(vardv[il], varvv[il], linekinds[il], label= linesn[il])
             else:
-                plt.plot(vardv[il], varvv[il], linekinds[N7lines], label= linesn[il],\
+                plt.plot(vardv[il], varvv[il], linekinds[il], label= linesn[il],\
                   color=coll[il])
 
@@ -5298 +5308 @@
             if mindv < xtrmdv[0]: xtrmdv[0] = mindv
             if maxdv > xtrmdv[1]: xtrmdv[1] = maxdv
-
-            if il == 6: N7lines = N7lines + 1
 
         plt.xlabel(dtit)