fix spectro buffer post-process

diff --git a/timeside/analyzer/spectrogram.py b/timeside/analyzer/spectrogram.py
index 040b98cfb74d42c5c256753cc0da1b8071654f37..44a779ac0dfaf9257be112aee798a1dbaf3da90b 100644 (file)
--- 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 0f8907f1ff674699a390077113dbc6172141563c..ac28b6c1f1c218e160a6ba7b2edd600aa1c008cc 100644 (file)
--- 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)
+