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 template
#include <util/containers/DArray.h>      // member
 
namespace Pscf {
 
   // Forward declaration
   class Edge;
 
   using namespace Util;

   template <class BT, class PT, typename WT = double>
   class PolymerTmpl : public PolymerSpecies<WT>
   {
 
   public:
 
      // Public member functions

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

      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);

 
      // Public typename aliases

      using BlockT = BT;

      using PropagatorT = PT;
 
   protected:

      PolymerTmpl();

      ~PolymerTmpl() override = default;

      void allocateBlocks() final;

      void readBlocks(std::istream& in) final;
 
      // Protected typename aliases (base classes)

      using SpeciesT = Species<WT>;

      using PolymerSpeciesT = PolymerSpecies<WT>;
 
   private:

      DArray<BT> blocks_;

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