From: Guillaume Pellerin <yomguy@parisson.com>
Date: Thu, 23 Oct 2014 12:23:01 +0000 (+0200)
Subject: fix spectro buffer post-process
X-Git-Tag: 0.6~2^2~4
X-Git-Url: https://git.parisson.com/?a=commitdiff_plain;h=c74f3a7ea6fe45623c358a0726111d011242e733;p=timeside.git

fix spectro buffer post-process
---

diff --git a/timeside/analyzer/spectrogram.py b/timeside/analyzer/spectrogram.py
index 040b98c..44a779a 100644
--- a/timeside/analyzer/spectrogram.py
+++ b/timeside/analyzer/spectrogram.py
@@ -134,7 +134,6 @@ class Spectrogram(Analyzer):
         nb_freq = spectrogram.data_object.value.shape[1]
         spectrogram.data_object.y_value = (np.arange(0, nb_freq) *
                                            self.samplerate() / self.fft_size)
-
         self.add_result(spectrogram)
 
 
diff --git a/timeside/analyzer/spectrogram_buffer.py b/timeside/analyzer/spectrogram_buffer.py
index 0f8907f..ac28b6c 100644
--- a/timeside/analyzer/spectrogram_buffer.py
+++ b/timeside/analyzer/spectrogram_buffer.py
@@ -100,6 +100,15 @@ class SpectrogramBuffer(Spectrogram):
             self.values.append('stft', stft)
             return frames, eod
 
+    def post_process(self):
+        spectrogram = self.new_result(data_mode='value', time_mode='framewise')
+        spectrogram.parameters = {'fft_size': self.fft_size}
+        spectrogram.data_object.value = self.values['stft']
+        nb_freq = spectrogram.data_object.value.shape[1]
+        spectrogram.data_object.y_value = (np.arange(0, nb_freq) *
+                                           self.samplerate() / self.fft_size)
+        self.add_result(spectrogram)
+
     def release(self):
         self.values.close()
 
@@ -107,4 +116,5 @@ class SpectrogramBuffer(Spectrogram):
 if __name__ == "__main__":
     import doctest
     import timeside
-    doctest.testmod(timeside.analyzer.spectrogram_buffer, verbose=True)
\ No newline at end of file
+    doctest.testmod(timeside.analyzer.spectrogram_buffer, verbose=True)
+