athos@610: // -*- c++ -*-
athos@610: #ifndef HUGO_MINLENGTHPATHS_H
athos@610: #define HUGO_MINLENGTHPATHS_H
athos@610: 
alpar@759: ///\ingroup flowalgs
athos@610: ///\file
athos@610: ///\brief An algorithm for finding k paths of minimal total length.
athos@610: 
athos@611: 
athos@610: //#include <hugo/dijkstra.h>
athos@610: //#include <hugo/graph_wrapper.h>
athos@610: #include <hugo/maps.h>
athos@610: #include <vector>
athos@610: #include <hugo/mincostflows.h>
athos@610: #include <for_each_macros.h>
athos@610: 
athos@610: namespace hugo {
athos@610: 
alpar@759: /// \addtogroup flowalgs
athos@610: /// @{
athos@610: 
athos@610:   ///\brief Implementation of an algorithm for finding k paths between 2 nodes 
athos@610:   /// of minimal total length 
athos@610:   ///
athos@610:   /// The class \ref hugo::MinLengthPaths "MinLengthPaths" implements
athos@610:   /// an algorithm for finding k edge-disjoint paths
athos@610:   /// from a given source node to a given target node in an
athos@610:   /// edge-weighted directed graph having minimal total weigth (length).
athos@610:   ///
athos@610:   ///\warning It is assumed that the lengths are positive, since the
athos@610:   /// general flow-decomposition is not implemented yet.
athos@610:   ///
athos@610:   ///\author Attila Bernath
athos@610:   template <typename Graph, typename LengthMap>
athos@610:   class MinLengthPaths{
athos@610: 
athos@610: 
athos@610:     typedef typename LengthMap::ValueType Length;
athos@610:     
athos@610:     typedef typename Graph::Node Node;
athos@610:     typedef typename Graph::NodeIt NodeIt;
athos@610:     typedef typename Graph::Edge Edge;
athos@610:     typedef typename Graph::OutEdgeIt OutEdgeIt;
athos@610:     typedef typename Graph::template EdgeMap<int> EdgeIntMap;
athos@610: 
athos@610:     typedef ConstMap<Edge,int> ConstMap;
athos@610: 
athos@610:     //Input
athos@610:     const Graph& G;
athos@610: 
athos@610:     //Auxiliary variables
athos@610:     //This is the capacity map for the mincostflow problem
athos@610:     ConstMap const1map;
athos@610:     //This MinCostFlows instance will actually solve the problem
athos@610:     MinCostFlows<Graph, LengthMap, ConstMap> mincost_flow;
athos@610: 
athos@610:     //Container to store found paths
athos@610:     std::vector< std::vector<Edge> > paths;
athos@610: 
athos@610:   public :
athos@610: 
athos@610: 
athos@610:     MinLengthPaths(Graph& _G, LengthMap& _length) : G(_G),
athos@610:       const1map(1), mincost_flow(_G, _length, const1map){}
athos@610: 
athos@610:     ///Runs the algorithm.
athos@610: 
athos@610:     ///Runs the algorithm.
athos@610:     ///Returns k if there are at least k edge-disjoint paths from s to t.
athos@610:    ///Otherwise it returns the number of found edge-disjoint paths from s to t.
athos@610:     int run(Node s, Node t, int k) {
athos@610: 
athos@610:       int i = mincost_flow.run(s,t,k);
athos@610:       
athos@610: 
athos@610: 
athos@610:       //Let's find the paths
athos@610:       //We put the paths into stl vectors (as an inner representation). 
athos@610:       //In the meantime we lose the information stored in 'reversed'.
athos@610:       //We suppose the lengths to be positive now.
athos@610: 
athos@610:       //We don't want to change the flow of mincost_flow, so we make a copy
athos@610:       //The name here suggests that the flow has only 0/1 values.
athos@610:       EdgeIntMap reversed(G); 
athos@610: 
hegyi@776:       //FOR_EACH_LOC(typename Graph::EdgeIt, e, G){
hegyi@776:       for(typename Graph::EdgeIt e=loopFirst(typename Graph::EdgeIt(), (G)); e!=INVALID; ++e){
athos@610: 	reversed[e] = mincost_flow.getFlow()[e];
athos@610:       }
athos@610:       
athos@610:       paths.clear();
athos@610:       //total_length=0;
athos@610:       paths.resize(k);
athos@610:       for (int j=0; j<i; ++j){
athos@610: 	Node n=s;
athos@610: 	OutEdgeIt e;
athos@610: 
athos@610: 	while (n!=t){
athos@610: 
athos@610: 
athos@610: 	  G.first(e,n);
athos@610: 	  
athos@610: 	  while (!reversed[e]){
hegyi@776: 	    ++e;
athos@610: 	  }
athos@610: 	  n = G.head(e);
athos@610: 	  paths[j].push_back(e);
athos@610: 	  //total_length += length[e];
athos@610: 	  reversed[e] = 1-reversed[e];
athos@610: 	}
athos@610: 	
athos@610:       }
athos@610:       return i;
athos@610:     }
athos@610: 
athos@610:     
athos@610:     ///This function gives back the total length of the found paths.
athos@610:     ///Assumes that \c run() has been run and nothing changed since then.
athos@610:     Length totalLength(){
athos@610:       return mincost_flow.totalLength();
athos@610:     }
athos@610: 
athos@610:     ///Returns a const reference to the EdgeMap \c flow. \pre \ref run() must
athos@610:     ///be called before using this function.
athos@610:     const EdgeIntMap &getFlow() const { return mincost_flow.flow;}
athos@610: 
athos@610:   ///Returns a const reference to the NodeMap \c potential (the dual solution).
athos@610:     /// \pre \ref run() must be called before using this function.
athos@610:     const EdgeIntMap &getPotential() const { return mincost_flow.potential;}
athos@610: 
athos@610:     ///This function checks, whether the given solution is optimal
athos@610:     ///Running after a \c run() should return with true
athos@610:     ///In this "state of the art" this only checks optimality, doesn't bother with feasibility
athos@610:     ///
athos@610:     ///\todo Is this OK here?
athos@610:     bool checkComplementarySlackness(){
athos@610:       return mincost_flow.checkComplementarySlackness();
athos@610:     }
athos@610: 
athos@610:     ///This function gives back the \c j-th path in argument p.
athos@610:     ///Assumes that \c run() has been run and nothing changed since then.
athos@610:     /// \warning It is assumed that \c p is constructed to be a path of graph \c G. If \c j is not less than the result of previous \c run, then the result here will be an empty path (\c j can be 0 as well).
athos@610:     template<typename DirPath>
athos@610:     void getPath(DirPath& p, size_t j){
athos@610:       
athos@610:       p.clear();
athos@610:       if (j>paths.size()-1){
athos@610: 	return;
athos@610:       }
athos@610:       typename DirPath::Builder B(p);
athos@610:       for(typename std::vector<Edge>::iterator i=paths[j].begin(); 
athos@610: 	  i!=paths[j].end(); ++i ){
athos@610: 	B.pushBack(*i);
athos@610:       }
athos@610: 
athos@610:       B.commit();
athos@610:     }
athos@610: 
athos@610:   }; //class MinLengthPaths
athos@610: 
athos@610:   ///@}
athos@610: 
athos@610: } //namespace hugo
athos@610: 
athos@610: #endif //HUGO_MINLENGTHPATHS_H