VertexIterator.tpp

/*
* 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 "VertexIterator.h"
#include <pscf/chem/PolymerSpecies.h>
#include <pscf/chem/Edge.h>
 
namespace Pscf {
 
   /*
   * Constructor
   */
   template <typename WT>
   VertexIterator<WT>::VertexIterator(PolymerSpecies<WT> const & polymer)
    : currentId_(-1),
      targetId_(-1),
      polymerPtr_(&polymer)
   {}
 
   /*
   * Destructor
   */
   template <typename WT>
   VertexIterator<WT>::~VertexIterator()
   {}
 
   /*
   * Initialize iterator.
   */
   template <typename WT>
   void VertexIterator<WT>::begin(int sourceId, int targetId)
   {
      currentId_ = sourceId;
      targetId_  = targetId;
   }
 
   /*
   * Increment vertex - update current vertex to next one in path.
   */
   template <typename WT>
   VertexIterator<WT>& VertexIterator<WT>::operator ++ ()
   {
      UTIL_CHECK(notEnd());
      Pair<int> propId = polymerPtr_->path(currentId_, targetId_);
      int edgeId = propId[0];
      int dirId = propId[1];
      UTIL_CHECK(edgeId >= 0);
      UTIL_CHECK(edgeId < polymerPtr_->nBlock());
      UTIL_CHECK(dirId >= 0);
      UTIL_CHECK(dirId < 2);
      Edge const & edge = polymerPtr_->edge(edgeId);
      if (dirId == 0) {
         currentId_ = edge.vertexId(1);
      } else {
         currentId_ = edge.vertexId(0);
      }
      return *this;
   }
 
   /*
   * Get the current vertex id.
   */ 
   template <typename WT>
   int VertexIterator<WT>::currentId() const
   {  return currentId_; }
 
   /*
   * Is the current vertex equal to the target:
   */
   template <typename WT>
   bool VertexIterator<WT>::isEnd() const
   {
      return (bool)( currentId_ == targetId_ );
   }
 
   /*
   * Is the current vertex not equal to the target.
   */
   template <typename WT>
   bool VertexIterator<WT>::notEnd() const
   {
      return (bool)( currentId_ != targetId_ );
   }
 
}