mostly finish rewrite

2026-06-28 10:10:27 -07:00
parent f10eb64332
commit ddb30d98f2
18 changed files with 2116 additions and 2350 deletions
@@ -0,0 +1,122 @@
+"""
+Filename: viewerlauncher.py
+Description: Analysis options launcher for FLARES
+
+Author: Tyler de Zeeuw
+License: GPL-3.0
+"""
+
+from PySide6.QtWidgets import  QPushButton, QWidget, QVBoxLayout
+from PySide6.QtCore import QTimer
+
+from src.analysis.exportcsv import ExportDataAsCSVViewerWidget
+from src.analysis.group import GroupViewerWidget
+from src.analysis.groupbrain import GroupBrainViewerWidget
+from src.analysis.groupfunctionalconnectivity import GroupFunctionalConnectivityWidget
+from src.analysis.participant import ParticipantViewerWidget
+from src.analysis.participantbrain import ParticipantBrainViewerWidget
+from src.analysis.participantfoldchannels import ParticipantFoldChannelsWidget
+from src.analysis.participantfunctionalconnectivity import ParticipantFunctionalConnectivityWidget
+from src.shared.shareddata import APP_NAME
+
+
+class ViewerLauncherWidget(QWidget):
+    def __init__(self, haemo_dict, config_dict, fig_bytes_dict, cha_dict, contrast_results_dict, df_ind, design_matrix, epochs_dict, folding_bypass):
+        super().__init__()
+        self.setWindowTitle(f"Viewer Launcher - {APP_NAME.upper()}")
+
+        group_dict = {
+            file_path: config.get("GROUP", "Unknown")
+            for file_path, config in config_dict.items()
+        }
+
+        def launch(func, btn, *args):
+            func(*args)
+            self._trigger_success(btn)
+
+        layout = QVBoxLayout(self)
+
+        btn1 = QPushButton("Open Participant Viewer")
+        btn1.clicked.connect(lambda: launch(self.open_participant_viewer, btn1, haemo_dict, fig_bytes_dict))
+        btn1.setEnabled(not folding_bypass)
+
+        btn2 = QPushButton("Open Participant Brain Viewer")
+        btn2.clicked.connect(lambda: launch(self.open_participant_brain_viewer, btn2, haemo_dict, cha_dict))
+        btn2.setEnabled(not folding_bypass)
+
+        btn3 = QPushButton("Open Participant Fold Channels Viewer")
+        btn3.clicked.connect(lambda: launch(self.open_participant_fold_channels_viewer, btn3, haemo_dict, cha_dict))
+
+        btn7 = QPushButton("Open Functional Connectivity Viewer [BETA]")
+        btn7.clicked.connect(lambda: launch(self.open_participant_functional_connectivity_viewer, btn7, haemo_dict, epochs_dict))
+        btn7.setEnabled(not folding_bypass)
+
+        btn8 = QPushButton("Open Group Functional Connectivity Viewer [BETA]")
+        btn8.clicked.connect(lambda: launch(self.open_group_functional_connectivity_viewer, btn8, haemo_dict, group_dict, config_dict))
+        btn8.setEnabled(not folding_bypass)
+
+        btn4 = QPushButton("Open Inter-Group Viewer")
+        btn4.clicked.connect(lambda: launch(self.open_group_viewer, btn4, haemo_dict, cha_dict, df_ind, design_matrix, contrast_results_dict, group_dict))
+        btn4.setEnabled(not folding_bypass)
+        
+        btn5 = QPushButton("Open Cross Group Brain Viewer")
+        btn5.clicked.connect(lambda: launch(self.open_group_brain_viewer, btn5, haemo_dict, df_ind, design_matrix, group_dict, contrast_results_dict))
+        btn5.setEnabled(not folding_bypass)
+        
+        btn6 = QPushButton("Open Export Data As CSV Viewer")
+        btn6.clicked.connect(lambda: launch(self.open_export_data_as_csv_viewer, btn6, haemo_dict, cha_dict, df_ind, design_matrix, group_dict, contrast_results_dict))
+        btn6.setEnabled(not folding_bypass)
+
+        layout.addWidget(btn1)
+        layout.addWidget(btn2)
+        layout.addWidget(btn3)
+        layout.addWidget(btn7)
+        layout.addWidget(btn8)
+        layout.addWidget(btn4)
+        layout.addWidget(btn5)
+        layout.addWidget(btn6)
+
+    def open_participant_viewer(self, haemo_dict, fig_bytes_dict):
+        self.participant_viewer = ParticipantViewerWidget(haemo_dict, fig_bytes_dict)
+        self.participant_viewer.show()
+
+    def open_participant_brain_viewer(self, haemo_dict, cha_dict):
+        self.participant_brain_viewer = ParticipantBrainViewerWidget(haemo_dict, cha_dict)
+        self.participant_brain_viewer.show()
+        
+    def open_participant_fold_channels_viewer(self, haemo_dict, cha_dict):
+        self.participant_fold_channels_viewer = ParticipantFoldChannelsWidget(haemo_dict, cha_dict)
+        self.participant_fold_channels_viewer.show()
+
+    def open_participant_functional_connectivity_viewer(self, haemo_dict, epochs_dict):
+        self.participant_brain_viewer = ParticipantFunctionalConnectivityWidget(haemo_dict, epochs_dict)
+        self.participant_brain_viewer.show()
+
+    def open_group_functional_connectivity_viewer(self, haemo_dict, group, config_dict):
+        self.participant_brain_viewer = GroupFunctionalConnectivityWidget(haemo_dict, group, config_dict)
+        self.participant_brain_viewer.show()
+
+    def open_group_viewer(self, haemo_dict, cha_dict, df_ind, design_matrix, contrast_results_dict, group):
+        self.participant_brain_viewer = GroupViewerWidget(haemo_dict, cha_dict, df_ind, design_matrix, contrast_results_dict, group)
+        self.participant_brain_viewer.show()
+
+    def open_group_brain_viewer(self, haemo_dict, df_ind, design_matrix, group, contrast_results_dict):
+        self.participant_brain_viewer = GroupBrainViewerWidget(haemo_dict, df_ind, design_matrix, group, contrast_results_dict)
+        self.participant_brain_viewer.show()
+
+    def open_export_data_as_csv_viewer(self, haemo_dict, cha_dict, df_ind, design_matrix, group, contrast_results_dict):
+        self.export_data_as_csv_viewer = ExportDataAsCSVViewerWidget(haemo_dict, cha_dict, df_ind, design_matrix, group, contrast_results_dict)
+        self.export_data_as_csv_viewer.show()
+
+    def _trigger_success(self, button):
+        """Temporarily adds a green checkmark to the button text."""
+        original_text = button.text()
+        button.setText(f"{original_text}  ✔")
+        button.setStyleSheet("color: green; font-weight: bold;")
+        
+        # Revert after 1 second
+        QTimer.singleShot(1000, lambda: self._revert_button(button, original_text))
+
+    def _revert_button(self, button, original_text):
+        button.setText(original_text)
+        button.setStyleSheet("")