// -*- c++ -*-

#include <iostream>

#include <vector>

#include <lemon/concept_check.h>

#include <lemon/concept/matrix_maps.h>

#include <lemon/concept/maps.h>

#include <lemon/concept/graph.h>

#include <lemon/matrix_maps.h>

#include <lemon/smart_graph.h>

#include "test_tools.h"

#include "graph_test.h"

#include "map_test.h"

using namespace lemon;

using namespace lemon::concept;

int main() {

  typedef SmartGraph Graph;

  typedef Graph::Node Node;

  { // checking MatrixMap for int

    typedef DynamicMatrixMap<Graph, Node, int> IntMatrixMap;

    checkConcept<ReferenceMatrixMap<Node, Node, int, 

      IntMatrixMap::Reference, IntMatrixMap::ConstReference>,

      IntMatrixMap>();

  }

  { // checking MatrixMap for bool

    typedef DynamicMatrixMap<Graph, Node, bool> BoolMatrixMap;

    checkConcept<ReferenceMatrixMap<Node, Node, bool, 

      BoolMatrixMap::Reference, BoolMatrixMap::ConstReference>,

      BoolMatrixMap>();

  }

  {

    Graph graph;

    typedef DynamicMatrixMap<Graph, Node, int> IntMatrixMap;

    IntMatrixMap matrix(graph);

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

      graph.addNode();

    }

    for (Graph::NodeIt it(graph); it != INVALID; ++it) {

      for (Graph::NodeIt jt(graph); jt != INVALID; ++jt) {

	int val = urandom(100);

	matrix.set(it, jt, val);

	check(matrix(it, jt) == val, "Wrong assign");

	check(matrix(it, jt) == matrixColMap(matrix, it)[jt], "Wrong colMap");

	check(matrix(it, jt) == matrixRowMap(matrix, jt)[it], "Wrong rowMap");

      }

    }

    const IntMatrixMap& cm = matrix;

    for (Graph::NodeIt it(graph); it != INVALID; ++it) {

      for (Graph::NodeIt jt(graph); jt != INVALID; ++jt) {

	check(cm(it, jt) == matrixColMap(cm, it)[jt], "Wrong colMap");

	check(cm(it, jt) == matrixRowMap(cm, jt)[it], "Wrong rowMap");

      }

    }

  }

  { // checking MatrixMap for int

    typedef DynamicSymMatrixMap<Graph, Node, int> IntMatrixMap;

    checkConcept<ReferenceMatrixMap<Node, Node, int, 

      IntMatrixMap::Reference, IntMatrixMap::ConstReference>,

      IntMatrixMap>();

  }

  { // checking MatrixMap for bool

    typedef DynamicSymMatrixMap<Graph, Node, bool> BoolMatrixMap;

    checkConcept<ReferenceMatrixMap<Node, Node, bool, 

      BoolMatrixMap::Reference, BoolMatrixMap::ConstReference>,

      BoolMatrixMap>();

  }

  {

    Graph graph;

    typedef DynamicSymMatrixMap<Graph, Node, int> IntMatrixMap;

    IntMatrixMap matrix(graph);

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

      graph.addNode();

    }

    for (Graph::NodeIt it(graph); it != INVALID; ++it) {

      for (Graph::NodeIt jt(graph); jt != INVALID; ++jt) {

	int val = urandom(100);

	matrix.set(it, jt, val);

	check(matrix(it, jt) == val, "Wrong assign");

	check(matrix(jt, it) == val, "Wrong assign");

	check(matrix(it, jt) == matrixColMap(matrix, it)[jt], "Wrong colMap");

	check(matrix(it, jt) == matrixRowMap(matrix, jt)[it], "Wrong rowMap");

      }

    }

    const IntMatrixMap& cm = matrix;

    for (Graph::NodeIt it(graph); it != INVALID; ++it) {

      for (Graph::NodeIt jt(graph); jt != INVALID; ++jt) {

	check(cm(it, jt) == matrixColMap(cm, it)[jt], "Wrong colMap");

	check(cm(it, jt) == matrixRowMap(cm, jt)[it], "Wrong rowMap");

      }

    }

  }

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

  return 0;

}