@@ -47,12 +47,6 @@ def activate(
47
47
48
48
SI_SORTERS = [s .replace ("_" , "." ) for s in si .sorters .sorter_dict .keys ()]
49
49
50
- SI_READERS = {
51
- "Open Ephys" : si .extractors .read_openephys ,
52
- "SpikeGLX" : si .extractors .read_spikeglx ,
53
- "Intan" : si .extractors .read_intan ,
54
- }
55
-
56
50
57
51
@schema
58
52
class PreProcessing (dj .Imported ):
@@ -108,9 +102,7 @@ def make(self, key):
108
102
output_dir = find_full_path (ephys .get_ephys_root_data_dir (), output_dir )
109
103
recording_dir = output_dir / sorter_name / "recording"
110
104
recording_dir .mkdir (parents = True , exist_ok = True )
111
- recording_file = (
112
- recording_dir / "si_recording.pkl"
113
- ) # recording cache to be created for each key
105
+ recording_file = recording_dir / "si_recording.pkl"
114
106
115
107
# Create SI recording extractor object
116
108
if acq_software == "SpikeGLX" :
@@ -125,12 +117,16 @@ def make(self, key):
125
117
assert len (oe_probe .recording_info ["recording_files" ]) == 1
126
118
data_dir = oe_probe .recording_info ["recording_files" ][0 ]
127
119
else :
128
- raise NotImplementedError (f"Not implemented for { acq_software } " )
120
+ si_extractor : si .extractors .neoextractors = (
121
+ si .extractors .extractorlist .recording_extractor_full_dict [
122
+ acq_software .replace (" " , "" ).lower ()
123
+ ]
124
+ ) # data extractor object
129
125
130
126
stream_names , stream_ids = si .extractors .get_neo_streams (
131
127
acq_software .strip ().lower (), folder_path = data_dir
132
128
)
133
- si_recording : si .BaseRecording = SI_READERS [acq_software ](
129
+ si_recording : si .BaseRecording = si_extractor [acq_software ](
134
130
folder_path = data_dir , stream_name = stream_names [0 ]
135
131
)
136
132
0 commit comments