gtk2_ardour/fft.h

   1 /*
   2     Copyright (C) 2008 Paul Davis
   3     Author: Sampo Savolainen
   4
   5     This program is free software; you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License as published by
   7     the Free Software Foundation; either version 2 of the License, or
   8     (at your option) any later version.
   9
  10     This program is distributed in the hope that it will be useful,
  11     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13     GNU General Public License for more details.
  14
  15     You should have received a copy of the GNU General Public License
  16     along with this program; if not, write to the Free Software
  17     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18
  19 */
  20
  21 #ifndef __ardour_fft_h__
  22 #define __ardour_fft_h__
  23
  24
  25 #include <iostream>
  26
  27 #include <stdio.h>
  28 #include <sys/types.h>
  29 #include <complex> // this needs to be included before fftw3.h
  30 #include <fftw3.h>
  31
  32
  33 #include <ardour/types.h>
  34
  35 class FFT
  36 {
  37         public:
  38                 FFT(uint32_t);
  39                 ~FFT();
  40
  41                 enum WindowingType {
  42                         NONE,
  43                         HANN
  44                 };
  45
  46                 void reset();
  47                 void analyze(ARDOUR::Sample *, WindowingType w = NONE);
  48                 void calculate();
  49
  50                 uint32_t bins() const { return _data_size; }
  51
  52                 float power_at_bin(uint32_t i) const { return _power_at_bin[i]; }
  53                 float phase_at_bin(uint32_t i) const { return _phase_at_bin[i]; }
  54
  55
  56         private:
  57
  58                 float *get_hann_window();
  59
  60                 uint32_t const _window_size;
  61                 uint32_t const _data_size;
  62                 uint32_t _iterations;
  63
  64                 float *_hann_window;
  65
  66                 float *_fftInput;
  67                 float *_fftOutput;
  68
  69                 float *_power_at_bin;
  70                 float *_phase_at_bin;
  71
  72                 fftwf_plan _plan;
  73 };
  74
  75 #endif