Changeset 1627 in lmdz_wrf

Timestamp:

Sep 10, 2017, 9:02:02 PM (8 years ago)

Author:

lfita

Message:

Adding:

• 'WRF_to_newCF': Function to pass a WRF original file to new CF-conventions file

Location:

trunk/tools

Files:

• nc_var.py (modified) (4 diffs)
• nc_var_tools.py (modified) (1 diff)

trunk/tools/nc_var.py

@@ -16 +16 @@
 ## e.g. # nc_var.py -o varaddattrk -f SouthAm_Climzones.nc -S 'longitude_of_central_meridian|-60.0|R' -v Mercator
 ## e.g. # nc_var.py -o setvar_asciivalues -S 'nombres_zonas.dat' -f SouthAm_Climzones.nc -v regionn
+## e.g. # nc_var.py -o WRF_to_newCF -f wrfout_d01_1995-01-01_00\:00\:00 -S 'XLONG:XLAT:Times:19491201000000:minutes' -v QVAPOR,T2,Q2
 
 ## e.g. ccrc468-17 # ./nc_var.py -v time -f 123/CCRC_NARCliM_Sydney_All_1990-1999_pr10max.nc -o out -S 1:-1
@@ -167 +168 @@
 # WRF_CFxtime_creation: Function to add a CF-convention time unit in a WRF file using variable 'XTIME'
 # WRF_toCF: Function to pass a WRF original file to CF-conventions
+# WRF_to_newCF: Function to pass a WRF original file to new CF-conventions file
 
 operations=['addDim', 'addVar', 'addvals', 'CDO_toCF', 'chdimname', 'changevartype', \
@@ -199 +201 @@
   'vrattr', 'WRF_d0Nref',                                                            \
   'WRF_CFlonlat_creation', 'WRF_CFtime_creation', 'WRF_CFxtime_creation',            \
-  'list_operations', 'WRF_toCF']
+  'list_operations', 'WRF_toCF', 'WRF_to_newCF']
 
 ### Options
@@ -491 +493 @@
 elif oper == 'WRF_toCF':
     ncvar.WRF_toCF(opts.values, opts.ncfile)
+elif oper == 'WRF_to_newCF':
+    ncvar.WRF_to_newCF(opts.values, opts.ncfile, opts.varname)
 else:
     print errormsg

trunk/tools/nc_var_tools.py

@@ -18537 +18537 @@
     print fname + ": successful written of '" + ofile + "' !!"
 
