rpg/scft/sweep/Sweep.h

#ifndef RPG_SWEEP_H
#define RPG_SWEEP_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/sweep/SweepTmpl.h>            // base class template
#include <rpg/scft/sweep/BasisFieldState.h>  // template argument
 
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 Sweep : public SweepTmpl< BasisFieldState<D> >
   {
 
   public:

      Sweep();

      Sweep(System<D>& system);

      ~Sweep();

      void setSystem(System<D>& system);

      virtual void readParameters(std::istream& in);
 
      // Public members inherited from base class template SweepTmpl
      using SweepTmpl< BasisFieldState<D> >::addParameterTypes;
      using SweepTmpl< BasisFieldState<D> >::addParameterType;
      using SweepTmpl< BasisFieldState<D> >::historyCapacity;
      using SweepTmpl< BasisFieldState<D> >::historySize;
      using SweepTmpl< BasisFieldState<D> >::nAccept;
      using SweepTmpl< BasisFieldState<D> >::state;
      using SweepTmpl< BasisFieldState<D> >::s;
      using SweepTmpl< BasisFieldState<D> >::c;
 
   protected:

      virtual void checkAllocation(BasisFieldState<D>& state);

      virtual void setup();

      virtual void setParameters(double sNew) = 0;

      virtual void extrapolate(double sNew);

      virtual int solve(bool isContinuation);

      virtual void reset();

      virtual void getSolution();

      virtual void cleanup();

      bool hasSystem()
      {  return (systemPtr_ != 0); }

      System<D>& system()
      {  return *systemPtr_; }

      bool writeCRGrid_;

      bool writeCBasis_;

      bool writeWRGrid_;
 
      // Protected members inherited from base classes
      using SweepTmpl< BasisFieldState<D> >::ns_;
      using SweepTmpl< BasisFieldState<D> >::baseFileName_;
      using SweepTmpl< BasisFieldState<D> >::initialize;
      using SweepTmpl< BasisFieldState<D> >::setCoefficients;
      using ParamComposite::readOptional;
 
   private:

      BasisFieldState<D> trial_;

      FSArray<double, 6> unitCellParameters_;

      std::ofstream logFile_;

      System<D>* systemPtr_;

      void outputSolution();

      void outputSummary(std::ostream&);
 
   };
 
} // namespace Rpg
} // namespace Pscf
#endif