# -*- coding: utf-8 -*-
#
-# Copyright (c) 2013-2014 Parisson SARL
-# Copyright (c) 2013-2014 Thomas Fillon
+# Copyright (c) 2013 Paul Brossier <piem@piem.org>
# This file is part of TimeSide.
# You should have received a copy of the GNU General Public License
# along with TimeSide. If not, see <http://www.gnu.org/licenses/>.
-# Author: Thomas Fillon <thomas@parisson.com>
-
+# Author: Paul Brossier <piem@piem.org>
from __future__ import division
from timeside.core import implements, interfacedoc
from timeside.analyzer.core import Analyzer
from timeside.api import IAnalyzer
from timeside.analyzer.preprocessors import downmix_to_mono, frames_adapter
-from timeside.tools.parameters import Int, HasTraits
from timeside.tools.buffering import BufferTable
+from ..tools.parameters import Int, HasTraits
+from timeside.tools.pyx.cfft import cfft
import numpy as np
class Spectrogram(Analyzer):
+
"""
- Spectrogram analyzer
+ Spectrogram image builder with an extensible buffer based on tables
Parameters
----------
pipe.run()
res = spectrogram.results['spectrogram_analyzer']
res.render()
-
-
"""
+
implements(IAnalyzer)
# Define Parameters
else:
self.fft_size = fft_size
- self.values = BufferTable()
+ self.values = []
@interfacedoc
def setup(self, channels=None, samplerate=None,
@downmix_to_mono
@frames_adapter
def process(self, frames, eod=False):
- stft = np.fft.rfft(frames, self.fft_size)
- self.values.append('stft', stft)
- return frames, eod
+ self.values.append(np.abs(np.fft.rfft(frames, self.fft_size)))
+ 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 = np.abs(self.values['stft'])
+ # spectrogram.data_object.value = self.values['spectrogram']
+ spectrogram.data_object.value = self.values
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()
if __name__ == "__main__":
import doctest
import timeside
- doctest.testmod(timeside.analyzer.spectrogram, verbose=True)
+ doctest.testmod(timeside.analyzer.spectrogram, verbose=True)
\ No newline at end of file
--- /dev/null
+# -*- coding: utf-8 -*-
+#
+# Copyright (c) 2013-2014 Parisson SARL
+# Copyright (c) 2013-2014 Thomas Fillon
+
+# This file is part of TimeSide.
+
+# TimeSide is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 2 of the License, or
+# (at your option) any later version.
+
+# TimeSide is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+
+# You should have received a copy of the GNU General Public License
+# along with TimeSide. If not, see <http://www.gnu.org/licenses/>.
+
+# Author: Thomas Fillon <thomas@parisson.com>
+
+from __future__ import division
+from timeside.core import implements, interfacedoc
+from timeside.analyzer.core import Analyzer
+from timeside.api import IAnalyzer
+from timeside.analyzer.preprocessors import downmix_to_mono, frames_adapter
+from timeside.tools.parameters import Int, HasTraits
+from timeside.tools.buffering import BufferTable
+from timeside.analyzer.spectrogram import Spectrogram
+
+import numpy as np
+
+
+class SpectrogramBuffer(Spectrogram):
+ """
+ Spectrogram image builder with an extensible buffer based on tables
+
+ Parameters
+ ----------
+ input_blocksize : int, optional
+ Blocksize of the input signal, default to 2048
+ input_stepsize : str, optional
+ The second parameter, default to half blocksize.
+ fft_size : int, optional
+ The size of the fft, default to blocksize.
+
+ Examples
+ --------
+ >>> import timeside
+ >>> from timeside.core import get_processor
+ >>> from timeside.tools.test_samples import samples
+ >>> audio_source = samples['sweep.wav']
+ >>> decoder = get_processor('file_decoder')(uri=audio_source)
+ >>> spectrogram = get_processor('spectrogram_analyzer')(input_blocksize=2048, input_stepsize=1024)
+ >>> pipe = (decoder | spectrogram)
+ >>> pipe.run()
+ >>> spectrogram.results.keys()
+ ['spectrogram_analyzer']
+ >>> result = spectrogram.results['spectrogram_analyzer']
+ >>> result.data.shape
+ (344, 1025)
+
+ .. plot::
+
+ import timeside
+ from timeside.core import get_processor
+ from timeside.tools.test_samples import samples
+ audio_source = samples['sweep.wav']
+ decoder = get_processor('file_decoder')(uri=audio_source)
+ spectrogram = get_processor('spectrogram_analyzer')(input_blocksize=2048,
+ input_stepsize=1024)
+ pipe = (decoder | spectrogram)
+ pipe.run()
+ res = spectrogram.results['spectrogram_analyzer']
+ res.render()
+ """
+
+ implements(IAnalyzer)
+
+ def __init__(self, input_blocksize=2048, input_stepsize=None,
+ fft_size=None):
+ super(SpectrogramBuffer, self).__init__()
+ self.values = BufferTable()
+
+ @staticmethod
+ @interfacedoc
+ def id():
+ return "spectrogram_analyzer_buffer"
+
+ @staticmethod
+ @interfacedoc
+ def name():
+ return "Spectrogram Analyzer with extensible buffer"
+
+ @downmix_to_mono
+ @frames_adapter
+ def process(self, frames, eod=False):
+ stft = np.fft.rfft(frames, self.fft_size)
+ self.values.append('stft', stft)
+ return frames, eod
+
+ def release(self):
+ self.values.close()
+
+
+if __name__ == "__main__":
+ import doctest
+ import timeside
+ doctest.testmod(timeside.analyzer.spectrogram_buffer, verbose=True)
\ No newline at end of file
projects / timeside.git / commitdiff