adding QProcess for executing scripts in the repro folder

pyrelacs/app.py

@@ -3,7 +3,7 @@ import sys
 import pathlib
 import ctypes
 
-from PyQt6.QtCore import QSize, QThreadPool, Qt
+from PyQt6.QtCore import QProcess, QSize, QThreadPool, Qt
 from PyQt6.QtWidgets import (
     QApplication,
     QGridLayout,
@@ -11,6 +11,7 @@ from PyQt6.QtWidgets import (
     QToolBar,
     QWidget,
     QMainWindow,
+    QPlainTextEdit,
 )
 import tomli
 import uldaq
@@ -27,7 +28,11 @@ class PyRelacs(QMainWindow):
     def __init__(self):
         super().__init__()
         self.setWindowTitle("PyRelacs")
+        self.setMinimumSize(1000, 1000)
+
         self.threadpool = QThreadPool()
+        # for starting a Qprocess
+        self.p = None
 
         self.daq_connect_button = QPushButton("Connect Daq")
         self.daq_connect_button.setCheckable(True)
@@ -37,17 +42,17 @@ class PyRelacs(QMainWindow):
         self.daq_disconnect_button.setCheckable(True)
         self.daq_disconnect_button.clicked.connect(self.disconnect_dac)
 
-        self.repro_button = QPushButton("Load Repros")
+        self.text = QPlainTextEdit()
-        self.repro_button.setCheckable(True)
+        self.text.setReadOnly(True)
-        self.repro_button.clicked.connect(self.repro)
 
         layout = QGridLayout()
         layout.addWidget(self.daq_connect_button, 0, 0)
         layout.addWidget(self.daq_disconnect_button, 0, 1)
-        layout.addWidget(self.repro_button, 1, 0)
+        layout.addWidget(self.text, 2, 0, 1, 2)
 
         self.toolbar = QToolBar("Repros")
         self.addToolBar(self.toolbar)
+        self.repro()
 
         self.setFixedSize(QSize(400, 300))
         widget = QWidget()
@@ -78,70 +83,59 @@ class PyRelacs(QMainWindow):
             log.debug("DAQ was not connected")
 
     def repro(self):
-        config_path = pathlib.Path(__file__).parent.resolve() / "repro.toml"
+        repos_path = pathlib.Path(__file__).parent / "repros"
-        if not config_path.is_file:
+        repos_names = list(repos_path.glob("*.py"))
-            log.error("repro.toml not found")
+        # exclude the repos.py file
-            QApplication.quit()
+        repos_names = [
-        repors = []
+            f.with_suffix("").name for f in repos_names if not f.name == "repos.py"
-
+        ]
-        with open(config_path, mode="rb") as fp:
+        for rep in repos_names:
-            self.config = tomli.load(fp)
+            individual_repro_button = QAction(rep, self)
-            for r in self.config:
-                repors.append(r)
-
-        individual_repro_button = QAction(f"{repors[0]}", self)
             individual_repro_button.setStatusTip("Button")
-        individual_repro_button.triggered.connect(self.run_repro)
+            individual_repro_button.triggered.connect(
+                lambda checked, n=rep: self.run_repro(n)
+            )
             self.toolbar.addAction(individual_repro_button)
 
-    def run_repro(self):
+    def message(self, s):
-        daq_thread = Worker(self.read_daq)
+        self.text.appendPlainText(s)
-        self.threadpool.start(daq_thread)
+
-
+    def run_repro(self, name_of_repo):
-    def on_scan_finished(self):
+        if self.p is None:
-        log.debug("Scan finished")
+            self.message(f"Executing process {name_of_repo}")
-
+            self.p = QProcess()
-    def read_daq(self, progress_callback):
+            self.p.setWorkingDirectory(str(pathlib.Path(__file__).parent / "repros/"))
-        log.debug("running repro")
+            # log.debug(pathlib.Path(__file__).parent / "repos")
-        time = np.arange(0, 10, 1 / 30_000.0)
+            self.p.readyReadStandardOutput.connect(self.handle_stdout)
-        data = 2 * np.sin(2 * np.pi * 10 * time)
+            self.p.readyReadStandardError.connect(self.handle_stderr)
-
+            self.p.stateChanged.connect(self.handle_state)
-        buffer = ctypes.c_double * len(time)
+            self.p.finished.connect(self.process_finished)
-        buffer_ai = uldaq.create_float_buffer(1, len(time))
+            self.p.start("python3", [f"{name_of_repo}" + ".py"])
-        data_c = buffer(*data)
+
-        log.debug(f"Created C_double data {data_c}")
+    def handle_stderr(self):
-
+        if self.p is not None:
-        ao_device = self.daq_device.get_ao_device()
+            data = self.p.readAllStandardError()
-        ai_device = self.daq_device.get_ai_device()
+            stderr = bytes(data).decode("utf8")
-
+            self.message(stderr)
-        er = ai_device.a_in_scan(
+
-            1,
+    def handle_stdout(self):
-            1,
+        if self.p is not None:
-            uldaq.AiInputMode.SINGLE_ENDED,
+            data = self.p.readAllStandardOutput()
-            uldaq.Range.BIP10VOLTS,
+            stdout = bytes(data).decode("utf8")
-            int(len(data)),
+            self.message(stdout)
-            30_000.0,
+
-            uldaq.ScanOption.EXTTRIGGER,
+    def handle_state(self, state):
-            uldaq.AInScanFlag.DEFAULT,
+        states = {
-            data=buffer_ai,
+            QProcess.ProcessState.NotRunning: "Not running",
-        )
+            QProcess.ProcessState.Starting: "Starting",
-        # ai_device.scan_wait(uldaq.WaitType.WAIT_UNTIL_DONE, timeout=-1)
+            QProcess.ProcessState.Running: "Running",
-
+        }
-        err = ao_device.a_out_scan(
+        state_name = states[state]
-            0,
+        self.message(f"State changed: {state_name}")
-            0,
+
-            uldaq.Range.BIP10VOLTS,
+    def process_finished(self):
-            int(len(data)),
+        self.text.appendPlainText("Process finished")
-            30_000.0,
+        self.p = None
-            uldaq.ScanOption.DEFAULTIO,
-            uldaq.AOutScanFlag.DEFAULT,
-            data_c,
-        )
-        # INFO: Need to wait for the acquistion
-        self.daq_disconnect_button.setDisabled(True)
-        self.daq_connect_button.setEnabled(True)
-        embed()
-        exit()
 
 
 if __name__ == "__main__":