diff --git a/DAS/__init__.py b/DAS/__init__.py index 8dd11bf..c7fc7c0 100644 --- a/DAS/__init__.py +++ b/DAS/__init__.py @@ -1,3 +1,4 @@ from DAS.simulator import * from DAS.configuration import * from DAS.shape import * +from DAS.visualizer import * diff --git a/DAS/visualizer.py b/DAS/visualizer.py new file mode 100644 index 0000000..881d658 --- /dev/null +++ b/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...") diff --git a/study.py b/study.py index c5eed14..438fba4 100644 --- a/study.py +++ b/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()