// -*- c++ -*-

#include <iostream>

#include <fstream>

#include <sage_graph.h>

#include <hugo/smart_graph.h>

#include <hugo/dimacs.h>

#include <hugo/time_measure.h>

//#include <graph_wrapper.h>

#include <hugo/max_flow.h>

#include <augmenting_flow.h>

//#include <preflow_res.h>

#include <for_each_macros.h>

#include <graph_concept.h>

using namespace hugo;

// Use a DIMACS max flow file as stdin.

// max_flow_demo < dimacs_max_flow_file

int main(int, char **) {

  typedef SageGraph MutableGraph;

  //typedef FullFeatureGraphConcept Graph;

  //typedef SmartGraph Graph;

  typedef SageGraph 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);

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

  Timer ts;

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

  MaxFlow<Graph, int, Graph::EdgeMap<int>, Graph::EdgeMap<int> > 

    max_flow_test(g, s, t, cap, flow);

  AugmentingFlow<Graph, int, Graph::EdgeMap<int>, Graph::EdgeMap<int> > 

    augmenting_flow_test(g, s, t, cap, flow);

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

  {

    std::cout << "preflow ..." << std::endl;

    ts.reset();

    max_flow_test.run();

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

    std::cout << "flow value: "<< max_flow_test.flowValue() << std::endl;

    max_flow_test.actMinCut(cut);

    FOR_EACH_LOC(Graph::EdgeIt, e, g) {

      if (cut[g.tail(e)] && !cut[g.head(e)] && !flow[e]==cap[e]) 

	std::cout << "Slackness does not hold!" << std::endl;

      if (!cut[g.tail(e)] && cut[g.head(e)] && flow[e]>0) 

	std::cout << "Slackness does not hold!" << std::endl;

    }

  }

  {

    std::cout << "preflow ..." << std::endl;

    FOR_EACH_LOC(Graph::EdgeIt, e, g) flow.set(e, 0);

    ts.reset();

    max_flow_test.preflow(MaxFlow<Graph, int, Graph::EdgeMap<int>, Graph::EdgeMap<int> >::GEN_FLOW);

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

    std::cout << "flow value: "<< max_flow_test.flowValue() << std::endl;

    FOR_EACH_LOC(Graph::EdgeIt, e, g) {

      if (cut[g.tail(e)] && !cut[g.head(e)] && !flow[e]==cap[e]) 

	std::cout << "Slackness does not hold!" << std::endl;

      if (!cut[g.tail(e)] && cut[g.head(e)] && flow[e]>0) 

	std::cout << "Slackness does not hold!" << std::endl;

    }

  }

//   {

//     std::cout << "wrapped preflow ..." << std::endl;

//     FOR_EACH_LOC(Graph::EdgeIt, e, g) flow.set(e, 0);

//     ts.reset();

//     pre_flow_res.run();

//     std::cout << "elapsed time: " << ts << std::endl;

//     std::cout << "flow value: "<< pre_flow_test.flowValue() << std::endl;

//   }

  {

    std::cout << "physical blocking flow augmentation ..." << std::endl;

    FOR_EACH_LOC(Graph::EdgeIt, e, g) flow.set(e, 0);

    ts.reset();

    int i=0;

    while (augmenting_flow_test.augmentOnBlockingFlow<MutableGraph>()) { ++i; }

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

    std::cout << "number of augmentation phases: " << i << std::endl; 

    std::cout << "flow value: "<< augmenting_flow_test.flowValue() << std::endl;

    FOR_EACH_LOC(Graph::EdgeIt, e, g) {

      if (cut[g.tail(e)] && !cut[g.head(e)] && !flow[e]==cap[e]) 

	std::cout << "Slackness does not hold!" << std::endl;

      if (!cut[g.tail(e)] && cut[g.head(e)] && flow[e]>0) 

	std::cout << "Slackness does not hold!" << std::endl;

    }

  }

//   {

//     std::cout << "on-the-fly blocking flow augmentation ..." << std::endl;

//     FOR_EACH_LOC(Graph::EdgeIt, e, g) flow.set(e, 0);

//     ts.reset();

//     int i=0;

//     while (augmenting_flow_test.augmentOnBlockingFlow2()) { ++i; }

//     std::cout << "elapsed time: " << ts << std::endl;

//     std::cout << "number of augmentation phases: " << i << std::endl; 

//     std::cout << "flow value: "<< augmenting_flow_test.flowValue() << std::endl;

//     FOR_EACH_LOC(Graph::EdgeIt, e, g) {

//       if (cut[g.tail(e)] && !cut[g.head(e)] && !flow[e]==cap[e]) 

// 	std::cout << "Slackness does not hold!" << std::endl;

//       if (!cut[g.tail(e)] && cut[g.head(e)] && flow[e]>0) 

// 	std::cout << "Slackness does not hold!" << std::endl;

//     }

//   }

  {

    std::cout << "on-the-fly shortest path augmentation ..." << std::endl;

    FOR_EACH_LOC(Graph::EdgeIt, e, g) flow.set(e, 0);

    ts.reset();

    int i=0;

    while (augmenting_flow_test.augmentOnShortestPath()) { ++i; }

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

    std::cout << "number of augmentation phases: " << i << std::endl; 

    std::cout << "flow value: "<< augmenting_flow_test.flowValue() << std::endl;

    FOR_EACH_LOC(Graph::EdgeIt, e, g) {

      if (cut[g.tail(e)] && !cut[g.head(e)] && !flow[e]==cap[e]) 

	std::cout << "Slackness does not hold!" << std::endl;

      if (!cut[g.tail(e)] && cut[g.head(e)] && flow[e]>0) 

	std::cout << "Slackness does not hold!" << std::endl;

    }

  }

  {

    std::cout << "on-the-fly shortest path augmentation ..." << std::endl;

    FOR_EACH_LOC(Graph::EdgeIt, e, g) flow.set(e, 0);

    ts.reset();

    int i=0;

    while (augmenting_flow_test.augmentOnShortestPath2()) { ++i; }

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

    std::cout << "number of augmentation phases: " << i << std::endl; 

    std::cout << "flow value: "<< augmenting_flow_test.flowValue() << std::endl;

    FOR_EACH_LOC(Graph::EdgeIt, e, g) {

      if (cut[g.tail(e)] && !cut[g.head(e)] && !flow[e]==cap[e]) 

	std::cout << "Slackness does not hold!" << std::endl;

      if (!cut[g.tail(e)] && cut[g.head(e)] && flow[e]>0) 

	std::cout << "Slackness does not hold!" << std::endl;

    }

  }

  return 0;

}