From c74f3a7ea6fe45623c358a0726111d011242e733 Mon Sep 17 00:00:00 2001 From: Guillaume Pellerin Date: Thu, 23 Oct 2014 14:23:01 +0200 Subject: [PATCH] fix spectro buffer post-process --- timeside/analyzer/spectrogram.py | 1 - timeside/analyzer/spectrogram_buffer.py | 12 +++++++++++- 2 files changed, 11 insertions(+), 2 deletions(-) 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) + -- 2.39.5