#include <iostream>
#include <vector>
#include <string>

#include <marci_list_graph.hh>
#include <marci_property_vector.hh>
#include <marci_bfs.hh>
#include <marci_max_flow.hh>

using namespace marci;

int main (int, char*[])
{
  typedef list_graph::node_iterator node_iterator;
  typedef list_graph::edge_iterator edge_iterator;
  typedef list_graph::each_node_iterator each_node_iterator;
  typedef list_graph::each_edge_iterator each_edge_iterator;
  typedef list_graph::out_edge_iterator out_edge_iterator;
  typedef list_graph::in_edge_iterator in_edge_iterator;
  typedef list_graph::sym_edge_iterator sym_edge_iterator;
  list_graph G;
  std::vector<node_iterator> vector_of_node_iterators;
  for(int i=0; i!=8; ++i) vector_of_node_iterators.push_back(G.add_node());
  for(int i=0; i!=8; ++i)
    for(int j=0; j!=8; ++j) {
      if ((i<j)&&(i+j)%3) G.add_edge(vector_of_node_iterators[i], vector_of_node_iterators[j]);
    }  

  std::cout << "We construct a directed graph on the node set {0,1,2,...,7}," <<std::endl << "i-->j is arc iff i<j and (i+j)%3." << std::endl;
  std::cout << "number of nodes: " << number_of(G.first_node()) << std::endl;

  for(each_node_iterator i=G.first_node(); i.valid(); ++i) {
    std::cout << "node " << G.id(i) << std::endl;
    std::cout << " outdegree (out_edge_iterator): " << number_of(G.first_out_edge(i)) << " "; 
    for(out_edge_iterator j=G.first_out_edge(i); j.valid(); ++j) { 
      std::cout << "(" << G.id(G.tail(j)) << "--" << G.id(j) << "->" << G.id(G.head(j)) << ") ";
    }
    std::cout << std::endl; 

    std::cout<< " ";
    for(out_edge_iterator j=G.first_out_edge(i); j.valid(); ++j) { 
      std::cout << G.a_node(j) << "->" << G.b_node(j) << " "; } 
    std::cout<<std::endl;

    std::cout << " indegree: (in_edge_oterator) " << number_of(G.first_in_edge(i)) << " ";
    for(in_edge_iterator j=G.first_in_edge(i); j.valid(); ++j) { 
      std::cout << j << " "; } 
    std::cout << std::endl;

    std::cout<< " ";
    for(in_edge_iterator j=G.first_in_edge(i); j.valid(); ++j) { 
      std::cout << G.a_node(j) << "->" << G.b_node(j) << " "; } 
    std::cout<<std::endl;

    std::cout << " degree: (sym_edge_iterator) " << number_of(G.first_sym_edge(i)) << " ";
    for(sym_edge_iterator j=G.first_sym_edge(i); j.valid(); ++j) { 
      std::cout << j << " "; } 
    std::cout<<std::endl;

    std::cout<< " ";
    for(sym_edge_iterator j=G.first_sym_edge(i); j.valid(); ++j) { 
      std::cout << G.a_node(j) << "->" << G.b_node(j) << " "; } 
    std::cout<<std::endl;
  }

  std::cout << "all edges: ";
  for(each_edge_iterator i=G.first_edge(); i.valid(); ++i) {
    std::cout << i << " ";
  }
  std::cout << std::endl;

  std::cout << "node property array test" << std::endl;
  node_property_vector<list_graph, int> my_property_vector(G);
  each_node_iterator v;
  G.get_first(v);
  my_property_vector.put(v, 42);
  my_property_vector.put(++G.first_node(), 314);
  my_property_vector.put(++++G.first_node(), 1956);
  my_property_vector.put(vector_of_node_iterators[3], 1989);
  my_property_vector.put(vector_of_node_iterators[4], 2003);
  my_property_vector.put(vector_of_node_iterators[7], 1978);
  std::cout << "some node property values..." << std::endl;
  for(each_node_iterator i=G.first_node(); i.valid(); ++i) {
    std::cout << my_property_vector.get(i) << std::endl;
  }
  int _i=1;
  int _ii=1;
  edge_property_vector<list_graph, int> my_edge_property(G);
  for(each_edge_iterator i=G.first_edge(); i.valid(); ++i) {
    my_edge_property.put(i, _i);
    _i*=_ii; ++_ii;
  }

  std::cout << "node and edge property values on the tails and heads of edges..." << std::endl;
  for(each_edge_iterator j=G.first_edge(); j.valid(); ++j) {
    std::cout << my_property_vector.get(G.tail(j)) << "--" << my_edge_property.get(j) << "-->" << my_property_vector.get(G.head(j)) << " ";
  }
  std::cout << std::endl;

  //std::cout << "the same for inedges of the nodes..." << std::endl;
  //k=0;
  //for(each_node_iterator i=G.first_node(); i.valid(); ++i) {
  //  for(in_edge_iterator j=G.first_in_edge(i); j.valid(); ++j) {
  //    std::cout << my_property_vector.get(G.tail(j)) << "-->" << my_property_vector.get(G.head(j)) << " ";
  //  }
  //  std::cout << std::endl;
  //}

  std::cout << "bfs from the first node" << std::endl;
  bfs<list_graph> bfs_test(G, G.first_node());
  bfs_test.run();
  std::cout << "reached: ";
  for(each_node_iterator i=G.first_node(); i.valid(); ++i) {
    std::cout << bfs_test.reached.get(i) << " ";
  }
  std::cout<<std::endl;
  std::cout << "dist: ";
  for(each_node_iterator i=G.first_node(); i.valid(); ++i) {
    std::cout << bfs_test.dist.get(i) << " ";
  }
  std::cout<<std::endl;


  std::cout << "augmenting path flow algorithm test..." << std::endl;
  list_graph flow_test;

  node_iterator s=flow_test.add_node();
  node_iterator v1=flow_test.add_node();
  node_iterator v2=flow_test.add_node();
  node_iterator v3=flow_test.add_node();
  node_iterator v4=flow_test.add_node();
  node_iterator t=flow_test.add_node();
  
  node_property_vector<list_graph, std::string> node_name(flow_test);
  node_name.put(s, "s");
  node_name.put(v1, "v1");
  node_name.put(v2, "v2");
  node_name.put(v3, "v3");
  node_name.put(v4, "v4");
  node_name.put(t, "t");

  edge_iterator s_v1=flow_test.add_edge(s, v1);
  edge_iterator s_v2=flow_test.add_edge(s, v2);
  edge_iterator v1_v2=flow_test.add_edge(v1, v2);
  edge_iterator v2_v1=flow_test.add_edge(v2, v1);
  edge_iterator v1_v3=flow_test.add_edge(v1, v3);
  edge_iterator v3_v2=flow_test.add_edge(v3, v2);
  edge_iterator v2_v4=flow_test.add_edge(v2, v4);
  edge_iterator v4_v3=flow_test.add_edge(v4, v3);
  edge_iterator v3_t=flow_test.add_edge(v3, t);
  edge_iterator v4_t=flow_test.add_edge(v4, t);

  edge_property_vector<list_graph, int> cap(flow_test);

  cap.put(s_v1, 16);
  cap.put(s_v2, 13);
  cap.put(v1_v2, 10);
  cap.put(v2_v1, 4);
  cap.put(v1_v3, 12);
  cap.put(v3_v2, 9);
  cap.put(v2_v4, 14);
  cap.put(v4_v3, 7);
  cap.put(v3_t, 20);
  cap.put(v4_t, 4);

  std::cout << "on directed graph graph" << std::endl; //<< flow_test;
  std::cout << "names and capacity values" << std::endl; 
  for(each_node_iterator i=flow_test.first_node(); i.valid(); ++i) { 
    std::cout << node_name.get(i) << ": ";
    std::cout << "out edges: ";
    for(out_edge_iterator j=flow_test.first_out_edge(i); j.valid(); ++j) 
      std::cout << node_name.get(flow_test.tail(j)) << "-"<< cap.get(j) << "->" << node_name.get(flow_test.head(j)) << " ";
    std::cout << "in edges: ";
    for(in_edge_iterator j=flow_test.first_in_edge(i); j.valid(); ++j) 
      std::cout << node_name.get(flow_test.tail(j)) << "-"<< cap.get(j) << "->" << node_name.get(flow_test.head(j)) << " ";
    std::cout << std::endl;
  }

  
  //for(each_node_iterator i=flow_test.first_node(); i.valid(); ++i) { 
  //  std::cout << i << " ";
  //}
  
  max_flow_type<list_graph, int> max_flow_test(flow_test, s, t, cap);
  max_flow_test.run();

  return 0;
}