Changeset 1008 in lmdz_wrf

Timestamp:

Aug 11, 2016, 5:17:55 PM (9 years ago)

Author:

lfita

Message:

Adding `cleaning_varsfile': Function to keep a list of varibales from a file

Location:

trunk/tools

Files:

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

trunk/tools/nc_var.py

@@ -15 +15 @@
 ## e.g. # nc_var.py -o model_characteristics -f ~/PY/wrfout_d01_2001-11-11_00:00:00 -S WRF
 ## e.g. # nc_var.py -o WRF_toCF -f ~/PY/wrfout_d01_2001-11-11_00:00:00 -S XLONG:XLAT:19491201000000:minutes
+## e.g. # nc_var.py -o cleaning_varsfile -f ~/PY/wrfout_d01_2001-11-11_00:00:00 -S T2,XLONG,XLAT,Times
 
 from optparse import OptionParser
@@ -27 +28 @@
 operations=['addvals', 'chdimname', 'changevartype', 'checkallvars',                 \
   'checkAllValues', 'checkNaNs',                                                     \
-  'chgtimestep', 'chvarname', 'compute_deaccum', 'compute_opersvarsfiles',           \
+  'chgtimestep', 'chvarname', 'cleaning_varsfile', 'compute_deaccum',                \
+  'compute_opersvarsfiles',                                                          \
   'compute_opervaralltime', 'compute_opervartimes', 'compute_tevolboxtraj',          \
   'computevar_model', 'DatesFiles',                                                  \
@@ -130 +132 @@
 elif oper == 'chvarname':
     ncvar.chvarname(opts.values, opts.ncfile, opts.varname)
+elif oper == 'cleaning_varsfile':
+    ncvar.cleaning_varsfile(opts.values, opts.ncfile)
 elif oper == 'compute_deaccum':
     ncvar.compute_deaccum(opts.values, opts.ncfile, opts.varname)

trunk/tools/nc_var_tools.py

@@ -22 +22 @@
 # chgtimestep: Function to change the values of a given time-step of a variable inside a netCDF for values from a nother file
 # chvarname: Changing the name of the variable
+# cleaning_varsfile: Function to keep a list of varibales from a file
 # cls_time_information: Classs to provide information about variable time
 # compute_deaccum: Function to compute deaccum: deaccumulation of a variable (VAR[t+1]-VAR[t])
@@ -16514 +16515 @@
     return
 
+def cleaning_varsfile(values, filen):
+    """ Function to keep a list of varibales from a file
+      values= ',' separated list of variables to keep
+      filen= name of the file
+    """
+    fname = 'cleaning_varsfile'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print WRF_toCF.__doc__
+        quit()
+
+    arguments = '[values]'
+    gen.check_arguments(fname, values, arguments, ':')
+
+    keepvars = values.split(',')
+
+    ofile = 'cleaning_varsfile.nc'
+
+    # opening source file
+    onc = NetCDFFile(filen, 'r')
+
+    # opening new file
+    newonc = NetCDFFile(ofile, 'w')
+
+    oldvars = onc.variables.keys()
+    for oldv in oldvars:
+        newdims = newonc.dimensions
+        newvars = newonc.variables.keys()
+
+        if not gen.searchInlist(newvars, 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)
+            newvar[:] = oldov[:]
+            for attrn in vinf.attributes:
+                attrv = oldov.getncattr(attrn)
+                newattr = ncvar.set_attribute(newvar,attrn,attrv)
+
+        newonc.sync()
+
+    newonc.sync()
+    # Global attributes
+    for gattrn in onc.ncattrs():
+        gattrval = onc.getncattr(gattr)
+        newattr = ncvar.set_attribute(newonc, gattr, gattrval)
+
+    newonc.sync()
+    newonc.close()
+    onc.close()
+
+    sout = sub.call('mv ' + filen + ' ' + ofile, shell=True)
+
+    return
+
+
 #quit()