timeside/analyzer/aubio_temporal.py: add beat confidence

diff --git a/timeside/analyzer/aubio_temporal.py b/timeside/analyzer/aubio_temporal.py
index f4c3ce4dbedb5c3052f50223b261cb2cc80d2e90..60b4c0b70b5843f6795db7a0c22b69d013800319 100644 (file)
--- a/timeside/analyzer/aubio_temporal.py
+++ b/timeside/analyzer/aubio_temporal.py
@@ -51,6 +51,7 @@ class AubioTemporal(Analyzer):
         self.block_read = 0
         self.onsets = []
         self.beats = []
+        self.beat_confidences = []
 
     @staticmethod
     @interfacedoc
@@ -78,6 +79,7 @@ class AubioTemporal(Analyzer):
             self.onsets += [self.o.get_last_s()]
         if self.t(frames):
             self.beats += [self.t.get_last_s()]
+            self.beat_confidences += [self.t.get_confidence()]
         self.block_read += 1
         return frames, eod
 
@@ -128,6 +130,18 @@ class AubioTemporal(Analyzer):
 
         self.pipe.results.add(beats)
 
+        #---------------------------------
+        #  Beat confidences: Event (time, value)
+        #---------------------------------
+        beat_confidences = self.new_result(data_mode='value', time_mode='event')
+        beat_confidences.id_metadata.id += '.' + "beat_confidence"
+        beat_confidences.id_metadata.name += " " + "Beat confidences"
+        beat_confidences.id_metadata.unit = None
+        beat_confidences.data_object.time = self.beats
+        beat_confidences.data_object.value = self.beat_confidences
+
+        self.pipe.results.add(beat_confidences)
+
         #---------------------------------
         #  BPM: Segment (time, duration, value)
         #---------------------------------