1 files changed, 89 insertions, 0 deletions
diff --git a/lib/lennart/SigrokCLIInterface.py b/lib/lennart/SigrokCLIInterface.py
new file mode 100644
index 0000000..b28a8a9
--- /dev/null
+++ b/lib/lennart/SigrokCLIInterface.py
@@ -0,0 +1,89 @@
+import subprocess
+import time
+
+from dfatool.lennart.SigrokInterface import SigrokInterface
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+class SigrokCLIInterface(SigrokInterface):
+    def __init__(
+        self,
+        bin_temp_file="temp/out.bin",
+        sample_rate=100_000,
+        fake=False,
+    ):
+        """
+        creates SigrokCLIInterface object. Uses the CLI Interface (Command: sigrok-cli)
+
+        :param bin_temp_file: temporary file for binary output
+        :param sample_rate: The sample rate of the Logic analyzer
+        :param fake: if it should use existing data
+        """
+        super(SigrokCLIInterface, self).__init__(sample_rate)
+        self.fake = fake
+        self.bin_temp_file = bin_temp_file
+        self.sigrok_cli_thread = None
+
+    def forceStopMeasure(self):
+        """
+        Force stopping measure, sometimes needs pkill for killing definitly
+        :return: None
+        """
+        self.sigrok_cli_thread.terminate()
+
+        try:
+            self.sigrok_cli_thread.wait(timeout=10)
+        except subprocess.TimeoutExpired:
+            logger.warning("sigrok-cli has not stopped. Killing it.")
+            self.sigrok_cli_thread.kill()
+
+        self.sigrok_cli_thread.communicate()
+        self.runOpenAnalyze()
+
+    def runMeasure(self):
+        """
+        starts the measurement, with waiting for done
+        """
+        if not self.fake:
+            self.runMeasureAsynchronous()
+        self.waitForAsynchronousMeasure()
+
+    def runMeasureAsynchronous(self):
+        """
+        starts the measurement, not waiting for done
+        """
+        shellcommand = [
+            "sigrok-cli",
+            "--output-file",
+            self.bin_temp_file,
+            "--output-format",
+            "binary",
+            "--continuous",
+            "-d",
+            self.driver,
+            "--config",
+            f"samplerate={self.sample_rate} Hz",
+        ]
+        self.sigrok_cli_thread = subprocess.Popen(shellcommand)
+
+    def waitForAsynchronousMeasure(self):
+        """
+        Wait until is command call is done
+        """
+        if not self.fake:
+            self.sigrok_cli_thread.wait()
+        self.runOpenAnalyze()
+
+    def runOpenAnalyze(self):
+        """
+        Opens the generated binary file and parses it byte by byte
+
+        """
+        in_file = open(self.bin_temp_file, "rb")  # opening for [r]eading as [b]inary
+        data = in_file.read()  # if you only wanted to read 512 bytes, do .read(512)
+        in_file.close()
+
+        for x in data:
+            self.analyzeData(x)