Changeset 918 in lmdz_wrf for trunk

Timestamp:

Jun 21, 2016, 9:50:11 AM (8 years ago)

Author:

lfita

Message:

Working version of `file_oper_alongdims' with the statistics variable

File:

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

trunk/tools/nc_var_tools.py

@@ -9035 +9035 @@
     newNvar = 0
     for vn in varns:
-        print "'" + vn + "' ... .. ."
+        print "    '" + vn + "' ... .. ."
         if not objnc.variables.has_key(vn):
             print errormsg
@@ -9089 +9089 @@
                   tuple(finaldims))
                 newNvar = newNvar + 1
-            print fname + ': Lluis; shapes:', newvar.shape,' finalvarvals:', finalvarvals.shape
             newvar[:] = finalvarvals[:]
 
@@ -9141 +9140 @@
     print '  ' + fname + ': Opering slice of variable dimension _______'
     for vn in dimvn:
-        print "' " + vn + "' ... .. ."
+        print "    '" + vn + "' ... .. ."
         if not objnc.variables.has_key(vn):
             print warnmsg
@@ -9235 +9234 @@
 # Statistics of the variable dimension along the dimensions operated in case 
 #   dimension after operation disapears
-    print '  ' + fname + ': Statisitcs of operated variable dimension _______'
     for vn in dimvn:
-        print "' " + vn + "' ... .. ."
         if not objnc.variables.has_key(vn):
             print warnmsg
@@ -9243 +9240 @@
               '" does not have variable "' + vn + '" skipping it !!'
             continue
-        varslice, varsliceddims = SliceVarDict(ov,dimslice)
-        varvals0 = ov[tuple(varslice)]
-        varvals = np.squeeze(varvals0)
-
-        varname = gen.variables_values(vn)[0]
-        if gen.searchInlist(oldvarattr, 'standard_name'):
-            stdname = ov.getncattr('standard_name')
-        else:
-            stdname = gen.variables_values(vn)[1]
-        
-        if gen.searchInlist(oldvarattr, 'long_name'):
-            lname = ov.getncattr('long_name')
-        else:
-            lname = gen.variables_values(vn)[4].replace('|',' ')
-
-        if gen.searchInlist(oldvarattr, 'units'):
-            uname = ov.getncattr('units')
-        else:
-            uname = gen.variables_values(vn)[5]
-
-        if len(varvals.shape) == 1:
-            # Statistics of the variable dimension along the dimensions operated
-            newvar = objnewnc.createVariable(vn+'stats', 'f4' )
-            newattr = basicvardef(newvar, stdname + 'statistics', lname +            \
-              ' statistics', uname)
-
-            newattr = set_attributek(newvar, 'Nvalues', len(varvals), 'I')
-            for op in basicoper:
-                finalvarv, finalds = operdim(varvals,varsliceddims,dimsoper,op)
-                newattr = set_attributek(newvar, op, finalvarv, 'R')
-            objnewnc.sync()
+        # Do we have to compute this variable?
+        ov = objnc.variables[vn]
+        tocompute = False
+        for vd in ov.dimensions:
+            if gen.searchInlist(dimsoper, vd): tocompute = True
+
+        if tocompute:
+            varslice, varsliceddims = SliceVarDict(ov,dimslice)
+            varvals0 = ov[tuple(varslice)]
+            varvals = np.squeeze(varvals0)
+
+            varname = gen.variables_values(vn)[0]
+            if gen.searchInlist(oldvarattr, 'standard_name'):
+                stdname = ov.getncattr('standard_name')
+            else:
+                stdname = gen.variables_values(vn)[1]
+            
+            if gen.searchInlist(oldvarattr, 'long_name'):
+                lname = ov.getncattr('long_name')
+            else:
+                lname = gen.variables_values(vn)[4].replace('|',' ')
+
+            if gen.searchInlist(oldvarattr, 'units'):
+                uname = ov.getncattr('units')
+            else:
+                uname = gen.variables_values(vn)[5]
+
+            if len(varvals.shape) == 1:
+                print '  ' + fname + ': Statisitcs of operated variable dimension _______'
+                print "    '" + vn + "' ... .. ."
+                # Statistics of the variable dimension along the dimensions operated
+                newvar = objnewnc.createVariable(vn+'stats', 'f4' )
+                newattr = basicvardef(newvar, stdname + 'statistics', lname +            \
+                  ' statistics', uname)
+
+                newattr = set_attributek(newvar, 'Nvalues', len(varvals), 'I')
+                for op in basicoper:
+                    finalvarv, finalds = operdim(varvals,varsliceddims,dimsoper,op)
+                    newattr = set_attributek(newvar, op, finalvarv, 'R')
+                objnewnc.sync()
 
 # Including operation dimensions on the new slice for the variables dimension
@@ -9287 +9293 @@
     print '  ' + fname + ': Adding slice of variable dimension _______'
     for vn in dimvn:
-        print "' " + vn + "' ... .. ."
+        print "    '" + vn + "' ... .. ."
         if not objnc.variables.has_key(vn):
             print warnmsg
@@ -9306 +9312 @@
             varvals0 = ov[tuple(varslice)]
             varvals = np.squeeze(varvals0)
-
-            print 'Lluis: variables to add:', varsliceddims
 
             for idv in range(len(varvals.shape)):