Changeset 1005 in lmdz_wrf for trunk

Timestamp:

Aug 11, 2016, 1:38:08 PM (9 years ago)

Author:

lfita

Message:

Adding `WRF_toCF': Function to pass a WRF original 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

@@ -14 +14 @@
 ## e.g. # nc_var.py -o computevar_model -f ~/PY/wrfout_d01_2001-11-11_00:00:00 -S hurs
 ## e.g. # nc_var.py -o model_characteristics -f ~/PY/wrfout_d01_2001-11-11_00:00:00 -S WRF
+## e.g. # nc_var.py -o WRF_toCF -f ~/PY/wrfout_d01_2001-11-11_00:00:00 -S XLONG:XLAT:19491201000000:minutes
 
 from optparse import OptionParser
@@ -51 +52 @@
   'vrattr', 'WRF_d0Nref',                                                            \
   'WRF_CFlonlat_creation', 'WRF_CFtime_creation', 'WRF_CFxtime_creation',            \
-  'list_operations']
+  'list_operations', 'WRF_toCF']
 
 ### Options
@@ -296 +297 @@
 elif oper == 'WRF_CFxtime_creation':
     ncvar.WRF_CFxtime_creation(opts.values, opts.ncfile, opts.varname)
+elif oper == 'WRF_toCF':
+    ncvar.WRF_toCF(opts.values, opts.ncfile)
 else:
     print errormsg

trunk/tools/nc_var_tools.py

@@ -135 +135 @@
 # WRF_CFxtime_creation: Function to add a CF-convention time unit in a WRF file using variable 'XTIME'
 # WRF_d0Nref: Function for the generation of an extra WRF domain from a given one
+# WRF_toCF: Function to pass a WRF original file to CF-conventions
 
 main = 'nc_var_tools.py'
@@ -13251 +13252 @@
                   str(fendvs[ivar]) + ',1); '
 
-    ofile = ofile=filen.split('.')[0] + ofiletile + '.nc'
+    ofile = filen.split('.')[0] + ofiletile + '.nc'
 
     ncoobj = NetCDFFile(ofile,'w')
@@ -16441 +16442 @@
     print fname + ": successful written of '" + ofile + "' !!"
 
+def WRF_toCF(values, ncfile):
+    """ Function to pass a WRF original file to CF-conventions
+      values= [wrfvl]:[wrfvL]:[refDateTime]:[Tunits]
+        [wrfvl]: WRF longitude var name (it might come from WPS!)
+        [wrfvL]: WRF latitude var name (it might come from WPS!)
+        [refDateTime]: date of reference for times (in [YYYY][MM][DD][HH][MI][SS] format)
+        [Tunits]: units for time variable
+      ncfile= file to transform 
+      variables= variables to transform
+    """
+    fname = 'WRF_toCF'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print WRF_toCF.__doc__
+        quit()
+
+    arguments = '[wrfvl]:[wrfvL]:[refDateTime]:[Tunits]'
+    gen.check_arguments(fname, values, arguments, ':')
+
+    ofile = 'CF_WRF.nc'
+
+    wrfvl = values.split(':')[0]
+    wrfvL = values.split(':')[1]
+    refDateTime = values.split(':')[2]
+    Tunits = values.split(':')[3]
+
+    # CFing time
+    WRF_CFtime_creation(refDateTime+','+Tunits, ncfile, 'time')
+    
+    # CFing lon,lat
+    WRF_CFlonlat_creation('lon,lat', ncfile, wrfvl+','+wrfvL)
+
+    ncf = NetCDFFile(ncfile,'a')
+    ncdims = ncf.dimensions
+
+    CFdims = {'west_east': 'lon', 'south_north': 'lat', 'Time': 'time'}
+
+    # CFing dimensions
+    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])
+
+    # dimension variables which do not need to attach CF attributes, otherwise values
+    #   retrieved from 'variable_values.dat'
+    dimvars = ['lon', 'lat', 'time', 'Times']
+
+    for vn in ncf.variables.keys():
+        if not gen.searchInlist(dimvars,vn):
+            print fname + ' Lluis vn:',vn
+            varobj = ncf.variables[vn]
+            varattrs = gen.variables_values(vn)
+
+            stn = varattrs[1]
+            lonn = varattrs[4].replace('|',' ')
+            un = varattrs[5].replace('|',' ')
+
+            newattr = set_attribute(varobj, 'standard_name', stn)
+            newattr = set_attribute(varobj, 'long_name', lonn)
+            newattr = set_attribute(varobj, 'units', un)
+            newattr = set_attribute(varobj, 'coordinates', 'lon lat')
+            ncf.sync()
+
+    ncf.sync()
+    ncf.close()
+
+    print fname + ": succesfull CFication of WRF file '" + ncfile + "' !!"
+
+    return
+
 #quit()