Skip to content

Commit f6dbcbd

Browse files
authored
Merge pull request #1497 from h-mayorquin/fix_plexon1_error
Plexon overflow fix
2 parents 1804b65 + 38fe0c5 commit f6dbcbd

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

neo/rawio/plexonrawio.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -95,14 +95,14 @@ def _parse_header(self):
9595
)
9696

9797
# dsp channels header = spikes and waveforms
98-
nb_unit_chan = global_header["NumDSPChannels"]
98+
nb_unit_chan = int(global_header["NumDSPChannels"])
9999
offset1 = np.dtype(GlobalHeader).itemsize
100100
dspChannelHeaders = np.memmap(
101101
self.filename, dtype=DspChannelHeader, mode="r", offset=offset1, shape=(nb_unit_chan,)
102102
)
103103

104104
# event channel header
105-
nb_event_chan = global_header["NumEventChannels"]
105+
nb_event_chan = int(global_header["NumEventChannels"])
106106
offset2 = offset1 + np.dtype(DspChannelHeader).itemsize * nb_unit_chan
107107
eventHeaders = np.memmap(
108108
self.filename,
@@ -113,7 +113,7 @@ def _parse_header(self):
113113
)
114114

115115
# slow channel header = signal
116-
nb_sig_chan = global_header["NumSlowChannels"]
116+
nb_sig_chan = int(global_header["NumSlowChannels"])
117117
offset3 = offset2 + np.dtype(EventChannelHeader).itemsize * nb_event_chan
118118
slowChannelHeaders = np.memmap(
119119
self.filename, dtype=SlowChannelHeader, mode="r", offset=offset3, shape=(nb_sig_chan,)
@@ -136,7 +136,9 @@ def _parse_header(self):
136136

137137
while pos < data.size:
138138
bl_header = data[pos : pos + 16].view(DataBlockHeader)[0]
139-
length = bl_header["NumberOfWaveforms"] * bl_header["NumberOfWordsInWaveform"] * 2 + 16
139+
number_of_waveforms = int(bl_header["NumberOfWaveforms"])
140+
number_of_words_in_waveform = int(bl_header["NumberOfWordsInWaveform"])
141+
length = (number_of_waveforms * number_of_words_in_waveform * 2 ) + 16
140142
bl_type = int(bl_header["Type"])
141143
chan_id = int(bl_header["Channel"])
142144
block_pos[bl_type][chan_id].append(pos)

0 commit comments

Comments
 (0)