Changeset 216 in lmdz_wrf for trunk/tools

Timestamp:

Jan 14, 2015, 5:27:05 PM (10 years ago)

Author:

lfita

Message:

Almost working version... up to file filling!

File:

• trunk/tools/iniaqua.py (modified) (6 diffs)

trunk/tools/iniaqua.py

@@ -17 +17 @@
 warnmsg='WARNING -- warning -- WARNING -- warning'
 
-filekinds = ['CF', 'WRF']
+filekinds = ['CF', 'LMDZ', 'WRF']
 
 ## g.e. # iniaqua.py -d 32,32,39 -p hybdrid -o WRF -t 19791201000000 -z tropo -q 2
@@ -243 +243 @@
 #     .... variables locales  ....
 #
-    xlon = np.zeros((dx+1), dtype=np.float32),
+    xlon = np.zeros((dx+1), dtype=np.float32)
     xprimm = np.zeros((dx+1), dtype=np.float32)
     xtild = np.zeros((nmax2), dtype=np.float32)
@@ -2367 +2367 @@
     return ran1
 
+def name_variables(filekind):
+    """ Function to provide name of the variables and their atributes as function of
+      the output type of file
+      filekind= kind of file
+        'CF': CF-convention
+        'LMDZ': LMDZ style
+        'WRF': WRF style
+    """
+    fname = 'name_variables'
+
+    dimnames = {}
+    varnames = {}
+
+# Standard dimensions
+    dimn = ['x','y','z','t']
+
+# Standard variables
+    varn = ['lon', 'lat', 'lev', 'time', 'temp', 'tsfc', 'u10m', 'v10m', 'u', 'v',   \
+      'zsfc', 'geopot', 'psfc', 'pres', 'H2Ov', 'H2Ol']
+
+# Standard variables' attribute names
+    stdattrn = ['standard_name', 'long_name', 'units']
+
+# Extra variables' attribute names
+#    kextrattrn = []
+
+# Dictionary with the values for each standard variable 
+#   kvarn['stdn'] = ['name', 'dims(list)', 'std_name', 'long_name', 'units', 'FillValue']
+#     None: No value
+
+    kdimn = {}
+    kvarn = {}
+    kattrn = {}
+    if filekind == 'CF':
+        kdimn['x'] = 'x'
+        kdimn['y'] = 'y'
+        kdimn['z'] = 'z'
+        kdimn['t'] = 'time'
+
+        kvarn['lon'] = ['lon',['x'],'longitude','Longitude','degrees_east',None]
+        kvarn['lat'] = ['lat',['y'],'latitude','Latitude','degrees_north',None]
+        kvarn['lev'] = ['lev',['z'],'levels','Levels','-',None]
+        kvarn['time'] = ['time',['t'],'time','Time',                                 \
+          'monutes since 1949-12-01 00:00:00', None]
+        kvarn['temp'] = ['ta',['t','z','y','x'],'air_temperature','Air temperature', \
+          'K',None]
+        kvarn['tsfc'] = ['tas',['t','y','x'],'air_temperature','Air temperature','K',None]
+        kvarn['u10m'] = ['uas',['t','y','x'],'eastward_wind','eastward wind','ms-1',None]
+        kvarn['v10m'] = ['vas',['t','y','x'],'northward_wind','northward wind','ms-1',None]
+        kvarn['u'] = ['ua',['t','z','y','x'],'eastward_wind','eastward wind','ms-1',None]
+        kvarn['v'] = ['va',['t','z','y','x'],'northward_wind','northward wind','ms-1',None]
+        kvarn['zsfc'] = ['zgs',['t','y','x'],'surface_geopotential_height',                      \
+          'surface geopotential height','m2s-2',None]
+        kvarn['geopot'] = ['zg',['t','z','y','x'],'geopotential_height','geopotential height',       \
+          'm2s-2',None]
+        kvarn['psfc'] = ['ps',['t','z','y','x'],'surface_air_pressure','surface pressure','Pa',None]
+        kvarn['pres'] = ['pres',['t','z','y','x'],'air_pressure','pressure','Pa',None]
+        kvarn['H2Ov'] = ['r',['t','z','y','x'],'water_mixing_ratio','water mixing ratio','kgkg-1',   \
+          None]
+        kvarn['H2Ol'] = ['c',['t','z','y','x'],'condensed_water_mixing_ratio',                       \
+          'condensed water mixing ratio','kgkg-1',None]
+
+        kattrn['standard_name'] = 'standard_name'
+        kattrn['long_name'] = 'long_name'
+        kattrn['units'] = 'units'
+
+        kextrattrn = ['']
+
+    elif filekind == 'LMDZ':
+        kdimn['x'] = 'x'
+        kdimn['y'] = 'y'
+        kdimn['z'] = 'presnivs'
+        kdimn['t'] = 'time_counter'
+
+        kvarn['lon'] = ['lon',['x'],'longitude','Longitude','degrees_east', None]
+        kvarn['lat'] = ['lat',['y'],'latitude','Latitude','degrees_north', None]
+        kvarn['lev'] = ['presnivs',['z'],'model_level_number','Vertical levels','Pa',    \
+          None]
+        kvarn['time'] = ['time_counter',['t'],'time','Time',                             \
+          'seconds since 1980-01-01 00:24:00', None]
+        kvarn['temp'] = ['temp',['t','z','y','x'],'Air temperature','Air temperature','K',           \
+          9.96921e+36]
+        kvarn['tsfc'] = ['t2m',['t','y','x'],'Temperature 2m','Temperature 2m','K',9.96921e+36]
+        kvarn['u10m'] = ['u10m',['t','y','x'],'Vent zonal 10m','Vent zonal 10m','m/s',9.96921e+36]
+        kvarn['v10m'] = ['v10m',['t','y','x'],'Vent meridien 10m','Vent meridien 10m','m/s',     \
+          9.96921e+36]
+        kvarn['u'] = ['vitu',['t','z','y','x'],'Zonal wind','Zonal wind','m/s',9.96921e+36]
+        kvarn['v'] = ['vitv',['t','z','y','x'],'Meridional wind','Meridional wind','m/s',9.96921e+36]
+        kvarn['zsfc'] = ['phis',['t','y','x'],'Surface geop.height','Surface geop.height',       \
+          'm2/s2',9.96921e+36]
+        kvarn['geopot'] = ['geop',['t','z','y','x'],'Geopotential height','Geopotential height',     \
+          'm2/s2',9.96921e+36]
+        kvarn['psfc'] = ['psol',['t','y','x'],'Surface Pressure','Surface Pressure','Pa',        \
+          9.96921e+36]
+        kvarn['pres'] = ['pres',['t','z','y','x'],'Air pressure','Air pressure','Pa',9.96921e+36]
+        kvarn['H2Ov'] = ['ovap',['t','z','y','x'],'Specific humidity','Specific humidity','kg/kg',   \
+          9.96921e+36]
+        kvarn['H2Ol'] = ['oliq',['t','z','y','x'],'Condensed water','Condensed water','kg/kg',       \
+          9.96921e+36]
+
+        kattrn['standard_name'] = 'standard_name'
+        kattrn['long_name'] = 'long_name'
+        kattrn['units'] = 'units'
+
+        kextrattrn = ['coordinates']
+
+    elif filekind == 'WRF':
+        kdimn['x'] = 'west_east'
+        kdimn['y'] = 'south_north'
+        kdimn['z'] = 'bottom_top'
+        kdimn['t'] = 'Time'
+
+        kvarn['lon'] = ['XLONG',['t','y','x'], None,'LONGITUDE, WEST IS NEGATIVE','degree_east', \
+          None]
+        kvarn['lat'] = ['XLAT',['t','y','x'], None,'LATITUDE, SOUTH IS NEGATIVE','degree_north', \
+          None]
+        kvarn['lev'] = ['ZNU',['t','z'],None,'eta values on half (mass) levels','',None]
+        kvarn['time'] = ['Times',['t','DateStrLen'],None,None,None,None]
+        kvarn['temp'] = ['T',['t','z','y','x'],None,'perturbation potential temperature (theta-t0)','K'\
+          ,None]
+        kvarn['tsfc'] = ['T2',['t','y','x'],None,'TEMP at 2 M','K',None]
+        kvarn['u10m'] = ['U10',['t','y','x'],None,'U at 10 M','m s-1',None]
+        kvarn['v10m'] = ['V10',['t','y','x'],None,'V at 10 M','m s-1',None]
+        kvarn['u'] = ['U',['t','z','y','x'],None,'x-wind component','m s-1',None]
+        kvarn['v'] = ['V',['t','z','y','x'],None,'y-wind component','m s-1',None]
+        kvarn['zsfc'] = [None,['t','y','x'],None,None,None,None]
+        kvarn['geopot'] = ['PHB',['t','z','y','x'],None,'perturbation geopotential','m2 s-2',None]
+        kvarn['psfc'] = [None,['t','y','x'],None,None,None,None]
+        kvarn['pres'] = ['P',['t','z','y','x'],None,'perturbation pressure','Pa',None]
+        kvarn['H2Ov'] = ['QVAPOR',['t','z','y','x'],None,'Water vapor mixing ratio','kg kg-1',None]
+        kvarn['H2Ol'] = ['QCLOUD',['t','z','y','x'],None,'Cloud water mixing ratio','kg kg-1',None]
+
+        kattrn['standard_name'] = None
+        kattrn['long_name'] = 'description'
+        kattrn['units'] = 'units'
+
+        kextrattrn = ['FieldType','MemoryOrder','stagger']
+
+#    elif filekind == 'OTHER':
+#        kdimn['x'] = ''
+#        kdimn['y'] = ''
+#        kdimn['z'] = ''
+#        kdimn['t'] = ''
+
+#        kvarn['lon'] = ['',['t','z','y','x'],'','','',]
+#        kvarn['lat'] = ['',['t','z','y','x'],'','','',]
+#        kvarn['lev'] = ['',['t','z','y','x'],'','','',]
+#        kvarn['time'] = ['',['t','z','y','x'],'','','',]
+#        kvarn['temp'] = ['',['t','z','y','x'],'','','',]
+#        kvarn['tsfc'] = ['',['t','y','x'],'','','',]
+#        kvarn['u10m'] = ['',['t','y','x'],'','','',]
+#        kvarn['v10m'] = ['',['t','y','x'],'','','',]
+#        kvarn['u'] = ['',['t','z','y','x'],'','','',]
+#        kvarn['v'] = ['',['t','z','y','x'],'','','',]
+#        kvarn['zsfc'] = ['',['t','y','x'],'','','',]
+#        kvarn['geopot'] = ['',['t','z','y','x'],'','','',]
+#        kvarn['psfc'] = ['',['t','y','x'],'','','',]
+#        kvarn['pres'] = ['',['t','z','y','x'],'','','',]
+#        kvarn['H2Ov'] = ['',['t','z','y','x'],'','','',]
+#        kvarn['H2Ol'] = ['',['t','z','y','x'],'','','',]
+
+#        kattrn['standard_name'] = ''
+#        kattrn['long_name'] = ''
+#        kattrn['units'] = ''
+
+#        kextrattrn = ['']
+    else:
+        print errormsg
+        print '  ' + fname + ": filekind '" + filekind + "' not ready !!"
+        quit(-1)
+
+    return kdimn, kvarn, kattrn, kextrattrn
+
+def generic_NetCDFfile(ncobj, dims, kfile, kdimns, kvarns, stdattrns, extrattrns):
+    """ Function to fill a generic NetCDF file
+      ncobj= NetCDF file object to which the variables have to be created
+      kfile= kind of file
+        'CF': CF-convention
+        'LMDZ': LMDZ style
+        'WRF': WRF style
+      dims= [dimx, dimy, dimz] dimensions of the file
+      kdimns= dictionary for the specific names for the standard dimensions (x,y,z,t)
+      kvarns= dictionary for the specific values for the standard variables
+        kvarn['stdn'] = ['name', 'dims(list)', 'std_name', 'long_name', 'units', 'FillValue']
+          None: No value
+      stdattrns= dictionary for the specific values for the variables' standard attributes
+      extrattrns= list for the specific values for the variables' extra attributes
+    """
+
+# Creation of dimensions
+##
+    newdim = ncobj.createDimension(kdimns['x'], dims[0])
+    newdim = ncobj.createDimension(kdimns['y'], dims[1])
+    newdim = ncobj.createDimension(kdimns['z'], dims[2])
+    newdim = ncobj.createDimension(kdimns['t'], None)
+
+    if kfile == 'WRF':
+        newdim = ncobj.createDimension('DateStrLen', 19)
+
+# Creation of variables
+##
+    for varn in kvarns.keys():
+        varvals = kvarns[varn]
+        dimsvar = []
+        for dimn in varvals[1]:
+            dimsvar.append(kdimns[dimn])
+
+        if varvals[5] is not None:
+            nerwvar = ncobj.createVariable(varvals[0], 'f4', tuple(dimsvar),         \
+              fill_value=np.float(varvals[5]))
+        else:
+            newvar = ncobj.createVariable(varvals[0], 'f4', tuple(dimsvar))
+
+# Attributes
+        for iattr in range(len(stdattrns)):
+            attrn = stdattrns[iattr]
+            if kattrns[attrn] is not None:
+                newvar.setncattr(kattrns[attrn], varvals[2+iattr])
+
+# Extra attributes
+        for iattr in range(len(extrattrns)):
+            attrn = extrrattrns[iattr]
+            if kfile == 'LMDZ':
+                if attrn == 'coordinates':
+                    attrv = ''
+                    for din in varvals[1]:
+                        attrv = attrv + kdimns[din] + ' '
+            elif kfile == 'WRF':
+                if attrn == 'FieldType':
+                    attrv = '104'
+                elif attrn == 'MemoryOrder':
+                    attrv = ''
+                    Ndims = len(varvals[1])
+                    for idim in range(Ndims,0,-1):
+                        if varvals[1][idim] != 'T':
+                            attrv = attrv + varvals[1][idim].uppercase
+                elif attrn == 'stagger':
+                    staggeredvars = {}
+                    staggeredvars['U'] = 'X'
+                    staggeredvars['V'] = 'Y'
+                    staggeredvars['PH'] = 'Z'
+
+                    if ncvar.searchInlist(staggeredvars.keys(),varvals[0]):
+                        attrv = staggeredvars[vavals[0]]
+
+            newvar.setncattr(extrattrns[iattr], attrv)
+
+        newvar.sync()
+
+    return
 
 ####### ###### ##### #### ### ## #
