Changeset 1530 in lmdz_wrf for trunk

Timestamp:

Apr 18, 2017, 11:06:50 PM (8 years ago)

Author:

lfita

Message:

Adding on `file_oper_alongdims':

• Addition of that variables which are not computed in output file
• Tacking into account if variable is masked

File:

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

trunk/tools/nc_var_tools.py

@@ -9464 +9464 @@
     file_oper_alongdims(values, ncfile, varn)
       values= [dimvals],[dimsoper],[opkind],[dimvn]
-        [dimvals]; [dimname1]@[val1]|[dimdname2]@[val2]|[...[dimnameN]@[valN]]
+        [dimvals]; [dimname1]|[val1]:[...[dimnameN]|[valN]]
           [value]; 
             * [integer]: which value of the dimension
@@ -9487 +9487 @@
         quit()
 
-    dimvals = values.split(',')[0].split('@')
+    dimvals = values.split(',')[0].split(':')
     dimsoper = gen.str_list(values.split(',')[1], ':')
     operkind = values.split(',')[2]
@@ -9554 +9554 @@
 
             varslice, varsliceddims = SliceVarDict(ov,dimslice)
-            # This is absolutely weirdm, but if not there is segmentation fault! 
+            # This is absolutely weird, but if I do not make this, there is segmentation fault! 
             if len(varsliceddims) == len(ov.shape):
-                varvals0 = np.zeros(tuple(ov.shape), dtype=np.float)
+                varvals00 = np.zeros(tuple(ov.shape), dtype=np.float)
                 for ix in range(ov.shape[0]):
-                    varvals0[ix,...] = ov[ix,...]
+                    varvals00[ix,...] = ov[ix,...]
+                if type(ov[:]) == type(ma.array((3), dtype=np.float)):
+                    print warnmsg
+                    print '    ' + fname + ': masked variale !!'
+                    varvals0 = ma.masked_array(varvals00, mask=ov[:].mask)
+                else:
+                    varvals0 = varvals00
             else:
                 varvals0 = ov[tuple(varslice)]
@@ -9638 +9644 @@
 
             newattr = set_attribute(newvar, 'orig_dimensions', origdimsS)
+            # Adding original attributes
+            for attrn in ov.ncattrs():
+                if attrn != '_FillValue':
+                    attrv = ov.getncattr(attrn)
+                    newattr = set_attribute(newvar,attrn,attrv)
+
             objnewnc.sync()
-    
+        else:
+            varinf = variable_inf(ov)
+            dimvars = varinf.dimns
+            shapevar = ov.shape
+            idim = 0
+            for dimn in dimvars:
+                if not gen.searchInlist(list(objnewnc.dimensions), dimn):
+                    if objnc.dimensions[dimn].isunlimited:
+                        newdim = objnewnc.createDimension(dimn,None)
+                    else:
+                        newdim = objnewnc.createDimension(dimn,shapevar[idim])
+                idim = idim + 1
+
+            # Adding that variables which are not computed via 'operation'
+            if varinf.FillValue is not None:
+                newvar = objnewnc.createVariable(vn, nctype(varinf.dtype),           \
+                  tuple(varinf.dimns), fill_value=varinf.FillValue)
+            else:
+                newvar = objnewnc.createVariable(vn, nctype(varinf.dtype),           \
+                  tuple(varinf.dimns))
+            newvar[:] = ov[:]
+            for attrn in ov.ncattrs():
+                if attrn != '_FillValue':
+                    attrv = ov.getncattr(attrn)
+                    newattr = set_attribute(newvar,attrn,attrv)
+            objnewnc.sync()
+
+
     # Statistics of the variable dimension along the dimensions operated
     basicoper = ['min', 'max', 'mean', 'mean2', 'sum', 'std', 'var']