Skip to content
Snippets Groups Projects
Commit 0df8f721 authored by Lukas Kugler's avatar Lukas Kugler
Browse files

minor

parent 5e5443fa
No related branches found
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
scripts/create_obsseq.py
+ 71
− 73
View file @ 0df8f721
......@@ -27,10 +27,6 @@ def add_timezone_UTC(t):
def get_dart_date(time_dt):
# assumes input is UTC!
#try:
# time_dt = add_timezone_UTC(time_dt)
#except:
# time_dt = time_dt.replace(tzinfo=dt.timezone.utc) # overwrites existing timezone!
days_since_010120 = 145731
ref_day = dt.datetime(2000, 1, 1, tzinfo=dt.timezone.utc)
dart_date_day = str((time_dt - ref_day).days + days_since_010120)
......@@ -102,6 +98,65 @@ def calc_obs_locations(n_obs, distance_between_obs_meters,
assert len(coords) == n_obs, (len(coords), n_obs)
return coords
def write_generic_obsseq(obs_name, obs_kind_nr, error_var, coords,
dart_date_day, secs_thatday, output_path):
n_obs_str = str(int(n_obs))
error_var = str(error_var)
line_obstypedef = obs_kind_nr+' '+obs_name
msg = """
obs_sequence
obs_kind_definitions
1
"""+line_obstypedef+"""
num_copies: 0 num_qc: 0
num_obs: """+n_obs_str+" max_num_obs: "+n_obs_str+"""
first: 1 last: """+n_obs_str
for i_obs in range(1, int(n_obs)+1):
lon = coords[i_obs-1][1]
lat = coords[i_obs-1][0]
hgt_m = str(coords[i_obs-1][2])+'.000'
lon_rad = str(degr_to_rad(lon))
lat_rad = str(degr_to_rad(lat))
# compile text
if i_obs < int(n_obs):
msg += """
OBS """+str(i_obs)+"""
-1 """+str(i_obs+1)+""" -1
obdef
loc3d
"""+lon_rad+""" """+lat_rad+""" """+hgt_m+""" 3
kind
"""+obs_kind_nr+"""
"""+secs_thatday+""" """+dart_date_day+"""
"""+error_var
if i_obs == int(n_obs): # last_observation
# compile text
msg += """
OBS """+str(i_obs)+"""
"""+str(i_obs-1)+""" -1 -1
obdef
loc3d
"""+lon_rad+""" """+lat_rad+""" """+hgt_m+""" 3
kind
"""+obs_kind_nr+"""
"""+secs_thatday+""" """+dart_date_day+"""
"""+error_var
fpath = output_path+'/obs_seq.in'
try:
os.remove(fpath)
except OSError:
pass
with open(fpath, 'w') as f:
f.write(msg)
print(fpath, 'saved.')
def sat(time_dt, channel_id, n_obs, error_var, distance_between_obs_meters,
output_path='./', fpath_obs_locations=False):
"""Create obs_seq.in
......@@ -136,7 +191,7 @@ def sat(time_dt, channel_id, n_obs, error_var, distance_between_obs_meters,
coords = calc_obs_locations(n_obs, distance_between_obs_meters,
coords_from_domaincenter=False,
fpath_obs_locations=False)
fpath_obs_locations=fpath_obs_locations)
time_dt = add_timezone_UTC(time_dt)
sun_az = str(get_azimuth(lat0, lon0, time_dt))
......@@ -218,7 +273,8 @@ def gen_coords(n_obs, distance_between_obs_meters, heights,
coords_from_domaincenter=True, fpath_obs_locations=False):
coords = calc_obs_locations(n_obs, distance_between_obs_meters,
coords_from_domaincenter=True, fpath_obs_locations=False)
coords_from_domaincenter=coords_from_domaincenter,
fpath_obs_locations=fpath_obs_locations)
# append height
coords2 = []
......@@ -227,7 +283,8 @@ def gen_coords(n_obs, distance_between_obs_meters, heights,
coords2.append(coords[i] + (hgt_m,))
n_obs = len(coords2)
print('effective number of observations (with vertical levels):', n_obs, ' on each level:', len(coords))
print('effective number of observations (with vertical levels):', n_obs,
' on each level:', len(coords))
return coords2
......@@ -238,11 +295,11 @@ def generic_obs(obs_type, time_dt, n_obs, error_var, distance_between_obs_meters
'RADAR': {'name': 'RADAR_REFLECTIVITY', 'nr': '37'}
}
coords_from_domaincenter = True
heights = np.arange(1000, 15001, 1000)
coords = gen_coords(n_obs, distance_between_obs_meters, heights,
coords_from_domaincenter=True, fpath_obs_locations=False)
coords_from_domaincenter=False,
fpath_obs_locations=fpath_obs_locations)
dart_date_day, secs_thatday = get_dart_date(time_dt)
print('secs, days:', secs_thatday, dart_date_day)
......@@ -254,65 +311,6 @@ def generic_obs(obs_type, time_dt, n_obs, error_var, distance_between_obs_meters
dart_date_day, secs_thatday, output_path)
def write_generic_obsseq(obs_name, obs_kind_nr, error_var, coords,
dart_date_day, secs_thatday, output_path):
n_obs_str = str(int(n_obs))
error_var = str(error_var)
line_obstypedef = obs_kind_nr+' '+obs_name
msg = """
obs_sequence
obs_kind_definitions
1
"""+line_obstypedef+"""
num_copies: 0 num_qc: 0
num_obs: """+n_obs_str+" max_num_obs: "+n_obs_str+"""
first: 1 last: """+n_obs_str
for i_obs in range(1, int(n_obs)+1):
lon = coords[i_obs-1][1]
lat = coords[i_obs-1][0]
hgt_m = str(coords[i_obs-1][2])+'.000'
lon_rad = str(degr_to_rad(lon))
lat_rad = str(degr_to_rad(lat))
# compile text
if i_obs < int(n_obs):
msg += """
OBS """+str(i_obs)+"""
-1 """+str(i_obs+1)+""" -1
obdef
loc3d
"""+lon_rad+""" """+lat_rad+""" """+hgt_m+""" 3
kind
"""+obs_kind_nr+"""
"""+secs_thatday+""" """+dart_date_day+"""
"""+error_var
if i_obs == int(n_obs): # last_observation
# compile text
msg += """
OBS """+str(i_obs)+"""
"""+str(i_obs-1)+""" -1 -1
obdef
loc3d
"""+lon_rad+""" """+lat_rad+""" """+hgt_m+""" 3
kind
"""+obs_kind_nr+"""
"""+secs_thatday+""" """+dart_date_day+"""
"""+error_var
fpath = output_path+'/obs_seq.in'
try:
os.remove(fpath)
except OSError:
pass
with open(fpath, 'w') as f:
f.write(msg)
print(fpath, 'saved.')
if __name__ == '__main__':
time_dt = dt.datetime(2008, 7, 30, 15, 30)
n_obs = 100
......
scripts/gen_synth_obs.py
+ 7
− 7
View file @ 0df8f721
......@@ -46,8 +46,7 @@ def read_prior_obs(f_obs_prior):
for j in range(5, 5+exp.n_ens):
prior_ens.append(float(obsseq[i+j].strip()))
OBSs.append(dict(observed=observed, truth=truth,
prior_ens=np.array(prior_ens)))
OBSs.append(dict(observed=observed, truth=truth, prior_ens=np.array(prior_ens)))
return OBSs
def edit_obserr_in_obsseq(fpath_obsseqin, OEs):
......@@ -88,6 +87,7 @@ def set_input_nml(sat_channel=False, just_prior_values=False):
rttov_nml = cluster.scriptsdir+'/../templates/obs_def_rttov.IR.nml'
append_file(cluster.dartrundir+'/input.nml', rttov_nml)
if __name__ == "__main__":
time = dt.datetime.strptime(sys.argv[1], '%Y-%m-%d_%H:%M')
......
scripts/run_obs_diag.py
+ 26
− 19
View file @ 0df8f721
......@@ -17,11 +17,12 @@ def run(folder_obs_seq_final):
f.write(fin)
f.write('\n')
for obserr_iszero in ['.true.', '.false.']:
print('ensure correct input.nml')
copy(cluster.scriptsdir+'/../templates/input.nml',
rundir_program+'/input.nml') #cluster.dartrundir+'/input.nml')
rundir_program+'/input.nml')
sed_inplace(rundir_program+'/input.nml', '<n_ens>', str(int(exp.n_ens)))
sed_inplace(rundir_program+'/input.nml', '<zero_error_obs>', obserr_iszero)
append_file(rundir_program+'/input.nml', cluster.scriptsdir+'/../templates/obs_def_rttov.VIS.nml')
# run obs_diag
......@@ -34,9 +35,15 @@ def run(folder_obs_seq_final):
pass
os.system('./obs_diag >& obs_diag.log') # caution, this overwrites obs_seq_to_netcdf
# move output to archive
outdir = '/'.join(folder_obs_seq_final.split('/')[:-1])
print('moving output to', outdir+'/obs_diag_output.nc')
copy(rundir_program+'/obs_diag_output.nc', outdir+'/obs_diag_output.nc')
if obserr_iszero == '.true.':
fout = '/obs_diag_wrt_truth.nc'
elif obserr_iszero == '.false.':
fout = '/obs_diag_wrt_obs.nc'
print('moving output to', outdir+fout)
copy(rundir_program+'/obs_diag_output.nc', outdir+fout)
print('running obs_seq_to_netcdf program')
shutil.copy(cluster.dart_srcdir+'/obs_seq_to_netcdf-bak', cluster.dart_srcdir+'/obs_seq_to_netcdf')
......
templates/input.nml
+ 1
− 1
View file @ 0df8f721
......@@ -316,7 +316,7 @@
print_mismatched_locs = .false.,
create_rank_histogram = .true.,
outliers_in_histogram = .true.,
use_zero_error_obs = .true.,
use_zero_error_obs = <zero_error_obs>,
verbose = .false.
/
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment