#include <lemon/smart_graph.h>
#include <lemon/kruskal.h>
#include <lemon/graph_reader.h>
#include <lemon/time_measure.h>
#include <lemon/graph_to_eps.h>

#include <lemon/steiner.h>

#include <cstdlib>
#include <cmath>

using namespace lemon;
using namespace lemon::dim2;

using namespace std;

int main(int argc, const char *argv[]) {
  std::string lgf = argc > 1 ? argv[1] : "steiner.lgf";
  std::string eps = argc > 2 ? argv[2] : "steiner.eps";

  SmartUGraph graph;
  SmartUGraph::NodeMap<bool> terminal(graph);
  SmartUGraph::NodeMap<int> label(graph);
  SmartUGraph::NodeMap<Point<double> > coord(graph);
  
  UGraphReader<SmartUGraph>(lgf, graph).
    readNodeMap("coordinates_x", xMap(coord)).
    readNodeMap("coordinates_y", yMap(coord)).
    readNodeMap("terminal", terminal).run();

  SmartUGraph::UEdgeMap<double> cost(graph);
  for (SmartUGraph::UEdgeIt it(graph); it != INVALID; ++it) {
    cost[it] = sqrt((coord[graph.target(it)] - 
                     coord[graph.source(it)]).normSquare());
  }
  
  SteinerTree<SmartUGraph> steiner(graph, cost);
  steiner.init();

  for (SmartUGraph::NodeIt it(graph); it != INVALID; ++it) {
    if (terminal[it]) {
      steiner.addTerminal(it);
    }
  }
  steiner.start();


  Palette nodepalette(0);
  nodepalette.add(Color(1.0, 1.0, 1.0));
  nodepalette.add(Color(0.0, 1.0, 0.0));
  nodepalette.add(Color(0.5, 0.5, 0.5));
  SmartUGraph::NodeMap<int> nodecolor(graph);
  for (SmartUGraph::NodeIt it(graph); it != INVALID; ++it) {
    if (steiner.terminal(it)) {
      nodecolor[it] = 1;
    } else if (steiner.steiner(it)) {
      nodecolor[it] = 2;
    } else {
      nodecolor[it] = 0;
    }
  }


  Palette edgepalette(0);
  edgepalette.add(Color(0.0, 0.0, 0.0));
  edgepalette.add(Color(1.0, 0.0, 0.0));
  SmartUGraph::UEdgeMap<int> edgecolor(graph);
  for (SmartUGraph::UEdgeIt it(graph); it != INVALID; ++it) {
    edgecolor[it] = steiner.tree(it) ? 1 : 0;
  }


  graphToEps(graph, eps).
    coords(coord).undirected().
    nodeScale(1.0).scaleToA4().
    nodeColors(composeMap(nodepalette, nodecolor)).
    edgeColors(composeMap(edgepalette, edgecolor)).
    nodeTexts(label).nodeTextSize(8).run();

  std::cout << "The tree constructed: " << eps << std::endl;
  std::cout << "The cost of the tree: " << steiner.treeValue() << std::endl;

  return 0;
}