EdgeIterator.h

#ifndef PSCF_EDGE_ITERATOR_H
#define PSCF_EDGE_ITERATOR_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>   
 
namespace Pscf { 
 
   using namespace Util;

   template <typename WT>
   class EdgeIterator 
   {
 
   public:

      EdgeIterator(PolymerSpecies<WT> const & polymer);
   
      ~EdgeIterator();

      void begin(int sourceId, int targetId);

      EdgeIterator<WT>& operator ++ ();

      int currentEdgeId() const;

      int currentDirectionId() const;

      int currentVertexId() const;

      bool isEnd() const;

      bool notEnd() const;
 
   private:
 
      // Index of current edge.
      int currentEdgeId_;
 
      // Direction index for the current edge.
      int currentDirectionId_;
 
      // Index of current vertex.
      int currentVertexId_;
 
      // Index of target edge.
      int targetEdgeId_;
 
      // Index of target vertex.
      int targetVertexId_;
 
      // Pointer to associated PolymerSpecies object.
      PolymerSpecies<WT> const * polymerPtr_;
 
   };
 
   extern template class EdgeIterator<double>;
 
}
#endif