/* -*- C++ -*-

  *

  * This file is a part of LEMON, a generic C++ optimization library

  *

  * Copyright (C) 2003-2008

  * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport

  * (Egervary Research Group on Combinatorial Optimization, EGRES).

  *

  * Permission to use, modify and distribute this software is granted

  * provided that this copyright notice appears in all copies. For

  * precise terms see the accompanying LICENSE file.

  *

  * This software is provided "AS IS" with no warranty of any kind,

  * express or implied, and with no claim as to its suitability for any

  * purpose.

  *

  */

 #ifndef LEMON_BITS_PRED_MAP_PATH_H

 #define LEMON_BITS_PRED_MAP_PATH_H

 namespace lemon {

   template <typename _Graph, typename _PredMap>

   class PredMapPath {

   public:

     typedef True RevPathTag;

     typedef _Graph Graph;

     typedef typename Graph::Edge Edge;

     typedef _PredMap PredMap;

     PredMapPath(const Graph& _graph, const PredMap& _predMap,

                 typename Graph::Node _target)

       : graph(_graph), predMap(_predMap), target(_target) {}

     int length() const {

       int len = 0;

       typename Graph::Node node = target;

       typename Graph::Edge edge;

       while ((edge = predMap[node]) != INVALID) {

         node = graph.source(edge);

         ++len;

       }

       return len;

     }

     bool empty() const {

       return predMap[target] != INVALID;

     }

     class RevEdgeIt {

     public:

       RevEdgeIt() {}

       RevEdgeIt(Invalid) : path(0), current(INVALID) {}

       RevEdgeIt(const PredMapPath& _path) 

         : path(&_path), current(_path.target) {

         if (path->predMap[current] == INVALID) current = INVALID;

       }

       operator const typename Graph::Edge() const {

         return path->predMap[current];

       }

       RevEdgeIt& operator++() {

         current = path->graph.source(path->predMap[current]);

         if (path->predMap[current] == INVALID) current = INVALID;

         return *this;

       }

       bool operator==(const RevEdgeIt& e) const { 

         return current == e.current; 

       }

       bool operator!=(const RevEdgeIt& e) const {

         return current != e.current; 

       }

       bool operator<(const RevEdgeIt& e) const { 

         return current < e.current; 

       }

     private:

       const PredMapPath* path;

       typename Graph::Node current;

     };

   private:

     const Graph& graph;

     const PredMap& predMap;

     typename Graph::Node target;

   };

   template <typename _Graph, typename _PredMatrixMap>

   class PredMatrixMapPath {

   public:

     typedef True RevPathTag;

     typedef _Graph Graph;

     typedef typename Graph::Edge Edge;

     typedef _PredMatrixMap PredMatrixMap;

     PredMatrixMapPath(const Graph& _graph, 

                       const PredMatrixMap& _predMatrixMap,

                       typename Graph::Node _source, 

                       typename Graph::Node _target)

       : graph(_graph), predMatrixMap(_predMatrixMap), 

         source(_source), target(_target) {}

     int length() const {

       int len = 0;

       typename Graph::Node node = target;

       typename Graph::Edge edge;

       while ((edge = predMatrixMap(source, node)) != INVALID) {

         node = graph.source(edge);

         ++len;

       }

       return len;

     }

     bool empty() const {

       return source != target;

     }

     class RevEdgeIt {

     public:

       RevEdgeIt() {}

       RevEdgeIt(Invalid) : path(0), current(INVALID) {}

       RevEdgeIt(const PredMatrixMapPath& _path) 

         : path(&_path), current(_path.target) {

         if (path->predMatrixMap(path->source, current) == INVALID) 

           current = INVALID;

       }

       operator const typename Graph::Edge() const {

         return path->predMatrixMap(path->source, current);

       }

       RevEdgeIt& operator++() {

         current = 

           path->graph.source(path->predMatrixMap(path->source, current));

         if (path->predMatrixMap(path->source, current) == INVALID) 

           current = INVALID;

         return *this;

       }

       bool operator==(const RevEdgeIt& e) const { 

         return current == e.current; 

       }

       bool operator!=(const RevEdgeIt& e) const {

         return current != e.current; 

       }

       bool operator<(const RevEdgeIt& e) const { 

         return current < e.current; 

       }

     private:

       const PredMatrixMapPath* path;

       typename Graph::Node current;

     };

   private:

     const Graph& graph;

     const PredMatrixMap& predMatrixMap;

     typename Graph::Node source;

     typename Graph::Node target;

   };

 }

 #endif