#define BOOST_TEST_MODULE bottleneck test

#include <boost/test/included/unit_test.hpp>
#include <random>
#include "../include/gudhi/Graph_matching.h"

#include <chrono>
#include <fstream>

using namespace Gudhi::bipartite_graph_matching;

int n1 = 81; // a natural number >0
int n2 = 180; // a natural number >0
double upper_bound = 400.5; // any real >0

BOOST_AUTO_TEST_CASE(global){
    std::uniform_real_distribution<double> unif1(0.,upper_bound);
    std::uniform_real_distribution<double> unif2(upper_bound/1000.,upper_bound/100.);
    std::default_random_engine re;
    std::vector< std::pair<double, double> > v1, v2;
    for (int i = 0; i < n1; i++) {
        double a = unif1(re);
        double b = unif1(re);
        double x = unif2(re);
        double y = unif2(re);
        v1.emplace_back(std::min(a,b), std::max(a,b));
        v2.emplace_back(std::min(a,b)+std::min(x,y), std::max(a,b)+std::max(x,y));
        if(i%5==0)
            v1.emplace_back(std::min(a,b),std::min(a,b)+x);
        if(i%3==0)
            v2.emplace_back(std::max(a,b),std::max(a,b)+y);
    }
    //99.5 and not 100 to avoid float errors.
    BOOST_CHECK(bottleneck_distance(v1, v2) <= upper_bound/99.5);
}

BOOST_AUTO_TEST_CASE(persistence_diagrams_graph){
    // Random construction
    std::uniform_real_distribution<double> unif(0.,upper_bound);
    std::default_random_engine re;
    std::vector< std::pair<double, double> > v1, v2;
    for (int i = 0; i < n1; i++) {
        double a = unif(re);
        double b = unif(re);
        v1.emplace_back(std::min(a,b), std::max(a,b));
    }
    for (int i = 0; i < n2; i++) {
        double a = unif(re);
        double b = unif(re);
        v2.emplace_back(std::min(a,b), std::max(a,b));
    }
    G::initialize(v1, v2, 0.);
    std::unique_ptr< std::vector<double> > d = std::move(G::sorted_distances());
    //
    BOOST_CHECK(!G::on_the_u_diagonal(n1-1));
    BOOST_CHECK(!G::on_the_u_diagonal(n1));
    BOOST_CHECK(!G::on_the_u_diagonal(n2-1));
    BOOST_CHECK(G::on_the_u_diagonal(n2));
    BOOST_CHECK(!G::on_the_v_diagonal(n1-1));
    BOOST_CHECK(G::on_the_v_diagonal(n1));
    BOOST_CHECK(G::on_the_v_diagonal(n2-1));
    BOOST_CHECK(G::on_the_v_diagonal(n2));
    //
    BOOST_CHECK(G::corresponding_point_in_u(0)==n2);
    BOOST_CHECK(G::corresponding_point_in_u(n1)==0);
    BOOST_CHECK(G::corresponding_point_in_v(0)==n1);
    BOOST_CHECK(G::corresponding_point_in_v(n2)==0);
    //
    BOOST_CHECK(G::size()==(n1+n2));
    //
    BOOST_CHECK((int) d->size() <= (n1+n2)*(n1+n2) - n1*n2 + 1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance(0,0))==1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance(0,n1-1))==1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance(0,n1))==1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance(0,n2-1))==1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance(0,n2))==1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance(0,(n1+n2)-1))==1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance(n1,0))==1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance(n1,n1-1))==1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance(n1,n1))==1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance(n1,n2-1))==1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance(n1,n2))==1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance(n1,(n1+n2)-1))==1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance((n1+n2)-1,0))==1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance((n1+n2)-1,n1-1))==1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance((n1+n2)-1,n1))==1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance((n1+n2)-1,n2-1))==1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance((n1+n2)-1,n2))==1);
    BOOST_CHECK(std::count(d->begin(), d->end(), G::distance((n1+n2)-1,(n1+n2)-1))==1);
}


