Changeset 2247 in lmdz_wrf for trunk/tools/nc_var_tools.py

Timestamp:

Nov 26, 2018, 3:10:55 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `dimrename': Renaming a dimension from a file

File:

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

trunk/tools/nc_var_tools.py

@@ -61 +61 @@
 # 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
+# dimrename: Renaming a dimension from a file
 # dimrm: Removing a dimension from a file
 # DimsLoop: Function to provide the shape of the dimensions which are not selected
@@ -1754 +1755 @@
 
   return
+
+ 
+def dimrename(ncfile, newdimn, dimn):
+    """ Renaming a dimension from a file
+    ncfile = netCDF object file
+    values= [newdimn], new name for the dimension
+    dimn = dimension name to change
+    """
+    import shutil as shu
+    fname = 'dimrename'
+
+    if newdimn == 'h':
+        print fname + '_____________________________________________________________'
+        print dimrename.__doc__
+        quit()
+
+    ncf = NetCDFFile(ncfile,'a')
+    if not ncf.dimensions.has_key(dimn):
+        print errormsg
+        print '  ' + fname + ": file does not have dimension name '" + dimn+ "' !!"
+        dimns = ncf.dimensions
+        dimns.sort()
+        print '    available ones:', dimns
+        ncf.close()
+        quit(-1)
+
+    if ncf.dimensions.has_key(newdimn):
+        print warnmsg
+        print '  ' + fname + ": file already have dimension name '" + newdimn+ "' !!"
+        print "    looping all over that variables with dimension name '"+dimn+"' !!"
+        for varn in ncf.variables.keys():
+            ovar = ncf.variables[varn]
+            if gen.searchInlist(ovar.dimensions,dimn):
+                vardims = list(ovar.dimensions)
+                idim = vardims.index(dimn)
+                vardims[idim] = newdimn
+                
+                newvar = ncf.createVariable(varn+'tmp', ovar.dtype, tuple(vardims))
+                newvar[:] = ovar[:]
+                for ncattr in ovar.ncattrs():
+                    attrv = ovar.getncattr(ncattr)
+                    newvar.setncattr(ncattr, attrv)
+                ncf.sync()
+                ncf.renameVariable(varn, varn+'old')
+                ncf.renameVariable(varn+'tmp', varn)
+                ncf.sync()
+    else:
+        newdim = ncf.renameDimension(dimn, newdimn)
+    ncf.sync()
+    ncf.close()
+
+    return
  
 def dimrm(ncfile, values, dimn):