rpg/fts/compressor/AmCompressor.h

#ifndef RPG_AM_COMPRESSOR_H
#define RPG_AM_COMPRESSOR_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 "Compressor.h"
#include <prdc/cuda/RField.h>         
#include <pscf/cuda/DeviceArray.h>
#include <pscf/iterator/AmIteratorTmpl.h>     
 
#include <util/containers/DArray.h>                 
 
namespace Pscf {
namespace Rpg {
 
   template <int D> class System;
 
   using namespace Util;
   using namespace Pscf::Prdc;
   using namespace Pscf::Prdc::Cuda;

   template <int D>
   class AmCompressor : public AmIteratorTmpl<Compressor<D>, 
                               DeviceArray<cudaReal> >
   {
 
   public:

      AmCompressor(System<D>& system);

      ~AmCompressor();

      void readParameters(std::istream& in);
 
      void setup(bool isContinuation);      
      
      
      int compress();    
      
      double computeLambda(double r);  
      
      int mdeCounter(); 
      
      void outputTimers(std::ostream& out) const;
      
      void clearTimers();
      
      // 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:
      int counter_;
      
      DArray< RField<D> > w0_;  

      bool isAllocated_;
      
      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();
      
 
   };
 
   // Explicit instantiation declarations
   extern template class AmCompressor<1>;
   extern template class AmCompressor<2>;
   extern template class AmCompressor<3>;
   
} // namespace Rpg
} // namespace Pscf
#endif