/* -*- C++ -*-

  *

  * This file is a part of LEMON, a generic C++ optimization library

  *

  * Copyright (C) 2003-2008

  * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport

  * (Egervary Research Group on Combinatorial Optimization, EGRES).

  *

  * Permission to use, modify and distribute this software is granted

  * provided that this copyright notice appears in all copies. For

  * precise terms see the accompanying LICENSE file.

  *

  * This software is provided "AS IS" with no warranty of any kind,

  * express or implied, and with no claim as to its suitability for any

  * purpose.

  *

  */

 #include <iostream>

 #include <lemon/smart_graph.h>

 #include <lemon/graph_reader.h>

 #include <lemon/ugraph_adaptor.h>

 #include <lemon/graph_to_eps.h>

 #include <lemon/dfs.h>

 #include <lemon/topology.h>

 /// \ingroup demos

 /// \file

 /// \brief Strongly connected orientation

 ///

 /// This example demonstrates the usage of the DirUGraphAdaptor,

 /// the DfsVisitor and some topology functions. The program reads

 /// an input undirected graph and with a DfsVisit it orients each edge

 /// to minimize the strongly connected components in the graph. At least

 /// it checks the result of the orientation with to topology functions.

 ///

 /// \include strongly_connected_orientation.cc

 using namespace lemon;

 using namespace std;

 typedef SmartUGraph UGraph;

 UGRAPH_TYPEDEFS(UGraph);

 Color color(bool c) {

   return c ? BLACK : RED;

 }

 template <typename DirMap>

 class OrientVisitor : public DfsVisitor<UGraph> {

 public:

   OrientVisitor(const UGraph& ugraph, DirMap& dirMap)

     : _ugraph(ugraph), _dirMap(dirMap), _processed(ugraph, false) {}

   void discover(const Edge& edge) {

     _processed.set(edge, true);

     _dirMap.set(edge, _ugraph.direction(edge));

   }

   void examine(const Edge& edge) {

     if (_processed[edge]) return;

     _processed.set(edge, true);

     _dirMap.set(edge, _ugraph.direction(edge));

   }  

 private:

   const UGraph& _ugraph;  

   DirMap& _dirMap;

   UGraph::UEdgeMap<bool> _processed;

 };

 int main() {

   cout << "Orientation of the strongly_connected_orientation.lgf "

        << "to be strongly connected" << endl;

   UGraph ugraph;

   UGraph::NodeMap<dim2::Point<double> > coords(ugraph);

   UGraphReader<UGraph>("strongly_connected_orientation.lgf", ugraph).

     readNodeMap("coords", coords).run();

   UGraph::UEdgeMap<bool> dmap(ugraph);

   typedef OrientVisitor<UGraph::UEdgeMap<bool> > Visitor;

   Visitor visitor(ugraph, dmap);

   DfsVisit<UGraph, Visitor> dfs(ugraph, visitor);

   dfs.run();

   typedef DirUGraphAdaptor<UGraph> DGraph;

   DGraph dgraph(ugraph, dmap);

   cout << "The result written into the " 

        << "strongly_connected_orientation.eps file" << endl;

   graphToEps(dgraph, "strongly_connected_orientation.eps").

     drawArrows().coords(coords).scaleToA4().enableParallel().

     autoNodeScale().autoEdgeWidthScale().run();

   int num_scc = countStronglyConnectedComponents(dgraph);

   int num_becc = countBiEdgeConnectedComponents(ugraph);

   if (num_scc != num_becc) {

     std::cerr << "Wrong Orientation" << std::endl;

   }

   return 0;

 }