// Boost.Geometry (aka GGL, Generic Geometry Library)

// Copyright (c) 2022, Oracle and/or its affiliates.

// Contributed and/or modified by Adam Wulkiewicz, on behalf of Oracle

// Licensed under the Boost Software License version 1.0.
// http://www.boost.org/users/license.html


#include <geometry_test_common.hpp>

#include <boost/geometry/algorithms/equals.hpp>
#include <boost/geometry/algorithms/intersection.hpp>
//#include <boost/geometry/geometries/adapted/boost_variant2.hpp>
#include <boost/geometry/geometries/geometries.hpp>
#include <boost/geometry/io/wkt/wkt.hpp>


using pt_t = bg::model::point<double, 2, bg::cs::cartesian>;
using ls_t = bg::model::linestring<pt_t>;
using po_t = bg::model::polygon<pt_t>;
using mpt_t = bg::model::multi_point<pt_t>;
using mls_t = bg::model::multi_linestring<ls_t>;
using mpo_t = bg::model::multi_polygon<po_t>;
using var_t = boost::variant<pt_t, ls_t, po_t, mpt_t, mls_t, mpo_t>;
//using var_t = boost::variant2::variant<pt_t, ls_t, po_t, mpt_t, mls_t, mpo_t>;
using gc_t = bg::model::geometry_collection<var_t>;

template <typename GC1, typename GC2>
inline void check_one(GC1 const& out, GC2 const& expected)
{
    std::size_t out_size = boost::size(out);
    std::size_t expected_size = boost::size(expected);
    BOOST_CHECK(out_size == expected_size);
    if (out_size == expected_size)
    {
        for (std::size_t i = 0; i < out_size; ++i)
        {
            bg::traits::iter_visit<GC1>::apply([&](auto const& o)
            {
                bg::traits::iter_visit<GC2>::apply([&](auto const& e)
                {
                    using o_t = typename bg::util::remove_cref<decltype(o)>::type;
                    using e_t = typename bg::util::remove_cref<decltype(e)>::type;
                    int oid = bg::geometry_id<o_t>::value;
                    int eid = bg::geometry_id<e_t>::value;
                    BOOST_CHECK_MESSAGE(oid == eid, oid << " and " << eid);
                    bool elements_equal = bg::equals(o, e);
                    BOOST_CHECK_MESSAGE(elements_equal, bg::wkt(o) << " and " << bg::wkt(e));
                }, expected.begin() + i);
            }, out.begin() + i);
        }
    }
}


void test_gc_gc_gc()
{
    const char* gc1_cstr = "GEOMETRYCOLLECTION("
        "MULTIPOLYGON(((0 0, 0 10, 10 10, 10 0, 0 0), (1 1, 5 1, 5 5, 1 5, 1 1)), ((3 3, 3 4, 4 4, 4 3, 3 3))),"
        "POLYGON((3 3, 3 10, 10 10, 10 3, 3 3)),"
        "MULTILINESTRING((0 0, 5 5), (5 5, 11 5), (5 5, 5 11)),"
        "LINESTRING(0 0, 2 2, 2 11),"
        "MULTIPOINT(0 0, 1 1, 2 2, 3 3, 4 4, 5 5, 6 6, 7 7, 8 8, 9 9, 10 10),"
        "POINT(11 11)"
    ")";
    const char* gc2_cstr = "GEOMETRYCOLLECTION("
        "POLYGON((4 0, 4 10, 6 10, 6 0, 4 0)),"
        "POLYGON((0 4, 0 6, 10 6, 10 4, 0 4)),"
        "POLYGON((3 3, 3 5, 5 5, 5 3, 3 3)),"
        "MULTILINESTRING((1 1, 6 6), (5 1, 5 9), (2 2, 9 2)),"
        "MULTIPOINT(9 9, 11 11, 1 9),"
        "POINT(3 2)"
    ")";
    const char* expected_cstr = "GEOMETRYCOLLECTION("
        "MULTIPOINT((1 1),(9 9),(1 9),(5 5),(5 5),(5 5),(5 5),(2 2),(3 3),(4 4),(6 6),(11 11)),"
        "MULTILINESTRING((3 5,5 5,5 3),(4 3,4 4),(4 4,3 4),(3 3,4 4),(5 5,6 6),(5 1,5 3),(5 5,5 9),(5 2,9 2),(4 4,5 5),(5 5,6 5),(5 9,5 10),(6 5,10 5),(1 1,3 3),(2 4,2 6)),"
        "POLYGON((6 6,10 6,10 4,6 4,6 3,6 0,4 0,4 1,5 1,5 3,4 3,3 3,3 5,1 5,1 4,0 4,0 6,4 6,4 10,6 10,6 6))"
    ")";

    gc_t gc1, gc2, expected;
    bg::read_wkt(gc1_cstr, gc1);
    bg::read_wkt(gc2_cstr, gc2);
    bg::read_wkt(expected_cstr, expected);

    gc_t out1, out2;
    bg::intersection(gc1, gc2, out1);
    bg::intersection(gc2, gc1, out2);

    check_one(out1, expected);
    check_one(out2, expected);
}

