LrAmPreCompressor.h

#ifndef RPG_LR_AM_COMPRESSOR_H
#define RPG_LR_AM_COMPRESSOR_H
 
/*
* PSCF - Polymer Self-Consistent Field Theory
*
* Copyright 2016 - 2022, The Regents of the University of Minnesota
* Distributed under the terms of the GNU General Public License.
*/
 
#include "Compressor.h"
#include <prdc/cuda/RField.h>    
#include <prdc/cuda/RFieldDft.h> 
#include <pscf/math/IntVec.h>
#include <pscf/cuda/DeviceArray.h>
#include <pscf/iterator/AmIteratorTmpl.h>      
#include <rpg/fts/compressor/intra/IntraCorrelation.h>            
 
namespace Pscf {
namespace Rpg
{
 
   template <int D> class System;
   template <int D> class IntraCorrelation;
   
   using namespace Util;
   using namespace Pscf::Prdc;
   using namespace Pscf::Prdc::Cuda;
 
   template <int D>
   class LrAmPreCompressor 
         : public AmIteratorTmpl<Compressor<D>, DeviceArray<cudaReal> >
   {
 
   public:
 
      LrAmPreCompressor(System<D>& system);
 
      ~LrAmPreCompressor();
 
      void readParameters(std::istream& in);
 
      void setup(bool isContinuation);      
      
      
      int compress();    
      
      void outputTimers(std::ostream& out);
 
      void clearTimers();
 
      double computeError(DeviceArray<cudaReal>& residTrial, 
                          DeviceArray<cudaReal>& fieldTrial,
                          std::string errorType,
                          int verbose);
      
      // Inherited public member functions
 
      using AmIteratorTmpl<Compressor<D>,DeviceArray<cudaReal> >::setClassName;
      
   protected:
  
      // Inherited protected members 
      using ParamComposite::readOptional;
      using Compressor<D>::system;
      using Compressor<D>::mdeCounter_;
 
   private:
 
      std::string errorType_;
      
      int itr_;
  
      DArray< RField<D> > w0_;  
      
      RField<D> error_;
      
      RField<D> resid_;
      
      RFieldDft<D> residK_;
     
      RField<D> intraCorrelation_;
      
      IntVec<D> kMeshDimensions_;
      
      int kSize_;
   
      DArray< RField<D> > wFieldTmp_;
 
      void setEqual(DeviceArray<cudaReal>& a, DeviceArray<cudaReal> const & b);
 
      double dotProduct(DeviceArray<cudaReal> const & a, DeviceArray<cudaReal> const & b);
 
      double maxAbs(DeviceArray<cudaReal> const & hist);
 
      void updateBasis(RingBuffer<DeviceArray<cudaReal> > & basis, 
                       RingBuffer<DeviceArray<cudaReal> > const & hists);
 
      void addHistories(DeviceArray<cudaReal>& trial, 
                        RingBuffer<DeviceArray<cudaReal> > const & basis, 
                        DArray<double> coeffs, 
                        int nHist);
 
      void addPredictedError(DeviceArray<cudaReal>& fieldTrial, 
                             DeviceArray<cudaReal> const & resTrial, 
                             double lambda);
 
      bool hasInitialGuess();
     
      int nElements();
 
      void getCurrent(DeviceArray<cudaReal>& curr);
 
      void evaluate();
 
      void getResidual(DeviceArray<cudaReal>& resid);
 
      void update(DeviceArray<cudaReal>& newGuess);
 
      void outputToLog();
   
      double computeLambda(double r);
      
      IntraCorrelation<D> intra_;
      
      bool isIntraCalculated_;
      
      bool isAllocated_;
      
   };
   
   #ifndef RPG_LR_AM_COMPRESSOR_TPP
   // Suppress implicit instantiation
   extern template class LrAmPreCompressor<1>;
   extern template class LrAmPreCompressor<2>;
   extern template class LrAmPreCompressor<3>;
   #endif
 
} // namespace Rpg
} // namespace Pscf
#endif