#include "test_tools.h"

 #include <hugo/smart_graph.h>

 #include <hugo/dijkstra.h>

 #include<hugo/skeletons/graph.h>

 #include<hugo/skeletons/maps.h>

 using namespace hugo;

 const int PET_SIZE =5;

 void check_Dijkstra_BinHeap_Compile() 

 {

   typedef int VType;

   typedef skeleton::StaticGraphSkeleton Graph;

   typedef Graph::Edge Edge;

   typedef Graph::Node Node;

   typedef Graph::EdgeIt EdgeIt;

   typedef Graph::NodeIt NodeIt;

   typedef skeleton::ReadMap<Edge,VType> LengthMap;

   typedef Dijkstra<Graph, LengthMap> DType;

   Graph G;

   Node n;

   Edge e;

   VType l;

   bool b;

   DType::DistMap d(G);

   DType::PredMap p(G);

   DType::PredNodeMap pn(G);

   LengthMap cap;

   DType dijkstra_test(G,cap);

   dijkstra_test.run(n);

   l  = dijkstra_test.dist(n);

   e  = dijkstra_test.pred(n);

   n  = dijkstra_test.predNode(n);

   d  = dijkstra_test.distMap();

   p  = dijkstra_test.predMap();

   pn = dijkstra_test.predNodeMap();

   b  = dijkstra_test.reached(n);

 }

 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 cap(G);

   PetStruct<Graph> ps = addPetersen(G,PET_SIZE);

   for(int i=0;i<PET_SIZE;i++) {

     cap[ps.outcir[i]]=4;

     cap[ps.incir[i]]=1;

     cap[ps.chords[i]]=10;

   }

   s=ps.outer[0];

   t=ps.inner[1];

   Dijkstra<Graph, LengthMap> 

 	dijkstra_test(G, cap);

   dijkstra_test.run(s);

   check(dijkstra_test.dist(t)==13,"Dijkstra found a wrong path.");

   for(EdgeIt e(G); e!=INVALID; ++e) {

     Node u=G.tail(e);

     Node v=G.head(e);

     check( !dijkstra_test.reached(u) ||

 	   (dijkstra_test.dist(v) - dijkstra_test.dist(u) <= cap[e]),

 	   "dist(head)-dist(tail)- edge_length= " 

 	   << dijkstra_test.dist(v) - dijkstra_test.dist(u) 

 	   - cap[e]);

   }

   ///\bug This works only for integer lengths

   for(NodeIt v(G); v!=INVALID; ++v){

     check(dijkstra_test.reached(v),"Each node should be reached.");

     if ( dijkstra_test.pred(v)!=INVALID ) {

       Edge e=dijkstra_test.pred(v);

       Node u=G.tail(e);

       check(u==dijkstra_test.predNode(v),"Wrong tree.");

       check(dijkstra_test.dist(v) - dijkstra_test.dist(u) == cap[e],

 	    "Wrong distance! Difference: " 

 	    << std::abs(dijkstra_test.dist(v) - dijkstra_test.dist(u) 

 			    - cap[e]));

     }

   }

 }