ComplexFFT.H

Go to the documentation of this file.

/* Copyright 2000-2018 Matt Flax <flatmax@flatmax.org>
   This file is part of GTK+ IOStream class set

   GTK+ IOStream 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.

   GTK+ IOStream 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 have received a copy of the GNU General Public License
   along with GTK+ IOStream
*/
#ifndef COMPLEXFFT_H_
#define COMPLEXFFT_H_

#include "fft/FFTCommon.H"
#include "fft/ComplexFFTData.H"

//class ComplexFFTData;

class ComplexFFT {
  fftw_plan fwdPlan, invPlan;
  void createPlan(void){
  if (data){
    //fftw3
    fwdPlan = fftw_plan_dft_1d(data->getSize(), data->in, data->out, FFTW_FORWARD, PLANTYPE);
    invPlan = fftw_plan_dft_1d(data->getSize(), data->out, data->in, FFTW_BACKWARD, PLANTYPE);
  }
}

  void destroyPlan(void){
  if (data){
    fftw_destroy_plan(fwdPlan);
    fftw_destroy_plan(invPlan);
  }
}

protected:
  //  int size;
  ComplexFFTData *data;
public:

  ComplexFFT(ComplexFFTData *d){
  //  std::cout <<"ComplexFFT init:"<<this<<std::endl;
  data=d;
  createPlan();
}

  virtual ~ComplexFFT(){
  destroyPlan();
}


  void switchData(ComplexFFTData *d){
  //fftw_cleanup();
  destroyPlan();
  data=d;
  createPlan();
}


  void fwdTransform(){
  if (!data)
    printf("ComplexFFT::fwdTransform : data not present, please switch data\n");
  else
    fftw_execute(fwdPlan);
  /*fftw_execute_dft(
          fwdPlan,
          data->in, data->out);

          }*/
}


  void invTransform(){
  if (!data)
    printf("ComplexFFT::invTransform : data not present, please switch data\n");
  else
    fftw_execute(invPlan);
  /*fftw_execute_dft(
          invPlan,
          data->in, data->out);

          }*/
}

};
#endif // COMPLEXFFT_H_