add pause button (#480)

* add pause button & logic * Update gui.png
int-brain-lab · Sep 11, 2023 · 25a7bdd · 25a7bdd
1 parent 6da0a84
commit 25a7bdd
Show file tree

Hide file tree

Showing 4 changed files with 245 additions and 144 deletions.
diff --git a/docs/source/gui.png b/docs/source/gui.png
diff --git a/iblrig/base_choice_world.py b/iblrig/base_choice_world.py
@@ -126,6 +126,8 @@ def _run(self):
             time_last_trial_end = time.time()
             self.trial_completed(self.bpod.session.current_trial.export())
             self.show_trial_log()
+            while self.paths.SESSION_FOLDER.joinpath('.pause').exists():
+                time.sleep(1)
             if self.paths.SESSION_FOLDER.joinpath('.stop').exists():
                 self.paths.SESSION_FOLDER.joinpath('.stop').unlink()
                 break

diff --git a/iblrig/gui/wizard.py b/iblrig/gui/wizard.py
@@ -10,6 +10,7 @@
 import webbrowser
 
 from PyQt5 import QtWidgets, QtCore, uic
+from PyQt5.QtWidgets import QStyle
 
 from one.api import ONE
 import iblrig_tasks
@@ -130,10 +131,20 @@ def __init__(self, *args, **kwargs):
         self.uiPushHelp.clicked.connect(self.help)
         self.uiPushFlush.clicked.connect(self.flush)
         self.uiPushStart.clicked.connect(self.startstop)
+        self.uiPushPause.clicked.connect(self.pause)
         self.uiPushConnect.clicked.connect(self.alyx_connect)
         self.lineEditSubject.textChanged.connect(self._filter_subjects)
         self.running_task_process = None
 
+        pixmapi = QStyle.SP_MediaPlay
+        self.uiPushStart.setIcon(self.style().standardIcon(pixmapi))
+        pixmapi = QStyle.SP_MediaPause
+        self.uiPushPause.setIcon(self.style().standardIcon(pixmapi))
+        pixmapi = QStyle.SP_BrowserReload
+        self.uiPushFlush.setIcon(self.style().standardIcon(pixmapi))
+        pixmapi = QStyle.SP_DialogHelpButton
+        self.uiPushHelp.setIcon(self.style().standardIcon(pixmapi))
+
     def model2view(self):
         # stores the current values in the model
         self.controller2model()
@@ -168,6 +179,17 @@ def _filter_subjects(self):
             result = [self.model.test_subject_name]
         self.uiComboSubject.setModel(QtCore.QStringListModel(result))
 
+    def pause(self):
+        match self.uiPushPause.isChecked():
+            case True:
+                print('Pausing after current trial ...')
+                if self.model.session_folder.exists():
+                    self.model.session_folder.joinpath('.pause').touch()
+            case False:
+                print('Resuming ...')
+                if self.model.session_folder.joinpath('.pause').exists():
+                    self.model.session_folder.joinpath('.pause').unlink()
+
     def startstop(self):
         match self.uiPushStart.text():
             case 'Start':
@@ -193,7 +215,10 @@ def startstop(self):
                 if self.running_task_process is None:
                     self.running_task_process = subprocess.Popen(cmd)
                 self.uiPushStart.setText('Stop')
+                pixmapi = QStyle.SP_MediaStop
+                self.uiPushStart.setIcon(self.style().standardIcon(pixmapi))
                 self.uiPushFlush.setEnabled(False)
+                self.uiPushPause.setEnabled(True)
             case 'Stop':
                 # if the process crashed catastrophically, the session folder might not exist
                 if self.model.session_folder.exists():
@@ -202,7 +227,10 @@ def startstop(self):
                 self.running_task_process.communicate()
                 self.running_task_process = None
                 self.uiPushStart.setText('Start')
+                pixmapi = QStyle.SP_MediaPlay
+                self.uiPushStart.setIcon(self.style().standardIcon(pixmapi))
                 self.uiPushFlush.setEnabled(True)
+                self.uiPushPause.setEnabled(False)
 
     def flush(self):
         try: