Add time vector case to 'get_durations'.

SpikeInterface · Jul 1, 2024 · 75af77b · 75af77b
1 parent 30f03b2
commit 75af77b
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/spikeinterface/core/baserecording.py b/src/spikeinterface/core/baserecording.py
@@ -233,8 +233,14 @@ def get_duration(self, segment_index=None) -> float:
             The duration in seconds
         """
         segment_index = self._check_segment_index(segment_index)
-        segment_num_samples = self.get_num_samples(segment_index=segment_index)
-        segment_duration = segment_num_samples / self.get_sampling_frequency()
+
+        if self.has_time_vector(segment_index):
+            times = self.get_times(segment_index)
+            segment_duration = times[-1] - times[0] + (1 / self.get_sampling_frequency())
+        else:
+            segment_num_samples = self.get_num_samples(segment_index=segment_index)
+            segment_duration = segment_num_samples / self.get_sampling_frequency()
+
         return segment_duration
 
     def get_total_duration(self) -> float:
@@ -246,7 +252,7 @@ def get_total_duration(self) -> float:
         float
             The duration in seconds
         """
-        duration = self.get_total_samples() / self.get_sampling_frequency()
+        duration = sum([self.get_duration(idx) for idx in self.get_num_segments()])
         return duration
 
     def get_memory_size(self, segment_index=None) -> int: