Skip to content

Commit

Permalink
fix largest culprit in speed issue
Browse files Browse the repository at this point in the history
  • Loading branch information
alexrockhill committed Apr 25, 2024
1 parent b5d4917 commit da88999
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions mne_gui_addons/_ieeg_locate.py
Original file line number Diff line number Diff line change
Expand Up @@ -992,7 +992,9 @@ def _update_ch_images(self, axis=None, draw=False):
def _update_ct_images(self, axis=None, draw=False):
"""Update the CT image(s)."""
for axis in range(3) if axis is None else [axis]:
ct_data = np.take(self._ct_data, self._current_slice[axis], axis=axis).T
ct_data = self._ct_data[
(slice(None),) * axis + (self._current_slice[axis],)
].T
# Threshold the CT so only bright objects (electrodes) are visible
ct_data[ct_data < self._ct_min_slider.value()] = np.nan
ct_data[ct_data > self._ct_max_slider.value()] = np.nan
Expand All @@ -1010,7 +1012,9 @@ def _update_mri_images(self, axis=None, draw=False):
if "mri" in self._images:
for axis in range(3) if axis is None else [axis]:
self._images["mri"][axis].set_data(
np.take(self._mr_data, self._current_slice[axis], axis=axis).T
self._mr_data[
(slice(None),) * axis + (self._current_slice[axis],)
].T
)
if draw:
self._draw(axis)
Expand Down

0 comments on commit da88999

Please sign in to comment.