Changeset 408 in lmdz_wrf for trunk/tools

Timestamp:

May 4, 2015, 11:42:22 AM (10 years ago)

Author:

lfita

Message:

Adding 'changevartype' Function to change the type of a variable (when possible)

Location:

trunk/tools

Files:

• nc_var.py (modified) (2 diffs)
• nc_var_tools.py (modified) (1 diff)

trunk/tools/nc_var.py

@@ -17 +17 @@
 import nc_var_tools as ncvar
 
-operations=['addvals', 'chdimname', 'checkallvars', 'checkAllValues', 'checkNaNs',   \
+operations=['addvals', 'chdimname', 'changevartype', 'checkallvars',                 \
+  'checkAllValues', 'checkNaNs',                                                     \
   'chgtimestep', 'chvarname', 'compute_deaccum', 'compute_opersvarsfiles',           \
   'compute_opervaralltime', 'compute_opervartimes', 'compute_tevolboxtraj',          \
@@ -99 +100 @@
 elif oper == 'chdimname':
     ncvar.chdimname(opts.values, opts.ncfile, opts.varname)
+elif oper == 'changevartype':
+    ncvar.changevartype(opts.values, opts.ncfile, opts.varname)
 elif oper == 'checkallvars':
     ncvar.checkallvars(opts.values, opts.ncfile)

trunk/tools/nc_var_tools.py

@@ -14150 +14150 @@
 
     return
-
 #increaseDimvar('shan:29:1:y', 'test.nc', 'var')
+
+def changevartype(values, ncfile, varn):
+    """ Function to change the type of a variable (when possible)
+      values=[newtype] type to use ('c', character; 'i', integer, 'f', float, 'd', double)
+      ncfile= netCDF file to use
+      varn= variable to change its type
+
+    """
+    import subprocess as sub
+    fname = 'changevartype'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print changevartype.__doc__
+        quit()
+
+    newtype = values
+
+    ofile = ncfile + '_new.nc'
+
+    onc = NetCDFFile(ncfile, 'r')
+
+    if not searchInlist(onc.variables, varn):
+        print errormsg
+        print '  ' + fname + ": file '"  + ncfile + "' has not variable '" + varn +  \
+          "' !!"
+        quit(-1)
+
+    dims = onc.dimensions
+    varns = onc.variables
+
+# Creation of a new file with the new variable dimensions
+    newnc = NetCDFFile(ofile, 'w')
+
+# Creation of dimensions
+    for dn in dims:
+        print 'adding dim:',dn,' ...'
+        if onc.dimensions[dn].isunlimited():
+            newdim = newnc.createDimension(dn, None)
+        else:
+            newdim = newnc.createDimension(dn, len(onc.dimensions[dn]))
+
+    newnc.sync()
+
+# Transforming variable
+    varo = onc.variables[varn]
+    vartype = varo.dtype
+    vardims = varo.dimensions
+    varatts = varo.ncattrs()
+    varshape = varo.shape
+    varv = varo[:]
+
+    if newtype == 'c':
+        print errormsg
+        print '  ' + fname + ": new type '" + newtype + "' not ready!!'"
+    elif newtype == 'i':
+        newvarv = varv.astype(int)
+    elif newtype == 'f':
+        newvarv = varv.astype(np.float)
+    elif newtype == 'd':
+        newvarv = varv.astype(np.float64)
+    else:
+        print errormsg
+        print '  ' + fname + ": type '" + newtpe + "' not ready !!"
+        print '    availables: i, f, d'
+        quit(-1) 
+    
+    if searchInlist(varatts, '_FillValue'):
+        fillv = varo.getncattr('_FillValue')
+        newvar = newnc.createVariable(varn, newtype, vardims, fill_value=fillv)
+    else:
+        newvar = newnc.createVariable(varn, newtype, vardims)
+
+    for nattr in varo.ncattrs():
+        if not nattr == '_FillValue':
+            nattrv = varo.getncattr(nattr)
+            newattr = newvar.setncattr(nattr, nattrv)
+
+    newvar[:] = newvarv
+
+# Adding variables
+    for vn in varns:
+        print 'adding variable:',vn,'...'
+        vno = onc.variables[vn]
+        if vn != varn:
+            vdim = vno.dimensions
+            vtype = vno.dtype
+            varatts = vno.ncattrs()
+
+            if searchInlist(varatts, '_FillValue'):
+                fillv = vno.getncattr('_FillValue')
+                newvar = newnc.createVariable(vn, vtype, vdim, fill_value=fillv)
+            else:
+                newvar = newnc.createVariable(vn, vtype, vdim)
+
+            for nattr in vno.ncattrs():
+                if not nattr == '_FillValue':
+                    nattrv = vno.getncattr(nattr)
+                    newattr = newvar.setncattr(nattr, nattrv)
+
+            newvar[:] = vno[:]
+
+    onc.close()
+    newnc.sync()
+    newnc.close()
+
+# Adding attributes
+    fgaddattr(ncfile, ofile)
+    sub.call(['mv',ofile,ncfile])
+
+    return
+
+#changevartype('i', 'test.nc', 'var')
+#quit()
 
 """operation to make: