pylance things

src/analysis/intragroupfunctionalconnectivity.py

@@ -0,0 +1,83 @@
+"""
+Filename: intragroupfunctionalconnectivity.py
+Description: Logic for the Intra-Group Functional Connectivity analysis window
+Note: Compliant with pylance strict type checking
+
+Author: Tyler de Zeeuw
+License: GPL-3.0
+"""
+
+# Built-in imports
+from pathlib import Path
+from typing import Any, cast
+
+# External library imports
+from PySide6.QtWidgets import QMessageBox
+
+from mne.io.base import BaseRaw
+
+from flares import run_group_functional_connectivity
+from src.shared.flaresbasewidget import IntraGroupUIMixin, FlaresBaseWidget
+from src.shared.shareddata import APP_NAME
+
+
+PARAMETERIZED_INDEXES: dict[int, list[dict[str, Any]]] = {
+    0: [
+        {
+            "key": "n_lines",
+            "label": "<Description>",
+            "default": "20",
+            "type": int,
+        },
+        {
+            "key": "vmin",
+            "label": "<Description>",
+            "default": "0.9",
+            "type": float,
+        },
+    ],
+}
+
+
+class IntraGroupFunctionalConnectivityWidget(IntraGroupUIMixin, FlaresBaseWidget):
+    def __init__(
+        self,
+        haemo_dict: dict[str | Path, BaseRaw],
+        group_dict: dict[str, str],
+        config_dict: dict[str, dict[str, Any]]
+    ) -> None:
+        
+        super().__init__("IntraGroupFunctionalConnectivity")
+        self.setWindowTitle(f"Intra-Group Functional Connectivity Viewer [BETA] - {APP_NAME.upper()}")
+        self.haemo_dict = haemo_dict
+        #self.group_dict = group_dict
+        self.config_dict = config_dict
+
+        QMessageBox.warning(self, f"Warning - {APP_NAME.upper()}", f"Functional Connectivity is still in development and the results should currently be taken with a grain of salt. "
+                    "By clicking OK, you accept that the images generated may not be factual.")
+
+        self.setup_intra_group_ui(["0 (Betas)",])
+
+
+    def process_request(self):
+        request = self.get_common_request_data(PARAMETERIZED_INDEXES)
+        if request is None:
+            return
+
+        (selected_event, selected_file_paths, selected_indexes, raw_params) = request   
+
+        param_values = cast(dict[int | str, dict[str, Any]], raw_params)
+            
+        for idx in selected_indexes:
+            if idx == 0:
+                params = param_values.get(idx, {})
+                n_lines = params.get("n_lines", None)
+                vmin = params.get("vmin", None)
+
+                if n_lines is None or vmin is None:
+                    print(f"Missing parameters for index {idx}, skipping.")
+                    continue
+                run_group_functional_connectivity(self.haemo_dict, self.config_dict, selected_file_paths, selected_event, 50, 0.5)            
+
+            else:
+                print(f"No method defined for index {idx}")