#include <iostream>

#include <smart_graph.h>

#include <dimacs.h>

#include <preflow.h>

#include <time_measure.h>

using namespace hugo;

int main(int, char **) {

  typedef SmartGraph Graph;

  typedef Graph::Node Node;

  typedef Graph::EdgeIt EdgeIt;

  Graph G;

  Node s, t;

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

  readDimacsMaxFlow(std::cin, G, s, t, cap);

  Timer ts;

  bool error=false;

  std::cout <<

    "\n  Testing preflow.h on a graph with " << 

    G.nodeNum() << " nodes and " << G.edgeNum() << " edges..."

	   << std::endl;

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

  Preflow<Graph, int> preflow_test(G, s, t, cap, flow);

  std::cout << "\nCalling run() (flow must be constant zero)..."<<std::endl;

  ts.reset();

  preflow_test.run();

  std::cout << "Elapsed time: " << ts << std::endl;

  Graph::NodeMap<bool> mincut(G);

  preflow_test.minMinCut(mincut); 

  int min_min_cut_value=0;

  Graph::NodeMap<bool> cut(G);

  preflow_test.minCut(cut); 

  int min_cut_value=0;

  Graph::NodeMap<bool> maxcut(G);

  preflow_test.maxMinCut(maxcut); 

  int max_min_cut_value=0;

  EdgeIt e;

  for(G.first(e); G.valid(e); G.next(e)) {

    int c=cap[e];

    if (mincut[G.tail(e)] && !mincut[G.head(e)]) min_min_cut_value+=c;

    if (cut[G.tail(e)] && !cut[G.head(e)]) min_cut_value+=c; 

    if (maxcut[G.tail(e)] && !maxcut[G.head(e)]) max_min_cut_value+=c;

  }

  std::cout << "\nChecking the result: " <<std::endl;  

  std::cout << "Flow value: "<< preflow_test.flowValue() << std::endl;

  std::cout << "Min cut value: "<< min_cut_value << std::endl;

  std::cout << "Min min cut value: "<< min_min_cut_value << std::endl;

  std::cout << "Max min cut value: "<< max_min_cut_value << 

    std::endl;

  if ( preflow_test.flowValue() == min_cut_value &&

       min_cut_value == min_min_cut_value &&

       min_min_cut_value == max_min_cut_value )

    std::cout << "They are equal. " <<std::endl;  

  else {

    std::cout << "ERROR! They are not equal! " <<std::endl;  

    error=true;

  }

  Preflow<Graph, int> preflow_test2(G, s, t, cap, flow);

  std::cout << "\n\nCalling preflow(GEN_FLOW) with the given maximum flow..."<<std::endl;

  ts.reset();

  preflow_test2.preflow(preflow_test2.GEN_FLOW);

  std::cout << "Elapsed time: " << ts << std::endl;

  Graph::NodeMap<bool> mincut2(G);

  preflow_test.minMinCut(mincut2); 

  int min_min_cut2_value=0;

  Graph::NodeMap<bool> cut2(G);

  preflow_test.minCut(cut2); 

  int min_cut2_value=0;

  Graph::NodeMap<bool> maxcut2(G);

  preflow_test.maxMinCut(maxcut2); 

  int max_min_cut2_value=0;

  for(G.first(e); G.valid(e); G.next(e)) {

    int c=cap[e];

    if (mincut2[G.tail(e)] && !mincut2[G.head(e)]) min_min_cut2_value+=c;

    if (cut2[G.tail(e)] && !cut2[G.head(e)]) min_cut2_value+=c; 

    if (maxcut2[G.tail(e)] && !maxcut2[G.head(e)]) max_min_cut2_value+=c;

  }

  std::cout << "\nThe given flow value is "

	    << preflow_test2.flowValue();

  if ( preflow_test2.flowValue() == min_cut2_value &&

       min_cut2_value == min_min_cut2_value &&

       min_min_cut2_value == max_min_cut2_value )

    std::cout <<", which is equal to all three min cut values." 

	      <<std::endl;  

  else {

    std::cout << "ERROR! It is not equal to all three min cut values! " 

	      <<std::endl;  

    error=true;

  }

  Graph::EdgeMap<int> flow3(G,0);

  Preflow<Graph, int> preflow_test3(G, s, t, cap, flow3);

  std::cout << "\n\nCalling preflowPhase0(PREFLOW) on the constant zero flow..."<<std::endl;

  ts.reset();

  preflow_test3.preflowPhase0(preflow_test3.PREFLOW);

  std::cout << "Elapsed time: " << ts << std::endl;

  Graph::NodeMap<bool> actcut3(G);

  std::cout << "\nCalling actMinCut()..."<<std::endl;

  preflow_test3.actMinCut(actcut3); 

  std::cout << "Calling preflowPhase1() on the given flow..."<<std::endl;

  ts.reset();

  preflow_test3.preflowPhase1();

  std::cout << "Elapsed time: " << ts << std::endl;

  int act_min_cut3_value=0;

  Graph::NodeMap<bool> mincut3(G);

  preflow_test.minMinCut(mincut3); 

  int min_min_cut3_value=0;

  Graph::NodeMap<bool> cut3(G);

  preflow_test.minCut(cut3); 

  int min_cut3_value=0;

  Graph::NodeMap<bool> maxcut3(G);

  preflow_test.maxMinCut(maxcut3); 

  int max_min_cut3_value=0;

  for(G.first(e); G.valid(e); G.next(e)) {

    int c=cap[e];

    if (mincut3[G.tail(e)] && !mincut3[G.head(e)]) min_min_cut3_value+=c;

    if (cut3[G.tail(e)] && !cut3[G.head(e)]) min_cut3_value+=c; 

    if (maxcut3[G.tail(e)] && !maxcut3[G.head(e)]) max_min_cut3_value+=c;

    if (actcut3[G.tail(e)] && !actcut3[G.head(e)]) act_min_cut3_value+=c;

  }

 std::cout << "\nThe min cut value given by actMinCut() after phase 0 is "<<

   act_min_cut3_value;

  if ( preflow_test3.flowValue() == min_cut3_value &&

       min_cut3_value == min_min_cut3_value &&

       min_min_cut3_value == max_min_cut3_value &&

       max_min_cut3_value == act_min_cut3_value ) {

    std::cout << 

      ", which is equal to the given flow value and to all three min cut values after phase 1." 

	      <<std::endl;  

  }

  else {

    std::cout << 

      "ERROR! It is not equal to the given flow value and to all three min cut values after phase 1! " 

	      <<std::endl;  

    error=true;

  }

  Graph::EdgeMap<int> flow4(G,0);

  Preflow<Graph, int> preflow_test4(G, s, t, cap, flow4);

  std::cout << 

    "\n\nCalling preflow(PREFLOW) with the constant 0 flow, the result is f..."

	    <<std::endl;

  preflow_test4.preflow(preflow_test4.PREFLOW);

  std::cout << "Swapping the source and the target, "<<std::endl;

  std::cout << "by calling resetSource(t) and resetTarget(s)..."

	    <<std::endl;

  preflow_test4.resetSource(t);

  preflow_test4.resetTarget(s);

  std::cout << 

    "Calling preflow(PREFLOW) to find a maximum t-s flow starting with flow f..."

	    <<std::endl;

  preflow_test4.preflow(preflow_test4.PREFLOW);

  Graph::NodeMap<bool> mincut4(G);

  preflow_test4.minMinCut(mincut4); 

  int min_min_cut4_value=0;

  Graph::NodeMap<bool> cut4(G);

  preflow_test4.minCut(cut4); 

  int min_cut4_value=0;

  Graph::NodeMap<bool> maxcut4(G);

  preflow_test4.maxMinCut(maxcut4); 

  int max_min_cut4_value=0;

  for(G.first(e); G.valid(e); G.next(e)) {

    int c=cap[e];

    if (mincut4[G.tail(e)] && !mincut4[G.head(e)]) min_min_cut4_value+=c;

    if (cut4[G.tail(e)] && !cut4[G.head(e)]) min_cut4_value+=c; 

    if (maxcut4[G.tail(e)] && !maxcut4[G.head(e)]) max_min_cut4_value+=c;

  }

  std::cout << "\nThe given flow value is "

	    << preflow_test4.flowValue();

  if ( preflow_test4.flowValue() == min_cut4_value &&

       min_cut4_value == min_min_cut4_value &&

       min_min_cut4_value == max_min_cut4_value )

    std::cout <<", which is equal to all three min cut values." 

	      <<std::endl;  

  else {

    std::cout << "ERROR! It is not equal to all three min cut values! " 

	      <<std::endl;  

    error=true;

  }

  Graph::EdgeMap<int> flow5(G,0);

  std::cout << "Resetting the stored flow to constant zero, by calling resetFlow..."

	    <<std::endl;

  preflow_test4.resetFlow(flow5);

  std::cout << 

    "Calling preflow(GEN_FLOW) to find a maximum t-s flow "<<std::endl;

  std::cout << 

    "starting with this constant zero flow..." <<std::endl;

  preflow_test4.preflow(preflow_test4.GEN_FLOW);

  Graph::NodeMap<bool> mincut5(G);

  preflow_test4.minMinCut(mincut5); 

  int min_min_cut5_value=0;

  Graph::NodeMap<bool> cut5(G);

  preflow_test4.minCut(cut5); 

  int min_cut5_value=0;

  Graph::NodeMap<bool> maxcut5(G);

  preflow_test4.maxMinCut(maxcut5); 

  int max_min_cut5_value=0;

  for(G.first(e); G.valid(e); G.next(e)) {

    int c=cap[e];

    if (mincut5[G.tail(e)] && !mincut5[G.head(e)]) min_min_cut5_value+=c;

    if (cut5[G.tail(e)] && !cut5[G.head(e)]) min_cut5_value+=c; 

    if (maxcut5[G.tail(e)] && !maxcut5[G.head(e)]) max_min_cut5_value+=c;

  }

  std::cout << "\nThe given flow value is "

	    << preflow_test4.flowValue();

  if ( preflow_test4.flowValue() == min_cut5_value &&

       min_cut5_value == min_min_cut5_value &&

       min_min_cut5_value == max_min_cut5_value )

    std::cout <<", which is equal to all three min cut values." 

	      <<std::endl<<std::endl;  

  else {

    std::cout << "ERROR! It is not equal to all three min cut values! " 

	      <<std::endl;  

    error=true;

  }

  if (error) std::cout <<"\nThere was some error in the results!\n"<<std::endl; 

  else std::cout <<"\nThere was no error in the results.\n"<<std::endl; 

  return 0;

}