PolymerTmpl.h

#ifndef PSCF_POLYMER_TMPL_H
#define PSCF_POLYMER_TMPL_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/chem/PolymerSpecies.h>    // base class
#include <util/param/ParamComposite.h>   // base class
 
#include <util/containers/Pair.h>        // member template
#include <util/containers/DArray.h>      // member template
 
namespace Pscf {
 
   class Edge;
   using namespace Util;

   template <class BT, class PT>
   class PolymerTmpl : public PolymerSpecies
   {
 
   public:
 
      // Public typename aliases

      using BlockT = BT;

      using PropagatorT = PT;
 
      // Public member functions

      PolymerTmpl();

      ~PolymerTmpl();

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

      virtual void solve(double phiTot = 1.0);


      Edge& edge(int id) final;

      Edge const& edge(int id) const final;

      BT& block(int id);

      BT const& block(int id) const ;

      PT& propagator(int blockId, int directionId);

      PT const & propagator(int blockId, int directionId) const;

      PT& propagator(int id);

 
      // Inherited public members
 
      using PolymerSpecies::vertex;
      using PolymerSpecies::propagatorId;
      using PolymerSpecies::path;
      using PolymerSpecies::nBlock;
      using PolymerSpecies::nVertex;
      using PolymerSpecies::nPropagator;
      using PolymerSpecies::length;
      using PolymerSpecies::nBead;
      using PolymerSpecies::type;
 
      using Species::phi;
      using Species::mu;
      using Species::q;
      using Species::ensemble;
 
   protected:

      void allocateBlocks() final;

      void readBlocks(std::istream& in) final;
 
   private:

      DArray<BT> blocks_;

      void isValid();
 
   };
 
   // Inline functions
 
   /*
   * Get a specified Edge (block descriptor) by non-const reference.
   */
   template <class BT, class PT>
   inline
   Edge& PolymerTmpl<BT,PT>::edge(int id)
   {  return blocks_[id]; }
 
   /*
   * Get a specified Edge (block descriptor) by const reference.
   */
   template <class BT, class PT>
   inline
   Edge const & PolymerTmpl<BT,PT>::edge(int id) const
   {  return blocks_[id]; }
 
   /*
   * Get a specified Block solver by non-const reference.
   */
   template <class BT, class PT>
   inline
   BT& PolymerTmpl<BT,PT>::block(int id)
   {  return blocks_[id]; }
 
   /*
   * Get a specified Block solver by const reference.
   */
   template <class BT, class PT>
   inline
   BT const & PolymerTmpl<BT,PT>::block(int id) const
   {  return blocks_[id]; }
 
}
#endif