scft/iterator/AmIteratorBasis.h

#ifndef RP_AM_ITERATOR_BASIS_H
#define RP_AM_ITERATOR_BASIS_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>   // direct base member 
#include <util/containers/DArray.h>          // indirect base member
 
namespace Pscf {
namespace Rp {
 
   using namespace Util;

   template <int D, class T>
   class AmIteratorBasis 
    : public AmIteratorTmpl<typename T::Iterator, DArray<double> >
   {
 
   public:

      using IteratorT = typename T::Iterator;

      using VectorT = DArray<double>;

      using AmIterTmplT = AmIteratorTmpl< IteratorT, VectorT >;

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

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

      AmIteratorBasis(typename T::System& system);

      virtual ~AmIteratorBasis();

      void setup(bool isContinuation) override;
 
      // Inherited protected members
      using IteratorT::system;
      using IteratorT::flexibleParams_;
 
   private:

      AmbdInteraction interaction_;

      double scaleStress_;
 
      // Private virtual functions that interact with parent system

      bool hasInitialGuess() override;

      int nElements() override;

      void getCurrent(VectorT& curr) override;

      void evaluate() override;

      void getResidual(VectorT& resid) override;

      void update(VectorT& newGuess) override;

      void outputToLog() override;
 
   };
 
} // namespace Rp
} // namespace Pscf
#endif