Add SelectSegmentEvent (#4575)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: tayheau

src/spikeinterface/core/baseevent.py

@@ -66,6 +66,11 @@ def add_event_segment(self, event_segment):
     def get_num_segments(self):
         return len(self._event_segments)
 
+    def select_segment(self, segment_indices: int | list[int]):
+        from .segmentutils import SelectSegmentEvent
+
+        return SelectSegmentEvent(self, segment_indices=segment_indices)
+
     def get_events(
         self,
         channel_id: int | str | None = None,

src/spikeinterface/core/segmentutils.py

@@ -1,5 +1,6 @@
 import numpy as np
 
+from .baseevent import BaseEvent
 from .baserecording import BaseRecording, BaseRecordingSegment
 from .basesorting import BaseSorting, BaseSortingSegment
 
@@ -604,3 +605,23 @@ def __init__(self, sorting: BaseSorting, segment_indices: int | list[int]):
 
 
 select_segment_sorting = define_function_from_class(source_class=SelectSegmentSorting, name="select_segment_sorting")
+
+
+class SelectSegmentEvent(BaseEvent):
+    def __init__(self, event: BaseEvent, segment_indices: int | list[int]):
+        BaseEvent.__init__(self, event.channel_ids, event.structured_dtype)
+
+        if isinstance(segment_indices, int):
+            segment_indices = [segment_indices]
+
+        num_segments = event.get_num_segments()
+
+        if not all(0 <= s < num_segments for s in segment_indices):
+            raise ValueError(f"'segment_index' must be between 0 and {num_segments - 1}")
+
+        for seg_idx in segment_indices:
+            seg = event._event_segments[seg_idx]
+            self.add_event_segment(seg)
+
+        self._parent = event
+        self._kwargs = {"event": event, "segment_indices": segment_indices}