Changeset 307 in lmdz_wrf

Timestamp:

Feb 26, 2015, 9:43:15 AM (10 years ago)

Author:

lfita

Message:

Moving some functions to their place

File:

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

trunk/tools/nc_var_tools.py

@@ -704 +704 @@
     return varvals
 
+def numVector_String(vec,char):
+    """ Function to transform a vector of numbers to a single string [char] separated
+    numVector_String(vec,char)
+      vec= vector with the numerical values
+      char= single character to split the values
+    >>> print numVector_String(np.arange(10),' ')
+    0 1 2 3 4 5 6 7 8 9
+    """
+    fname = 'numVector_String'
+
+    if vec == 'h':
+        print fname + '_____________________________________________________________'
+        print numVector_String.__doc__
+        quit()
+
+    Nvals = len(vec)
+
+    string=''
+    for i in range(Nvals):
+        if i == 0:
+            string = str(vec[i])
+        else:
+            string = string + char + str(vec[i])
+
+    return string
+
+def index_mat(mat,val):
+    """ Function to provide the coordinates of a given value inside a matrix
+    index_mat(mat,val)
+      mat= matrix with values
+      val= value to search
+    >>> index_mat(np.arange(27).reshape(3,3,3),22)
+    [2 1 1]
+    """
+
+    fname = 'index_mat'
+
+    matshape = mat.shape
+
+    matlist = list(mat.flatten())
+    ifound = matlist.index(val)
+
+    Ndims = len(matshape)
+    valpos = np.zeros((Ndims), dtype=int)
+    baseprevdims = np.zeros((Ndims), dtype=int)
+
+    for dimid in range(Ndims):
+        baseprevdims[dimid] = np.product(matshape[dimid+1:Ndims])
+        if dimid == 0:
+            alreadyplaced = 0
+        else:
+            alreadyplaced = np.sum(baseprevdims[0:dimid]*valpos[0:dimid])
+        valpos[dimid] = int((ifound - alreadyplaced )/ baseprevdims[dimid])
+
+    return valpos
 
 ####### ###### ##### #### ### ## #
@@ -11252 +11307 @@
 #  '001/full_concatenated.nc', 'PSFC')
 
-def numVector_String(vec,char):
-    """ Function to transform a vector of numbers to a single string [char] separated
-    numVector_String(vec,char)
-      vec= vector with the numerical values
-      char= single character to split the values
-    >>> print numVector_String(np.arange(10),' ')
-    0 1 2 3 4 5 6 7 8 9
-    """
-    fname = 'numVector_String'
-
-    if vec == 'h':
-        print fname + '_____________________________________________________________'
-        print numVector_String.__doc__
-        quit()
-
-    Nvals = len(vec)
-
-    string=''
-    for i in range(Nvals):
-        if i == 0:
-            string = str(vec[i])
-        else:
-            string = string + char + str(vec[i])
-
-    return string
-
 def interpolate_locs(locs,coords,kinterp):
     """ Function to provide interpolate locations on a given axis
@@ -12588 +12617 @@
 
 #netcdf_fold_concatenation('/media/data2/etudes/WRF_LMDZ/WL_HyMeX_HighRes/medic950116,Time', 'wrfout_d03', 'Times,XLONG,XLAT,T2,TH2,PSFC,U10,V10,MAPFAC_M')
-
-def index_mat(mat,val):
-    """ Function to provide the coordinates of a given value inside a matrix
-    index_mat(mat,val)
-      mat= matrix with values
-      val= value to search
-    >>> index_mat(np.arange(27).reshape(3,3,3),22)
-    [2 1 1]
-    """
-
-    fname = 'index_mat'
-
-    matshape = mat.shape
-
-    matlist = list(mat.flatten())
-    ifound = matlist.index(val)
-
-    Ndims = len(matshape)
-    valpos = np.zeros((Ndims), dtype=int)
-    baseprevdims = np.zeros((Ndims), dtype=int)
-
-    for dimid in range(Ndims):
-        baseprevdims[dimid] = np.product(matshape[dimid+1:Ndims])
-        if dimid == 0:
-            alreadyplaced = 0
-        else:
-            alreadyplaced = np.sum(baseprevdims[0:dimid]*valpos[0:dimid])
-        valpos[dimid] = int((ifound - alreadyplaced )/ baseprevdims[dimid])
-
-    return valpos
 
 def netcdf_concatenation(ncfile):