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

Timestamp:

Oct 10, 2018, 4:20:23 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `consecutive_list': Function to provide a list of consecutive values
• `Latin_Greek': Function to pass an ASCII latin character to its equivalent greek

File:

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

trunk/tools/generic_tools.py

@@ -84 +84 @@
 # colhex_dec: Function to pas a hexadecimal (#[RR][GG][BB]; 00-64, 0A-FF) color to decimal ([0.,1.])
 # color_deg: Function to generate a degradation of colors in rgb base
+# consecutive_list: Function to provide a list of consecutive values
 # contflow: Function to bring back the increment in j,i grid points according to a trip: (inflow directions)
 # create_LateX_figs: Function to create a LaTeX from a folder with multiple plots from different values
@@ -126 +127 @@
 # latex_fig_array: Function to add an array of figures to an existing tex file
 # latex_text: Function to transform a text to LaTeX following style rules
+# Latin_Greek: Function to pass an ASCII latin character to its equivalent greek
 # linearint_weights: Function to provide the weights for a linear interpolation of a value between a couple of values as a weighted distance mean
 # linearint_3x3weights: Function to provide the weights for a linear interpolation of a value inside a 3x3 matrix of values as a weighted distance mean
@@ -11695 +11697 @@
       raise TypeError, "expected integer, got %s" % type(input)
    if not 0 < input < 4000:
-      raise ValueError, "Argument must be between 1 and 3999"   
+      raise ValueError, "Argument must be between 1 and 3999"
    ints = (1000, 900,  500, 400, 100,  90, 50,  40, 10,  9,   5,  4,   1)
    if romanchar == 'U':
@@ -14370 +14372 @@
 #    print islc, ':', itime, '-', etime, '<>', tvals, '=', tstat[islc]
    
+def Latin_Greek(char):
+    """ Function to pass an ASCII latin character to its equivalent greek
+      char: latin character to pass
+    >>> Latin_Greek('a')
+    $\alpha$
+    >>> Latin_Greek('z')
+    $\omega$
+    """
+    fname = 'Latin_Greek'
+
+    greekequiv = {'a': '$\\alpha$', 'b': '$\\beta$', 'c': '$\\gamma$',               \
+     'd': '$\\delta$', 'e': '$\\epsilon$', 'f': '$\\zeta$', 'g': '$\\eta$',          \
+     'h': '$\\theta$', 'i': '$\\iota$', 'j': '$\\kappa$', 'k': '$\\lambda$',         \
+     'l': '$\\mu$', 'm': '$\\nu $', 'n': '$\\xi$', 'o': '$\\omicron$', 'p': '$\\pi$',\
+     'q': '$\\(san)$', 'r': '$\\(koppa)$', 's': '$\\rho$', 't': '$\\sigma$',         \
+     'u': '$\\tau$', 'v': '$\\upsilon$', 'w': '$\\phi$', 'x': '$\\chi$',             \
+     'y': '$\\psi$', 'z': '$\\omega$'}
+
+    greek = greekequiv[char]
+
+    return greek
+
+def consecutive_list(Nvals, kind):
+    """ Function to provide a list of consecutive values
+      Nvals: number of values
+      kind: kind of values
+        'ALPHA': list of consecutive characters upper case
+        'alpha': list of consecutive characters lower case
+        'greek': list of consecutive greek characters lower case (LaTeX)
+        'numberF': list of consecutive float numbers
+        'numberI': list of consecutive integer numbers
+        'ROMAN': list of consecutive uppercase roman numbers
+        'roman': list of consecutive lowercase roman numbers
+    >>> consecutive_list(10, 'ALPHA')
+    ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J']
+    >>> consecutive_list(10, 'alpha')
+    ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j']
+    >>> consecutive_list(10, 'greek')
+    ['$\\alpha$', '$\\beta$', '$\\gamma$', '$\\delta$', '$\\epsilon$', '$\\zeta$', '$\\eta$', '$\\theta$', '$\\iota$', '$\\kappa$']
+    >>> consecutive_list(10, 'numberI')
+    [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
+    >>> consecutive_list(10, 'roman')
+    ['i', 'ii', 'iii', 'iv', 'v', 'vi', 'vii', 'viii', 'ix', 'x']
+    """
+    fname = 'consecutive_list'
+
+    availkind = ['ALPHA', 'alpha', 'greek', 'numberF', 'numberI', 'ROMAN', 'roman']
+
+    if kind == 'ALPHA':
+        vals = []
+        for ic in range(Nvals):
+            vals.append(chr(65+ic))
+    elif kind == 'alpha':
+        vals = []
+        for ic in range(Nvals):
+            vals.append(chr(97+ic))
+    elif kind == 'greek':
+        vals = []
+        for ic in range(Nvals):
+            vals.append(Latin_Greek(chr(97+ic)))
+    elif kind == 'numberF':
+        vals = list(np.arange(Nvals, drype=np.float))
+    elif kind == 'numberI':
+        vals = range(Nvals)
+    elif kind == 'ROMAN':
+        vals = []
+        for ic in range(Nvals):
+            vals.append(int_to_roman(ic+1, 'U'))
+    elif kind == 'roman':
+        vals = []
+        for ic in range(Nvals):
+            vals.append(int_to_roman(ic+1, 'l'))
+    else:
+        print errormsg
+        print '  ' + fname + ": kind '" + kind + "' not ready !!"
+        print '    avaiable ones:', availkind
+        quit(-1)
+
+    return vals
+
 #quit()
 
+