BOOST_AUTO_TEST_CASE(planar_neighbors_finder) {
    Planar_neighbors_finder pnf = Planar_neighbors_finder(1.);
    for(int v_point_index=0; v_point_index<n1; v_point_index+=2)
        pnf.add(v_point_index);
    //
    BOOST_CHECK(pnf.contains(0));
    BOOST_CHECK(!pnf.contains(1));
    BOOST_CHECK(pnf.contains(2));
    BOOST_CHECK(!pnf.contains(3));
    //
    pnf.remove(0);
    pnf.remove(1);
    //
    BOOST_CHECK(!pnf.contains(0));
    BOOST_CHECK(!pnf.contains(1));
    BOOST_CHECK(pnf.contains(2));
    BOOST_CHECK(!pnf.contains(3));
    //
    int v_point_index_1 = pnf.pull_near(n2/2);
    BOOST_CHECK((v_point_index_1 == -1) || (G::distance(n2/2,v_point_index_1)<=1.));
    BOOST_CHECK(!pnf.contains(v_point_index_1));
    std::list<int> l = *pnf.pull_all_near(n2/2);
    bool v = true;
    for(auto it = l.cbegin(); it != l.cend(); ++it)
        v = v && (G::distance(n2/2,*it)>1.);
    BOOST_CHECK(v);
    int v_point_index_2 = pnf.pull_near(n2/2);
    BOOST_CHECK(v_point_index_2 == -1);
}


BOOST_AUTO_TEST_CASE(neighbors_finder) {
    Neighbors_finder nf = Neighbors_finder(1.);
    for(int v_point_index=1; v_point_index<((n2+n1)*9/10); v_point_index+=2)
        nf.add(v_point_index);
    //
    int v_point_index_1 = nf.pull_near(n2/2);
    BOOST_CHECK((v_point_index_1 == -1) || (G::distance(n2/2,v_point_index_1)<=1.));
    std::list<int> l = *nf.pull_all_near(n2/2);
    bool v = true;
    for(auto it = l.cbegin(); it != l.cend(); ++it)
        v = v && (G::distance(n2/2,*it)>1.);
    BOOST_CHECK(v);
    int v_point_index_2 = nf.pull_near(n2/2);
    BOOST_CHECK(v_point_index_2 == -1);
}

BOOST_AUTO_TEST_CASE(layered_neighbors_finder) {
    Layered_neighbors_finder lnf = Layered_neighbors_finder(1.);
    for(int v_point_index=1; v_point_index<((n2+n1)*9/10); v_point_index+=2)
        lnf.add(v_point_index, v_point_index % 7);
    //
    int v_point_index_1 = lnf.pull_near(n2/2,6);
    BOOST_CHECK((v_point_index_1 == -1) || (G::distance(n2/2,v_point_index_1)<=1.));
    int v_point_index_2 = lnf.pull_near(n2/2,6);
    BOOST_CHECK(v_point_index_2 == -1);
    v_point_index_1 = lnf.pull_near(n2/2,0);
    BOOST_CHECK((v_point_index_1 == -1) || (G::distance(n2/2,v_point_index_1)<=1.));
    v_point_index_2 = lnf.pull_near(n2/2,0);
    BOOST_CHECK(v_point_index_2 == -1);
}

BOOST_AUTO_TEST_CASE(graph_matching) {
    Graph_matching m1;
    m1.set_r(0.);
    int e  = 0;
    while (m1.multi_augment())
        ++e;
    BOOST_CHECK(e <= 2*sqrt(2*(n1+n2)));
    Graph_matching m2 = m1;
    BOOST_CHECK(!m2.multi_augment());
    m2.set_r(upper_bound);
    e  = 0;
    while (m2.multi_augment())
        ++e;
    BOOST_CHECK(e <= 2*sqrt(2*(n1+n2)));
    BOOST_CHECK(m2.perfect());
    BOOST_CHECK(!m1.perfect());
}

/*
BOOST_AUTO_TEST_CASE(chrono) {
    std::ofstream objetfichier;
    objetfichier.open("results.csv", std::ios::out);

    for(int n =50; n<=1000; n+=100){
        std::uniform_real_distribution<double> unif1(0.,upper_bound);
        std::uniform_real_distribution<double> unif2(upper_bound/1000.,upper_bound/100.);
        std::default_random_engine re;
        std::vector< std::pair<double, double> > v1, v2;
        for (int i = 0; i < n; i++) {
            double a = unif1(re);
            double b = unif1(re);
            double x = unif2(re);
            double y = unif2(re);
            v1.emplace_back(std::min(a,b), std::max(a,b));
            v2.emplace_back(std::min(a,b)+std::min(x,y), std::max(a,b)+std::max(x,y));
            if(i%5==0)
                v1.emplace_back(std::min(a,b),std::min(a,b)+x);
            if(i%3==0)
                v2.emplace_back(std::max(a,b),std::max(a,b)+y);
        }

        std::chrono::steady_clock::time_point start = std::chrono::steady_clock::now();
        double b = bottleneck_distance(v1,v2);
        std::chrono::steady_clock::time_point end = std::chrono::steady_clock::now();

        typedef std::chrono::duration<int,std::milli> millisecs_t;
        millisecs_t duration(std::chrono::duration_cast<millisecs_t>(end-start));
        objetfichier << n << ";" << duration.count() << ";" << b << std::endl;
    }
    objetfichier.close();
}
*/