LArFFTWPlan.cxx

Go to the documentation of this file.

#include "lardata/Utilities/LArFFTWPlan.h"

using std::string;
std::mutex util::LArFFTWPlan::mutex_;

util::LArFFTWPlan::LArFFTWPlan(int transformSize, const std::string& option)
  : fSize(transformSize), fOption(option)
{

  std::lock_guard<std::mutex> lock(mutex_);

  fFreqSize = fSize / 2 + 1;
  fN = new int[1];
  fN[0] = fSize;

  fIn = fftw_malloc(sizeof(double) * fSize);
  fOut = fftw_malloc(sizeof(fftw_complex) * fFreqSize);
  fPlan = (void*)fftw_plan_dft_r2c(1, fN, (double*)fIn, (fftw_complex*)fOut, MapFFTWOption());

  rIn = fftw_malloc(sizeof(fftw_complex) * fFreqSize);
  rOut = fftw_malloc(sizeof(double) * fSize);
  rPlan = (void*)fftw_plan_dft_c2r(1, fN, (fftw_complex*)rIn, (double*)rOut, MapFFTWOption());
}

util::LArFFTWPlan::~LArFFTWPlan()
{
  fftw_destroy_plan((fftw_plan)fPlan);
  fPlan = 0;
  fftw_free(fIn);
  fIn = 0;
  fftw_free((fftw_complex*)fOut);
  fOut = 0;

  fftw_destroy_plan((fftw_plan)rPlan);
  rPlan = 0;
  fftw_free((fftw_complex*)rIn);
  rIn = 0;
  fftw_free(rOut);
  rOut = 0;

  delete[] fN;
  fN = 0;
}

unsigned int util::LArFFTWPlan::MapFFTWOption()
{
  std::transform(fOption.begin(), fOption.end(), fOption.begin(), ::toupper);
  if (fOption.find("ES") != string::npos) return FFTW_ESTIMATE;
  if (fOption.find("M") != string::npos) return FFTW_MEASURE;
  if (fOption.find("P") != string::npos) return FFTW_PATIENT;
  if (fOption.find("EX") != string::npos) return FFTW_EXHAUSTIVE;
  return FFTW_ESTIMATE;
}