Changeset 1360 in lmdz_wrf

Timestamp:

Dec 2, 2016, 3:32:12 PM (9 years ago)

Author:

lfita

Message:

Adding `getvars_tofile': Function to get variables from a file and pass them to another one

Location:

trunk/tools

Files:

• nc_var.py (modified) (4 diffs)
• nc_var_tools.py (modified) (2 diffs)

trunk/tools/nc_var.py

@@ -33 +33 @@
 ## e.g. # nc_var.py -o compute_opersvarsfiles -S 'west_east|XLONG|-1;south_north|XLAT|-1;Time|Times|3@add|wrfout_d01_2001-11-11_00:00:00|T2%west_east|XLONG|-1;south_north|XLAT|-1;Time|Times|3@subc,273.15|wrfout_d01_2001-11-11_00:00:00|None' -v 'tempC,air!temperature,C'
 ## e.g. # nc_var.py -o compute_opersvarsfiles -S 'lon|lon|-1;lat|lat|-1;time_counter|time_counter|-1@forwrdderiv,1,1,2|/ccc/store/cont003/gen7593/fitaborl/etudes/DynamicoESM/aquaplanet/AR40/19800101000000-19810101000000/histday.nc|t2m' -v 'tasderiv,x-derivative|of|air|temperature,K
-
+## e.g. # nc_var.py -o getvars_tofile -S soils_param_Zobler_textXIOS.nc -f soils_param_colorXIOS.nc -v 'soiltext,soiltexttypes'
 
 from optparse import OptionParser
@@ -83 +83 @@
 # 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
+# getvars_tofile: Function to get variables from a file and pass them to another one
 # grattr: Function to read a global atribute
 # grmattr: Removing a global attribute
@@ -158 +159 @@
   'flipdim', 'fvaradd', 'gaddattrk', 'gaddattr', 'get_attribute',                    \
   'get_namelist_vars', 'get_Variables',                                              \
-  'getvalues_lonlat', 'grattr',                                                      \
+  'getvalues_lonlat', 'getvars_tofile', 'grattr',                                    \
   'grmattr', 'idims', 'igattrs', 'increaseDimvar', 'isgattrs',                       \
   'isvattrs', 'ivars', 'ivattrs', 'LMDZ_toCF', 'maskvar', 'model_characteristics',   \
@@ -326 +327 @@
 elif oper == 'getvalues_lonlat':
     ncvar.getvalues_lonlat(opts.values, opts.ncfile)
+elif oper == 'getvars_tofile':
+    ncvar.getvars_tofile(opts.values, opts.ncfile, opts.varname)
 elif oper == 'grattr':
     ncvar.grattr(opts.values, opts.ncfile)

trunk/tools/nc_var_tools.py

@@ -71 +71 @@
 # 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
+# getvars_tofile: Function to get variables from a file and pass them to another one
 # grattr: Function to read a global atribute
 # grmattr: Removing a global attribute
@@ -17958 +17959 @@
     return
 
+def getvars_tofile(values,ncfile,varname):
+    """ Function to get variables from a file and pass them to another one
+      values= [getfile]
+        getfile: name of the file from which variables have to be taken
+      ncfile= name of the file to receive the variables
+      varname= ',' list of variables to get ('all', for all variables)
+      NOTE: all coincident dimensions, variables will not be taken
+    """
+    fname = 'getvars_tofile'
+    
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print getvars_tofile.__doc__
+        quit()
+
+    #arguments = '[getfile]'
+    #gen.check_arguments(fname, values, arguments, ',')
+
+    #getfile = values.split(',')[0]
+    getfile = values
+
+    if not os.path.isfile(getfile):
+        print errormsg
+        print '  ' + fname + ": file from which get variables '" + getfile +         \
+          "' does not exist !!"
+        quit(-1)
+
+    ionc = NetCDFFile(getfile, 'r')
+    ronc = NetCDFFile(ncfile, 'a')
+
+    # Getting list of variables to get
+    ivarns = ionc.variables.keys()
+    if varname == 'all':
+        varns = list(ivarns)
+    else:
+        varns = varname.split(',')
+
+    print '  ' + fname + ": getting variables from file '" + getfile + "' to '" +    \
+      ncfile + ' _______'
+    rvarns = ronc.variables.keys()
+    for vn in varns:
+       rdimns = list(ronc.dimensions)
+       if not gen.searchInlist(ivarns,vn):
+           print errormsg
+           print '  '  + fname + ": file '" + getfile + "' does not have variable '"+\
+             vn + "' !!"
+           print '    available variables:', ivarns
+           quit(-1)
+       if not gen.searchInlist(rvarns,vn):
+           print '    '  + fname + ": adding variable '" + vn + "' ..." 
+           
+           iovar = ionc.variables[vn]
+           vardims = list(iovar.dimensions)
+           for vd in vardims:
+               if not gen.searchInlist(rdimns,vd):
+                   ovd = ionc.dimensions[vd]
+                   if ovd.isunlimited():
+                       dsize = None
+                   else:
+                       dsize = len(ovd)
+
+                   print '    '  + fname + ": adding dimension '" + vd + "' (",      \
+                     dsize, ') ...' 
+                   ronc.createDimension(vd,dsize)
+           varvals = iovar[:]
+           vartype = iovar.dtype
+           varattrs = iovar.ncattrs()
+           if gen.searchInlist(varattrs, '_FillValue'):
+               fillval = iovar.getncattr('_FillValue')
+               newvar = ronc.createVariable(vn, vartype, vardims, fill_value=fillval)
+               varattrs.remove('_FillValue')
+           else:
+               newvar = ronc.createVariable(vn, vartype, vardims)
+           newvar[:] = varvals[:]
+           for attrn in varattrs:
+               attrv = iovar.getncattr(attrn)
+               newattr = set_attribute(newvar,attrn,attrv)
+           ronc.sync() 
+
+    ionc.close()
+
+    # Global values
+    newattr = set_attribute(ronc, 'PyNCplot', 'http://www.xn--llusfb-5va.cat/python/PyNCplot/')
+    newattr = set_attribute(ronc, 'script', fname)
+    newattr = set_attribute(ronc, 'get_file', getfile)
+    newattr = set_attribute(ronc, 'get_variables', varname)
+
+    ronc.sync()
+    ronc.close()
+
+    return
 
 #quit()