Changeset 1455 in lmdz_wrf for trunk/tools

Timestamp:

Feb 24, 2017, 7:21:27 PM (8 years ago)

Author:

lfita

Message:

Adding

• `slice_createArray': Function to create a variable from a list which contains slicing-variable values

File:

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

trunk/tools/nc_var_tools.py

@@ -121 +121 @@
 # SliceVar: Function to slice a given variable throughout a given list of dimensions
 # SliceVarDict: Function to slice a given variable throughout a dictionary
+# slice_createArray: Function to create a variable from a list which contains slicing-variable values
 # slice_variable: Function to return a slice of a given variable according to values to its dimensions
 # sorttimesmat: Function to sort the time values of a given file
@@ -18984 +18985 @@
     return
 
+def slice_createArray(listslice, typev=np.float):
+    """ Function to create a variable from a list which contains slicing-variable values
+      listslice= list with ordered slice insrtuctions to be used to slice a variable
+      typev= type of the array to create
+    >>> slice_createArray([1, slice(0,3,None)], typev=np.float)
+    [ 0.  0.  0.]
+    >>> slice_createArray([slice(0,20,3), slice(0,3,None)], typev=type(True))
+    [[False False False]
+     [False False False]
+     [False False False]
+     [False False False]
+     [False False False]
+     [False False False]]
+    """
+
+    dimsize = []
+    for slc in listslice:
+        if type(slc) == type(slice(0,1,None)):
+            iv = slc.start
+            ev = slc.stop
+            sv = slc.step
+            if sv is not None:
+                Lslc = (ev - iv)/sv
+            else:
+                Lslc = ev - iv
+        elif type(slc) == int or type(slc) == np.int64:
+            Lslc = 0
+        if Lslc != 0: dimsize.append(Lslc)
+
+    if typev == int:
+        arrayvalues = np.zeros(tuple(dimsize), dtype=int)
+    elif typev == np.float:
+        arrayvalues = np.zeros(tuple(dimsize), dtype=np.float)
+    elif typev == np.float64:
+        arrayvalues = np.zeros(tuple(dimsize), dtype=np.float64)
+    elif typev == type(True):
+        arrayvalues = np.zeros(tuple(dimsize), dtype=bool)
+    else:
+        print errormsg
+        print '  ' + fname + ': type', typev, 'not ready !!'
+        quit(-1)
+
+    return arrayvalues
+
 #quit()