Changeset 1062 in lmdz_wrf for trunk

Timestamp:

Aug 30, 2016, 3:46:56 PM (9 years ago)

Author:

lfita

Message:

Adding `CDO_toCF': Function to pass a CDO output file to CF-conventions

Location:

trunk/tools

Files:

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

trunk/tools/nc_var.py

@@ -17 +17 @@
 ## e.g. # nc_var.py -o cleaning_varsfile -f ~/PY/wrfout_d01_2001-11-11_00:00:00 -S T2,XLONG,XLAT,Times
 ## e.g. # nc_var.py -o LMDZ_toCF -f LMDZ/AR40/vas_histins_1-1.nc
+## e.g. # nc_var.py -o CDO_toCF -f LMDZ/AR40/hfss_reproj-histins.nc
 ## e.g. # nc_var.py -o pinterp -f wrfout_d01_2001-11-11_00\:00\:00 -S 100000.:97500.:95000.:92500.:90000.:85000.:80000.:75000.:70000.:65000.:60000.:55000.:50000.:45000.:40000.:35000.:30000.:25000.:20000.:15000.:10000.:5000.:2500.:1000.:500.:250.,1,0 -v WRFt,WRFrh
 ## e.g. # nc_var.py -o pinterp -f /media/ExtDiskC_ext4/DATA/etudes/WRF_LMDZ/WaquaL_highres/short_copies/LMDZ/AR40/histins_19790101000000-19790304000000_short.nc -S 100000.:97500.:95000.:92500.:90000.:85000.:80000.:75000.:70000.:65000.:60000.:55000.:50000.:45000.:40000.:35000.:30000.:25000.:20000.:15000.:10000.:5000.:2500.:1000.:500.:250.,1,0 -v temp,ovap
@@ -29 +30 @@
 import generic_tools as gen
 
-operations=['addvals', 'chdimname', 'changevartype', 'checkallvars',                 \
+operations=['addvals', 'CDO_toCF', 'chdimname', 'changevartype', 'checkallvars',     \
   'checkAllValues', 'checkNaNs',                                                     \
   'chgtimestep', 'chvarname', 'cleaning_varsfile', 'compute_deaccum',                \
@@ -121 +122 @@
 if oper == 'addvals':
     ncvar.addvals(opts.values, opts.ncfile, opts.varname)
+elif oper == 'CDO_toCF':
+    ncvar.CDO_toCF(opts.ncfile)
 elif oper == 'chdimname':
     ncvar.chdimname(opts.values, opts.ncfile)

trunk/tools/nc_var_tools.py

@@ -1419 +1419 @@
 
   ncf = NetCDFFile(ncfile,'a')
-  ncvars = ncf.variables
+  ncvars = ncf.variables.keys()
   ncf.close()
 
@@ -1425 +1425 @@
       print '  ' + fname + ": File '" + ncfile + "' does not have variable: '" +     \
         var + "' !!"
-      ncf.close()
+      print '  variables:', ncvars
       quit(-1)
 
@@ -16760 +16760 @@
     return
 
+def CDO_toCF(ncfile):
+    """ Function to pass a CDO output file to CF-conventions
+      ncfile= file to transform 
+      variables= variables to transform
+    """
+    fname = 'CDO_toCF'
+
+    if ncfile == 'h':
+        print fname + '_____________________________________________________________'
+        print CDO_toCF.__doc__
+        quit()
+
+    ncf = NetCDFFile(ncfile,'a')
+    ncdims = ncf.dimensions.keys()
+    ncvars = ncf.variables.keys()
+
+    # renaming variables 'lon, lat'
+    CFvardims = {'lon': 'lon2', 'lat': 'lat2'}
+    for dimvarn in CFvardims.keys():
+        if gen.searchInlist(ncvars,dimvarn):
+            newn = CFvardims[dimvarn]
+            print warnmsg
+            print fname + ": renaming variable '" + dimvarn + "' as '" + newn + "' !!"
+            newname = ncf.renameVariable(dimvarn,newn)
+    ncf.sync()
+
+    # CFing dimensions
+    CFdims = {'x': 'lon', 'y': 'lat'}
+    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()
+    # CFing variables 'lon, lat'
+    olon2 = ncf.variables['lon2']
+    dimslon2 = olon2.dimensions
+    newvar = ncf.createVariable('lon','f8',dimslon2)
+    newvar[:] = olon2[:]
+    olonattr = olon2.ncattrs()
+    for attrn in olonattr:
+        attrv = olon2.getncattr(attrn)
+        set_attribute(newvar,attrn,attrv)
+
+    olat2 = ncf.variables['lat2']
+    dimslat2 = olat2.dimensions
+    newvar = ncf.createVariable('lat','f8',dimslat2)
+    newvar[:] = olat2[:]
+    olatattr = olat2.ncattrs()
+    for attrn in olatattr:
+        attrv = olat2.getncattr(attrn)
+        set_attribute(newvar,attrn,attrv)
+
+    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()
+
+    # We should removing spurious lon2, lat2, but this would make too much...
+
+    ncf.close()
+
+    print fname + ": succesfull CFication of CDO file '" + ncfile + "' !!"
+
+    return
+
 def cleaning_varsfile(values, filen):
     """ Function to keep a list of varibales from a file