marci@301: // -*- c++ -*-
marci@301: #ifndef HUGO_BFS_ITERATOR_H
marci@301: #define HUGO_BFS_ITERATOR_H
marci@301: 
marci@301: #include <queue>
marci@301: #include <stack>
marci@301: #include <utility>
marci@301: 
marci@301: namespace hugo {
marci@301: 
marci@303:   template <typename Graph, /*typename OutEdgeIt,*/ 
marci@303: 	    typename ReachedMap/*=typename Graph::NodeMap<bool>*/ >
marci@360:   class BfsIterator {
marci@303:   protected:
marci@303:     typedef typename Graph::Node Node;
marci@303:     typedef typename Graph::OutEdgeIt OutEdgeIt;
marci@303:     const Graph* graph;
marci@301:     std::queue<Node> bfs_queue;
marci@301:     ReachedMap& reached;
marci@301:     bool b_node_newly_reached;
marci@301:     OutEdgeIt actual_edge;
marci@301:     bool own_reached_map;
marci@301:   public:
marci@360:     BfsIterator(const Graph& _graph, ReachedMap& _reached) : 
marci@303:       graph(&_graph), reached(_reached), 
marci@301:       own_reached_map(false) { }
marci@360:     BfsIterator(const Graph& _graph) : 
marci@303:       graph(&_graph), reached(*(new ReachedMap(*graph /*, false*/))), 
marci@301:       own_reached_map(true) { }
marci@360:     ~BfsIterator() { if (own_reached_map) delete &reached; }
marci@301:     void pushAndSetReached(Node s) { 
marci@301:       reached.set(s, true);
marci@301:       if (bfs_queue.empty()) {
marci@301: 	bfs_queue.push(s);
marci@303: 	graph->first(actual_edge, s);
marci@303: 	if (graph->valid(actual_edge)) { 
marci@303: 	  Node w=graph->bNode(actual_edge);
marci@303: 	  if (!reached[w]) {
marci@301: 	    bfs_queue.push(w);
marci@301: 	    reached.set(w, true);
marci@301: 	    b_node_newly_reached=true;
marci@301: 	  } else {
marci@301: 	    b_node_newly_reached=false;
marci@301: 	  }
marci@301: 	} 
marci@301:       } else {
marci@301: 	bfs_queue.push(s);
marci@301:       }
marci@301:     }
marci@360:     BfsIterator<Graph, /*OutEdgeIt,*/ ReachedMap>& 
marci@301:     operator++() { 
marci@303:       if (graph->valid(actual_edge)) { 
marci@303: 	graph->next(actual_edge);
marci@303: 	if (graph->valid(actual_edge)) {
marci@303: 	  Node w=graph->bNode(actual_edge);
marci@303: 	  if (!reached[w]) {
marci@301: 	    bfs_queue.push(w);
marci@301: 	    reached.set(w, true);
marci@301: 	    b_node_newly_reached=true;
marci@301: 	  } else {
marci@301: 	    b_node_newly_reached=false;
marci@301: 	  }
marci@301: 	}
marci@301:       } else {
marci@301: 	bfs_queue.pop(); 
marci@301: 	if (!bfs_queue.empty()) {
marci@303: 	  graph->first(actual_edge, bfs_queue.front());
marci@303: 	  if (graph->valid(actual_edge)) {
marci@303: 	    Node w=graph->bNode(actual_edge);
marci@303: 	    if (!reached[w]) {
marci@301: 	      bfs_queue.push(w);
marci@301: 	      reached.set(w, true);
marci@301: 	      b_node_newly_reached=true;
marci@301: 	    } else {
marci@301: 	      b_node_newly_reached=false;
marci@301: 	    }
marci@301: 	  }
marci@301: 	}
marci@301:       }
marci@301:       return *this;
marci@301:     }
marci@301:     bool finished() const { return bfs_queue.empty(); }
marci@301:     operator OutEdgeIt () const { return actual_edge; }
marci@301:     bool isBNodeNewlyReached() const { return b_node_newly_reached; }
marci@303:     bool isANodeExamined() const { return !(graph->valid(actual_edge)); }
marci@301:     Node aNode() const { return bfs_queue.front(); }
marci@303:     Node bNode() const { return graph->bNode(actual_edge); }
marci@301:     const ReachedMap& getReachedMap() const { return reached; }
marci@301:     const std::queue<Node>& getBfsQueue() const { return bfs_queue; }
marci@301:   };  
marci@301: 
marci@303:   template <typename Graph, /*typename OutEdgeIt,*/ 
marci@303: 	    typename ReachedMap/*=typename Graph::NodeMap<bool>*/ >
marci@360:   class DfsIterator {
marci@303:   protected:
marci@303:     typedef typename Graph::Node Node;
marci@303:     typedef typename Graph::OutEdgeIt OutEdgeIt;
marci@303:     const Graph* graph;
marci@301:     std::stack<OutEdgeIt> dfs_stack;
marci@301:     bool b_node_newly_reached;
marci@301:     OutEdgeIt actual_edge;
marci@301:     Node actual_node;
marci@301:     ReachedMap& reached;
marci@301:     bool own_reached_map;
marci@301:   public:
marci@360:     DfsIterator(const Graph& _graph, ReachedMap& _reached) : 
marci@303:       graph(&_graph), reached(_reached), 
marci@301:       own_reached_map(false) { }
marci@360:     DfsIterator(const Graph& _graph) : 
marci@303:       graph(&_graph), reached(*(new ReachedMap(*graph /*, false*/))), 
marci@301:       own_reached_map(true) { }
marci@360:     ~DfsIterator() { if (own_reached_map) delete &reached; }
marci@301:     void pushAndSetReached(Node s) { 
marci@301:       actual_node=s;
marci@301:       reached.set(s, true);
marci@301:       OutEdgeIt e;
marci@303:       graph->first(e, s);
marci@301:       dfs_stack.push(e); 
marci@301:     }
marci@360:     DfsIterator<Graph, /*OutEdgeIt,*/ ReachedMap>& 
marci@301:     operator++() { 
marci@301:       actual_edge=dfs_stack.top();
marci@301:       //actual_node=G.aNode(actual_edge);
marci@303:       if (graph->valid(actual_edge)/*.valid()*/) { 
marci@303: 	Node w=graph->bNode(actual_edge);
marci@301: 	actual_node=w;
marci@303: 	if (!reached[w]) {
marci@301: 	  OutEdgeIt e;
marci@303: 	  graph->first(e, w);
marci@301: 	  dfs_stack.push(e);
marci@301: 	  reached.set(w, true);
marci@301: 	  b_node_newly_reached=true;
marci@301: 	} else {
marci@303: 	  actual_node=graph->aNode(actual_edge);
marci@303: 	  graph->next(dfs_stack.top());
marci@301: 	  b_node_newly_reached=false;
marci@301: 	}
marci@301:       } else {
marci@301: 	//actual_node=G.aNode(dfs_stack.top());
marci@301: 	dfs_stack.pop();
marci@301:       }
marci@301:       return *this;
marci@301:     }
marci@301:     bool finished() const { return dfs_stack.empty(); }
marci@301:     operator OutEdgeIt () const { return actual_edge; }
marci@301:     bool isBNodeNewlyReached() const { return b_node_newly_reached; }
marci@303:     bool isANodeExamined() const { return !(graph->valid(actual_edge)); }
marci@301:     Node aNode() const { return actual_node; /*FIXME*/}
marci@301:     Node bNode() const { return G.bNode(actual_edge); }
marci@301:     const ReachedMap& getReachedMap() const { return reached; }
marci@301:     const std::stack<OutEdgeIt>& getDfsStack() const { return dfs_stack; }
marci@301:   };
marci@301: 
marci@301: } // namespace hugo
marci@301: 
marci@301: #endif //HUGO_BFS_ITERATOR_H