// Boost.Geometry
// Unit Test

// Copyright (c) 2021 Oracle and/or its affiliates.
// Contributed and/or modified by Adam Wulkiewicz, on behalf of Oracle

// Use, modification and distribution is subject to 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 <iterator>
#include <vector>

#include <geometry_test_common.hpp>

#include <boost/geometry/core/visit.hpp>
#include <boost/geometry/geometries/adapted/boost_any.hpp>
#include <boost/geometry/geometries/adapted/boost_variant.hpp>
#include <boost/geometry/geometries/adapted/boost_variant2.hpp>
#include <boost/geometry/geometries/adapted/std_any.hpp>
#include <boost/geometry/geometries/adapted/std_variant.hpp>
#include <boost/geometry/geometries/geometries.hpp>
#include <boost/geometry/util/type_traits.hpp>


using point_t = bg::model::point<double, 2, bg::cs::cartesian>;
using linestring_t = bg::model::linestring<point_t>;
using polygon_t = bg::model::polygon<point_t>;


namespace boost { namespace geometry { namespace traits
{

template <>
struct geometry_types<boost::any>
{
    typedef util::type_sequence<point_t, linestring_t, polygon_t> type;
};

#ifndef BOOST_NO_CXX17_HDR_ANY

template <>
struct geometry_types<std::any>
{
    typedef util::type_sequence<point_t, linestring_t, polygon_t> type;
};

#endif // BOOST_NO_CXX17_HDR_ANY

}}}


