// Copyright 2018 Hans Dembinski
//
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt
// or copy at http://www.boost.org/LICENSE_1_0.txt)

#include <boost/core/lightweight_test.hpp>
#include <boost/histogram/axis/integer.hpp>
#include <boost/histogram/axis/ostream_operators.hpp>
#include <boost/histogram/axis/variable.hpp>
#include <boost/histogram/histogram.hpp>
#include <boost/histogram/indexed.hpp>
#include <boost/histogram/literals.hpp>
#include <vector>
#include "utility_histogram.hpp"

using namespace boost::histogram;
using namespace boost::histogram::literals;

template <typename Tag>
void run_tests() {
  // histogram iterator 1D
  {
    auto h = make(Tag(), axis::integer<>(0, 3));
    h(weight(2), 0);
    h(1);
    h(1);

    auto ind = indexed(h);
    auto it = ind.begin();
    BOOST_TEST_EQ(it->size(), 1);

    BOOST_TEST_EQ(it->operator[](0), 0);
    BOOST_TEST_EQ(it->value, 2);
    BOOST_TEST_EQ(it->bin(0), h.axis()[0]);
    ++it;
    BOOST_TEST_EQ(it->operator[](0), 1);
    BOOST_TEST_EQ(it->value, 2);
    BOOST_TEST_EQ(it->bin(0), h.axis()[1]);
    ++it;
    BOOST_TEST_EQ(it->operator[](0), 2);
    BOOST_TEST_EQ(it->value, 0);
    BOOST_TEST_EQ(it->bin(0), h.axis()[2]);
    ++it;
    BOOST_TEST_EQ(it->operator[](0), 3);
    BOOST_TEST_EQ(it->value, 0);
    BOOST_TEST_EQ(it->bin(0), h.axis()[3]);
    ++it;
    BOOST_TEST_EQ(it->operator[](0), -1);
    BOOST_TEST_EQ(it->value, 0);
    BOOST_TEST_EQ(it->bin(0), h.axis()[-1]);
    ++it;
    BOOST_TEST(it == ind.end());
  }

  // histogram iterator 2D
  {
    auto h = make_s(Tag(), std::vector<int>(), axis::integer<>(0, 1),
                    axis::integer<int, axis::null_type, axis::option_type::none>(2, 4));
    h(weight(2), 0, 2);
    h(-1, 2);
    h(1, 3);

    BOOST_TEST_EQ(axis::traits::extend(h.axis(0)), 3);
    BOOST_TEST_EQ(axis::traits::extend(h.axis(1)), 2);

    auto ind = indexed(h);
    auto it = ind.begin();
    BOOST_TEST_EQ(it->size(), 2);

    BOOST_TEST_EQ(it->operator[](0), 0);
    BOOST_TEST_EQ(it->operator[](1), 0);
    BOOST_TEST_EQ(it->bin(0_c), h.axis(0_c)[0]);
    BOOST_TEST_EQ(it->bin(1_c), h.axis(1_c)[0]);
    BOOST_TEST_EQ(it->value, 2);
    ++it;
    BOOST_TEST_EQ(it->operator[](0), 1);
    BOOST_TEST_EQ(it->operator[](1), 0);
    BOOST_TEST_EQ(it->bin(0), h.axis(0)[1]);
    BOOST_TEST_EQ(it->bin(1), h.axis(1)[0]);
    BOOST_TEST_EQ(it->value, 0);
    ++it;
    BOOST_TEST_EQ(it->operator[](0), -1);
    BOOST_TEST_EQ(it->operator[](1), 0);
    BOOST_TEST_EQ(it->bin(0_c), h.axis(0_c)[-1]);
    BOOST_TEST_EQ(it->bin(1_c), h.axis(1_c)[0]);
    BOOST_TEST_EQ(it->value, 1);
    ++it;
    BOOST_TEST_EQ(it->operator[](0), 0);
    BOOST_TEST_EQ(it->operator[](1), 1);
    BOOST_TEST_EQ(it->bin(0), h.axis(0)[0]);
    BOOST_TEST_EQ(it->bin(1), h.axis(1)[1]);
    BOOST_TEST_EQ(it->value, 0);
    ++it;
    BOOST_TEST_EQ(it->operator[](0), 1);
    BOOST_TEST_EQ(it->operator[](1), 1);
    BOOST_TEST_EQ(it->bin(0_c), h.axis(0_c)[1]);
    BOOST_TEST_EQ(it->bin(1_c), h.axis(1_c)[1]);
    BOOST_TEST_EQ(it->value, 1);
    ++it;
    BOOST_TEST_EQ(it->operator[](0), -1);
    BOOST_TEST_EQ(it->operator[](1), 1);
    BOOST_TEST_EQ(it->bin(0), h.axis(0)[-1]);
    BOOST_TEST_EQ(it->bin(1), h.axis(1)[1]);
    BOOST_TEST_EQ(it->value, 0);
    ++it;
    BOOST_TEST(it == ind.end());
  }

  {
    auto ax = axis::variable<>({0.0, 0.1, 0.3, 0.6});
    auto ay = axis::integer<int>(0, 2);
    auto az = ax;
    auto h = make_s(Tag(), std::vector<int>(), ax, ay, az);

    // fill uniformly
    for (auto x : indexed(h)) {
      h(x.bin(0).center(), x.bin(1).value(), x.bin(2).center());
    }

    for (auto x : indexed(h)) {
      BOOST_TEST_EQ(x.density(), x.value / (x.bin(0).width() * x.bin(2).width()));
    }
  }
}

int main() {
  run_tests<static_tag>();
  run_tests<dynamic_tag>();

  return boost::report_errors();
}