From 1e26f26d49d922aff3707ae7fe4536e991fc7539 Mon Sep 17 00:00:00 2001
From: Youngjoon Lee <5462944+youngjoon-lee@users.noreply.github.com>
Date: Mon, 9 Sep 2024 14:49:38 +0900
Subject: [PATCH] aggregate *_coeff_stats.csv files as well

---
 mixnet-rs/ordering/src/bin/aggregate.rs | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/mixnet-rs/ordering/src/bin/aggregate.rs b/mixnet-rs/ordering/src/bin/aggregate.rs
index 322979f..887ddbc 100644
--- a/mixnet-rs/ordering/src/bin/aggregate.rs
+++ b/mixnet-rs/ordering/src/bin/aggregate.rs
@@ -46,9 +46,11 @@ fn aggregate(path: &str) {
             );
             add_stats_columns(&mut df, entry.path().join("latency_stats.csv"), "latency_");
 
-            let coeff_stats_path = entry.path().join("coeff_stats.csv");
-            if coeff_stats_path.exists() {
-                add_stats_columns(&mut df, coeff_stats_path, "");
+            for prefix in ["strong", "casual", "weak"] {
+                let coeff_stats_path = entry.path().join(format!("{}_coeff_stats.csv", prefix));
+                if coeff_stats_path.exists() {
+                    add_stats_columns(&mut df, coeff_stats_path, &format!("{}_coeff_", prefix));
+                }
             }
 
             dataframes.push(df);