Add visualizer class

DAS/__init__.py

@@ -1,3 +1,4 @@
 from DAS.simulator import *
 from DAS.configuration import *
 from DAS.shape import *
+from DAS.visualizer import *

DAS/visualizer.py

@@ -0,0 +1,22 @@
+#!/bin/python3
+
+class Visualizer:
+
+    def __init__(self, execID):
+        self.execID = execID
+        self.folderPath = "results/"+self.execID
+
+    def plottingData(self):
+        data = []
+        print("Getting data from the folder...")
+        return data
+
+    def similarKeys(self, data):
+        filteredKeys = []
+        print("Getting filtered keys from data...")
+        return filteredKeys
+
+    def plotHeatmaps(self):
+        data = self.plottingData()
+        filteredKeys = self.similarKeys(data)
+        print("Plotting heatmaps...")

study.py

@@ -51,6 +51,10 @@ def study():
             res.dump(execID)
         sim.logger.info("Results dumped into results/%s/" % (execID), extra=sim.format)
 
+    visualization = 1
+    if visualization:
+        vis = Visualizer(execID)
+        vis.plotHeatmaps()
 
 
 study()