QFFTWPlan.cc

Go to the documentation of this file.

#include "QFFTWPlan.hh"
 
 
Diana::QFFTWRealComplexPlan::QFFTWRealComplexPlan(Diana::QVector &realData,Diana::QVectorC &complexData,QFFT::FFTDirection direction)
  : fSize(realData.Size()),
    fRealData(realData.fData,realData.fSize,realData.fStride),
    fComplexData((gsl_complex *)complexData.fData,complexData.fSize,complexData.fStride)
{
  Build(direction);
}
 
// Diana::QFFTWRealComplexPlan::QFFTWRealComplexPlan(size_t size,Diana::QVector &realData,Diana::QVectorC &complexData,QFFT::FFTDirection direction)
//   : fSize(size),
//     fRealData(realData.fData,realData.fSize,realData.fStride),
//     fComplexData((gsl_complex *)complexData.fData,complexData.fSize,complexData.fStride)
// {
//   fRealData.Resize(size);
//   fComplexData.Resize(size/2+1);
//   Build(direction);
// }
 
Diana::QFFTWRealComplexPlan::QFFTWRealComplexPlan(Diana::QVector &realComplexData,QFFT::FFTDirection direction)
  : fSize(realComplexData.Size()),
    fRealData(realComplexData.fData,realComplexData.fSize,realComplexData.fStride),
    fComplexData((gsl_complex *)realComplexData.fData,realComplexData.fSize,realComplexData.fStride)
{
  Build(direction);
}
 
// Diana::QFFTWRealComplexPlan::QFFTWRealComplexPlan(size_t size,Diana::QVector &realComplexData,QFFT::FFTDirection direction)
//   : fSize(size),
//     fRealData(realComplexData.fData,realComplexData.fSize,realComplexData.fStride),
//     fComplexData((gsl_complex *)realComplexData.fData,realComplexData.fSize,realComplexData.fStride)
// {
//   fRealData.Resize(size);
//   fComplexData.Resize(size/2+1);
//   Build(direction);
// }
  
Diana::QFFTWRealComplexPlan::~QFFTWRealComplexPlan() {
#ifdef _HAVE_FFTW3_
  fftw_destroy_plan(fFFTPlan);
#endif
}
 
void Diana::QFFTWRealComplexPlan::Execute() {
#ifdef _HAVE_FFTW3_
  fftw_execute(fFFTPlan);
#endif
}
 
Diana::QFFTWRealComplexPlan Diana::QFFTWRealComplexPlan::CreateForwardPlan(Diana::QVector &realData,Diana::QVectorC &complexData) {
  return QFFTWRealComplexPlan(realData,complexData,QFFT::kForward);
}
 
Diana::QFFTWRealComplexPlan Diana::QFFTWRealComplexPlan::CreateBackwardPlan(Diana::QVectorC &complexData,Diana::QVector &realData) {
  return QFFTWRealComplexPlan(realData,complexData,QFFT::kBackward);
}
 
// Diana::QFFTWRealComplexPlan Diana::QFFTWRealComplexPlan::CreateForwardPlan(size_t size,Diana::QVector &realData,Diana::QVectorC &complexData) {
//   return QFFTWRealComplexPlan(size,realData,complexData,QFFT::kForward);
// }
 
// Diana::QFFTWRealComplexPlan Diana::QFFTWRealComplexPlan::CreateBackwardPlan(size_t size,Diana::QVectorC &complexData,Diana::QVector &realData) {
//   return QFFTWRealComplexPlan(size,realData,complexData,QFFT::kBackward);
// }
 
void Diana::QFFTWRealComplexPlan::Build(QFFT::FFTDirection direction) {
  if(fComplexData.Size() < fRealData.Size()/2+1)
    throw QError(QERR_SIZE_NOT_MATCH,__FILE__,__LINE__,"Complex data vector is too small for the ");
#ifdef _HAVE_FFTW3_
  if(direction == QFFT::kForward)
    fFFTPlan = fftw_plan_many_dft_r2c(1,&fSize,1,fRealData.fData,NULL,fRealData.fStride,0,
                                      (fftw_complex *)fComplexData.fData,NULL,fComplexData.fStride,0,
                                      FFTW_MEASURE);
  if(direction == QFFT::kBackward)
    fFFTPlan = fftw_plan_many_dft_c2r(1,&fSize,1,(fftw_complex *)fComplexData.fData,NULL,fComplexData.fStride,0,
                                      fRealData.fData,NULL,fRealData.fStride,0,
                                      FFTW_MEASURE);
#else
  throw QError(QERR_UNKNOWN_ERR,__FILE__,__LINE__,
               "Package mathtools not compiled with FFTW3 support!");
#endif
}