Changeset 1046 in lmdz_wrf for trunk

Timestamp:

Aug 26, 2016, 2:47:39 PM (9 years ago)

Author:

lfita

Message:

Adding `LMDZ_toCF': Function to pass a LMDZ original file to CF-conventions
Removing some spurious 'Lluis' check printings

Location:

trunk/tools

Files:

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

trunk/tools/nc_var.py

@@ -16 +16 @@
 ## e.g. # nc_var.py -o WRF_toCF -f ~/PY/wrfout_d01_2001-11-11_00:00:00 -S XLONG:XLAT:19491201000000:minutes
 ## 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
 
 from optparse import OptionParser
@@ -41 +42 @@
   'getvalues_lonlat', 'grattr',                                                      \
   'grmattr', 'idims', 'igattrs', 'increaseDimvar', 'isgattrs', 'isvattrs', 'ivars',  \
-  'ivattrs', 'maskvar', 'model_characteristics',                                     \
+  'ivattrs', 'LMDZ_toCF', 'maskvar', 'model_characteristics',                        \
   'ncreplace', 'ncstepdiff', 'netcdf_concatenation', 'netcdf_fold_concatenation',    \
   'netcdf_fold_concatenation_HMT', 'Partialmap_Entiremap',                           \
@@ -213 +214 @@
             print opern + '_______ ______ _____ ____ ___ __ _'
             print getattr(object, opern).__doc__
+elif oper == 'LMDZ_toCF':
+    ncvar.LMDZ_toCF(opts.ncfile)
 elif oper == 'maskvar':
     ncvar.maskvar(opts.values, opts.ncfile, opts.varname)

trunk/tools/nc_var_tools.py

@@ -5714 +5714 @@
                 dimsize = (iendv - ibegv + 1)/iintv - 1
             else:
-                dimsize == 0                       
+                dimsize = 0
                 if begv == -9:
                     ibegv = len(objdim) - 1
@@ -5746 +5746 @@
 
     ncoobj.sync()
-    print fname + 'Lluis; dims:', ncoobj.dimensions
 # Creating variables
 ## 
-    print fname + '; Lluis dimsecslice',dimsecslice
 
     filedims = ncoobj.dimensions
@@ -5757 +5755 @@
 
         vardims = varobj.dimensions
-        print '  ' + fname + ' Lluis; vardims:', vardims
 
         if not gen.searchInlist(list(vardims),dimn):
@@ -5798 +5795 @@
             varfil = None
         vartype = varobj.dtype
-        print '  ' + fname + '; Lluis filevardims:', filevardims
 
         if varfil is not None:
@@ -5806 +5802 @@
             newvar = ncoobj.createVariable(varns, vartype, tuple(filevardims))
         ncoobj.sync()
-        print fname + ' Lluis newvardims:', newvar.dimensions
 
         newvar[:] = varvals
@@ -5832 +5827 @@
 
     onc = NetCDFFile(ofile, 'r')
-    print fname + 'Lluis: ', onc.variables['lon']
     onc.close()
 
@@ -16112 +16106 @@
             self.vardxn = 'lon'
             self.vardyn = 'lat'
-            self.vardzn = 'pres'
+            self.vardzn = 'presnivs'
             self.vardtn = 'time_counter'
             self.vardsn = None
@@ -16565 +16559 @@
     return
 
+def LMDZ_toCF(ncfile):
+    """ Function to pass a LMDZ original file to CF-conventions
+      ncfile= file to transform 
+      variables= variables to transform
+    """
+    fname = 'LMDZ_toCF'
+
+    if ncfile == 'h':
+        print fname + '_____________________________________________________________'
+        print LMDZ_toCF.__doc__
+        quit()
+
+    ofile = 'CF_LMDZ.nc'
+  
+    ncf = NetCDFFile(ncfile,'a')
+    ncdims = ncf.dimensions.keys()
+    ncvars = ncf.variables.keys()
+
+#    # CFing dimension-variables
+#    CFvardims = {'time_counter': 'time', 'presnivs': 'pres'}
+#    for varn in CFvardims.keys():
+#        if gen.searchInlist(ncvars,varn):
+#            # Making sure CF vardimension is not in the file
+#            if not gen.searchInlist(ncvars,CFvardims[varn]):
+#                newname = ncf.renameVariable(varn,CFvardims[varn])
+
+#    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 LMDZ file '" + ncfile + "' !!"
+
+    return
+
 def cleaning_varsfile(values, filen):
     """ Function to keep a list of varibales from a file