From 6a7146cc80f6acdeb9ad74aa2248863675dc2462 Mon Sep 17 00:00:00 2001
From: Youngjoon Lee <5462944+youngjoon-lee@users.noreply.github.com>
Date: Fri, 2 Aug 2024 22:24:25 +0900
Subject: [PATCH] session 2 for experiment 4

---
 mixnet/queuesim/paramset.py | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/mixnet/queuesim/paramset.py b/mixnet/queuesim/paramset.py
index a591f93..707bc6b 100644
--- a/mixnet/queuesim/paramset.py
+++ b/mixnet/queuesim/paramset.py
@@ -186,6 +186,7 @@ def __build_session_2_parameter_sets(
         peering_degree_list = [4, 8, 16]
         min_queue_size_list = [10, 50, 100]
         transmission_rate_list = [1, 10, 100]
+        num_senders_list = [num_nodes // 10, num_nodes // 5, num_nodes // 2]
         num_iterations = 20
 
         match exp_id:
@@ -211,6 +212,35 @@ def __build_session_2_parameter_sets(
                             num_iterations=num_iterations,
                         )
                     )
+            case ExperimentID.EXPERIMENT_4:
+                for (
+                    peering_degree,
+                    min_queue_size,
+                    transmission_rate,
+                    num_senders,
+                ) in itertools.product(
+                    peering_degree_list,
+                    min_queue_size_list,
+                    transmission_rate_list,
+                    num_senders_list,
+                ):
+                    for num_sent_msgs in [
+                        min_queue_size // 2,
+                        min_queue_size,
+                        min_queue_size * 2,
+                    ]:
+                        sets.append(
+                            ParameterSet(
+                                num_nodes=num_nodes,
+                                peering_degree=peering_degree,
+                                min_queue_size=min_queue_size,
+                                transmission_rate=transmission_rate,
+                                num_sent_msgs=num_sent_msgs,
+                                num_senders=num_senders,
+                                queue_type=queue_type,
+                                num_iterations=num_iterations,
+                            )
+                        )
             case _:
                 raise NotImplementedError(
                     f"Experiment {exp_id} not implemented for session 2"