rp/solvers/Polymer.h

#ifndef RP_POLYMER_H
#define RP_POLYMER_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/solvers/PolymerTmpl.h>    // base class template
#include <util/containers/FSArray.h>     // member template
 
// Forward declarations
namespace Util {
   template <typename T> class DArray;
}
 
namespace Pscf {
namespace Rp {
 
   using namespace Util;

   template <int D, class T>
   class Polymer 
     : public PolymerTmpl<typename T::Block, typename T::Propagator, double>
   {
 
   public:
 
      // Public type name aliases

      using BlockT = typename T::Block;

      using PropagatorT = typename T::Propagator;

      using PolymerTmplT = PolymerTmpl<BlockT, PropagatorT, double>;

      using typename PolymerTmplT::PolymerSpeciesT;

      using typename PolymerTmplT::SpeciesT;
 
      // Public member functions

      void setNParams(int nParam);

      void clearUnitCellData();

      void compute(DArray<typename T::RField> const & wFields,
                   double phiTot = 1.0);

      void computeStress();

      double stress(int n) const;
 
      // Inherited non-dependent public member functions
      using PolymerTmplT::edge;
      using PolymerTmplT::block;
      using PolymerTmplT::propagator;
      using PolymerSpeciesT::vertex;
      using PolymerSpeciesT::propagatorId;
      using PolymerSpeciesT::path;
      using PolymerSpeciesT::nBlock;
      using PolymerSpeciesT::nVertex;
      using PolymerSpeciesT::nPropagator;
      using PolymerSpeciesT::length;
      using PolymerSpeciesT::nBead;
      using PolymerSpeciesT::type;
      using SpeciesT::phi;
      using SpeciesT::mu;
      using SpeciesT::q;
      using SpeciesT::ensemble;
      using SpeciesT::setPhi;
      using SpeciesT::setMu;
 
   protected:

      Polymer();

      ~Polymer();
 
   private:

      FSArray<double, 6> stress_;

      int nParam_;
 
      // Restrict access to inherited functions
      using PolymerTmplT::solve;
      using SpeciesT::setQ;
 
   };

   template <int D, class T> inline
   double Polymer<D,T>::stress(int n) const
   {  return stress_[n]; }
 
} // namespace Rp
} // namespace Pscf
#endif