From 0c89705d5c1bca5559fa85b1425620e1c934045b Mon Sep 17 00:00:00 2001
From: lkugler <lukas.kugler@gmail.com>
Date: Wed, 7 Jun 2023 14:57:28 +0200
Subject: [PATCH] updating from master
---
dartwrf/assim_synth_obs.py | 42 +++++++++++++++++++++++++-------------
1 file changed, 28 insertions(+), 14 deletions(-)
diff --git a/dartwrf/assim_synth_obs.py b/dartwrf/assim_synth_obs.py
index 271a393..9d42cc4 100755
--- a/dartwrf/assim_synth_obs.py
+++ b/dartwrf/assim_synth_obs.py
@@ -15,13 +15,17 @@ from dartwrf.server_config import cluster
wrfout_format = 'wrfout_d01_%Y-%m-%d_%H:%M:%S'
-def link_nature_to_dart_truth(time):
- """Set a symlink from the WRFout file to be used as nature to the run_DART folder
-
- Args:
- time (dt.datetime): Time at which observations will be made
+def _prepare_DART_grid_template():
+ # DART needs a wrfinput file as a template for the grid
+ # No data will be read from this file, but the grid information must match exactly.
+ symlink(cluster.dartrundir + "/prior_ens1/wrfout_d01",
+ cluster.dartrundir + "/wrfinput_d01")
+
+def _copy_nature_to_dart(time):
+ """Copies wrfout_d01 from nature run to DART directory
+
+ TODO: This is a bit of a hack, because it is not explicit about where to take the nature from.
"""
- # get wrfout_d01 from nature run
# find the file in any init directory
fformat = 'wrfout_d01_%Y-%m-%d_%H:%M:%S'
f_nat = glob.glob(cluster.archive_base + '/' + exp.nature_expname + '/*/1/'+time.strftime(fformat))[0]
@@ -33,16 +37,24 @@ def link_nature_to_dart_truth(time):
print("linked", f_nat, "to", cluster.dart_rundir + "/wrfout_d01")
f_wrfout_nature = time.strftime(exp.nature+'/'+wrfout_format)
- if os.path.exists(f_wrfout_nature):
- print("linking nature to DART & georeferencing")
- shutil.copy(f_wrfout_nature, cluster.dartrundir + "/wrfout_d01")
- print("linked", f_wrfout_nature, "to", cluster.dartrundir + "/wrfout_d01")
- if cluster.geo_em_for_WRF_ideal:
- wrfout_add_geo.run(cluster.geo_em_for_WRF_ideal, cluster.dart_rundir + "/wrfout_d01")
+ assert os.path.exists(f_wrfout_nature)
- else: # if nature is not available due to any reason
- print('-> has no nature, not copying nature')
+ print("linking nature to DART & georeferencing")
+ shutil.copy(f_wrfout_nature, cluster.dartrundir + "/wrfout_d01")
+ print("linked", f_wrfout_nature, "to", cluster.dartrundir + "/wrfout_d01")
+ if cluster.geo_em_for_WRF_ideal:
+ wrfout_add_geo.run(cluster.geo_em_for_WRF_ideal, cluster.dart_rundir + "/wrfout_d01")
+def prepare_nature_dart(time):
+ """Prepares DART nature (wrfout_d01) if available
+
+ Args:
+ time (dt.datetime): Time at which observations will be made
+ """
+ try:
+ _copy_nature_to_dart(time)
+ except FileExistsError: # if nature is not available due to any reason
+ print('-> has no nature, not copying nature')
def prepare_prior_ensemble(assim_time, prior_init_time, prior_valid_time, prior_path_exp):
@@ -119,6 +131,8 @@ def write_list_of_outputfiles():
def filter(nproc=12):
+ _prepare_DART_grid_template()
+
print("time now", dt.datetime.now())
print("running filter")
os.chdir(cluster.dart_rundir)
--
GitLab