Skip to content
Snippets Groups Projects
Commit dcc0bf6f authored by David Jean du Preez (local)'s avatar David Jean du Preez (local)
Browse files

fix pm10 unit and improve output

parent 01b683a9
Branches main
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
dynamic/palm_dynamic.py
+ 4
− 3
View file @ dcc0bf6f
...@@ -427,9 +427,10 @@ for wrf_file in wrf_files_proc: ...@@ -427,9 +427,10 @@ for wrf_file in wrf_files_proc:
v_out[:] = regridder.regrid(v_wrf[...,regridder.ys,regridder.xs]) v_out[:] = regridder.regrid(v_wrf[...,regridder.ys,regridder.xs])
# PM10 # PM10
elif varname == 'PM10': elif varname == 'PM10':
v_wrf = f_wrf.variables[varname] v_wrf = f_wrf.variables[varname]
v_out = f_out.createVariable(varname.lower(), 'f4', v_wrf.dimensions) v_wrf_val = f_wrf.variables[varname][:]*1e-9
v_out[:] = regridder.regrid(v_wrf[...,regridder.ys,regridder.xs]) v_out = f_out.createVariable(varname.lower(), 'f4', v_wrf.dimensions)
v_out[:] = regridder.regrid(v_wrf_val[...,regridder.ys,regridder.xs])
else: else:
# other dynamical & chemical variables # other dynamical & chemical variables
v_wrf = f_wrf.variables[varname] v_wrf = f_wrf.variables[varname]
......
dynamic/palm_dynamic_output.py
+ 31
− 25
View file @ dcc0bf6f
...@@ -179,27 +179,28 @@ def palm_dynamic_output(wrf_files, interp_files, dynamic_driver_file, times_sec, ...@@ -179,27 +179,28 @@ def palm_dynamic_output(wrf_files, interp_files, dynamic_driver_file, times_sec,
for var in dynam_chem_variables: for var in dynam_chem_variables:
if (var == 'pt' or var == 'qv'): if (var == 'pt' or var == 'qv'):
_val_init_var = outfile.createVariable('init_atmosphere_'+ var, "f4", ("z", "y", "x"),fill_value=fillvalue_float) _val_init_var = outfile.createVariable('init_atmosphere_'+ var, "f4", ("z", "y", "x"),fill_value=fillvalue_float)
_val_init_var.setncattr('lod', 2)
elif (var == 'soil_m' or var == 'soil_t'): elif (var == 'soil_m' or var == 'soil_t'):
_val_init_var = outfile.createVariable('init_'+ var,"f4", ("zsoil", "y", "x"),fill_value=fillvalue_float) _val_init_var = outfile.createVariable('init_'+ var,"f4", ("zsoil", "y", "x"),fill_value=fillvalue_float)
_val_init_var.setncattr('lod', 2)
elif var == 'u': elif var == 'u':
_val_init_var = outfile.createVariable('init_atmosphere_'+ var, "f4", ("z", "y", "xu"),fill_value=fillvalue_float) _val_init_var = outfile.createVariable('init_atmosphere_'+ var, "f4", ("z", "y", "xu"),fill_value=fillvalue_float)
_val_init_var.setncattr('lod', 2)
elif var == 'v': elif var == 'v':
_val_init_var = outfile.createVariable('init_atmosphere_'+ var, "f4", ("z", "yv", "x"),fill_value=fillvalue_float) _val_init_var = outfile.createVariable('init_atmosphere_'+ var, "f4", ("z", "yv", "x"),fill_value=fillvalue_float)
_val_init_var.setncattr('lod', 2)
elif var == 'w': elif var == 'w':
_val_init_var = outfile.createVariable('init_atmosphere_'+ var, "f4", ("zw", "y", "x"),fill_value=fillvalue_float) _val_init_var = outfile.createVariable('init_atmosphere_'+ var, "f4", ("zw", "y", "x"),fill_value=fillvalue_float)
_val_init_var.setncattr('lod', 2)
elif var in wrfchem_spec: elif var in wrfchem_spec:
_val_init_var = outfile.createVariable('init_atmosphere_'+ var.upper(), "f4", ("z",),fill_value=fillvalue_float) _val_init_var = outfile.createVariable('init_atmosphere_'+ var.upper(), "f4", ("z",),fill_value=fillvalue_float)
else:
_val_init_var = outfile.createVariable('init_atmosphere_'+ var, "f4",('z',),fill_value=fillvalue_float)
# add attributes
if var in atmos_var:
_val_init_var.setncattr('lod', 2)
else:
_val_init_var.setncattr('lod', 1)
if (var == 'PM10' or var == 'PM2_5_DRY'): if (var == 'PM10' or var == 'PM2_5_DRY'):
_val_init_var.setncattr('lod', 1)
_val_init_var.setncattr('units','kg/m3') _val_init_var.setncattr('units','kg/m3')
else: else:
_val_init_var.setncattr('lod', 1)
_val_init_var.setncattr('units','ppm') _val_init_var.setncattr('units','ppm')
# create time dependent variables # create time dependent variables
if not nested_domain: if not nested_domain:
if (var == 'soil_m' or var == 'soil_t'): if (var == 'soil_m' or var == 'soil_t'):
...@@ -262,7 +263,6 @@ def palm_dynamic_output(wrf_files, interp_files, dynamic_driver_file, times_sec, ...@@ -262,7 +263,6 @@ def palm_dynamic_output(wrf_files, interp_files, dynamic_driver_file, times_sec,
_val_ls_forcing.setncattr('units', 'kg/m3') _val_ls_forcing.setncattr('units', 'kg/m3')
else: else:
_val_ls_forcing.setncattr('units', 'ppm') _val_ls_forcing.setncattr('units', 'ppm')
#outfile.close()
# function - create dynamical & chemical variables # function - create dynamical & chemical variables
add_interpDim(dynam_chem_variables) add_interpDim(dynam_chem_variables)
...@@ -275,31 +275,37 @@ def palm_dynamic_output(wrf_files, interp_files, dynamic_driver_file, times_sec, ...@@ -275,31 +275,37 @@ def palm_dynamic_output(wrf_files, interp_files, dynamic_driver_file, times_sec,
outfile = netCDF4.Dataset(dynamic_driver_file, "r+", format="NETCDF4") outfile = netCDF4.Dataset(dynamic_driver_file, "r+", format="NETCDF4")
# initialization variables # initialization variables
for var in dynam_chem_variables: for var in dynam_chem_variables:
if (var == 'soil_m' or var == 'soil_t'): if var == 'soil_m':
init_var = infile.variables['init_'+var] init_var = infile.variables['init_'+var]
_val_init_var = outfile.variables['init_'+var] _val_init_var = outfile.variables['init_'+var]
for k in range(0,_val_init_var.shape[0]):
_val_init_var[k, :, :] = init_var[0, k, :, :] * sma[:, :]
elif var == 'soil_t':
init_var = infile.variables['init_'+var]
_val_init_var = outfile.variables['init_'+var]
_val_init_var[:, :, :] = init_var[0, :, :, :]
elif (var == 'pt' or var == 'qv' or var == 'w' or var == 'u' or var == 'v'):
init_var = infile.variables['init_atmosphere_'+ var]
_val_init_var = outfile.variables['init_atmosphere_'+ var]
if var == 'u':
_val_init_var[:, :, :] = init_var[0, :, :, 1:]
elif var == 'v':
_val_init_var[:, :, :] = init_var[0, :, 1:, :]
elif (var == 'pt' or var == 'qv' or var == 'w'):
_val_init_var[:, :, :] = init_var[0, :, :, :]
elif var in wrfchem_spec: elif var in wrfchem_spec:
if var == 'PM10': if var == 'PM10':
init_var = infile.variables['init_atmosphere_'+ var.lower()] init_var = infile.variables['init_atmosphere_'+ var.lower()]
_val_init_var = outfile.variables['init_atmosphere_'+ var] _val_init_var = outfile.variables['init_atmosphere_'+ var]
_val_init_var[:] = init_var[0,:,:,:].mean(axis=(1,2))
else: else:
init_var = infile.variables['init_atmosphere_'+ var] init_var = infile.variables['init_atmosphere_'+ var]
_val_init_var = outfile.variables['init_atmosphere_'+ var.upper()] _val_init_var = outfile.variables['init_atmosphere_'+ var.upper()]
else: _val_init_var[:] = init_var[0,:,:,:].mean(axis=(1,2))
init_var = infile.variables['init_atmosphere_'+ var]
_val_init_var = outfile.variables['init_atmosphere_'+ var]
# write values
if (var == 'pt' or var == 'qv' or var == 'w' or var == 'soil_t'):
_val_init_var[:, :, :] = init_var[0, :, :, :]
elif var == 'u':
_val_init_var[:, :, :] = init_var[0, :, :, 1:]
elif var == 'v':
_val_init_var[:, :, :] = init_var[0, :, 1:, :]
elif var == 'soil_m':
for k in range(0,_val_init_var.shape[0]):
_val_init_var[k, :, :] = init_var[0, k, :, :] * sma[:, :]
if var in wrfchem_spec:
_val_init_var[:] = init_var[0,:,:,:].mean(axis=(1,2))
infile.close() infile.close()
outfile.close() outfile.close()
...@@ -331,7 +337,7 @@ def palm_dynamic_output(wrf_files, interp_files, dynamic_driver_file, times_sec, ...@@ -331,7 +337,7 @@ def palm_dynamic_output(wrf_files, interp_files, dynamic_driver_file, times_sec,
else: else:
if (var == 'h2so4' or var=='hno3' or var== 'nh3' or var== 'ocnv' or var== 'ocsv' or var == 'no' or var == 'no2' or var == 'no3'): if (var == 'h2so4' or var=='hno3' or var== 'nh3' or var== 'ocnv' or var== 'ocsv' or var == 'no' or var == 'no2' or var == 'no3'):
var = var.upper() var = var.upper()
init_var = infile.variables['init_atmosphere_'+ var.lower()] init_var = infile.variables['init_atmosphere_'+ var.lower()]
for side in boundary: for side in boundary:
_val_ls_forcing_var = outfile.variables['ls_forcing_'+ side +'_'+var] _val_ls_forcing_var = outfile.variables['ls_forcing_'+ side +'_'+var]
......
dynamic/palm_wrf_utils.py
+ 1
− 11
View file @ dcc0bf6f
...@@ -366,7 +366,7 @@ def palm_wrf_vertical_interp(infile, outfile, wrffile, z_levels, z_levels_stag, ...@@ -366,7 +366,7 @@ def palm_wrf_vertical_interp(infile, outfile, wrffile, z_levels, z_levels_stag,
if (spec == 'PM10' or spec == 'PM2_5_DRY'): if (spec == 'PM10' or spec == 'PM2_5_DRY'):
chem_data_raw = nc_infile.variables[spec.lower()][0] chem_data_raw = nc_infile.variables[spec.lower()][0]
chem_data_raw = np.r_[chem_data_raw[0:1], chem_data_raw] chem_data_raw = np.r_[chem_data_raw[0:1], chem_data_raw]
chem_data = interpolate_1d(z_levels, height, (chem_data_raw)*1e-9) chem_data = interpolate_1d(z_levels, height, (chem_data_raw))
vdata = nc_outfile.createVariable('init_atmosphere_'+spec.lower(),"f4",("Time", "z","south_north","west_east")) vdata = nc_outfile.createVariable('init_atmosphere_'+spec.lower(),"f4",("Time", "z","south_north","west_east"))
vdata[0,:,:,:] = chem_data vdata[0,:,:,:] = chem_data
# other chemical species # other chemical species
...@@ -377,16 +377,6 @@ def palm_wrf_vertical_interp(infile, outfile, wrffile, z_levels, z_levels_stag, ...@@ -377,16 +377,6 @@ def palm_wrf_vertical_interp(infile, outfile, wrffile, z_levels, z_levels_stag,
vdata = nc_outfile.createVariable('init_atmosphere_'+spec,"f4",("Time", "z","south_north","west_east")) vdata = nc_outfile.createVariable('init_atmosphere_'+spec,"f4",("Time", "z","south_north","west_east"))
vdata[0,:,:,:] = chem_data vdata[0,:,:,:] = chem_data
# if (spec == 'PM10' or spec == 'PM2_5_DRY'):
# chem_data = interpolate_1d(z_levels, height, (chem_data_raw)*1e-9)
# # other chemical species
# else:
# chem_data = interpolate_1d(z_levels, height, chem_data_raw)
#vdata = nc_outfile.createVariable('init_atmosphere_'+spec,"f4",("Time", "z","south_north","west_east"))
#vdata[0,:,:,:] = chem_data
chem_from_wrfchem(wrfchem_spec) chem_from_wrfchem(wrfchem_spec)
# ======================== AEROSOLS ===================================== # ======================== AEROSOLS =====================================
# aerosol mass fraction for each aerosol species - interpolate # aerosol mass fraction for each aerosol species - interpolate
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment