Changeset 1879 in lmdz_wrf for trunk/tools

Timestamp:

Apr 3, 2018, 4:45:50 PM (7 years ago)

Author:

lfita

Message:

Adding:

• `dimrm': Removing a dimension from a file

Location:

trunk/tools

Files:

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

trunk/tools/nc_var.py

@@ -87 +87 @@
 # DataSetSection_multivars: Function to get a section (values along multiple variables) of a given data-set
 # DYNAMICO_toCF: Function to pass a DYNAMICO original file to CF-conventions
+# dimrm: Removing a dimension from a file
 # dimToUnlimited: Operation to create an unlimited dimension from an existing one
 # dimVar_creation: Function to add a 1D variable with the size of a given dimension in a file
@@ -183 +184 @@
   'computevar_model', 'curve_section', 'DatesFiles',                                 \
   'DataSetSection', 'DataSetSection_multidims', 'DataSetSection_multivars',          \
-  'DYNAMICO_toCF', 'dimToUnlimited', 'dimVar_creation',                              \
+  'DYNAMICO_toCF', 'dimrm', 'dimToUnlimited', 'dimVar_creation',                     \
   'fattradd',                                                                        \
   'fdimadd', 'fgaddattr', 'field_stats', 'field_stats_dim', 'file_creation',         \
@@ -332 +333 @@
 elif oper == 'DYNAMICO_toCF':
     ncvar.DYNAMICO_toCF(opts.values, opts.ncfile)
+elif oper == 'dimrm':
+    ncvar.dimrm(opts.ncfile, opts.values, opts.varname)
 elif oper == 'dimToUnlimited':
     ncvar.dimToUnlimited(opts.values, opts.ncfile)

trunk/tools/nc_var_tools.py

@@ -57 +57 @@
 # DataSetSection_multivars: Function to get a section (values along multiple variables) of a given data-set
 # DatesFiles: Function to find different time values on a series of WRF files in a folder
+# dimrm: Removing a dimension from a file
 # DimsLoop: Function to provide the shape of the dimensions which are not selected
 # DYNAMICO_toCF: Function to pass a DYNAMICO original file to CF-conventions
@@ -1635 +1636 @@
   fname = 'varrm'
 
-  if var == 'h':
+  if var == 'h' or var is None:
       print fname + '_____________________________________________________________'
       print varrm.__doc__
@@ -1665 +1666 @@
   return
  
+def dimrm(ncfile, values, dimn):
+    """ Removing a dimension from a file
+    ncfile = netCDF object file
+    values= [varsAction], which action to be done on that variables which contain the 
+      given dimension
+        'renamedim',[newname]: create a new dimension and its varibale-dimension with 
+          name [newname] for that varibales with the diemnsion to remove
+        'removevar': remove varibale with the given dimension
+    dimn = dimension name to remove
+    """
+    import shutil as shu
+    fname = 'dimrm'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print dimrm.__doc__
+        quit()
+
+    ncf = NetCDFFile(ncfile,'a')
+    dims = ncf.dimensions
+    ncvars = ncf.variables.keys()
+    ncf.close()
+
+    if not gen.searchInlist(dims, var):
+        print '  ' + fname + ": File '" + ncfile + "' does not have dimension: '" +  \
+          dimn + "' !!"
+        print '  dimensions:', dims
+        quit(-1)
+
+    if values[0:9] == 'renamedim':
+        newdimn = values.split(',')[1]
+        ncf = NetCDFFile(ncfile,'a')
+        if gen.searchInlist(dims,newdimn):
+            print errormsg
+            print '  ' + fname + ": file '" + ncfile + "' already has a " +          \
+              "dimension labelled '" + newdimn + "' !!"
+            print '    pick up another name'
+            ncf.close()
+            quit()
+        newdim = ncf.renameDimension(dimn, newdimn)
+        ncf.sync()
+        # Checking for the existence of a dimension-variable of the same name
+        if gen.searchInlist(ncvars, dimn):
+            if gen.searchInlist(ncvars,newdimn):
+                print errormsg
+                print '  ' + fname + ": file '" + ncfile + "' already has a " +      \
+                  "variable labelled '" + newdimn + "' !!"
+                print '    pick up another name'
+                ncf.close()
+                quit()
+            newvar = ncf.renameVariable(dimn, newdimn)
+            ncf.sync()
+    elif values[0:9] == 'removevar':
+        ncf = NetCDFFile(ncfile,'r')
+        tmpncf = NetCDFFile('tmp_py.nc' , 'w')
+        gtmpattr = set_attribute(tmpncf, 'copy', 'temporal')
+        tmpncf.sync()
+
+        for varn in ncf.variables.keys():
+            vdimn = varn.dimensions
+            if not gen.searchInlist(vdimn,dimn):
+                ncf.close()
+                fvaradd(ncfile + ',' + varn, 'tmp_py.nc')
+                ncf = NetCDFFile(ncfile,'r')
+        ncf.close()
+        fgaddattr(ncfile, 'tmp_py.nc')
+        shu.copyfile('tmp_py.nc', ncfile)
+        os.remove('tmp_py.nc')
+
+    return
+
 def seasmean(timename, filename, varn):
     """ Function to compute the seasonal mean of a variable