// -*- mode:C++ -*-

 #include<math.h>

 #include<hugo/list_graph.h>

 #include<hugo/smart_graph.h>

 #include<hugo/dijkstra.h>

 #include<hugo/preflow.h>

 #include"bench_tools.h"

 using namespace std;

 using namespace hugo;

 inline int numOfOnes(int n,int dim)

 {

   int s=0;

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

     s+=n%2;

     n>>=1;

   }

   return s;

 }

 inline int numOfZeros(int n,int dim)

 {

   int s=dim;

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

     s-=n&1;

     n>>=1;

   }

   return s;

 }

 int main(int argc, char *argv[])

 {

   //  typedef ListGraph Graph;

   typedef SmartGraph Graph;

   ///\bug GRAPH_TYPEDEF_FACTORY(Graph);

   GRAPH_TYPEDEF_FACTORY_NOTYPENAME(Graph);

   Graph G;

   Timer T;

   if(argc!=2) {

     cout << "Usage: " << argv[0] << " dim\n";

     return 1;

   }

   int dim=atoi(argv[1]);

 //   cout << "Creating Hipercube ("<< (1<<dim) << " nodes, "

 //        << dim*(1<<dim) << " edges):";

   T.reset();

   vector<Node> nodes;

   addBiDirHiperCube(G,dim,nodes);

   PrintTime("GENGRAPH",T);

   T.reset();

   Graph::EdgeMap<int> map(G);

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

     Primes P;

     for(int i=0;i<dim*(1<<dim);i++) P();

     //  for(EdgeIt e(G);G.valid(e);G.next(e)) map[e]=P();

     Edge te;

     for(int i=0;i<dim*(1<<dim);i++) {

       te.setToId(((long long int)(i)*93505)%(dim*(1<<dim)));

       //    map[Edge(((long long int)(i)*2987)%(dim*(1<<dim)))]=P();

       map[te]=P();

     }

 //     for(int i=0;i<(1<<dim);i++) {

 //       int mul= (1<<(numOfZeros(i,dim)/4));

 //       for(OutEdgeIt e(G,nodes[i]);G.valid(e);G.next(e))

 // 	map[e]*=mul;

 //     }

   }

   PrintTime("GENLENGTHS",T);

   T.reset();

   {

     Dijkstra<Graph> Dij(G,map);

     for(int i=0;i<10;i++)

       Dij.run(nodes[0]);

   }

   PrintTime("DIJKSTRA",T);

   T.reset();

   {

    Graph::EdgeMap<int> flow(G);

     Preflow<Graph,int> MF(G,nodes[0],nodes[1<<dim-1],map,flow);

     for(int i=0;i<10;i++)

       MF.run(MF.NO_FLOW);

   }

   PrintTime("PREFLOW",T);

 }