From 92b67786220cbfe82ee5e3f71889e04e032df037 Mon Sep 17 00:00:00 2001
From: Sudipta Basak <sudiptab2100@gmail.com>
Date: Tue, 4 Jun 2024 21:05:34 +0000
Subject: [PATCH] Message sent boxen plot updated

---
 DAS/visualizor.py | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/DAS/visualizor.py b/DAS/visualizor.py
index 1506f4b..9fb0fd5 100644
--- a/DAS/visualizor.py
+++ b/DAS/visualizor.py
@@ -680,10 +680,21 @@ class Visualizor:
         conf["title"] = "Number of Messages Sent by Nodes"
         conf["xlabel"] = "Node Type"
         conf["ylabel"] = "Number of Messages Sent"
-        n1 = int(result.numberNodes * result.class1ratio)
-        data = [result.msgSentCount[1: n1], result.msgSentCount[n1+1: ]]
-        labels = ["Class 1", "Class 2"]
-        sns.boxenplot(data=data, palette="Set2", ax=plt.gca())
+        data = []
+        nodeClasses, nodeRanges = self.__getNodeRanges(result.shape)
+        _start = 1
+        for _range in nodeRanges:
+            data.append(result.msgSentCount[_start: _range])
+            _start = _range
+        _values, _categories = [], []
+        for _d, _nc in zip(data, nodeClasses):
+            _values += _d
+            _categories += [f'Class {_nc}'] * len(_d)
+        data = pd.DataFrame({
+            'values': _values,
+            'category': _categories
+        })
+        sns.boxenplot(x='category', y='values', data=data, width=0.8)
         plt.xlabel(conf["xlabel"], fontsize=12)
         plt.ylabel(conf["ylabel"], fontsize=12)
         plt.title(conf["title"], fontsize=14)