// -*- c++ -*-

#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>();

  ///Everything is OK

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

  return 0;

}