template <typename DynamicGeometry>
void test_all()
{
    DynamicGeometry dg = point_t(1, 2);
    bg::traits::visit<DynamicGeometry>::apply([](auto g)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g)>::value);
        static_assert(! std::is_const<decltype(g)>::value,
            "Copy of mutable must be visited as mutable.");
    }, dg);
    bg::traits::visit<DynamicGeometry>::apply([](auto const g)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g)>::value);
        static_assert(std::is_const<decltype(g)>::value,
            "Const copy of mutable must be visited as const.");
    }, dg);
    bg::traits::visit<DynamicGeometry>::apply([](auto & g)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g)>::value);
        static_assert(! std::is_const<std::remove_reference_t<decltype(g)>>::value,
            "Reference of mutable must be visited as mutable.");
    }, dg);
    bg::traits::visit<DynamicGeometry>::apply([](auto const& g)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g)>::value);
        static_assert(std::is_const<std::remove_reference_t<decltype(g)>>::value,
            "Const reference of mutable must be visited as const.");
    }, dg);

    DynamicGeometry const cdg = point_t(3, 4);
    bg::traits::visit<DynamicGeometry>::apply([](auto g)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g)>::value);
        static_assert(! std::is_const<decltype(g)>::value,
            "Copy of const must be visited as mutable.");
    }, cdg);
    bg::traits::visit<DynamicGeometry>::apply([](auto const g)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g)>::value);
        static_assert(std::is_const<decltype(g)>::value,
            "Const copy of const must be visited as const.");
    }, cdg);
    bg::traits::visit<DynamicGeometry>::apply([](auto & g)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g)>::value);
        static_assert(std::is_const<std::remove_reference_t<decltype(g)>>::value,
            "Reference of const must be visited as const.");
    }, cdg);
    bg::traits::visit<DynamicGeometry>::apply([](auto const& g)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g)>::value);
        static_assert(std::is_const<std::remove_reference_t<decltype(g)>>::value,
            "Const reference of const must be visited as const.");
    }, cdg);

    bg::traits::visit<DynamicGeometry>::apply([](auto && g)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g)>::value);
        static_assert(! std::is_const<std::remove_reference_t<decltype(g)>>::value,
            "Forward must remain mutable.");
        static_assert(! std::is_rvalue_reference<decltype(g)>::value,
            "Forward must collapse to lvalue reference.");
    }, dg);
    bg::traits::visit<DynamicGeometry>::apply([](auto && g)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g)>::value);
        static_assert(std::is_const<std::remove_reference_t<decltype(g)>>::value,
            "Must remain const.");
        static_assert(! std::is_rvalue_reference<decltype(g)>::value,
            "Forward must collapse to lvalue reference.");
    }, cdg);
    bg::traits::visit<DynamicGeometry>::apply([](auto && g)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g)>::value);
        static_assert(! std::is_const<std::remove_reference_t<decltype(g)>>::value,
            "Moved from must remain mutable.");
        static_assert(std::is_rvalue_reference<decltype(g)>::value,
            "Moved from must dynamic geometry must be visited as rvalue reference.");
    }, std::move(dg));
    bg::traits::visit<DynamicGeometry>::apply([](auto && g)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g)>::value);
        static_assert(std::is_const<std::remove_reference_t<decltype(g)>>::value,
            "Moved from must remain const.");
        static_assert(std::is_rvalue_reference<decltype(g)>::value,
            "Moved from const dynamic geometry must be visited as rvalue reference.");
    }, std::move(cdg));
    bg::traits::visit<DynamicGeometry>::apply([](auto && g)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g)>::value);
        static_assert(! std::is_const<std::remove_reference_t<decltype(g)>>::value,
            "Forwarded temporary dynamic geometry must be mutable.");
        static_assert(std::is_rvalue_reference<decltype(g)>::value,
            "Forwarded temporary dynamic geometry must be visited as rvalue reference.");
    }, DynamicGeometry{ point_t(1, 2) });


    bg::traits::visit<DynamicGeometry, DynamicGeometry>::apply([](auto && g1, auto && g2)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g1)>::value);
        BOOST_CHECK(bg::util::is_point<decltype(g2)>::value);
        static_assert(! std::is_const<std::remove_reference_t<decltype(g1)>>::value,
            "Forward must remain mutable.");
        static_assert(std::is_const<std::remove_reference_t<decltype(g2)>>::value,
            "Forward must remain const.");
        static_assert(! std::is_rvalue_reference<decltype(g1)>::value,
            "Forward must collapse to lvalue reference.");
        static_assert(! std::is_rvalue_reference<decltype(g2)>::value,
            "Forward must collapse to lvalue reference.");
    }, dg, cdg);

    bg::traits::visit<DynamicGeometry, DynamicGeometry>::apply([](auto && g1, auto && g2)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g1)>::value);
        BOOST_CHECK(bg::util::is_point<decltype(g2)>::value);
        static_assert(! std::is_const<std::remove_reference_t<decltype(g1)>>::value,
            "Moved from must remain mutable.");
        static_assert(std::is_const<std::remove_reference_t<decltype(g2)>>::value,
            "Moved from must remain const.");
        static_assert(std::is_rvalue_reference<decltype(g1)>::value,
            "Moved from must be visited as rvalue reference.");
        static_assert(std::is_rvalue_reference<decltype(g2)>::value,
            "Moved from must be visited as rvalue reference.");
    }, std::move(dg), std::move(cdg));

    bg::traits::visit<DynamicGeometry, DynamicGeometry>::apply([](auto && g1, auto && g2)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g2)>::value);
        BOOST_CHECK(bg::util::is_point<decltype(g1)>::value);
        static_assert(! std::is_const<std::remove_reference_t<decltype(g2)>>::value,
            "Forward must remain mutable.");
        static_assert(std::is_const<std::remove_reference_t<decltype(g1)>>::value,
            "Forward must remain const.");
        static_assert(! std::is_rvalue_reference<decltype(g2)>::value,
            "Forward must collapse to lvalue reference.");
        static_assert(! std::is_rvalue_reference<decltype(g1)>::value,
            "Forward must collapse to lvalue reference.");
    }, cdg, dg);

    bg::traits::visit<DynamicGeometry, DynamicGeometry>::apply([](auto && g1, auto && g2)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g2)>::value);
        BOOST_CHECK(bg::util::is_point<decltype(g1)>::value);
        static_assert(! std::is_const<std::remove_reference_t<decltype(g2)>>::value,
            "Moved from must remain mutable.");
        static_assert(std::is_const<std::remove_reference_t<decltype(g1)>>::value,
            "Moved from must remain const.");
        static_assert(std::is_rvalue_reference<decltype(g2)>::value,
            "Moved from must be visited as rvalue reference.");
        static_assert(std::is_rvalue_reference<decltype(g1)>::value,
            "Moved from must be visited as rvalue reference.");
    }, std::move(cdg), std::move(dg));


    std::vector<DynamicGeometry> v = { DynamicGeometry{ point_t(1, 2) } };
    std::vector<DynamicGeometry> const cv = { DynamicGeometry{ point_t(1, 2) } };

    bg::traits::iter_visit<std::vector<DynamicGeometry>>::apply([](auto && g)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g)>::value);
        static_assert(! std::is_const<std::remove_reference_t<decltype(g)>>::value,
            "Forward must remain mutable.");
        static_assert(! std::is_rvalue_reference<decltype(g)>::value,
            "Forward must collapse to lvalue reference.");
    }, v.begin());

    bg::traits::iter_visit<std::vector<DynamicGeometry>>::apply([](auto && g)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g)>::value);
        static_assert(std::is_const<std::remove_reference_t<decltype(g)>>::value,
            "Forward must remain const.");
        static_assert(! std::is_rvalue_reference<decltype(g)>::value,
            "Forward must collapse to lvalue reference.");
    }, cv.begin());

    bg::traits::iter_visit<std::vector<DynamicGeometry>>::apply([](auto && g)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g)>::value);
        static_assert(! std::is_const<std::remove_reference_t<decltype(g)>>::value,
            "Moved from must remain mutable.");
        static_assert(std::is_rvalue_reference<decltype(g)>::value,
            "Moved from must be visited as rvalue reference.");
    }, std::make_move_iterator(v.begin()));

    bg::traits::iter_visit<std::vector<DynamicGeometry>>::apply([](auto && g)
    {
        BOOST_CHECK(bg::util::is_point<decltype(g)>::value);
        static_assert(std::is_const<std::remove_reference_t<decltype(g)>>::value,
            "Moved from must remain const.");
        static_assert(std::is_rvalue_reference<decltype(g)>::value,
            "Moved from must be visited as lvalue reference.");
    }, std::make_move_iterator(cv.begin()));
}

int test_main(int, char* [])
{
    test_all<boost::any>();
    test_all<boost::variant<point_t, linestring_t, polygon_t>>();
    test_all<boost::variant2::variant<point_t, linestring_t, polygon_t>>();

#ifndef BOOST_NO_CXX17_HDR_ANY
    test_all<std::any>();
#endif
#ifndef BOOST_NO_CXX17_HDR_VARIANT
    test_all<std::variant<point_t, linestring_t, polygon_t>>();
#endif

    return 0;
}