/*
//@HEADER
// ************************************************************************
//
//                        Kokkos v. 3.0
//       Copyright (2020) National Technology & Engineering
//               Solutions of Sandia, LLC (NTESS).
//
// Under the terms of Contract DE-NA0003525 with NTESS,
// the U.S. Government retains certain rights in this software.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
// 1. Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
//
// 2. Redistributions in binary form must reproduce the above copyright
// notice, this list of conditions and the following disclaimer in the
// documentation and/or other materials provided with the distribution.
//
// 3. Neither the name of the Corporation nor the names of the
// contributors may be used to endorse or promote products derived from
// this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY NTESS "AS IS" AND ANY
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL NTESS OR THE
// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
// Questions? Contact Christian R. Trott (crtrott@sandia.gov)
//
// ************************************************************************
//@HEADER
*/

#include <gtest/gtest.h>

#include <stdexcept>
#include <sstream>
#include <iostream>

#include <Kokkos_Core.hpp>

namespace Test {

void test_is_specialization_of() {
  using Kokkos::Impl::is_specialization_of;
  static_assert(is_specialization_of<Kokkos::pair<float, int>, Kokkos::pair>{},
                "");
  static_assert(!is_specialization_of<Kokkos::View<int*>, Kokkos::pair>{}, "");
  static_assert(is_specialization_of<Kokkos::View<int*>, Kokkos::View>{}, "");
  // NOTE Not removing cv-qualifiers
  static_assert(!is_specialization_of<Kokkos::View<int*> const, Kokkos::View>{},
                "");
  // NOTE Would not compile because Kokkos::Array takes a non-type template
  // parameter
  // static_assert(is_specialization_of<Kokkos::Array<int, 4>, Kokkos::Array>{},
  // "");
  // But this is fine of course
  static_assert(!is_specialization_of<Kokkos::Array<float, 2>, Kokkos::pair>{},
                "");
}

inline void test_utilities() {
  using namespace Kokkos::Impl;

  {
    using i = integer_sequence<int>;
    using j = make_integer_sequence<int, 0>;

    static_assert(std::is_same<i, j>::value, "Error: make_integer_sequence");
    static_assert(i::size() == 0u, "Error: integer_sequence.size()");
  }

  {
    using i = integer_sequence<int, 0>;
    using j = make_integer_sequence<int, 1>;

    static_assert(std::is_same<i, j>::value, "Error: make_integer_sequence");
    static_assert(i::size() == 1u, "Error: integer_sequence.size()");

    static_assert(integer_sequence_at<0, i>::value == 0,
                  "Error: integer_sequence_at");

    static_assert(at(0, i{}) == 0, "Error: at(unsigned, integer_sequence)");
  }

  {
    using i = integer_sequence<int, 0, 1>;
    using j = make_integer_sequence<int, 2>;

    static_assert(std::is_same<i, j>::value, "Error: make_integer_sequence");
    static_assert(i::size() == 2u, "Error: integer_sequence.size()");

    static_assert(integer_sequence_at<0, i>::value == 0,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<1, i>::value == 1,
                  "Error: integer_sequence_at");

    static_assert(at(0, i{}) == 0, "Error: at(unsigned, integer_sequence)");
    static_assert(at(1, i{}) == 1, "Error: at(unsigned, integer_sequence)");
  }

  {
    using i = integer_sequence<int, 0, 1, 2>;
    using j = make_integer_sequence<int, 3>;

    static_assert(std::is_same<i, j>::value, "Error: make_integer_sequence");
    static_assert(i::size() == 3u, "Error: integer_sequence.size()");

    static_assert(integer_sequence_at<0, i>::value == 0,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<1, i>::value == 1,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<2, i>::value == 2,
                  "Error: integer_sequence_at");

    static_assert(at(0, i{}) == 0, "Error: at(unsigned, integer_sequence)");
    static_assert(at(1, i{}) == 1, "Error: at(unsigned, integer_sequence)");
    static_assert(at(2, i{}) == 2, "Error: at(unsigned, integer_sequence)");
  }

  {
    using i = integer_sequence<int, 0, 1, 2, 3>;
    using j = make_integer_sequence<int, 4>;

    static_assert(std::is_same<i, j>::value, "Error: make_integer_sequence");
    static_assert(i::size() == 4u, "Error: integer_sequence.size()");

    static_assert(integer_sequence_at<0, i>::value == 0,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<1, i>::value == 1,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<2, i>::value == 2,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<3, i>::value == 3,
                  "Error: integer_sequence_at");

    static_assert(at(0, i{}) == 0, "Error: at(unsigned, integer_sequence)");
    static_assert(at(1, i{}) == 1, "Error: at(unsigned, integer_sequence)");
    static_assert(at(2, i{}) == 2, "Error: at(unsigned, integer_sequence)");
    static_assert(at(3, i{}) == 3, "Error: at(unsigned, integer_sequence)");
  }

  {
    using i = integer_sequence<int, 0, 1, 2, 3, 4>;
    using j = make_integer_sequence<int, 5>;

    static_assert(std::is_same<i, j>::value, "Error: make_integer_sequence");
    static_assert(i::size() == 5u, "Error: integer_sequence.size()");

    static_assert(integer_sequence_at<0, i>::value == 0,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<1, i>::value == 1,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<2, i>::value == 2,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<3, i>::value == 3,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<4, i>::value == 4,
                  "Error: integer_sequence_at");

    static_assert(at(0, i{}) == 0, "Error: at(unsigned, integer_sequence)");
    static_assert(at(1, i{}) == 1, "Error: at(unsigned, integer_sequence)");
    static_assert(at(2, i{}) == 2, "Error: at(unsigned, integer_sequence)");
    static_assert(at(3, i{}) == 3, "Error: at(unsigned, integer_sequence)");
    static_assert(at(4, i{}) == 4, "Error: at(unsigned, integer_sequence)");
  }

  {
    using i = integer_sequence<int, 0, 1, 2, 3, 4, 5>;
    using j = make_integer_sequence<int, 6>;

    static_assert(std::is_same<i, j>::value, "Error: make_integer_sequence");
    static_assert(i::size() == 6u, "Error: integer_sequence.size()");

    static_assert(integer_sequence_at<0, i>::value == 0,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<1, i>::value == 1,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<2, i>::value == 2,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<3, i>::value == 3,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<4, i>::value == 4,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<5, i>::value == 5,
                  "Error: integer_sequence_at");

    static_assert(at(0, i{}) == 0, "Error: at(unsigned, integer_sequence)");
    static_assert(at(1, i{}) == 1, "Error: at(unsigned, integer_sequence)");
    static_assert(at(2, i{}) == 2, "Error: at(unsigned, integer_sequence)");
    static_assert(at(3, i{}) == 3, "Error: at(unsigned, integer_sequence)");
    static_assert(at(4, i{}) == 4, "Error: at(unsigned, integer_sequence)");
    static_assert(at(5, i{}) == 5, "Error: at(unsigned, integer_sequence)");
  }

  {
    using i = integer_sequence<int, 0, 1, 2, 3, 4, 5, 6>;
    using j = make_integer_sequence<int, 7>;

    static_assert(std::is_same<i, j>::value, "Error: make_integer_sequence");
    static_assert(i::size() == 7u, "Error: integer_sequence.size()");

    static_assert(integer_sequence_at<0, i>::value == 0,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<1, i>::value == 1,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<2, i>::value == 2,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<3, i>::value == 3,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<4, i>::value == 4,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<5, i>::value == 5,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<6, i>::value == 6,
                  "Error: integer_sequence_at");

    static_assert(at(0, i{}) == 0, "Error: at(unsigned, integer_sequence)");
    static_assert(at(1, i{}) == 1, "Error: at(unsigned, integer_sequence)");
    static_assert(at(2, i{}) == 2, "Error: at(unsigned, integer_sequence)");
    static_assert(at(3, i{}) == 3, "Error: at(unsigned, integer_sequence)");
    static_assert(at(4, i{}) == 4, "Error: at(unsigned, integer_sequence)");
    static_assert(at(5, i{}) == 5, "Error: at(unsigned, integer_sequence)");
    static_assert(at(6, i{}) == 6, "Error: at(unsigned, integer_sequence)");
  }

  {
    using i = integer_sequence<int, 0, 1, 2, 3, 4, 5, 6, 7>;
    using j = make_integer_sequence<int, 8>;

    static_assert(std::is_same<i, j>::value, "Error: make_integer_sequence");
    static_assert(i::size() == 8u, "Error: integer_sequence.size()");

    static_assert(integer_sequence_at<0, i>::value == 0,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<1, i>::value == 1,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<2, i>::value == 2,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<3, i>::value == 3,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<4, i>::value == 4,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<5, i>::value == 5,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<6, i>::value == 6,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<7, i>::value == 7,
                  "Error: integer_sequence_at");

    static_assert(at(0, i{}) == 0, "Error: at(unsigned, integer_sequence)");
    static_assert(at(1, i{}) == 1, "Error: at(unsigned, integer_sequence)");
    static_assert(at(2, i{}) == 2, "Error: at(unsigned, integer_sequence)");
    static_assert(at(3, i{}) == 3, "Error: at(unsigned, integer_sequence)");
    static_assert(at(4, i{}) == 4, "Error: at(unsigned, integer_sequence)");
    static_assert(at(5, i{}) == 5, "Error: at(unsigned, integer_sequence)");
    static_assert(at(6, i{}) == 6, "Error: at(unsigned, integer_sequence)");
    static_assert(at(7, i{}) == 7, "Error: at(unsigned, integer_sequence)");
  }

  {
    using i = integer_sequence<int, 0, 1, 2, 3, 4, 5, 6, 7, 8>;
    using j = make_integer_sequence<int, 9>;

    static_assert(std::is_same<i, j>::value, "Error: make_integer_sequence");
    static_assert(i::size() == 9u, "Error: integer_sequence.size()");

    static_assert(integer_sequence_at<0, i>::value == 0,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<1, i>::value == 1,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<2, i>::value == 2,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<3, i>::value == 3,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<4, i>::value == 4,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<5, i>::value == 5,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<6, i>::value == 6,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<7, i>::value == 7,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<8, i>::value == 8,
                  "Error: integer_sequence_at");

    static_assert(at(0, i{}) == 0, "Error: at(unsigned, integer_sequence)");
    static_assert(at(1, i{}) == 1, "Error: at(unsigned, integer_sequence)");
    static_assert(at(2, i{}) == 2, "Error: at(unsigned, integer_sequence)");
    static_assert(at(3, i{}) == 3, "Error: at(unsigned, integer_sequence)");
    static_assert(at(4, i{}) == 4, "Error: at(unsigned, integer_sequence)");
    static_assert(at(5, i{}) == 5, "Error: at(unsigned, integer_sequence)");
    static_assert(at(6, i{}) == 6, "Error: at(unsigned, integer_sequence)");
    static_assert(at(7, i{}) == 7, "Error: at(unsigned, integer_sequence)");
    static_assert(at(8, i{}) == 8, "Error: at(unsigned, integer_sequence)");
  }

  {
    using i = integer_sequence<int, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9>;
    using j = make_integer_sequence<int, 10>;

    static_assert(std::is_same<i, j>::value, "Error: make_integer_sequence");
    static_assert(i::size() == 10u, "Error: integer_sequence.size()");

    static_assert(integer_sequence_at<0, i>::value == 0,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<1, i>::value == 1,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<2, i>::value == 2,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<3, i>::value == 3,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<4, i>::value == 4,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<5, i>::value == 5,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<6, i>::value == 6,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<7, i>::value == 7,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<8, i>::value == 8,
                  "Error: integer_sequence_at");
    static_assert(integer_sequence_at<9, i>::value == 9,
                  "Error: integer_sequence_at");

    static_assert(at(0, i{}) == 0, "Error: at(unsigned, integer_sequence)");
    static_assert(at(1, i{}) == 1, "Error: at(unsigned, integer_sequence)");
    static_assert(at(2, i{}) == 2, "Error: at(unsigned, integer_sequence)");
    static_assert(at(3, i{}) == 3, "Error: at(unsigned, integer_sequence)");
    static_assert(at(4, i{}) == 4, "Error: at(unsigned, integer_sequence)");
    static_assert(at(5, i{}) == 5, "Error: at(unsigned, integer_sequence)");
    static_assert(at(6, i{}) == 6, "Error: at(unsigned, integer_sequence)");
    static_assert(at(7, i{}) == 7, "Error: at(unsigned, integer_sequence)");
    static_assert(at(8, i{}) == 8, "Error: at(unsigned, integer_sequence)");
    static_assert(at(9, i{}) == 9, "Error: at(unsigned, integer_sequence)");
  }

  {
    using i  = make_integer_sequence<int, 5>;
    using r  = reverse_integer_sequence<i>;
    using gr = integer_sequence<int, 4, 3, 2, 1, 0>;

    static_assert(std::is_same<r, gr>::value,
                  "Error: reverse_integer_sequence");
  }

  {
    using s = make_integer_sequence<int, 10>;
    using e = exclusive_scan_integer_sequence<s>;
    using i = inclusive_scan_integer_sequence<s>;

    using ge = integer_sequence<int, 0, 0, 1, 3, 6, 10, 15, 21, 28, 36>;
    using gi = integer_sequence<int, 0, 1, 3, 6, 10, 15, 21, 28, 36, 45>;

    static_assert(e::value == 45, "Error: scan value");
    static_assert(i::value == 45, "Error: scan value");

    static_assert(std::is_same<e::type, ge>::value, "Error: exclusive_scan");
    static_assert(std::is_same<i::type, gi>::value, "Error: inclusive_scan");
  }
}

template <std::size_t... Idxs, class... Args>
std::size_t do_comma_emulation_test(std::integer_sequence<std::size_t, Idxs...>,
                                    Args... args) {
  // Count the bugs, since ASSERT_EQ is a statement and not an expression
  std::size_t bugs = 0;
  // Ensure in-order evaluation
  std::size_t i = 0;
  KOKKOS_IMPL_FOLD_COMMA_OPERATOR(bugs += std::size_t(Idxs != i++) /*, ...*/);
  // Ensure expansion of multiple packs works
  KOKKOS_IMPL_FOLD_COMMA_OPERATOR(bugs += std::size_t(Idxs != args) /*, ...*/);
  return bugs;
}

TEST(utilities, comma_operator_emulation) {
  ASSERT_EQ(
      0, do_comma_emulation_test(std::make_index_sequence<5>{}, 0, 1, 2, 3, 4));
}

}  // namespace Test