// -*- c++ -*-

 // Use a DIMACS max flow file as stdin.

 // sub_graph_wrapper_demo < dimacs_max_flow_file

 // This program computes a maximum number of edge-disjoint shortest paths

 // between s and t.

 #include <iostream>

 #include <fstream>

 #include <hugo/smart_graph.h>

 #include <hugo/dijkstra.h>

 #include <hugo/maps.h>

 #include <hugo/graph_wrapper.h>

 #include <hugo/dimacs.h>

 #include <hugo/preflow.h>

 #include <tight_edge_filter_map.h>

 using namespace hugo;

 using std::cout;

 using std::endl;

 int main()

 {    

   typedef SmartGraph Graph;

   typedef Graph::Edge Edge;

   typedef Graph::Node Node;

   typedef Graph::EdgeIt EdgeIt;

   typedef Graph::NodeIt NodeIt;

   typedef Graph::EdgeMap<int> LengthMap;

   Graph g;

   Node s, t;

   LengthMap length(g);

   readDimacs(std::cin, g, length, s, t);

   cout << "edges with lengths (of form tail--length->head): " << endl;

   for(EdgeIt e(g); e!=INVALID; ++e) 

     cout << " " << g.id(g.tail(e)) << "--" 

 	 << length[e] << "->" << g.id(g.head(e)) << endl;

   cout << "s: " << g.id(s) << " t: " << g.id(t) << endl;

   typedef Dijkstra<Graph, LengthMap> Dijkstra;

   Dijkstra dijkstra(g, length);

   dijkstra.run(s);

   // This map returns true exactly for those edges which are 

   // tight w.r.t the length funcion and the potential 

   // given by the dijkstra algorithm.

   typedef TightEdgeFilterMap<Graph, const Dijkstra::DistMap, LengthMap> 

     TightEdgeFilter;

   TightEdgeFilter tight_edge_filter(g, dijkstra.distMap(), length);

   ConstMap<Node, bool> const_true_map(true);

   // This graph contains exaclty the tight edges.

   typedef SubGraphWrapper<Graph, ConstMap<Node, bool>, TightEdgeFilter> SubGW;

   SubGW gw(g, const_true_map, tight_edge_filter);

   ConstMap<Edge, int> const_1_map(1);

   Graph::EdgeMap<int> flow(g, 0);

   // Max flow between s and t in the graph of tight edges.

   Preflow<SubGW, int, ConstMap<Edge, int>, Graph::EdgeMap<int> > 

     preflow(gw, s, t, const_1_map, flow);

   preflow.run();

   cout << "edges of the maximum number of edge-disjoint shortest s-t paths: " 

        << endl;

   for(EdgeIt e(g); e!=INVALID; ++e) 

     if (flow[e])

       cout << " " << g.id(g.tail(e)) << "--" 

 	   << length[e] << "->" << g.id(g.head(e)) << endl;

   return 0;

 }