+def WRF_to_newCF(values, ncfile, variables):
+    """ Function to pass a WRF original file to new CF-conventions file
+      values= [wlonn]:[wlatn]:[wtimen]:[refDateTime]:[Tunits]
+        [wlonn]: WRF longitude var name (it might come from WPS!)
+        [wlatn]: WRF latitude var name (it might come from WPS!)
+        [wtimen]: WRF time var name
+        [refDateTime]: date of reference for times (in [YYYY][MM][DD][HH][MI][SS] format)
+        [Tunits]: units for time variable
+      ncfile= file to transform 
+      variables= ',' list of variables to transform ('all', for all variables)
+    """
+    fname = 'WRF_to_newCF'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print WRF_to_newCF.__doc__
+        quit()
+
+    arguments = '[wlonn]:[wlatn]:[wtimen]:[refDateTime]:[Tunits]'
+    gen.check_arguments(fname, values, arguments, ':')
+
+    ofile = 'CF_WRF.nc'
+
+    # New names for certain dimensions
+    dimsrename = {'west_east': 'lon', 'south_north': 'lat', 'Time':'time'}
+
+    # Variables to not be included
+    varsNO = ['XLONG', 'XLONG_M', 'XLAT', 'XLAT_M', 'Times']
+
+    onewnc = NetCDFFile(ofile, 'w')
+
+    wlonn = values.split(':')[0]
+    wlatn = values.split(':')[1]
+    wtimen = values.split(':')[2]
+    refDateTime = values.split(':')[3]
+    Tunits = values.split(':')[4]
+
+    wrfnc = NetCDFFile(ncfile, 'r')
+  
+    if variables == 'all':
+        varns = wrnc.variables.keys()
+    else:
+        varns = gen.str_list(variables, ',')
+
+    # lon, lat
+    if not gen.searchInlist(wrfnc.variables.keys(),wlonn):
+        print errormsg
+        print '  ' + fname + ": file '" + ncfile + "' has not lon variable '" +      \
+          wlonn + "' !!!"
+        quit(-1)
+
+    if  not gen.searchInlist(wrfnc.variables.keys(),wlatn):
+        print errormsg
+        print '  ' + fname + ": file '" + ncfile + "' has not lat variable '" +      \
+          wlatn + "' !!!"
+        quit(-1)
+
+    if  not gen.searchInlist(wrfnc.dimensions.keys(),'west_east'):
+        print errormsg
+        print '  ' + fname + ": file '" + ncfile + "' has not 'west_east' dimension !!!"
+        quit(-1)
+
+    if  not gen.searchInlist(wrfnc.dimensions.keys(),'south_north'):
+        print errormsg
+        print '  ' + fname + ": file '" + ncfile + "' has not 'south_north' dimension !!!"
+        quit(-1)
+
+    lonobj = wrfnc.variables[wlonn]
+    lonv = lonobj[:]
+
+    latobj = wrfnc.variables[wlatn]
+    latv = latobj[:]
+
+    if len(lonv.shape) == 3:
+        cflons = np.zeros((lonv.shape[1],lonv.shape[2]), dtype=np.float)
+        cflons = np.where(lonv[0,:,:] < 0., lonv[0,:,:] + 360., lonv[0,:,:])
+        cflats = np.zeros((latv.shape[1],latv.shape[2]), dtype=np.float)
+        cflats = latv[0,:,:]
+        dimx = lonv.shape[2]
+        dimy = lonv.shape[1]
+    elif len(lonv.shape) == 2:
+        cflons = np.zeros((lonv.shape), dtype=np.float)
+        cflons = np.where(lonv < 0., lonv + 360., lonv)
+        cflats = latv
+        dimx = lonv.shape[1]
+        dimy = lonv.shape[0]
+    elif len(lonv.shape) == 1:
+        cflons = np.zeros((lonv.shape), dtype=np.float)
+        cflons = np.where(lonv < 0., lonv + 360., lonv)
+        cflats = latv
+        dimx = lonv.shape[0]
+        dimy = latv.shape[0]
+    else:
+        print errormsg
+        print '  ' + fname + ": original longitude variable '" + varn + "' shape:" + \
+          lonv.shape + "not ready !!!"
+        quit(-1)
+
+    # Creation of CF-dimensions
+    newdim = onewnc.createDimension('lon', dimx)
+    newdim = onewnc.createDimension('lat', dimy)
+    newdim = onewnc.createDimension('time', None)
+
+    # dimension-variables
+    newvar = onewnc.createVariable('lon','f8',('lat','lon'))
+    newattr = basicvardef(newvar, 'longitude','Longitude','degrees_East')
+    newvar[:] = cflons
+    newattr = set_attribute(newvar, 'axis', 'X')
+    newattr = set_attribute(newvar, '_CoordinateAxisType', 'Lon')
+
+    newvar = onewnc.createVariable('lat','f8',('lat','lon'))
+    newattr = basicvardef(newvar, 'latitude','Latitude','degrees_North')
+    newvar[:] = cflats
+    newattr = set_attribute(newvar, 'axis', 'Y')
+    newattr = set_attribute(newvar, '_CoordinateAxisType', 'Lat')
+
+    onewnc.sync()
+
+    # CFing time
+    if  not gen.searchInlist(wrfnc.variables.keys(), wtimen):
+        print errormsg
+        print '  ' + fname + ": file '" + ncfile + "' has not time variable '" +     \
+          wtimen + "' !!!"
+        quit(-1)
+
+    if  not gen.searchInlist(wrfnc.dimensions.keys(), 'Time'):
+        print errormsg
+        print '  ' + fname + ": file '" + ncfile + "' has not 'Time' dimension !!!"
+        quit(-1)
+
+    timeobj = wrfnc.variables[wtimen]
+    timev = timeobj[:]
+
+    fdates = wrfnc.variables['Times'][:]
+    cftimes = []
+    for it in range(fdates.shape[0]):
+        wrfdates = gen.datetimeStr_conversion(fdates[it,:], 'WRFdatetime', 'matYmdHMS')
+        cftimes.append(gen.realdatetime1_CFcompilant(wrfdates, refDateTime, Tunits))
+
+    yrref=refDateTime[0:4]
+    monref=refDateTime[4:6]
+    dayref=refDateTime[6:8]
+    horref=refDateTime[8:10]
+    minref=refDateTime[10:12]
+    secref=refDateTime[12:14]
+
+    tunit = Tunits + ' since ' + yrref + '/' + monref + '/' + dayref + ' ' +         \
+      horref + ':' + minref + ':' + secref
+
+    newvar = onewnc.createVariable('time','f8',('time'))
+    newattr = basicvardef(newvar, 'time','Time',tunit)
+    newvar[:] = cftimes
+    newattr = set_attribute(newvar, 'axis', 'T')
+    newattr = set_attribute(newvar, '_CoordinateAxisType', 'Time')
+    newattr = set_attribute(newvar, 'calendar', 'gregorian')
+
+    onewnc.sync()
+
+    # Removing not desired variables
+    for vrmn in varsNO:
+        if gen.searchInlist(varns, vrmn): varns.remove(vrmn)
+
+    # Looping on variables
+    for vn in varns:
+        newvars = onewnc.variables.keys()
+        newdims = list(onewnc.dimensions)
+        if not gen.searchInlist(newvars,vn):
+            varobj = wrfnc.variables[vn]
+            varattrs = gen.variables_values(vn)
+            vvatrs = varobj.ncattrs()
+
+            newvard = []
+            for dimn in varobj.dimensions:
+                if dimsrename.has_key(dimn):
+                    newvard.append(dimsrename[dimn])
+                else:
+                    newvard.append(str(dimn))
+                    if not gen.searchInlist(newdims, dimn): add_dims(wrfnc,onewnc,[str(dimn)])
+
+
+            cfn = varattrs[0]
+            stn = varattrs[1]
+            lonn = varattrs[4].replace('|',' ')
+            un = varattrs[5].replace('|',' ')
+
+            newvar = onewnc.createVariable(cfn, nctype(varobj.dtype), tuple(newvard))
+            newvar[:] = varobj[:]
+
+            newattr = set_attribute(newvar, 'standard_name', stn)
+            newattr = set_attribute(newvar, 'long_name', lonn)
+            # Original attributes
+            for attrn in vvatrs:
+                attrv = varobj.getncattr(attrn)
+                if attrn == 'units':
+                    if un != attrv:
+                        print warnmsg
+                        print '  ' + fname + ": WRF variable '" + vn + "' has different " +  \
+                          " units '" + attrv + "' than CF '" + un + "' !!"
+                        print '    keeping WRF ones'
+                        newattr = set_attribute(newvar, 'units', attrv)
+                    else:
+                        newattr = set_attribute(newvar, 'units', un)
+
+            newattr = set_attribute(newvar, 'coordinates', 'lon lat')
+            onewnc.sync()
+
+    onewnc.close()
+    wrfnc.close()
+
+    print fname + ": succesfull creation of file '" + ofile + "' !!"
+
+    return
+
 def WRF_toCF(values, ncfile):
     """ Function to pass a WRF original file to CF-conventions