From 201fc08da025b1e2c37c98bc0c713cd27cdaa07b Mon Sep 17 00:00:00 2001
From: Sudipta Basak <sudiptab2100@gmail.com>
Date: Tue, 4 Jun 2024 09:30:12 +0000
Subject: [PATCH] duplicate data plot fixed

---
 DAS/visualizor.py | 22 +++++++++++++---------
 1 file changed, 13 insertions(+), 9 deletions(-)

diff --git a/DAS/visualizor.py b/DAS/visualizor.py
index c5b2d35..a4565fb 100644
--- a/DAS/visualizor.py
+++ b/DAS/visualizor.py
@@ -1035,11 +1035,12 @@ class Visualizor:
 
     def plotDupData(self, result, plotPath):
         """Plots the percentage of nodes ready in the network"""
-        vector1 = result.metrics["progress"]["Dup class1 mean"]
-        vector2 = result.metrics["progress"]["Dup class2 mean"]
-        for i in range(len(vector1)):
-            vector1[i] = (vector1[i] * 8 * (1000/self.config.stepDuration) * self.config.segmentSize) / 1000000
-            vector2[i] = (vector2[i] * 8 * (1000/self.config.stepDuration) * self.config.segmentSize) / 1000000
+        vectors = {}
+        for nc in result.shape.nodeClasses:
+            if nc != 0: vectors[nc] = result.metrics["progress"][f"Dup class{nc} mean"]
+        for _k in vectors.keys():
+            for i in range(len(list(vectors.values())[0])):
+                vectors[_k][i] = (vectors[_k][i] * 8 * (1000/self.config.stepDuration) * self.config.segmentSize) / 1000000
         conf = {}
         attrbs = self.__get_attrbs__(result)
         nodeTypes = self.__getNodeTypes__(attrbs['ntypes'])
@@ -1056,12 +1057,15 @@ class Visualizor:
         conf["type"] = "plot"
         conf["legLoc"] = 2
         conf["desLoc"] = 2
-        conf["colors"] = ["c-", "m-"]
-        conf["labels"] = ["Solo stakers", "Staking pools"]
+        # conf["colors"] = ["c-", "m-"]
+        conf["labels"] = []
+        conf["data"] = []
+        for _k, _v in vectors.items():
+            conf["labels"].append(f"Node Class: {_k}")
+            conf["data"].append(_v)
         conf["xlabel"] = "Time (ms)"
         conf["ylabel"] = "Bandwidth (MBits/s)"
-        conf["data"] = [vector1, vector2]
-        conf["xdots"] = [x*self.config.stepDuration for x in range(len(vector1))]
+        conf["xdots"] = [x*self.config.stepDuration for x in range(len(list(vectors.values())[0]))]
         conf["path"] = plotPath+"/dupData.png"
         maxi = 0
         for v in conf["data"]: