scft/iterator/AmIteratorGrid.h

#ifndef RP_AM_ITERATOR_GRID_H
#define RP_AM_ITERATOR_GRID_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/iterator/AmIteratorTmpl.h>    // base class
#include <pscf/iterator/AmbdInteraction.h>   // member
#include <iostream>
 
namespace Pscf {
namespace Rp {
 
   using namespace Util;

   template <int D, class T>
   class AmIteratorGrid
    : public AmIteratorTmpl< typename T::Iterator, typename T::Vector >
   {
 
   public:

      void readParameters(std::istream& in) override;

      void outputTimers(std::ostream& out) const override;
 
   protected:

      AmIteratorGrid(typename T::System& system);

      virtual ~AmIteratorGrid();

      void setup(bool isContinuation) override;

      using IteratorT = typename T::Iterator;

      using VectorT = typename T::Vector;

      using AmIterTmplT = AmIteratorTmpl< IteratorT, VectorT >;
 
      // Inherited protected members
      using IteratorT::system;
      using IteratorT::flexibleParams_;
 
   private:

      AmbdInteraction interaction_;

      double scaleStress_;
 
      // Private overridden virtual functions

      int nElements() override;

      bool hasInitialGuess() override;

      void getCurrent(VectorT& curr) override;

      void evaluate() override;

      void getResidual(VectorT& resid) override;

      void update(VectorT& newGuess) override;

      void outputToLog() override;
 
      // Private type aliases
      using RealT = double;
      using RFieldT = typename T::RField;
 
      //template <typename Data> 
      //using HostArrayT = (typename T::HostDArray)<Data>;
 
   };
 
} // namespace Rp
} // namespace Pscf
#endif