void test_gc_g_gc()
{
    const char* gc_cstr = "GEOMETRYCOLLECTION("
        "MULTIPOLYGON(((0 0, 0 10, 10 10, 10 0, 0 0), (1 1, 5 1, 5 5, 1 5, 1 1)), ((3 3, 3 4, 4 4, 4 3, 3 3))),"
        "POLYGON((3 3, 3 10, 10 10, 10 3, 3 3)),"
        "MULTILINESTRING((0 0, 5 5), (5 5, 11 5), (5 5, 5 11)),"
        "LINESTRING(0 0, 2 2, 2 11),"
        "MULTIPOINT(0 0, 1 1, 2 2, 3 3, 4 4, 5 5, 6 6, 7 7, 8 8, 9 9, 10 10),"
        "POINT(11 11)"
    ")";
    const char* po_cstr = "POLYGON((4 0, 4 10, 6 10, 6 0, 4 0))";
    const char* expected_cstr = "GEOMETRYCOLLECTION("
        "MULTIPOINT((4 4),(5 5),(6 6)),"
        "MULTILINESTRING((4 3,4 4),(4 4,5 5),(5 5,6 5),(5 5,5 10)),"
        "POLYGON((6 10,6 3,6 0,4 0,4 1,5 1,5 3,4 3,4 10,6 10))"
    ")";

    gc_t gc, expected;
    po_t po;
    bg::read_wkt(gc_cstr, gc);
    bg::read_wkt(po_cstr, po);
    bg::read_wkt(expected_cstr, expected);

    gc_t out;
    bg::intersection(gc, po, out);

    check_one(out, expected);
}

void test_g_gc_gc()
{
    const char* mpo_cstr = "MULTIPOLYGON(((0 0, 0 10, 10 10, 10 0, 0 0), (1 1, 5 1, 5 5, 1 5, 1 1)), ((3 3, 3 4, 4 4, 4 3, 3 3)))";
    const char* gc_cstr = "GEOMETRYCOLLECTION("
        "POLYGON((4 0, 4 10, 6 10, 6 0, 4 0)),"
        "POLYGON((0 4, 0 6, 10 6, 10 4, 0 4)),"
        "POLYGON((3 3, 3 5, 5 5, 5 3, 3 3)),"
        "MULTILINESTRING((1 1, 6 6), (5 1, 5 9), (2 2, 9 2)),"
        "MULTIPOINT(9 9, 11 11, 1 9),"
        "POINT(3 2)"
    ")";
    const char* expected_cstr = "GEOMETRYCOLLECTION("
        "MULTIPOINT((1 1),(9 9),(1 9)),"
        "MULTILINESTRING((3 5,5 5,5 3),(4 3,4 4),(4 4,3 4),(3 3,4 4),(5 5,6 6),(5 1,5 3),(5 5,5 9),(5 2,9 2)),"
        "MULTIPOLYGON(((3 4,4 4,4 3,3 3,3 4)),((6 6,10 6,10 4,6 4,6 0,4 0,4 1,5 1,5 5,4 5,1 5,1 4,0 4,0 6,4 6,4 10,6 10,6 6)))"
    ")";

    mpo_t mpo;
    gc_t gc, expected;    
    bg::read_wkt(mpo_cstr, mpo);
    bg::read_wkt(gc_cstr, gc);    
    bg::read_wkt(expected_cstr, expected);

    gc_t out;
    bg::intersection(mpo, gc, out);

    check_one(out, expected);
}

void test_g_g_gc()
{
    const char* mpo_cstr = "MULTIPOLYGON(((0 0, 0 10, 10 10, 10 0, 0 0), (1 1, 5 1, 5 5, 1 5, 1 1)), ((3 3, 3 4, 4 4, 4 3, 3 3)))";
    const char* po_cstr = "POLYGON((4 0, 4 10, 6 10, 6 0, 4 0))";
    const char* expected_cstr = "GEOMETRYCOLLECTION("
        "LINESTRING(4 3,4 4),"
        "POLYGON((4 10,6 10,6 0,4 0,4 1,5 1,5 5,4 5,4 10))"
    ")";

    mpo_t mpo;
    po_t po;
    gc_t expected;    
    bg::read_wkt(mpo_cstr, mpo);
    bg::read_wkt(po_cstr, po);
    bg::read_wkt(expected_cstr, expected);

    gc_t out;
    bg::intersection(mpo, po, out);

    check_one(out, expected);
}

int test_main(int, char* [])
{
    test_gc_gc_gc();
    test_gc_g_gc();
    test_g_gc_gc();
    test_g_g_gc();

    return 0;
}