From 14345dcdbe213366a37c18c1c71d37948c1d4e31 Mon Sep 17 00:00:00 2001 From: Guillaume Pellerin Date: Wed, 8 Jan 2014 17:17:30 +0100 Subject: [PATCH] fix fft_size arg for spectrogram --- timeside/analyzer/spectrogram.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/timeside/analyzer/spectrogram.py b/timeside/analyzer/spectrogram.py index 3357448..2cc29c2 100644 --- a/timeside/analyzer/spectrogram.py +++ b/timeside/analyzer/spectrogram.py @@ -29,10 +29,7 @@ import numpy as np class Spectrogram(Analyzer): implements(IAnalyzer) - values = [] - FFT_SIZE = 2048 - - def __init__(self, blocksize=2048, stepsize=None): + def __init__(self, blocksize=2048, stepsize=None, fft_size=None): super(Spectrogram, self).__init__() self.input_blocksize = blocksize @@ -41,12 +38,19 @@ class Spectrogram(Analyzer): else: self.input_stepsize = blocksize / 2 + if not fft_size: + self.FFT_SIZE = blocksize + else: + self.FFT_SIZE = fft_size + + values = [] + @interfacedoc def setup(self, channels=None, samplerate=None, blocksize=None, totalframes=None): super(Spectrogram, self).setup(channels, samplerate, blocksize, totalframes) - + @staticmethod @interfacedoc def id(): -- 2.39.5