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

Timestamp:

Jan 15, 2016, 4:02:25 PM (9 years ago)

Author:

lfita

Message:

Fixing some errors

File:

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

trunk/tools/nc_var_tools.py

@@ -17172 +17172 @@
     fname = 'varDimension'
 
-    if not oncf.dimensions._has_key(dname):
+    if not oncf.dimensions.has_key(dname):
         print errormsg
         print '  ' + fname + ": NetCDF object does not have dimension '" + dname + "' !!"
@@ -17188 +17188 @@
     else:
         print warnmsg
-        print '  ' + fname ": Any  1D variables with dimension '" + dname + "' !!"
+        print '  ' + fname + ": Any  1D variables with dimension '" + dname + "' !!"
 
 # Dpouble precision
@@ -17196 +17196 @@
     if len(d1var) > 1:
         d1vardouble = []
-        for vn in d1var
+        for vn in d1var:
             ovn = oncf.variables[vn]
-            if ovn.dtype == np.float64(1.): d1vardouble.append(vn)
-        if len(d1vardouble) != 1:
+            if ovn.dtype == type(np.float64(1.)): d1vardouble.append(vn)
+        if len(d1vardouble) > 1:
             print errormsg
-            print '  ' + fname + ": More than 1D double variable with dimension '" + dname + "' !!"
+            print '  ' + fname + ": Found more than 1D double variable with " +      \
+              "dimension '" + dname + "' !!"
+            print '    found:', d1vardouble
             quit(-1)
+        elif len(d1vardouble) == 0:
+            print warnmsg
+            print '  ' + fname + ": Found any 1D double variable with dimension '" + \
+              dname + "' !!"
+            dimvarname = None
         else:
             dimvarname = d1vardouble[0]
@@ -17210 +17217 @@
     return dimvarname
 
-def ovar_onc(ovar, oncf):
+def ovar_onc(incf, ovar, oncf):
     """ Function to copy an object variable to a nother netcdf object
+      incf: input netCDF object
       ovar= variable object
       oncf= netCDF object
@@ -17217 +17225 @@
     fname = 'ovar_onc'
 
-    varname = ovar.name
-    varadims = ovar.dimensions
+    varname = ovar._name
+
+    if oncf.variables.has_key(varname):
+        print warnmsg
+        print '  ' + fname + ": netCDF object already has variable '" + ovar.name +  \
+          "' !!"
+        print '    doing noting'
+        return
+
+    vardims = ovar.dimensions
     vartype = ovar.dtype
+
+    for dn in vardims:
+        if not searchInlist(oncf.dimensions, dn):
+            if incf.dimensions[dn].isunlimited():
+                newdim = oncf.createDimension(dn, None)
+            else:
+                newdim = oncf.createDimension(dn, len(incf.dimensions[dn]))
  
     newvar = oncf.createVariable(varname, vartype, vardims)
-    newvar = ovar[:]
-
-    for ncatr in ovar.ncattrs():
-        ncatrv = ovar.get_ncattribute(ncattr)
-        set_ncatrribute(newvar, ncatr, ncatrv)
+    newvar[:] = ovar[:]
+
+    for attrn in ovar.ncattrs():
+        attrv = ovar.getncattr(attrn)
+        set_attribute(newvar, attrn, attrv)
 
     oncf.sync()
@@ -17234 +17257 @@
 def SpatialWeightedMean(values, filen, varn):
     """ Function to compute the spatial mean using weights from a netCDF file
-      values= [weightskind],[xdimname],[ydimname]
+      values= [weightskind],[xdimname],[ydimname],[addvars]
         [weightskind]: type of weights:
           * 'varnames',[varname],[oper]: using a variable [varname] with an operation [oper] as area size
@@ -17245 +17268 @@
         [xdimname]: name of the dimension for the x-axis
         [ydimname]: name of the dimension for the y-axis
+        [addvars]: ':', separetd list of name of variables to add to the output file 
       filen= name of the netCDF file
       varn= name of the variable
@@ -17260 +17284 @@
     xdimname = values.split(',')[3]
     ydimname = values.split(',')[4]
+    addvars = values.split(',')[5]
 
     ofile = 'SpatialWeightedMean_' + weightk + '.nc'
 
     if weightk == 'varnames':
-        arguments = '[weightk],[varname],[oper],[xdimname],[ydimname]'
+        arguments = '[weightk],[varname],[oper],[xdimname],[ydimname],[addvars]'
         check_arguments(fname, len(values.split(',')), arguments, ',',               \
           len(arguments.split(',')))
@@ -17498 +17523 @@
         vardim = varDimension(onc, dim)
         if vardim is not None:
-            ovar_nc(onc.variables[vardim], onewnc)
+            ovar_onc(onc, onc.variables[vardim], onewnc)
             
-
 # Output variable
 ##
@@ -17508 +17532 @@
       longvarname, iovar.getncattr('units'))
     newvar[:] = newvals
+
+    onewnc.sync()
 
 # Spatial weight
@@ -17517 +17543 @@
     basicvardef(newvar, 'spatialweight', 'space weight ' + longvarname, '-')
     newvar[:] = outweightvals
+
+    onewnc.sync()
+
+# Additional variables
+##
+    for vn in addvars.split(':'):
+        if not onc.variables.has_key(vn):
+            print errormsg
+            print '  ' + fname + ": file '" + filen + "' does not have variable '" + \
+              vn + "' !!"
+            quit(-1)
+        ovar_onc(onc, onc.variables[vn], onewnc)
+    onewnc.sync()
 
 # Global attributes
@@ -17531 +17570 @@
 
     onc.close()
+    onewnc.sync()
+    onewnc.close()
+
 
     print fname + ": Successfull written of file: '" + ofile + "' !!"