Changeset 479 in lmdz_wrf for trunk/tools

Timestamp:

Jun 12, 2015, 11:42:21 AM (10 years ago)

Author:

lfita

Message:

Adding `low/upthres@oper' on 'valmod' tomake an operation according to thresholds

File:

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

trunk/tools/nc_var_tools.py

@@ -870 +870 @@
           lowthres: if [val] < [modval1]; val = [modval2]
           upthres: if [val] > [modval1]; val = [modval2]
+          lowthres@oper: if [val] < [modval1]; val = [oper] (operation as [modval2],[modval3])
+          upthres@oper: if [val] > [modval1]; val = [oper] (operation as [modval2],[modval3])
           potc: [val] ** [modval1]
     """
@@ -897 +899 @@
         varVal2 = np.where(varVal[:] > float(valsS[1]), float(valsS[2]), varVal[:])
         varVal[:] = varVal2
+    elif modins == 'lowthres@oper':
+        if valS[2] == 'sumc': varVal2 = np.where(varVal[:] < float(valsS[1]),        \
+           varVal[:] + float(valsS[2]), varVal[:])
+        elif valS[2] == 'subc': varVal2 = np.where(varVal[:] < float(valsS[1]),      \
+           varVal[:] - float(valsS[2]), varVal[:])
+        elif valS[2] == 'mulc': varVal2 = np.where(varVal[:] < float(valsS[1]),      \
+           varVal[:] * float(valsS[2]), varVal[:])
+        elif valS[2] == 'divc': varVal2 = np.where(varVal[:] < float(valsS[1]),      \
+           varVal[:] / float(valsS[2]), varVal[:])
+        elif valS[2] == 'potc': varVal2 = np.where(varVal[:] < float(valsS[1]),      \
+           varVal[:] ** float(valsS[2]), varVal[:])
+        else:
+            print errormsg
+            print '  ' + fname + ": Operation to modify values '" + modins +         \
+              "' is not defined !!'
+            quit(-1)
+    elif modins == 'upthres@oper':
+        if valS[2] == 'sumc': varVal2 = np.where(varVal[:] > float(valsS[1]),        \
+           varVal[:] + float(valsS[2]), varVal[:])
+        elif valS[2] == 'subc': varVal2 = np.where(varVal[:] > float(valsS[1]),      \
+           varVal[:] - float(valsS[2]), varVal[:])
+        elif valS[2] == 'mulc': varVal2 = np.where(varVal[:] > float(valsS[1]),      \
+           varVal[:] * float(valsS[2]), varVal[:])
+        elif valS[2] == 'divc': varVal2 = np.where(varVal[:] > float(valsS[1]),      \
+           varVal[:] / float(valsS[2]), varVal[:])
+        elif valS[2] == 'potc': varVal2 = np.where(varVal[:] > float(valsS[1]),      \
+           varVal[:] ** float(valsS[2]), varVal[:])
+        else:
+            print errormsg
+            print '  ' + fname + ": Operation to modify values '" + modins +         \
+              "' is not defined !!'
+            quit(-1)
     elif modins == 'potc':
         varVal[:] = varVal[:] ** modval
     else: 
         print errormsg
-        print '  valmodoper: Operation to modify values ' + modins + ' is not defined !!!'
-        print errormsg
+        print '  ' + fname + ": Operation to modify values '" + modins +             \
+          "' is not defined !!'
         quit(-1)