Changeset 1880 in lmdz_wrf

Timestamp:

Apr 3, 2018, 6:04:51 PM (7 years ago)

Author:

lfita

Message:

Fixing `dimrm'

File:

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

trunk/tools/nc_var_tools.py

@@ -1685 +1685 @@
 
     ncf = NetCDFFile(ncfile,'a')
-    dims = ncf.dimensions
-    ncvars = ncf.variables.keys()
+    dims = list(ncf.dimensions)
+    ncvars = list(ncf.variables.keys())
     ncf.close()
 
-    if not gen.searchInlist(dims, var):
+    if not gen.searchInlist(dims, dimn):
         print '  ' + fname + ": File '" + ncfile + "' does not have dimension: '" +  \
           dimn + "' !!"
@@ -1697 +1697 @@
     if values[0:9] == 'renamedim':
         newdimn = values.split(',')[1]
+        print '  ' +fname+ ": renaming dimension '" + dimn + "' to '" + newdimn + "'"
         ncf = NetCDFFile(ncfile,'a')
         if gen.searchInlist(dims,newdimn):
@@ -1719 +1720 @@
             ncf.sync()
     elif values[0:9] == 'removevar':
+        print '  ' +fname+ ": rmoving dimension '" + dimn + "' and variables with it"
         ncf = NetCDFFile(ncfile,'r')
         tmpncf = NetCDFFile('tmp_py.nc' , 'w')
@@ -1725 +1727 @@
 
         for varn in ncf.variables.keys():
-            vdimn = varn.dimensions
+            ovn = ncf.variables[varn]
+            vdimn = ovn.dimensions
             if not gen.searchInlist(vdimn,dimn):
                 ncf.close()
                 fvaradd(ncfile + ',' + varn, 'tmp_py.nc')
                 ncf = NetCDFFile(ncfile,'r')
+            else:
+                print '    removing variable:', varn
         ncf.close()
         fgaddattr(ncfile, 'tmp_py.nc')