/*    This file is part of the Gudhi Library - https://gudhi.inria.fr/ - which is released under MIT.
 *    See file LICENSE or go to https://gudhi.inria.fr/licensing/ for full license details.
 *    Author(s):       Pawel Dlotko
 *
 *    Copyright (C) 2016 Inria
 *
 *    Modification(s):
 *      - YYYY/MM Author: Description of the modification
 */

#define BOOST_TEST_DYN_LINK
#define BOOST_TEST_MODULE "Persistence_intervals_test"
#include <boost/test/unit_test.hpp>
#include <gudhi/reader_utils.h>
#include "gudhi/Persistence_intervals.h"
#include <gudhi/common_persistence_representations.h>
#include <gudhi/Unitary_tests_utils.h>

#include <iostream>

using namespace Gudhi;
using namespace Gudhi::Persistence_representations;

BOOST_AUTO_TEST_CASE(check_min_max_function) {
  Persistence_intervals p("data/file_with_diagram");
  std::pair<double, double> min_max_ = p.get_x_range();

  GUDHI_TEST_FLOAT_EQUALITY_CHECK(min_max_.first, 0.0290362, Gudhi::Persistence_representations::epsi);
  GUDHI_TEST_FLOAT_EQUALITY_CHECK(min_max_.second, 0.994537, Gudhi::Persistence_representations::epsi);
}

BOOST_AUTO_TEST_CASE(check_length_of_dominant_intervals) {
  Persistence_intervals p("data/file_with_diagram");
  std::vector<double> dominant_ten_intervals_length = p.length_of_dominant_intervals(10);
  std::vector<double> dominant_intervals_length;
  dominant_intervals_length.push_back(0.862625);
  dominant_intervals_length.push_back(0.800893);
  dominant_intervals_length.push_back(0.762061);
  dominant_intervals_length.push_back(0.756501);
  dominant_intervals_length.push_back(0.729367);
  dominant_intervals_length.push_back(0.718177);
  dominant_intervals_length.push_back(0.708395);
  dominant_intervals_length.push_back(0.702844);
  dominant_intervals_length.push_back(0.700468);
  dominant_intervals_length.push_back(0.622177);
  for (size_t i = 0; i != dominant_ten_intervals_length.size(); ++i) {
    GUDHI_TEST_FLOAT_EQUALITY_CHECK(dominant_ten_intervals_length[i], dominant_intervals_length[i],
                                    Gudhi::Persistence_representations::epsi);
  }
}
BOOST_AUTO_TEST_CASE(check_dominant_intervals) {
  Persistence_intervals p("data/file_with_diagram");
  std::vector<std::pair<double, double> > ten_dominant_intervals = p.dominant_intervals(10);

  std::vector<std::pair<double, double> > templ;
  templ.push_back(std::pair<double, double>(0.114718, 0.977343));
  templ.push_back(std::pair<double, double>(0.133638, 0.93453));
  templ.push_back(std::pair<double, double>(0.104599, 0.866659));
  templ.push_back(std::pair<double, double>(0.149798, 0.906299));
  templ.push_back(std::pair<double, double>(0.247352, 0.976719));
  templ.push_back(std::pair<double, double>(0.192675, 0.910852));
  templ.push_back(std::pair<double, double>(0.191836, 0.900231));
  templ.push_back(std::pair<double, double>(0.284998, 0.987842));
  templ.push_back(std::pair<double, double>(0.294069, 0.994537));
  templ.push_back(std::pair<double, double>(0.267421, 0.889597));

  for (size_t i = 0; i != ten_dominant_intervals.size(); ++i) {
    GUDHI_TEST_FLOAT_EQUALITY_CHECK(ten_dominant_intervals[i].first, templ[i].first,
                                    Gudhi::Persistence_representations::epsi);
    GUDHI_TEST_FLOAT_EQUALITY_CHECK(ten_dominant_intervals[i].second, templ[i].second,
                                    Gudhi::Persistence_representations::epsi);
  }
}

