From 56e0b1def60c5c17aeb26a40536603fb335526ff Mon Sep 17 00:00:00 2001
From: Andreas Tsouchlos <an.tsouchlos@gmail.com>
Date: Mon, 5 Dec 2022 16:42:37 +0100
Subject: [PATCH] Added omega and K as metadata

---
 sw/simulate_2d_BER.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/sw/simulate_2d_BER.py b/sw/simulate_2d_BER.py
index e28cc20..b83e7d2 100644
--- a/sw/simulate_2d_BER.py
+++ b/sw/simulate_2d_BER.py
@@ -60,6 +60,7 @@ def task_func(params):
 
 
 def get_params(code_name: str):
+    """In this function all parameters for the simulation are defined."""
     # Define global simulation parameters
 
     H_file = f"res/{code_name}.alist"
@@ -87,21 +88,24 @@ def get_params(code_name: str):
                 {"decoder": decoder, "max_iterations": max_iterations,
                  "SNR": SNR, "gamma": gamma, "n": n, "k": k})
 
-    return task_params
+    return omega, K, task_params
 
 
 def configure_new_simulation(sim_mgr: SimulationManager, code_name: str,
                              sim_name: str) -> None:
     sim = GenericMultithreadedSimulator()
 
-    sim.task_params = get_params(code_name)
+    omega, K, task_params = get_params(code_name)
+
+    sim.task_params = task_params
     sim.task_func = task_func
     sim.format_func = partial(misc.pgf_reformat_data_3d, x_param_name="SNR",
                               y_param_name="gamma",
                               z_param_names=["BER", "FER", "DFR",
                                              "num_iterations"])
 
-    sim_mgr.configure_simulation(simulator=sim, name=sim_name)
+    sim_mgr.configure_simulation(simulator=sim, name=sim_name,
+                                 additional_metadata={"omega": omega, "K": K})
 
 
 def main():