Changeset 1295 in lmdz_wrf for trunk/tools

Timestamp:

Nov 9, 2016, 8:36:07 PM (9 years ago)

Author:

lfita

Message:

Adding `DYNAMICO_toCF': Function to pass a DYNAMICO original file to CF-conventions

Location:

trunk/tools

Files:

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

trunk/tools/nc_var.py

@@ -64 +64 @@
 # DataSetSection_multidims: Function to get a section (values along multiple dimensions) of a given data-set
 # 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
 # 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
@@ -151 +152 @@
   'computevar_model', 'DatesFiles',                                                  \
   'DataSetSection', 'DataSetSection_multidims', 'DataSetSection_multivars',          \
-  'dimToUnlimited', 'dimVar_creation',                                               \
+  'DYNAMICO_toCF', 'dimToUnlimited', 'dimVar_creation',                              \
   'fattradd',                                                                        \
   'fdimadd', 'fgaddattr', 'field_stats', 'field_stats_dim', 'file_creation',         \
@@ -286 +287 @@
 elif oper == 'DatesFiles':
     ncvar.DatesFiles(opts.values, opts.ncfile, opts.varname)
+elif oper == 'DYNAMICO_toCF':
+    ncvar.DYNAMICO_toCF(opts.values, opts.ncfile)
 elif oper == 'dimToUnlimited':
     ncvar.dimToUnlimited(opts.values, opts.ncfile)

trunk/tools/nc_var_tools.py

@@ -48 +48 @@
 # DatesFiles: Function to find different time values on a series of WRF files in a folder
 # 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
 # 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
@@ -17642 +17643 @@
     return
 
+
+def DYNAMICO_toCF(values,ncfile):
+    """ Function to pass a DYNAMICO original file to CF-conventions
+      values= [CFtimeUnits]
+        [CFtimeUnits]: [tunits] since [YYYYMMDDHHMISS], CF time-units to which variable time should be used ('!' for spaces)
+      ncfile= file to transform 
+    """
+    fname = 'DYNAMICO_toCF'
+
+    if ncfile == 'h':
+        print fname + '_____________________________________________________________'
+        print DYNAMICO_toCF.__doc__
+        quit()
+
+    CFT = values.replace('!',' ')
+
+    ofile = 'CF_DYNAMICO.nc'
+  
+    ncf = NetCDFFile(ncfile,'a')
+    ncdims = ncf.dimensions.keys()
+    ncvars = ncf.variables.keys()
+
+    # CFing variable 'pres'
+    if gen.searchInlist(ncvars,'pres'):
+        print warnmsg
+        print fname + ": renaming variable 'pres' as 'p' !!"
+        newname = ncf.renameVariable('pres','p')
+    ncf.sync()
+
+    # Re-aranging time-axis
+    tvaro = ncf.variables['time_counter']
+    tvals = tvaro[:]
+    tunits = tvaro.getncattr('units')
+
+    RefDate = CFT.split(' ')[2]
+    RefDateS = RefDate[0:4] + '-' + RefDate[4:6] + '-' + RefDate[6:8] + ' ' +        \
+      RefDate[8:10] + ':' + RefDate[10:12] + ':' + RefDate[12:14]
+    CFtimeUnits = ' '.join(CFT.split(' ')[0:2]) + ' '+ RefDateS
+
+    newtvals = gen.coincident_CFtimes(tvals, CFtimeUnits, tunits)
+    tvaro[:] = newtvals
+    newattr = set_attribute(tvaro, 'units', CFtimeUnits)
+    newattr = set_attribute(tvaro, 'time_origin', CFtimeUnits)
+
+    ncf.sync()
+
+    # CFing dimensions
+    CFdims = {'time_counter': 'time', 'presnivs': 'pres'}
+    for dimn in CFdims.keys():
+        if gen.searchInlist(ncdims,dimn):
+            # Making sure CF dimension is not in the file
+            if not gen.searchInlist(ncdims,CFdims[dimn]):
+                newname = ncf.renameDimension(dimn,CFdims[dimn])
+    ncf.sync()
+
+    ncvars = ncf.variables.keys()
+    # Re-arranging coordinates attribute for all the variables
+    for varn in ncvars:
+        varobj = ncf.variables[varn]
+        varattrs = varobj.ncattrs()
+        if gen.searchInlist(varattrs, unicode('coordinates')):
+            attrv = varobj.getncattr('coordinates')
+            newcoord = str(attrv)
+            for dimn in CFdims.keys():
+                newcoord = newcoord.replace(dimn, CFdims[dimn])
+            newattr = set_attribute(varobj, 'coordinates', newcoord)
+
+    ncf.sync()
+    ncf.close()
+
+    print fname + ": succesfull CFication of DYNAMICO file '" + ncfile + "' !!"
+
+    return
+
 def CDO_toCF(ncfile):
     """ Function to pass a CDO output file to CF-conventions