//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

// XFAIL: c++98, c++03

// <cuda/std/iterator>
// template <class C> constexpr auto begin(C& c) -> decltype(c.begin());
// template <class C> constexpr auto begin(const C& c) -> decltype(c.begin());
// template <class C> constexpr auto cbegin(const C& c) -> decltype(cuda::std::begin(c)); // C++14
// template <class C> constexpr auto cend(const C& c) -> decltype(cuda::std::end(c));     // C++14
// template <class C> constexpr auto end  (C& c) -> decltype(c.end());
// template <class C> constexpr auto end  (const C& c) -> decltype(c.end());
// template <class E> constexpr reverse_iterator<const E*> rbegin(initializer_list<E> il);
// template <class E> constexpr reverse_iterator<const E*> rend  (initializer_list<E> il);
//
// template <class C> auto constexpr rbegin(C& c) -> decltype(c.rbegin());                 // C++14
// template <class C> auto constexpr rbegin(const C& c) -> decltype(c.rbegin());           // C++14
// template <class C> auto constexpr rend(C& c) -> decltype(c.rend());                     // C++14
// template <class C> constexpr auto rend(const C& c) -> decltype(c.rend());               // C++14
// template <class T, size_t N> reverse_iterator<T*> constexpr rbegin(T (&array)[N]);      // C++14
// template <class T, size_t N> reverse_iterator<T*> constexpr rend(T (&array)[N]);        // C++14
// template <class C> constexpr auto crbegin(const C& c) -> decltype(cuda::std::rbegin(c));      // C++14
// template <class C> constexpr auto crend(const C& c) -> decltype(cuda::std::rend(c));          // C++14
//
//  All of these are constexpr in C++17

#include <cuda/std/iterator>
#include <cuda/std/cassert>
#if defined(_LIBCUDACXX_HAS_VECTOR)
#include <cuda/std/vector>
#endif
#include <cuda/std/array>
#if defined(_LIBCUDACXX_HAS_LIST)
#include <cuda/std/list>
#endif
#include <cuda/std/initializer_list>

#include "test_macros.h"

// cuda::std::array is explicitly allowed to be initialized with A a = { init-list };.
// Disable the missing braces warning for this reason.
#include "disable_missing_braces_warning.h"

template<typename C>
__host__ __device__
void test_const_container( const C & c, typename C::value_type val ) {
    assert ( cuda::std::begin(c)   == c.begin());
    assert (*cuda::std::begin(c)   ==  val );
    assert ( cuda::std::begin(c)   != c.end());
    assert ( cuda::std::end(c)     == c.end());
#if TEST_STD_VER > 11
    assert ( cuda::std::cbegin(c)  == c.cbegin());
    assert ( cuda::std::cbegin(c)  != c.cend());
    assert ( cuda::std::cend(c)    == c.cend());
    assert ( cuda::std::rbegin(c)  == c.rbegin());
    assert ( cuda::std::rbegin(c)  != c.rend());
    assert ( cuda::std::rend(c)    == c.rend());
    assert ( cuda::std::crbegin(c) == c.crbegin());
    assert ( cuda::std::crbegin(c) != c.crend());
    assert ( cuda::std::crend(c)   == c.crend());
#endif
    }

template<typename T>
__host__ __device__
void test_const_container( const cuda::std::initializer_list<T> & c, T val ) {
    assert ( cuda::std::begin(c)   == c.begin());
    assert (*cuda::std::begin(c)   ==  val );
    assert ( cuda::std::begin(c)   != c.end());
    assert ( cuda::std::end(c)     == c.end());
#if TEST_STD_VER > 11
//  initializer_list doesn't have cbegin/cend/rbegin/rend
//  but cuda::std::cbegin(),etc work (b/c they're general fn templates)
//     assert ( cuda::std::cbegin(c)  == c.cbegin());
//     assert ( cuda::std::cbegin(c)  != c.cend());
//     assert ( cuda::std::cend(c)    == c.cend());
//     assert ( cuda::std::rbegin(c)  == c.rbegin());
//     assert ( cuda::std::rbegin(c)  != c.rend());
//     assert ( cuda::std::rend(c)    == c.rend());
//     assert ( cuda::std::crbegin(c) == c.crbegin());
//     assert ( cuda::std::crbegin(c) != c.crend());
//     assert ( cuda::std::crend(c)   == c.crend());
#endif
    }

template<typename C>
__host__ __device__
void test_container( C & c, typename C::value_type val ) {
    assert ( cuda::std::begin(c)   == c.begin());
    assert (*cuda::std::begin(c)   ==  val );
    assert ( cuda::std::begin(c)   != c.end());
    assert ( cuda::std::end(c)     == c.end());
#if TEST_STD_VER > 11
    assert ( cuda::std::cbegin(c)  == c.cbegin());
    assert ( cuda::std::cbegin(c)  != c.cend());
    assert ( cuda::std::cend(c)    == c.cend());
    assert ( cuda::std::rbegin(c)  == c.rbegin());
    assert ( cuda::std::rbegin(c)  != c.rend());
    assert ( cuda::std::rend(c)    == c.rend());
    assert ( cuda::std::crbegin(c) == c.crbegin());
    assert ( cuda::std::crbegin(c) != c.crend());
    assert ( cuda::std::crend(c)   == c.crend());
#endif
    }

