Skip to content

Commit 092b3f6

Browse files
committed
merge with neo OE updates
2 parents 2d6257d + f7ef91f commit 092b3f6

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

neo/rawio/openephysbinaryrawio.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -476,10 +476,18 @@ def explore_folder(dirname, experiment_names=None):
476476
# now create all_streams, nb_block, nb_segment_per_block (from first recording Node)
477477
recording_node = folder_structure[list(folder_structure.keys())[0]]
478478
nb_block = len(recording_node['experiments'])
479-
for block_index, experiment in enumerate(recording_node['experiments']):
479+
# natural sort experiment names so that block_index sequentially indicate experiments
480+
experiment_names = [e['name'] for e in recording_node['experiments']]
481+
experiment_order = np.argsort([int(ename.replace('experiment', '')) for ename in experiment_names])
482+
for block_index, exp_index in enumerate(experiment_order):
483+
experiment = recording_node['experiments'][exp_index]
480484
nb_segment_per_block[block_index] = len(experiment['recordings'])
481485
all_streams[block_index] = {}
482-
for seg_index, recording in enumerate(experiment['recordings']):
486+
# natural sort recording names so that seg_index sequentially indicate recordings
487+
recording_names = [r['name'] for r in experiment['recordings']]
488+
recording_order = np.argsort([int(rname.replace('recording', '')) for rname in recording_names])
489+
for seg_index, rec_index in enumerate(recording_order):
490+
recording = experiment['recordings'][rec_index]
483491
all_streams[block_index][seg_index] = {}
484492
for stream_type in recording['streams']:
485493
all_streams[block_index][seg_index][stream_type] = {}

0 commit comments

Comments
 (0)