Changeset 911 in lmdz_wrf for trunk/tools

Timestamp:

Jun 20, 2016, 1:40:59 PM (8 years ago)

Author:

lfita

Message:

Removing spurious prints

File:

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

trunk/tools/generic_tools.py

@@ -41 +41 @@
 # significant_decomposition: Function to decompose a given number by its signifcant potencies
 # singleline_printing_class: Function to print all the values of a given class in a single line to be parseavel
+# stagger_unstagger: Function to de-stagger a variable
 # str_list: Function to obtain a list from a string givin a split character
 # subbasin_point: Function to provide sub-basins given a grid point following a matrix of trips
@@ -463 +464 @@
     if totunits == 'second':
         itime = dt.datetime(int(iyrS),int(imoS),int(idaS),int(ihoS),int(imiS),int(iseS))
-        print 'Lluis ', edate[12:15], '00'
         if edate[12:15] != '00':
             etime = dt.datetime(int(eyrS), int(emoS), int(edaS), int(ehoS), int(emiS), int(eseS)+1)
@@ -628 +628 @@
     CFvar_DIAGvar(varn)
       [varn]= CF name of the variable
-    >>> CFvar_MODvar('pr')
+    >>> CFvar_DIAGvar('pr')
     {'RAINTOT': ['RAINC', 'RAINNC']}
-    >>> CFvar_MODvar('hurs')
+    >>> CFvar_DIAGvar('hurs')
     {'TSrhs': ['psfc', 't', 'q'], 'LMDZrhs': ['psol', 't2m', 'q2m'], 'WRFrhs': ['PSFC', 'T2', 'Q2']}
     """
@@ -6442 +6442 @@
 #                quit(-1)
             if np.mod(iptpos,fracs) == 0: 
-                print 'Lluis syncronizing!'
                 onc.sync()
     if out:
@@ -7866 +7865 @@
     return listv
 
+def stagger_unstagger(varv, dndims, udims):
+    """ Function to de-stagger a variable
+      varv= values of the variable to de-stagger
+      dndims = dn[x/y/z/t] list of the names of the dimenions of the variable      
+      udims = ud[x/y/z/t] unstaggered dictionary with the dimenions of the variable
+
+    >>> mat = np.arange((20), dtype=np.float).reshape(5,4)
+    >>> dmat = ['dystaggered','dx']
+    >>> unstagdims = {'dy': 4, 'dx': 4}
+    >>> stagger_unstagger(mat, dmat, unstagdims)
+    [[  2.   3.   4.   5.]
+     [  6.   7.   8.   9.]
+     [ 10.  11.  12.  13.]
+     [ 14.  15.  16.  17.]]
+    """
+    fname = 'stagger_unstagger'
+
+    udndims = list(udims.keys())
+
+    idim = 0
+    vdstaggered = {}
+    newdims = []
+
+    idim = 0
+    for vd in dndims:
+        idim = dndims.index(vd)
+        if searchInlist(udndims, vd):
+            if varv.shape[idim] - 1 == udims[vd]:
+                vdstaggered[vd] =  varv.shape[idim] - 1
+            else:
+                vdstaggered[vd] = varv.shape[idim]
+        else:
+            vdstaggered[vd] = varv.shape[idim] - 1
+            print '  ' + fname + ": new dimension '" + vd + "' with length :",       \
+              varv.shape[idim], 'as staggered dim'
+        newdims.append(vdstaggered[vd])
+
+    if len(vdstaggered) > 1:
+        destagger_varv = np.zeros(tuple(newdims), dtype=np.float)
+
+        ndim = 0
+        upstagslc = []
+        downstagslc = []
+        for vd in vdstaggered.keys():
+            idim = dndims.index(vd)
+            if vdstaggered[vd] != varv.shape[idim]:
+                upstagslc.append(slice(1,varv.shape[idim]))
+                downstagslc.append(slice(0,varv.shape[idim]-1))
+            else:
+                upstagslc.append(slice(0,varv.shape[idim]))
+                downstagslc.append(slice(0,varv.shape[idim]))
+        destagger_varv = 0.5*(varv[tuple(upstagslc)] + varv[tuple(downstagslc)])
+    
+        return destagger_varv
+    else:
+        return varv
+
 #quit()