template<typename T>
__host__ __device__
void test_container( cuda::std::initializer_list<T> & c, T val ) {
    assert ( cuda::std::begin(c)   == c.begin());
    assert (*cuda::std::begin(c)   ==  val );
    assert ( cuda::std::begin(c)   != c.end());
    assert ( cuda::std::end(c)     == c.end());
#if TEST_STD_VER > 11
//  initializer_list doesn't have cbegin/cend/rbegin/rend
//     assert ( cuda::std::cbegin(c)  == c.cbegin());
//     assert ( cuda::std::cbegin(c)  != c.cend());
//     assert ( cuda::std::cend(c)    == c.cend());
//     assert ( cuda::std::rbegin(c)  == c.rbegin());
//     assert ( cuda::std::rbegin(c)  != c.rend());
//     assert ( cuda::std::rend(c)    == c.rend());
//     assert ( cuda::std::crbegin(c) == c.crbegin());
//     assert ( cuda::std::crbegin(c) != c.crend());
//     assert ( cuda::std::crend(c)   == c.crend());
#endif
    }

template<typename T, size_t Sz>
__host__ __device__
void test_const_array( const T (&array)[Sz] ) {
    assert ( cuda::std::begin(array)  == array );
    assert (*cuda::std::begin(array)  ==  array[0] );
    assert ( cuda::std::begin(array)  != cuda::std::end(array));
    assert ( cuda::std::end(array)    == array + Sz);
#if TEST_STD_VER > 11
    assert ( cuda::std::cbegin(array) == array );
    assert (*cuda::std::cbegin(array) == array[0] );
    assert ( cuda::std::cbegin(array) != cuda::std::cend(array));
    assert ( cuda::std::cend(array)   == array + Sz);
#endif
    }

int main(int, char**) {
#if defined(_LIBCUDACXX_HAS_VECTOR)
    cuda::std::vector<int> v; v.push_back(1);
#endif
#if defined(_LIBCUDACXX_HAS_LIST)
    cuda::std::list<int> l;   l.push_back(2);
#endif
    cuda::std::array<int, 1> a; a[0] = 3;
    cuda::std::initializer_list<int> il = { 4 };

#if defined(_LIBCUDACXX_HAS_VECTOR)
    test_container ( v, 1 );
#endif
#if defined(_LIBCUDACXX_HAS_LIST)
    test_container ( l, 2 );
#endif
    test_container ( a, 3 );
    test_container ( il, 4 );

#if defined(_LIBCUDACXX_HAS_VECTOR)
    test_const_container ( v, 1 );
#endif
#if defined(_LIBCUDACXX_HAS_LIST)
    test_const_container ( l, 2 );
#endif
    test_const_container ( a, 3 );
    test_const_container ( il, 4 );

    static constexpr int arrA [] { 1, 2, 3 };
    test_const_array ( arrA );
#if TEST_STD_VER > 11
    constexpr const int *b = cuda::std::cbegin(arrA);
    constexpr const int *e = cuda::std::cend(arrA);
    static_assert(e - b == 3, "");
#endif

#if TEST_STD_VER > 14
    {
        typedef cuda::std::array<int, 5> C;
        constexpr const C c{0,1,2,3,4};

        static_assert ( c.begin()   == cuda::std::begin(c), "");
        static_assert ( c.cbegin()  == cuda::std::cbegin(c), "");
        static_assert ( c.end()     == cuda::std::end(c), "");
        static_assert ( c.cend()    == cuda::std::cend(c), "");

        static_assert ( c.rbegin()  == cuda::std::rbegin(c), "");
        static_assert ( c.crbegin() == cuda::std::crbegin(c), "");
        static_assert ( c.rend()    == cuda::std::rend(c), "");
        static_assert ( c.crend()   == cuda::std::crend(c), "");

        static_assert ( cuda::std::begin(c)   != cuda::std::end(c), "");
        static_assert ( cuda::std::rbegin(c)  != cuda::std::rend(c), "");
        static_assert ( cuda::std::cbegin(c)  != cuda::std::cend(c), "");
        static_assert ( cuda::std::crbegin(c) != cuda::std::crend(c), "");

        static_assert ( *c.begin()  == 0, "");
        static_assert ( *c.rbegin()  == 4, "");

        static_assert ( *cuda::std::begin(c)   == 0, "" );
        static_assert ( *cuda::std::cbegin(c)  == 0, "" );
        static_assert ( *cuda::std::rbegin(c)  == 4, "" );
        static_assert ( *cuda::std::crbegin(c) == 4, "" );
    }

    {
        static constexpr const int c[] = {0,1,2,3,4};

        static_assert ( *cuda::std::begin(c)   == 0, "" );
        static_assert ( *cuda::std::cbegin(c)  == 0, "" );
        static_assert ( *cuda::std::rbegin(c)  == 4, "" );
        static_assert ( *cuda::std::crbegin(c) == 4, "" );
    }
#endif

  return 0;
}
