Add fNIRS support to coregistration gui (#10122)

mne-tools · Dec 13, 2021 · e0aa5fe · e0aa5fe
1 parent 9917316
commit e0aa5fe
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/mne/gui/_coreg.py b/mne/gui/_coreg.py
@@ -772,23 +772,24 @@ def _add_head_shape_points(self):
     def _add_eeg_channels(self):
         if self._eeg_channels:
             eeg = ["original"]
-            picks = pick_types(self._info, eeg=(len(eeg) > 0))
+            picks = pick_types(self._info, eeg=(len(eeg) > 0), fnirs=True)
             if len(picks) > 0:
-                eeg_actors = _plot_sensors(
+                actors = _plot_sensors(
                     self._renderer, self._info, self._to_cf_t, picks,
-                    meg=False, eeg=eeg, fnirs=False, warn_meg=False,
-                    head_surf=self._head_geo, units='m',
+                    meg=False, eeg=eeg, fnirs=["sources", "detectors"],
+                    warn_meg=False, head_surf=self._head_geo, units='m',
                     sensor_opacity=self._defaults["sensor_opacity"],
                     orient_glyphs=self._orient_glyphs,
                     scale_by_distance=self._scale_by_distance,
                     project_points=self._project_eeg,
                     surf=self._head_geo)
-                eeg_actors = eeg_actors["eeg"]
+                sens_actors = actors["eeg"]
+                sens_actors.extend(actors["fnirs"])
             else:
-                eeg_actors = None
+                sens_actors = None
         else:
-            eeg_actors = None
-        self._update_actor("eeg_channels", eeg_actors)
+            sens_actors = None
+        self._update_actor("eeg_channels", sens_actors)
 
     def _add_head_surface(self):
         bem = None