BOOST_AUTO_TEST_CASE(check_histogram_of_lengths) {
  Persistence_intervals p("data/file_with_diagram");
  std::vector<size_t> histogram = p.histogram_of_lengths(10);
  std::vector<size_t> template_histogram;
  template_histogram.push_back(10);
  template_histogram.push_back(5);
  template_histogram.push_back(3);
  template_histogram.push_back(4);
  template_histogram.push_back(4);
  template_histogram.push_back(3);
  template_histogram.push_back(6);
  template_histogram.push_back(1);
  template_histogram.push_back(7);
  template_histogram.push_back(1);
  template_histogram.push_back(1);
  for (size_t i = 0; i != histogram.size(); ++i) {
    BOOST_CHECK(histogram[i] == template_histogram[i]);
  }
}

BOOST_AUTO_TEST_CASE(check_cumulative_histograms_of_lengths) {
  Persistence_intervals p("data/file_with_diagram");
  std::vector<size_t> cumulative_histogram = p.cumulative_histogram_of_lengths(10);
  std::vector<size_t> template_cumulative_histogram;
  template_cumulative_histogram.push_back(10);
  template_cumulative_histogram.push_back(15);
  template_cumulative_histogram.push_back(18);
  template_cumulative_histogram.push_back(22);
  template_cumulative_histogram.push_back(26);
  template_cumulative_histogram.push_back(29);
  template_cumulative_histogram.push_back(35);
  template_cumulative_histogram.push_back(36);
  template_cumulative_histogram.push_back(43);
  template_cumulative_histogram.push_back(44);
  template_cumulative_histogram.push_back(45);

  for (size_t i = 0; i != cumulative_histogram.size(); ++i) {
    BOOST_CHECK(cumulative_histogram[i] == template_cumulative_histogram[i]);
  }
}
BOOST_AUTO_TEST_CASE(check_characteristic_function_of_diagram) {
  Persistence_intervals p("data/file_with_diagram");
  std::pair<double, double> min_max_ = p.get_x_range();
  std::vector<double> char_funct_diag = p.characteristic_function_of_diagram(min_max_.first, min_max_.second);
  std::vector<double> template_char_funct_diag;
  template_char_funct_diag.push_back(0.370665);
  template_char_funct_diag.push_back(0.84058);
  template_char_funct_diag.push_back(1.24649);
  template_char_funct_diag.push_back(1.3664);
  template_char_funct_diag.push_back(1.34032);
  template_char_funct_diag.push_back(1.31904);
  template_char_funct_diag.push_back(1.14076);
  template_char_funct_diag.push_back(0.991259);
  template_char_funct_diag.push_back(0.800714);
  template_char_funct_diag.push_back(0.0676303);

  for (size_t i = 0; i != char_funct_diag.size(); ++i) {
    GUDHI_TEST_FLOAT_EQUALITY_CHECK(char_funct_diag[i], template_char_funct_diag[i],
                                    Gudhi::Persistence_representations::epsi);
  }
}

BOOST_AUTO_TEST_CASE(check_cumulative_characteristic_function_of_diagram) {
  Persistence_intervals p("data/file_with_diagram");
  std::pair<double, double> min_max_ = p.get_x_range();
  std::vector<double> cumul_char_funct_diag =
      p.cumulative_characteristic_function_of_diagram(min_max_.first, min_max_.second);
  std::vector<double> template_char_funct_diag_cumul;

  template_char_funct_diag_cumul.push_back(0.370665);
  template_char_funct_diag_cumul.push_back(1.21125);
  template_char_funct_diag_cumul.push_back(2.45774);
  template_char_funct_diag_cumul.push_back(3.82414);
  template_char_funct_diag_cumul.push_back(5.16446);
  template_char_funct_diag_cumul.push_back(6.4835);
  template_char_funct_diag_cumul.push_back(7.62426);
  template_char_funct_diag_cumul.push_back(8.61552);
  template_char_funct_diag_cumul.push_back(9.41623);
  template_char_funct_diag_cumul.push_back(9.48386);

  for (size_t i = 0; i != cumul_char_funct_diag.size(); ++i) {
    GUDHI_TEST_FLOAT_EQUALITY_CHECK(cumul_char_funct_diag[i], template_char_funct_diag_cumul[i],
                                    Gudhi::Persistence_representations::epsi);
  }
}

