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

Timestamp:

Feb 8, 2017, 7:53:47 PM (8 years ago)

Author:

lfita

Message:

adding `auto_val': Function to provide a value following an 'auto' configuration
adding type('S') to `retype'
adding `rectangular_spiral': Function to provide a rectangular spiral (along x,y-axis values) of values

File:

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

trunk/tools/generic_tools.py

@@ -45 +45 @@
 # ASCII_LaTeX: Function to transform from an ASCII character to LaTeX codification
 # ASCII_to: Function to provide the equivalence text to a given lenguage omitting certain characters
+# auto_val: Function to provide a value following an 'auto' configuration
 # Capturing: Class to capture the standard output from a function
 # CFmonthU_daysU: Function to transform from a CF date series with units as 'months since [DATE]' to 'days since [DATE]'
@@ -105 +106 @@
 # radius_angle: Function to generate a matrix with the angle at a given point
 # radius_dist: Function to generate a matrix with the distance at a given point
+# rectangular_spiral: Function to provide a rectangular spiral (along x,y-axis values) of values
 # replace_list: Function to replace a value in a given list
 # roman_to_int: Convert a roman numeral to an integer
@@ -5087 +5089 @@
         else:
             newval = True
-    elif vtype == '|S1':
+    elif vtype == '|S1' or vtype == type('s'):
         newval = str(val)
     else:
@@ -11258 +11260 @@
     return stringv
 
+def auto_val(valchk, aval):
+    """ Function to provide a value following an 'auto' configuration
+      valchk= value to take (if != 'auto'), value is transformed to the type(aval)
+      aval= value to take when it is configured as 'auto'
+      >>> auto_val('auto', 3.)
+      3.
+      >>> auto_val('2.3456', 3.)
+      2.3456
+    """
+    fname = 'auto_val'
+
+    if valchk == 'auto':
+        autoval = aval
+    else:
+        autoval = retype(valchk, type(aval))
+
+    return autoval
+
+def rectangular_spiral(Ntimes, dunits):
+    """ Function to provide a rectangular spiral (along x,y-axis values) of values
+      Ntimes: number of loops of the spiral (increasing by 1 along each axis
+      dunits: units interval from which assign a value for each spiral value
+    >>> rectangular_spiral(3, 3600.)
+    array([ 0.,  1.,  1.,  0., -1., -1., -1.,  0.,  1.,  2.,  2.,  0., -2., -2., -2.,  
+        0 array([ 0.,  0.,  1.,  1.,  1.,  0., -1., -1., -1.,  0.,  2.,  2.,  2.,
+        0., -2., -2., -2.,  0.,  3.,  3.,  3.,  0., -3., -3., -3.]), [0.0, 3600.0, 
+        7200.0, 10800.0, 14400.0, 18000.0, 21600.0, 25200.0, 28800.0, 32400.0, 36000.0, 
+        39600.0, 43200.0, 46800.0, 50400.0, 54000.0, 57600.0, 61200.0, 64800.0, 68400.0, 
+        72000.0, 75600.0, 79200.0, 82800.0, 86400.0])
+    """
+    fname = 'rectangular_spiral'
+
+    # Wind test values
+    uwindtestv0 = [1., 1., 0., -1., -1., -1., 0., 1. ]
+    vwindtestv0 = [0., 1., 1., 1., 0., -1., -1., -1. ]
+
+    Nvals = len(uwindtestv0)
+
+    uwindtestv = np.zeros((Nvals*Ntimes+1), dtype=np.float)
+    vwindtestv = np.zeros((Nvals*Ntimes+1), dtype=np.float)
+
+    for i in range(Ntimes):
+        uwindtestv[i*Nvals+1:(i+1)*Nvals+1] = np.array(uwindtestv0)*(i+1)
+        vwindtestv[i*Nvals+1:(i+1)*Nvals+1] = np.array(vwindtestv0)*(i+1)
+
+    dimt = len(uwindtestv)
+    twindtestv = []
+    for it in range(dimt):
+        twindtestv.append(dtsecs*it)
+
+    return uwindtestv, vwindtestv, twindtestv
+
 #quit()