// -*- C++ -*-
//===----------------------------------------------------------------------===//
//
// Copyright (C) Intel Corporation
//
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
// This file incorporates work covered by the following copyright and permission
// notice:
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
//
//===----------------------------------------------------------------------===//

// <iterator>

// move_iterator

// template <class U>
//   requires HasAssign<Iter, const U&>
//   move_iterator&
//   operator=(const move_iterator<U>& u);
//
//  constexpr in C++17

#include "support/test_config.h"

#include <oneapi/dpl/iterator>
#include <oneapi/dpl/type_traits>

#include "support/test_iterators.h"
#include "support/utils.h"

template <class It, class U>
bool
test(U u)
{
    const dpl::move_iterator<U> r2(u);
    dpl::move_iterator<It> r1;
    dpl::move_iterator<It>& rr = r1 = r2;
    auto ret = (r1.base() == u);
    ret &= (&rr == &r1);
    return ret;
}

struct Base
{
};
struct Derived : Base
{
};

bool
kernel_test()
{
    sycl::queue deviceQueue = TestUtils::get_test_queue();
    bool ret = true;
    {
        sycl::range<1> numOfItems{1};
        sycl::buffer<bool, 1> buffer1(&ret, numOfItems);
        deviceQueue.submit([&](sycl::handler& cgh) {
            auto ret_access = buffer1.get_access<sycl::access::mode::write>(cgh);
            cgh.single_task<class KernelTest>([=]() {
                Derived d;

                ret_access[0] &= test<input_iterator<Base*>>(input_iterator<Derived*>(&d));
                ret_access[0] &= test<forward_iterator<Base*>>(forward_iterator<Derived*>(&d));
                ret_access[0] &= test<bidirectional_iterator<Base*>>(bidirectional_iterator<Derived*>(&d));
                ret_access[0] &= test<random_access_iterator<const Base*>>(random_access_iterator<Derived*>(&d));
                ret_access[0] &= test<Base*>(&d);

                {
                    using BaseIter = dpl::move_iterator<const Base*>;
                    using DerivedIter = dpl::move_iterator<const Derived*>;
                    constexpr const Derived* p = nullptr;
                    constexpr DerivedIter it1 = dpl::make_move_iterator(p);
                    constexpr BaseIter it2 = (BaseIter{nullptr} = it1);
                    static_assert(it2.base() == p);
                }
            });
        });
    }
    return ret;
}

int
main()
{
    auto ret = kernel_test();
    EXPECT_TRUE(ret, "Wrong result of move_iterator and operator==(...) in kernel_test()");

    return TestUtils::done();
}
