1 files changed, 119 insertions, 0 deletions
diff --git a/lib/lennart/EnergyInterface.py b/lib/lennart/EnergyInterface.py
new file mode 100644
index 0000000..2b23667
--- /dev/null
+++ b/lib/lennart/EnergyInterface.py
@@ -0,0 +1,119 @@
+import re
+import subprocess
+
+from dfatool.lennart.DataInterface import DataInterface
+
+
+class EnergyInterface(DataInterface):
+    def __init__(
+        self,
+        duration_seconds=10,
+        console_output=False,
+        temp_file="temp/energytrace.log",
+        fake=False,
+    ):
+        """
+        class is not used in embedded into dfatool.
+
+        :param duration_seconds: seconds the EnergyTrace should be running
+        :param console_output: if EnergyTrace output should be printed to the user
+        :param temp_file: file path for temporary file
+        :param fake: if already existing file should be used
+        """
+        self.energytrace = None
+        self.duration_seconds = duration_seconds
+        self.console_output = console_output
+        self.temp_file = temp_file
+        self.fake = fake
+
+    def runMeasure(self):
+        """
+        starts the measurement, with waiting for done
+        """
+        if self.fake:
+            return
+        self.runMeasureAsynchronously()
+        self.waitForAsynchronousMeasure()
+
+    def runMeasureAsynchronously(self):
+        """
+        starts the measurement, not waiting for done
+        """
+        if self.fake:
+            return
+        self.energytrace = subprocess.Popen(
+            "msp430-etv --save %s %s %s"
+            % (
+                self.temp_file,
+                self.duration_seconds,
+                "" if self.console_output else "> /dev/null",
+            ),
+            shell=True,
+        )
+        print(
+            "msp430-etv --save %s %s %s"
+            % (
+                self.temp_file,
+                self.duration_seconds,
+                "" if self.console_output else "> /dev/null",
+            )
+        )
+
+    def waitForAsynchronousMeasure(self):
+        """
+        Wait until is command call is done
+        """
+        if self.fake:
+            return
+        self.energytrace.wait()
+
+    def getData(self):
+        """
+        cleans the string data and creates int list
+        :return: list of data, in format [[int,int,int,int], [int,int,int,int], ... ]
+        """
+        energytrace_log = open(self.temp_file)
+        lines = energytrace_log.readlines()[21:]
+        data = []
+        for line in lines:
+            if "MSP430_DisableEnergyTrace" in line:
+                break
+            else:
+                data.append([int(i) for i in line.split()])
+        return data
+
+    @classmethod
+    def getDataFromString(cls, string, delimiter="\\n"):
+        """
+        Parsing the data from string
+
+        :param string: input string which will be parsed
+        :param delimiter: for normal file its \n
+        :return: list of data, in format [[int,int,int,int], [int,int,int,int], ... ]
+        """
+        lines = string.split(delimiter)[21:]
+        data = []
+        for line in lines:
+            if "MSP430_DisableEnergyTrace" in line:
+                break
+            else:
+                data.append([int(i) for i in line.split()])
+        return data
+
+    def setFile(self, path):
+        """
+        changeing the temporary file
+
+        :param path: file path of new temp file
+        :return: None
+        """
+        self.temp_file = path
+        pass
+
+    def forceStopMeasure(self):
+        """
+        force stops the Measurement, with signals
+        :return: None
+        """
+        self.energytrace.send_signal(subprocess.signal.SIGINT)
+        stdout, stderr = self.energytrace.communicate(timeout=15)