@@ -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