Changeset 1993 in lmdz_wrf for trunk/tools/generic_tools.py

Timestamp:

Jul 31, 2018, 8:22:11 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `auto_val_list': Function to provide a list of values following an 'auto' configuration or a single value

File:

• trunk/tools/generic_tools.py (modified) (2 diffs)

trunk/tools/generic_tools.py

@@ -58 +58 @@
 # ASCII_WIKItab: Function to pass a line to WIKI table format
 # auto_val: Function to provide a value following an 'auto' configuration
+# auto_val_list: Function to provide a list of values following an 'auto' configuration or a single value
 # Capturing: Class to capture the standard output from a function
 # CF_gribequiv: Function to provide the GRIB variable code number from a CF name
@@ -11981 +11982 @@
     return autoval
 
+def auto_val_list(listchk, Nvals, alist):
+    """ Function to provide a list of values following an 'auto' configuration or a single value
+      listchk= list to take (if != 'auto'), value is transformed to the type(aval[0])
+        *NOTE: if it is a single value, output will be Nvals copies of the single value
+      Nvals= number of values to provide. 
+      alist= list to take when it is configured as 'auto'
+      >>> auto_val('auto', 3, drawing_tools.colorsauto)
+      ['#FF0000', '#00FF00', '#0000FF']
+      >>> auto_val('#00FFAA', 3, drawing_tools.colorsauto)
+      ['#00FFAA', '#00FFAA', '#00FFAA']
+      >>> auto_val_list(['2.3','2','3.4'], 3, np.arange(3.))
+      [2.2999999999999998, 2.0, 3.3999999999999999]
+    """
+    fname = 'auto_val_list'
+
+    if type(listchk) == type('S'):
+        if listchk == 'auto':
+            autoval = alist[0:Nvals]
+        else:
+            autoval = []
+            for i in range(Nvals): autoval.append(retype(listchk, type(alist[0])))
+    else:
+        autoval = []
+        for i in range(Nvals): autoval.append(retype(listchk[i], type(alist[0])))
+
+    return autoval
+
+#colorsauto = ['#FF0000', '#00FF00', '#0000FF', '#FF00FF', '#00FFFF', '#FFAA00',      \
+#  '#AA0000', '#00AA00', '#0000AA', '#AA00AA', '#00AAAA', '#AA3200', '#AA0032',       \
+#  '#32AA00', '#3200AA', '#00AA32', '#0032AA', '#32FF00', '#3200FF', '#0032FF',       \
+#  '#FF3200', '#FF0032', '#00FF32', '#323200', '#320032', '#003232']
+#print auto_val_list('auto', 3, colorsauto)
+#print auto_val_list('#00FFAA', 3, colorsauto)
+#print auto_val_list(['2.3','2','3.4'], 3, np.arange(3.))
+
+quit()
+
+
+
+
 def rectangular_spiral(Ntimes, dunits):
     """ Function to provide a rectangular spiral (along x,y-axis values) of values