Skip to content

Commit 4e367d7

Browse files
author
Thinh Nguyen
committed
added loading of electrode location for new openephys format
1 parent 07604e2 commit 4e367d7

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

element_array_ephys/readers/openephys.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -198,6 +198,15 @@ def __init__(self, processor, probe_index=0):
198198
self.probe_info = processor['EDITOR']['NP_PROBE'][probe_index]
199199
self.probe_SN = self.probe_info['@probe_serial_number']
200200
self.probe_model = _probe_model_name_mapping[self.probe_info['@probe_name']]
201+
202+
if 'ELECTRODE_XPOS' in self.probe_info:
203+
self.probe_info['ELECTRODE_XPOS'] = {int(re.search(r'\d+$', k).group()): int(v)
204+
for k, v in self.probe_info.pop('ELECTRODE_XPOS').items()}
205+
self.probe_info['ELECTRODE_YPOS'] = {int(re.search(r'\d+$', k).group()): int(v)
206+
for k, v in self.probe_info.pop('ELECTRODE_YPOS').items()}
207+
self.probe_info['ELECTRODE_SHANK'] = {int(re.search(r'\d+$', k).group()): int(v)
208+
for k, v in self.probe_info['CHANNELS'].items()}
209+
201210
self._channels_connected = {int(re.search(r'\d+$', k).group()): 1
202211
for k in self.probe_info.pop('CHANNELS')}
203212

0 commit comments

Comments
 (0)