BOOST_AUTO_TEST_CASE(check_compute_persistent_betti_numbers) {
  Persistence_intervals p("data/file_with_diagram");
  std::vector<std::pair<double, size_t> > pbns;
  pbns.push_back(std::pair<double, size_t>(0.0290362, 1));
  pbns.push_back(std::pair<double, size_t>(0.0307676, 2));
  pbns.push_back(std::pair<double, size_t>(0.0366312, 3));
  pbns.push_back(std::pair<double, size_t>(0.0544614, 4));
  pbns.push_back(std::pair<double, size_t>(0.0920033, 5));
  pbns.push_back(std::pair<double, size_t>(0.104599, 6));
  pbns.push_back(std::pair<double, size_t>(0.114718, 7));
  pbns.push_back(std::pair<double, size_t>(0.117379, 8));
  pbns.push_back(std::pair<double, size_t>(0.123493, 9));
  pbns.push_back(std::pair<double, size_t>(0.133638, 10));
  pbns.push_back(std::pair<double, size_t>(0.137798, 9));
  pbns.push_back(std::pair<double, size_t>(0.149798, 10));
  pbns.push_back(std::pair<double, size_t>(0.155421, 11));
  pbns.push_back(std::pair<double, size_t>(0.158443, 12));
  pbns.push_back(std::pair<double, size_t>(0.176956, 13));
  pbns.push_back(std::pair<double, size_t>(0.183234, 12));
  pbns.push_back(std::pair<double, size_t>(0.191069, 13));
  pbns.push_back(std::pair<double, size_t>(0.191333, 14));
  pbns.push_back(std::pair<double, size_t>(0.191836, 15));
  pbns.push_back(std::pair<double, size_t>(0.192675, 16));
  pbns.push_back(std::pair<double, size_t>(0.208564, 17));
  pbns.push_back(std::pair<double, size_t>(0.218425, 18));
  pbns.push_back(std::pair<double, size_t>(0.219902, 17));
  pbns.push_back(std::pair<double, size_t>(0.23233, 16));
  pbns.push_back(std::pair<double, size_t>(0.234558, 17));
  pbns.push_back(std::pair<double, size_t>(0.237166, 16));
  pbns.push_back(std::pair<double, size_t>(0.247352, 17));
  pbns.push_back(std::pair<double, size_t>(0.267421, 18));
  pbns.push_back(std::pair<double, size_t>(0.268093, 19));
  pbns.push_back(std::pair<double, size_t>(0.278734, 18));
  pbns.push_back(std::pair<double, size_t>(0.284722, 19));
  pbns.push_back(std::pair<double, size_t>(0.284998, 20));
  pbns.push_back(std::pair<double, size_t>(0.294069, 21));
  pbns.push_back(std::pair<double, size_t>(0.306293, 22));
  pbns.push_back(std::pair<double, size_t>(0.322361, 21));
  pbns.push_back(std::pair<double, size_t>(0.323152, 22));
  pbns.push_back(std::pair<double, size_t>(0.371021, 23));
  pbns.push_back(std::pair<double, size_t>(0.372395, 24));
  pbns.push_back(std::pair<double, size_t>(0.387744, 25));
  pbns.push_back(std::pair<double, size_t>(0.435537, 26));
  pbns.push_back(std::pair<double, size_t>(0.462911, 25));
  pbns.push_back(std::pair<double, size_t>(0.483569, 26));
  pbns.push_back(std::pair<double, size_t>(0.489209, 25));
  pbns.push_back(std::pair<double, size_t>(0.517115, 24));
  pbns.push_back(std::pair<double, size_t>(0.522197, 23));
  pbns.push_back(std::pair<double, size_t>(0.532665, 22));
  pbns.push_back(std::pair<double, size_t>(0.545262, 23));
  pbns.push_back(std::pair<double, size_t>(0.587227, 22));
  pbns.push_back(std::pair<double, size_t>(0.593036, 23));
  pbns.push_back(std::pair<double, size_t>(0.602647, 24));
  pbns.push_back(std::pair<double, size_t>(0.605044, 25));
  pbns.push_back(std::pair<double, size_t>(0.621962, 24));
  pbns.push_back(std::pair<double, size_t>(0.629449, 23));
  pbns.push_back(std::pair<double, size_t>(0.636719, 22));
  pbns.push_back(std::pair<double, size_t>(0.64957, 21));
  pbns.push_back(std::pair<double, size_t>(0.650781, 22));
  pbns.push_back(std::pair<double, size_t>(0.654951, 23));
  pbns.push_back(std::pair<double, size_t>(0.683489, 24));
  pbns.push_back(std::pair<double, size_t>(0.687172, 23));
  pbns.push_back(std::pair<double, size_t>(0.69703, 22));
  pbns.push_back(std::pair<double, size_t>(0.701174, 21));
  pbns.push_back(std::pair<double, size_t>(0.717623, 22));
  pbns.push_back(std::pair<double, size_t>(0.722023, 21));
  pbns.push_back(std::pair<double, size_t>(0.722298, 20));
  pbns.push_back(std::pair<double, size_t>(0.725347, 19));
  pbns.push_back(std::pair<double, size_t>(0.73071, 18));
  pbns.push_back(std::pair<double, size_t>(0.758355, 17));
  pbns.push_back(std::pair<double, size_t>(0.770913, 18));
  pbns.push_back(std::pair<double, size_t>(0.790833, 17));
  pbns.push_back(std::pair<double, size_t>(0.821211, 16));
  pbns.push_back(std::pair<double, size_t>(0.849305, 17));
  pbns.push_back(std::pair<double, size_t>(0.853669, 16));
  pbns.push_back(std::pair<double, size_t>(0.866659, 15));
  pbns.push_back(std::pair<double, size_t>(0.872896, 16));
  pbns.push_back(std::pair<double, size_t>(0.889597, 15));
  pbns.push_back(std::pair<double, size_t>(0.900231, 14));
  pbns.push_back(std::pair<double, size_t>(0.903847, 13));
  pbns.push_back(std::pair<double, size_t>(0.906299, 12));
  pbns.push_back(std::pair<double, size_t>(0.910852, 11));
  pbns.push_back(std::pair<double, size_t>(0.93453, 10));
  pbns.push_back(std::pair<double, size_t>(0.944757, 9));
  pbns.push_back(std::pair<double, size_t>(0.947812, 8));
  pbns.push_back(std::pair<double, size_t>(0.959154, 7));
  pbns.push_back(std::pair<double, size_t>(0.975654, 6));
  pbns.push_back(std::pair<double, size_t>(0.976719, 5));
  pbns.push_back(std::pair<double, size_t>(0.977343, 4));
  pbns.push_back(std::pair<double, size_t>(0.980129, 3));
  pbns.push_back(std::pair<double, size_t>(0.987842, 2));
  pbns.push_back(std::pair<double, size_t>(0.990127, 1));
  pbns.push_back(std::pair<double, size_t>(0.994537, 0));

  std::vector<std::pair<double, size_t> > pbns_new = p.compute_persistent_betti_numbers();
  for (size_t i = 0; i != pbns.size(); ++i) {
    GUDHI_TEST_FLOAT_EQUALITY_CHECK(pbns[i].first, pbns_new[i].first, Gudhi::Persistence_representations::epsi);
    BOOST_CHECK(pbns[i].second == pbns_new[i].second);
  }
}

BOOST_AUTO_TEST_CASE(check_k_n_n) {
  Persistence_intervals p("data/file_with_diagram");
  std::vector<double> knn = p.k_n_n(5);
  std::vector<double> knn_template;
  knn_template.push_back(1.04208);
  knn_template.push_back(1.00344);
  knn_template.push_back(0.979395);
  knn_template.push_back(0.890643);
  knn_template.push_back(0.874769);
  knn_template.push_back(0.845787);
  knn_template.push_back(0.819713);
  knn_template.push_back(0.803984);
  knn_template.push_back(0.799864);
  knn_template.push_back(0.786945);

  for (size_t i = 0; i != knn.size(); ++i) {
    GUDHI_TEST_FLOAT_EQUALITY_CHECK(knn[i], knn_template[i], Gudhi::Persistence_representations::epsi);
  }
}