From 0cae56c9b438fb2d116882d5cc8b99cec3c51b07 Mon Sep 17 00:00:00 2001
From: Youngjoon Lee <5462944+youngjoon-lee@users.noreply.github.com>
Date: Fri, 5 Jul 2024 23:56:48 +0900
Subject: [PATCH] convert node state enum to integer in the DataFrame to
 minimize the size of CSV file

---
 mixnet/sim/state.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/mixnet/sim/state.py b/mixnet/sim/state.py
index b2cce2c..8b63618 100644
--- a/mixnet/sim/state.py
+++ b/mixnet/sim/state.py
@@ -24,7 +24,10 @@ class AllNodeStates:
     def analyze(self):
         df = pandas.DataFrame(self._table).transpose()
         df.columns = [f"Node-{i}" for i in range(len(self._table))]
+        # Convert NodeState enum to their integer values for analysis
+        df = df.map(lambda state: state.value)
         print(df)
+
         csv_path = f"all_node_states_{datetime.now().isoformat(timespec="seconds")}.csv"
         df.to_csv(csv_path)
         print(f"\nSaved DataFrame to {csv_path}\n")