Skip to content

Commit a2ca303

Browse files
committed
edits _seg_t_stops just for slice
1 parent 5a17bdc commit a2ca303

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

neo/rawio/blackrockrawio.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -412,7 +412,7 @@ def _parse_header(self):
412412
for k, (data, ev_ids) in self.nev_data.items():
413413
segment_mask = ev_ids == data_bl
414414
if data[segment_mask].size > 0:
415-
t = (data[segment_mask][-1]['timestamp'] + 1) / self.__nev_basic_header[
415+
t = (data[segment_mask][-1]['timestamp']) / self.__nev_basic_header[
416416
'timestamp_resolution']
417417
max_nev_time = max(max_nev_time, t)
418418
if max_nev_time > t_stop:
@@ -450,8 +450,7 @@ def _parse_header(self):
450450
# Calculate t_start and t_stop for each segment in seconds
451451
resolution = self.__nev_basic_header['timestamp_resolution']
452452
self._seg_t_starts = [v / float(resolution) for k, v in sorted(min_nev_times.items())]
453-
self._seg_t_stops = [(v + 1) / float(resolution)
454-
for k, v in sorted(max_nev_times.items())]
453+
self._seg_t_stops = [v / float(resolution) for k, v in sorted(max_nev_times.items())]
455454
self._nb_segment = len(self._seg_t_starts)
456455
self._sigs_t_starts = [None] * self._nb_segment
457456

@@ -614,7 +613,8 @@ def _get_timestamp_slice(self, timestamp, seg_index, t_start, t_stop):
614613
if t_start is None:
615614
t_start = self._seg_t_starts[seg_index]
616615
if t_stop is None:
617-
t_stop = self._seg_t_stops[seg_index]
616+
t_stop = self._seg_t_stops[seg_index] + 1 / float(
617+
self.__nev_basic_header['timestamp_resolution'])
618618

619619
if t_start is None:
620620
ind_start = None

0 commit comments

Comments
 (0)