#include <lemon/time_measure.h>
#include <lemon/smart_graph.h>
#include <lemon/graph_utils.h>
#include <lemon/maps.h>
#include <lemon/radix_sort.h>

#include "test_tools.h"


#include <vector>
#include <algorithm>
#include <cmath>

using namespace std;
using namespace lemon;

void checkRadixSort() {
  {
    int n = 10000;
    vector<int> data1(n), data2(n);
    for (int i = 0; i < n; ++i) {
      data1[i] = data2[i] = (int)(1000 * (rand() / (RAND_MAX + 1.0))) - 500;
    }
    radixSort(data1.begin(), data1.end());
    sort(data2.begin(), data2.end());
    for (int i = 0; i < n; ++i) {
      check(data1[i] == data2[i], "Test failed");
    }
  } {
    int n = 10000;
    vector<unsigned char> data1(n), data2(n);
    for (int i = 0; i < n; ++i) {
      data1[i] = data2[i] = (int)(200 * (rand() / (RAND_MAX + 1.0)));
    }
    radixSort(data1.begin(), data1.end());
    sort(data2.begin(), data2.end());
    for (int i = 0; i < n; ++i) {
      check(data1[i] == data2[i], "Test failed");
    }
  }
}


void checkCounterSort() {
  {
    int n = 10000;
    vector<int> data1(n), data2(n);
    for (int i = 0; i < n; ++i) {
      data1[i] = data2[i] = (int)(1000 * (rand() / (RAND_MAX + 1.0))) - 500;
    }
    counterSort(data1.begin(), data1.end());
    sort(data2.begin(), data2.end());
    for (int i = 0; i < n; ++i) {
      check(data1[i] == data2[i], "Test failed");
    } 
  } {
    int n = 10000;
    vector<unsigned char> data1(n), data2(n);
    for (int i = 0; i < n; ++i) {
      data1[i] = data2[i] = (int)(200 * (rand() / (RAND_MAX + 1.0)));
    }
    counterSort(data1.begin(), data1.end());
    sort(data2.begin(), data2.end());
    for (int i = 0; i < n; ++i) {
      check(data1[i] == data2[i], "Test failed");
    } 
  }
}

void checkSorts() {
  checkRadixSort();
  checkCounterSort();
}

void checkGraphRadixSort() {
  typedef SmartGraph Graph;
  typedef Graph::Node Node;
  typedef Graph::Edge Edge;

  const int n = 100;
  const int e = (int)(n * log((double)n));

  Graph graph;
  vector<Node> nodes;

  for (int i = 0; i < n; ++i) {
    nodes.push_back(graph.addNode());
  }
  vector<Edge> edges;
  for (int i = 0; i < e; ++i) {
    int s = (int)(n * (double)rand() / (RAND_MAX + 1.0));
    int t = (int)(n * (double)rand() / (RAND_MAX + 1.0));
    edges.push_back(graph.addEdge(nodes[s], nodes[t]));
  }

  radixSort(edges.begin(), edges.end(), 
	    mapFunctor(composeMap(IdMap<Graph, Node>(graph), 
				  sourceMap(graph))));

  Graph::EdgeMap<bool> was(graph, false);

  for (int i = 0; i < (int)edges.size(); ++i) {
    check(!was[edges[i]], "Test failed");
    was[edges[i]] = true;
  }

  for (int i = 1; i < (int)edges.size(); ++i) {
    check(graph.id(graph.source(edges[i - 1])) <= 
	  graph.id(graph.source(edges[i])), "Test failed");
  }

}

void checkGraphCounterSort() {
  typedef SmartGraph Graph;
  typedef Graph::Node Node;
  typedef Graph::Edge Edge;

  const int n = 100;
  const int e = (int)(n * log((double)n));

  Graph graph;
  vector<Node> nodes;

  for (int i = 0; i < n; ++i) {
    nodes.push_back(graph.addNode());
  }
  vector<Edge> edges;
  for (int i = 0; i < e; ++i) {
    int s = (int)(n * (double)rand() / (RAND_MAX + 1.0));
    int t = (int)(n * (double)rand() / (RAND_MAX + 1.0));
    edges.push_back(graph.addEdge(nodes[s], nodes[t]));
  }

  counterSort(edges.begin(), edges.end(), 
	      mapFunctor(composeMap(IdMap<Graph, Node>(graph), 
				    sourceMap(graph))));

  counterSort(edges.begin(), edges.end(), 
	      mapFunctor(composeMap(IdMap<Graph, Node>(graph), 
				    targetMap(graph))));

  Graph::EdgeMap<bool> was(graph, false);

  for (int i = 0; i < (int)edges.size(); ++i) {
    check(!was[edges[i]], "Test failed");
    was[edges[i]] = true;
  }

  for (int i = 1; i < (int)edges.size(); ++i) {
    check(graph.id(graph.target(edges[i - 1])) < 
	  graph.id(graph.target(edges[i])) || 
	  (graph.id(graph.target(edges[i - 1])) ==
	   graph.id(graph.target(edges[i])) &&
	   graph.id(graph.source(edges[i - 1])) <= 
	   graph.id(graph.source(edges[i]))), "Test failed");
  }

}

void checkGraphSorts() {
  checkGraphRadixSort();
  checkGraphCounterSort();
}

int main() {
  checkSorts();
  checkGraphSorts();
  return 0;
}