fix fft_size arg for spectrogram

diff --git a/timeside/analyzer/spectrogram.py b/timeside/analyzer/spectrogram.py
index 3357448d9fa731765eedf0d4f4fe1c59f0432eae..2cc29c28dc2ee20bffca6cdc50bb27812d641888 100644 (file)
--- 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():