Changeset 1009 in lmdz_wrf

Timestamp:

Aug 11, 2016, 6:19:51 PM (8 years ago)

Author:

lfita

Message:

Fixing `WRF_toCF', right sort of change dim-names and CF-variables
Fixing `cleaning_varsfile': right variables to keep and only warning when variable is not in original file

File:

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

trunk/tools/nc_var_tools.py

@@ -6 +6 @@
 # Importing generic tools file 'generic_tools.py'
 import generic_tools as gen
+import subprocess as sub
 
 ####### Contents:
@@ -16469 +16470 @@
     refDateTime = values.split(':')[2]
     Tunits = values.split(':')[3]
-
-    # CFing time
-    WRF_CFtime_creation(refDateTime+','+Tunits, ncfile, 'time')
-    
+  
     # CFing lon,lat
     WRF_CFlonlat_creation('lon,lat', ncfile, wrfvl+','+wrfvL)
@@ -16484 +16482 @@
     for dimn in CFdims.keys():
         if gen.searchInlist(ncdims,dimn):
+            print fname + ' Lluis dimn:', dimn, 'CFdims:', CFdims[dimn]
             # Making sure CF dimension is not in the file
             if not gen.searchInlist(ncdims,CFdims[dimn]):
                 newname = ncf.renameDimension(dimn,CFdims[dimn])
+
+    ncf.sync()
+    ncf.close()
+
+    # CFing time
+    WRF_CFtime_creation(refDateTime+','+Tunits, ncfile, 'time')
+
+    ncf = NetCDFFile(ncfile,'a')
 
     # dimension variables which do not need to attach CF attributes, otherwise values
@@ -16541 +16548 @@
 
     oldvars = onc.variables.keys()
-    for oldv in oldvars:
+
+    notinfile = list(set(keepvars).difference(set(oldvars)))
+    if len(notinfile) > 0:
+        print warnmsg
+        print '  ' + fname + ': variables to keep:', notinfile, "are not in file '"+ \
+          filen + "' !!"
+        print '    variables to keep:', keepvars
+        print '    variables in file:', oldvars
+
+    for oldv in keepvars:
         newdims = newonc.dimensions
         newvars = newonc.variables.keys()
 
-        if not gen.searchInlist(newvars, oldv):
+        if not gen.searchInlist(newvars, oldv) and gen.searchInlist(oldvars, oldv):
+            print '  ' + fname + ": keeping variable '" + oldv + "' ..."
             oldov = onc.variables[oldv]
             oldvdim = oldov.dimensions
-            vinf = ncvar.variable_inf(onc,oldv)
-
-            if not gen.searchInlist(newdims,olvdim):
-                odim = onc.dimensions[olvdim]
-                if odim.isunlimited:
-                    newdim = newonc.createDimension(None)
-                else:
-                    newdim = newonc.createDimension(len(odim))
-
-            newvar = newonc.createVariable(oldv, ncvar.nctype(vinf.dtype), vind.dimns)
+            vinf = variable_inf(oldov)
+
+            for oldd in oldvdim:
+                if not gen.searchInlist(newdims,oldd):
+                    odim = onc.dimensions[oldd]
+                    if odim.isunlimited():
+                        newdim = newonc.createDimension(oldd, None)
+                    else:
+                        newdim = newonc.createDimension(oldd, len(odim))
+
+            newvar = newonc.createVariable(oldv, nctype(vinf.dtype), vinf.dimns)
             newvar[:] = oldov[:]
             for attrn in vinf.attributes:
                 attrv = oldov.getncattr(attrn)
-                newattr = ncvar.set_attribute(newvar,attrn,attrv)
+                newattr = set_attribute(newvar,attrn,attrv)
 
         newonc.sync()
@@ -16568 +16586 @@
     # Global attributes
     for gattrn in onc.ncattrs():
-        gattrval = onc.getncattr(gattr)
-        newattr = ncvar.set_attribute(newonc, gattr, gattrval)
+        gattrval = onc.getncattr(gattrn)
+        newattr = set_attribute(newonc, gattrn, gattrval)
 
     newonc.sync()
@@ -16575 +16593 @@
     onc.close()
 
-    sout = sub.call('mv ' + filen + ' ' + ofile, shell=True)
+    sout = sub.call('mv ' + ofile + ' ' + filen, shell=True)
+    print fname + ": succesfully cleaned '" + filen + "' !!"
 
     return
 
-
 #quit()