/* -*- C++ -*-

  *

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

  *

  * Copyright (C) 2003-2006

  * 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 <vector>

 #include <lemon/graph_utils.h>

 #include <lemon/list_graph.h>

 #include <lemon/smart_graph.h>

 #include <lemon/full_graph.h>

 #include "test_tools.h"

 #include "graph_utils_test.h"

 using namespace lemon;

 template<class Graph>

 void checkSnapDeg() 

 {

   Graph g;

   typename Graph::Node n1=g.addNode();

   typename Graph::Node n2=g.addNode();

   InDegMap<Graph> ind(g);

   g.addEdge(n1,n2);

   typename Graph::Snapshot snap(g);

   OutDegMap<Graph> outd(g);

   check(ind[n1]==0 && ind[n2]==1, "Wrong InDegMap value.");

   check(outd[n1]==1 && outd[n2]==0, "Wrong OutDegMap value.");

   g.addEdge(n1,n2);

   g.addEdge(n2,n1);

   check(ind[n1]==1 && ind[n2]==2, "Wrong InDegMap value.");

   check(outd[n1]==2 && outd[n2]==1, "Wrong OutDegMap value.");

   snap.restore();

   check(ind[n1]==0 && ind[n2]==1, "Wrong InDegMap value.");

   check(outd[n1]==1 && outd[n2]==0, "Wrong OutDegMap value.");

 }

 int main() {

   ///\file

   { // checking list graph

     checkGraphCounters<ListGraph>();

     checkFindEdge<ListGraph>();

   }

   { // checking smart graph

     checkGraphCounters<SmartGraph>();

     checkFindEdge<SmartGraph>();

   }

   {

     int num = 5;

     FullGraph fg(num);

     check(countNodes(fg) == num, "FullGraph: wrong node number.");

     check(countEdges(fg) == num*num, "FullGraph: wrong edge number.");

     for (FullGraph::NodeIt src(fg); src != INVALID; ++src) {

       for (FullGraph::NodeIt trg(fg); trg != INVALID; ++trg) {

 	ConEdgeIt<FullGraph> con(fg, src, trg);

 	check(con != INVALID, "There is no connecting edge.");

 	check(fg.source(con) == src, "Wrong source.");

 	check(fg.target(con) == trg, "Wrong target.");

 	check(++con == INVALID, "There is more connecting edge.");

       }

     }

   }

   //check In/OutDegMap (and Snapshot feature)

   checkSnapDeg<ListGraph>();

   checkSnapDeg<SmartGraph>();

   {

     const int nodeNum = 10;

     const int edgeNum = 100;

     ListGraph graph;

     InDegMap<ListGraph> inDeg(graph);

     OutDegMap<ListGraph> outDeg(graph);

     std::vector<ListGraph::Node> nodes(nodeNum);

     for (int i = 0; i < nodeNum; ++i) {

       nodes[i] = graph.addNode();

     }

     std::vector<ListGraph::Edge> edges(edgeNum);

     for (int i = 0; i < edgeNum; ++i) {

       edges[i] = 

 	graph.addEdge(nodes[urandom(nodeNum)], nodes[urandom(nodeNum)]);

     }

     for (int i = 0; i < nodeNum; ++i) {

       check(inDeg[nodes[i]] == countInEdges(graph, nodes[i]), 

 	    "Wrong in degree map");

     }

     for (int i = 0; i < nodeNum; ++i) {

       check(outDeg[nodes[i]] == countOutEdges(graph, nodes[i]), 

 	    "Wrong in degree map");

     }

   }

   ///Everything is OK

   std::cout << __FILE__ ": All tests passed.\n";

   return 0;

 }