@@ -2428 +2678 @@
 if opts.okind == 'CF':
     varnames = ['zg', 'ta', 'ua', 'va', 'pres', 'r']
+    timev = float(opts.time)
 # Reference time from 1949-12-01 00:00:00 UTC
     timev = ncvar.datetimeStr_conversion(opts.time,'YmdHMS','cfTime')
@@ -2552 +2803 @@
       apolnorth, apolsouth)
 else:
-    pks, pk, pkf = exner_milieu(ip1jmp1, dimx, dimy, dimz, press, pres, beta)
+    pks, pk, pkf = exner_milieu(dimx, dimy, dimz, press, pres, beta)
 
 masse = massdair(dimx,dimy,dimz,pres,airesurge)
@@ -2599 +2850 @@
 ncf = NetCDFFile(ofile, 'w')
 
-# Dimensions creation
-ncf.createDimension(dimnames[0], None)
-ncf.createDimension(dimnames[1], dimz)
-ncf.createDimension(dimnames[2], dimy)
-ncf.createDimension(dimnames[3], dimx)
+# File structure creation
+filedimns, filevarns, filevarattr, filevarxtrattr = name_variables(opts.okind)
+generic_NetCDFfile(ncf,[dimx,dimy,dimz], opts.okind, filedimns, filevarns,           \
+  filevarattr, filevarxtrattr)
+
+# File filling
+for varn in ncf.variables.keys():
+    varobj = ncf.variables[varn]
+    if varn == 'lon': varobj[:] = lon
+    elif varn == 'lat': varobj[:] = lat
+    elif varn == 'lev': varobj[:] = presnivs
+    elif varn == 'time': varobj[:] = timev
+    elif varn == 'temp': varobj[0,:,:,:] = theta[:]
+    elif varn == 'tsfc': varobj[:] = theta[0,:,:]
+    elif varn == 'u10m': varobj[:] = ucov[0,:,:]
+    elif varn == 'v10m': varobj[:] = vcov[0,:,:]
+    elif varn == 'u': varobj[0,:,:,:] = ucov[:]
+    elif varn == 'v': varobj[0,:,:,:] = vcov[:]
+    elif varn == 'zsfc': varobj[0,:,:] = phisfc[:]
+    elif varn == 'geopot': varobj[0,:,:] = phisall[:]
+    elif varn == 'psfc': varobj[0,:,:] = press[:]
+    elif varn == 'pres': varobj[0,:,:,:] = pres[:]
+    elif varn == 'H2Ov': varobj[0,:,:,:] = q[:,:,:0]
+    elif varn == 'H2Ol': varobj[0,:,:,:] = q[:,:,:1]
+
 
 ncf.sync()