pspc/field/FFT.h

#ifndef PSPC_FFT_H
#define PSPC_FFT_H
 
/*
* PSCF Package 
*
* Copyright 2016 - 2022, The Regents of the University of Minnesota
* Distributed under the terms of the GNU General Public License.
*/
 
#include <pspc/field/RField.h>
#include <pspc/field/RFieldDft.h>
#include <pscf/math/IntVec.h>
#include <util/global.h>
 
#include <fftw3.h>
 
namespace Pscf {
namespace Pspc {
 
   using namespace Util;
   using namespace Pscf;
 
   template <int D>
   class FFT 
   {
 
   public:
 
      FFT();
 
      virtual ~FFT();
 
      void setup(IntVec<D> const & meshDimensions);
 
      void setup(RField<D>& rField, RFieldDft<D>& kField);
 
      void forwardTransform(RField<D> const & in, RFieldDft<D>& out) const;
 
      void inverseTransform(RFieldDft<D>& in, RField<D>& out) const;
 
      void inverseTransformSafe(RFieldDft<D> const & in, RField<D>& out) const;
 
      IntVec<D> const & meshDimensions() const;
 
      bool isSetup() const;
 
   private:
 
      mutable RField<D> rFieldCopy_;
 
      mutable RFieldDft<D> kFieldCopy_;
 
      IntVec<D> meshDimensions_;
 
      int rSize_;
 
      int kSize_;
 
      fftw_plan fPlan_;
 
      fftw_plan iPlan_;
 
      bool isSetup_;
 
      void makePlans(RField<D>& rField, RFieldDft<D>& kField);
 
   };
 
   // Declarations of explicit specializations
 
   template <>
   void FFT<1>::makePlans(RField<1>& rField, RFieldDft<1>& kField);
 
   template <>
   void FFT<2>::makePlans(RField<2>& rField, RFieldDft<2>& kField);
 
   template <>
   void FFT<3>::makePlans(RField<3>& rField, RFieldDft<3>& kField);
 
   /*
   * Has this object been setup?
   */
   template <int D>
   inline bool FFT<D>::isSetup() const
   {  return isSetup_; }
 
   /*
   * Return the dimensions of the grid for which this was allocated.
   */
   template <int D>
   inline IntVec<D> const & FFT<D>::meshDimensions() const
   {  return meshDimensions_; }
 
   #ifndef PSPC_FFT_TPP
   // Suppress implicit instantiation
   extern template class FFT<1>;
   extern template class FFT<2>;
   extern template class FFT<3>;
   #endif
 
} // namespace Pscf::Pspc
} // namespace Pscf
#endif