fts/compressor/IntraCorrelation.h

#ifndef RP_INTRACORRELATION_H
#define RP_INTRACORRELATION_H
 
/*
* PSCF - Polymer Self-Consistent Field
*
* Copyright 2015 - 2025, The Regents of the University of Minnesota
* Distributed under the terms of the GNU General Public License.
*/
 
#include <pscf/math/IntVec.h>         // member
#include <util/containers/DArray.h>   // member
 
// Forward declaration
namespace Pscf {
   namespace Correlation {
      template <typename WT> class Mixture;
   }
}
 
namespace Pscf {
namespace Rp {
 
   using namespace Util;

   template <int D, class T>
   class IntraCorrelation
   {
 
   public:

      void computeOmegaTotal(Array<typename T::Real>& correlations);
 
   protected:

      IntraCorrelation(typename T::System const & system);

      ~IntraCorrelation();

      void getMeshDimensions();

      void computeGsq();
 
      /*
      * Get the size of the k-space mesh. 
      */
      int kSize() const;

      typename T::System const & system() const;
 
   private:
 
      using RealT = typename T::Real;

      typename T::System const * systemPtr_;

      Correlation::Mixture<RealT>* correlationMixturePtr_;

      DArray<RealT> Gsq_;

      IntVec<D> meshDimensions_;

      IntVec<D> kMeshDimensions_;

      int kSize_;
 
      using FFTT = typename T::FFT;
 
   };
 
   // Get the number of wavevectors in the k-space mesh.
   template <int D, class T> inline
   int IntraCorrelation<D,T>::kSize() const
   {  return kSize_; }
 
   // Get the parent system by const reference.
   template <int D, class T> inline
   typename T::System const & IntraCorrelation<D,T>::system() const
   {  return *systemPtr_; }
 
} // namespace Rp
} // namespace Pscf
#endif