Changeset 2846 in lmdz_wrf for trunk/tools/nc_var_tools.py

Timestamp:

Jul 7, 2020, 6:48:20 AM (4 years ago)

Author:

lfita

Message:

Fixing? mask_var

File:

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

trunk/tools/nc_var_tools.py

@@ -4447 +4447 @@
     attr = newvar.setncattr('mask', 'variable masked using ' + values.split(':')[2]+ \
       ' of variable ' + maskvarn + ' from file ' + maskfilename)
+    attr = set_attributek(newvar, 'mask_value', maskval, 'F')
     newvar[:] = varobj[:]
 
@@ -4476 +4477 @@
 
     #slices = gen.provide_slices(varinf.dimns, varinf.dims, list(varmaskinf.dimns))
-    slices = gen.provide_slices(varinf.dimns, varinf.dims, vdimns)
+    print varinf.dimns, varinf.dims, 'Lluis vdimns', vdimns
+    slices = gen.provide_slices(list(varinf.dimns), varinf.dims, vdimns)
 
     if maskvalue != 'mask':
         maskTOuse = ma.masked_equal(maskvalues, maskval)
+        print 'Lluis maskTOuse', maskTOuse.mask
         maskTOusev = maskTOuse.mask
     else:
         maskTOusev = maskvalues.mask
 
+    print 'Lluis runningsize', runningsize
     for ir in range(runningsize):
         islice = slices[ir]
         varvals = varobj[tuple(islice)]
         maskvarvals = np.where(maskTOusev, fillValue, varvals)
+        print islice, newvar.name, 'Lluis maskvarvals', maskvarvals
         newvar[tuple(islice)] = maskvarvals
+    newvar[:] = np.where(maskvalues == maskval, fillValue, varobj[:])
 
     ncfnew.sync()
+   
+    add_globattrs(ncf,ncfnew,'all')
+    add_global_PyNCplot(ncfnew, 'nc_var_tools', fname, '1.1')
+    
     ncfnew.close()
     ncf.close()