Changeset 964 in lmdz_wrf for trunk/tools

Timestamp:

Jun 28, 2016, 5:30:05 PM (8 years ago)

Author:

lfita

Message:

Adding `get_Variables': Function to get a list of variables from an existing file

File:

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

trunk/tools/nc_var_tools.py

@@ -54 +54 @@
 # get_namelist_vars: Function to get namelist-like  values ([varname] = [value])
 # get_str_nc: Function to get string values in a netCDF variable as a chains of 1char values
+# get_Variables: Function to get a list of variables from an existing file
 # getvalues_lonlat: Function to retrieve the values from the closest grid point to a set of longitude, latitude values
 # grattr: Function to read a global atribute
@@ -16341 +16342 @@
     return
 
+def get_Variables(values, ncfile):
+    """ Function to get a list of variables from an existing file
+      values = ',' separated list of names of variables to get
+      ncfile= WRF file to use
+    """
+    fname = 'get_Variables'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print get_Variables.__doc__
+        quit()
+
+    variables = values.split(',')
+
+    ofile = 'get_Variables.nc'
+
+    onewnc = NetCDFFile(ofile, 'w')
+
+    onc = NetCDFFile(ncfile, 'r')
+    filevars = list(onc.varibles.keys)
+
+    for var in variables:
+        if not gen.searchInlist(filevars, var):
+            print errormsg
+            print '  ' + fname + ": file '" + ncfile + "' does not have variable '" +\
+              var + "' !!"
+            print '    variables in file:', filevars
+            quit(-1)
+
+        print '  ' + fname + ": getting variable '" + var + "'..."
+        ovar = onc.variables[var]
+        varinf = variable_inf(ovar)
+        vardims = varind.dmns
+
+        newfiledims = list(onewnc.dimensions.keys)
+
+        for vdim in vardims:
+            if not gen.searchInlist(newfiledims, vdim):
+                print '  ' + fname + ": including dimension '" + vdim + "'..."
+                if onc.dimensions[vdim].isunlimted:
+                    newdim = onewnc.createDimension(vdim, None)
+                else:
+                    newdim = onewnc.createDimension(vdim, len(onc.dimensions[vdim]))
+
+       if varinf.FillValue is not None:
+           newvar = onewnc.creatrVariable(var, nctype(varinf.dtype), tuple(vardims), \
+             fillvalue = varinf.FillValue) 
+       else:
+           newvar = onewnc.creatrVariable(var, nctype(varinf.dtype), tuple(vardims)) 
+
+       for attrn in varinf.atttributes:
+           attrv = ovar.getattribute(attrn)
+           newattr = set_attribute(newvar, attrn, attrv)
+
+       onewnc.sync()
+
+    # Global attributes
+    for attrn in onc.ncattrs:
+       attrv = ovar.getattribute(attrn)
+       newattr = set_attribute(newvar, attrn, attrv)
+    
+    onewnc.sync()
+    onc.close()
+
+    onewnc.close()
+
+    print fname + ": successful written of '" + ncfile + "' !!"
+
 #quit()