libs/Music/CFFTW3.cpp

   1 #include "CFFTW3.h"
   2
   3 #include <assert.h>
   4 using namespace std;
   5 #include <CppAddons/CAMath.h>
   6 using namespace Math;
   7
   8 CFFTW3::CFFTW3(bool forward)
   9 {
  10         m_size = 0;
  11         m_out = m_in = NULL;
  12         m_plan = NULL;
  13         m_forward = forward;
  14 }
  15 CFFTW3::CFFTW3(int n)
  16 {
  17         resize(n);
  18 }
  19 void CFFTW3::resize(int n)
  20 {
  21         assert(n>0);
  22
  23         m_size = n;
  24
  25         m_out = m_in = NULL;
  26
  27         m_in = new fftw_complex[m_size];
  28         m_out = new fftw_complex[m_size];
  29         in.resize(m_size);
  30         out.resize(m_size);
  31
  32         //  | FFTW_PRESERVE_INPUT
  33         if(m_forward)
  34                 m_plan = fftw_plan_dft_1d(m_size, m_in, m_out, FFTW_FORWARD, FFTW_MEASURE);
  35         else
  36                 m_plan = fftw_plan_dft_1d(m_size, m_in, m_out, FFTW_BACKWARD, FFTW_MEASURE);
  37 }
  38
  39 void CFFTW3::execute()
  40 {
  41         execute(in, out);
  42 }
  43
  44 void CFFTW3::execute(const vector<double>& in, vector<std::complex<double> >& out)
  45 {
  46         assert(int(in.size())>=m_size);
  47
  48         for(int i=0; i<m_size; i++)
  49         {
  50                 m_in[i][0] = in[i];
  51                 m_in[i][1] = 0.0;
  52         }
  53
  54         fftw_execute(m_plan);
  55
  56         if(int(out.size())<m_size)
  57                 out.resize(m_size);
  58
  59         for(int i=0; i<m_size; i++)
  60                 out[i] = make_complex(m_out[i]);
  61 }
  62
  63 CFFTW3::~CFFTW3()
  64 {
  65         if(m_plan)      fftw_destroy_plan(m_plan);
  66         if(m_in)        delete[] m_in;
  67         if(m_out)       delete[] m_out;
  68 }