add prefix for simulate

CDCgov · Jul 3, 2024 · b666062 · b666062
1 parent c133b65
commit b666062
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 1 deletion.
diff --git a/pipeline/src/simulate/generate_truthdata.jl b/pipeline/src/simulate/generate_truthdata.jl
@@ -21,13 +21,15 @@ the `simulate` function to generate the truth data. This is the default method.
 """
 function generate_truthdata(
         truth_data_config, pipeline::AbstractEpiAwarePipeline; plot = true,
-        datadir_str = "truth_data", prefix = "truth_data")
+        datadir_str = "truth_data")
     default_params = make_default_params(pipeline)
     config = TruthSimulationConfig(
         truth_process = default_params["Rt"], gi_mean = truth_data_config["gi_mean"],
         gi_std = truth_data_config["gi_std"], logit_daily_ascertainment = default_params["logit_daily_ascertainment"],
         cluster_factor = default_params["cluster_factor"], I0 = default_params["I0"])
 
+    prefix = _simulate_prefix(pipeline)
+
     truthdata, truthfile = produce_or_load(
         simulate, config, datadir(datadir_str); prefix = prefix)
     if plot

diff --git a/pipeline/src/simulate/simulate.jl b/pipeline/src/simulate/simulate.jl
@@ -1,2 +1,3 @@
 include("TruthSimulationConfig.jl")
+include("simulate_prefix.jl")
 include("generate_truthdata.jl")
diff --git a/pipeline/src/simulate/simulate_prefix.jl b/pipeline/src/simulate/simulate_prefix.jl
@@ -0,0 +1,12 @@
+"""
+Internal method for setting the prefix for the truth data file name.
+"""
+_simulate_prefix(pipeline::AbstractEpiAwarePipeline) = "truth_data"
+
+_simulate_prefix(pipeline::SmoothOutbreakPipeline) = "truth_data_smooth_outbreak"
+
+_simulate_prefix(pipeline::MeasuresOutbreakPipeline) = "truth_data_measures_outbreak"
+
+_simulate_prefix(pipeline::SmoothEndemicPipeline) = "truth_data_smooth_endemic"
+
+_simulate_prefix(pipeline::RoughEndemicPipeline) = "truth_data_rough_endemic"