#include <mbgl/test/util.hpp>

#include <mbgl/gfx/headless_frontend.hpp>
#include <mbgl/map/map.hpp>
#include <mbgl/map/map_options.hpp>
#include <mbgl/storage/resource_options.hpp>
#include <mbgl/style/layers/fill_layer.hpp>
#include <mbgl/style/style.hpp>
#include <mbgl/util/io.hpp>
#include <mbgl/util/mat4.hpp>
#include <mbgl/util/run_loop.hpp>

#if MLN_DRAWABLE_RENDERER

#include <mbgl/style/layers/custom_drawable_layer.hpp>
#include <mbgl/util/constants.hpp>

#include <memory>
#include <cmath>

class TestDrawableLayer : public mbgl::style::CustomDrawableLayerHost {
public:
    void initialize() override {}

    void update(Interface& interface) override {
        // if we have built our drawable(s) already, either update or skip
        if (interface.getDrawableCount()) return;

        // set tile
        interface.setTileID({11, 327, 791});

        // add polylines
        {
            using namespace mbgl;

            constexpr auto numLines = 6;
            Interface::LineOptions options[numLines]{
                {/*color=*/Color::red(),
                 /*blur=*/0.0f,
                 /*opacity=*/1.0f,
                 /*gapWidth=*/0.0f,
                 /*offset=*/0.0f,
                 /*width=*/8.0f,
                 {}},
                {/*color=*/Color::blue(),
                 /*blur=*/4.0f,
                 /*opacity=*/1.0f,
                 /*gapWidth=*/2.0f,
                 /*offset=*/-1.0f,
                 /*width=*/4.0f,
                 {}},
                {/*color=*/Color(1.f, 0.5f, 0, 0.5f),
                 /*blur=*/16.0f,
                 /*opacity=*/1.0f,
                 /*gapWidth=*/1.0f,
                 /*offset=*/2.0f,
                 /*width=*/16.0f,
                 {}},
                {/*color=*/Color(1.f, 1.f, 0, 0.3f),
                 /*blur=*/2.0f,
                 /*opacity=*/1.0f,
                 /*gapWidth=*/1.0f,
                 /*offset=*/-2.0f,
                 /*width=*/2.0f,
                 {}},
                {/*color=*/Color::black(),
                 /*blur=*/0.5f,
                 /*opacity=*/0.5f,
                 /*gapWidth=*/1.0f,
                 /*offset=*/0.5f,
                 /*width=*/0.5f,
                 {}},
                {/*color=*/Color(1.f, 0, 1.f, 0.2f),
                 /*blur=*/24.0f,
                 /*opacity=*/0.5f,
                 /*gapWidth=*/1.0f,
                 /*offset=*/-5.0f,
                 /*width=*/24.0f,
                 {}},
            };
            for (auto& opt : options) {
                opt.geometry.beginCap = style::LineCapType::Round;
                opt.geometry.endCap = style::LineCapType::Round;
                opt.geometry.joinType = style::LineJoinType::Round;
            }

            constexpr auto numPoints = 100;
            GeometryCoordinates polyline;
            for (auto ipoint{0}; ipoint < numPoints; ++ipoint) {
                polyline.emplace_back(ipoint * util::EXTENT / numPoints,
                                      std::sin(ipoint * 2 * M_PI / numPoints) * util::EXTENT / numLines / 2.f);
            }

            for (auto index{0}; index < numLines; ++index) {
                for (auto& p : polyline) {
                    p.y += util::EXTENT / numLines;
                }

                // set property values
                interface.setLineOptions(options[index]);

                // add polyline
                interface.addPolyline(polyline);
            }
        }

        // add fill polygon
        {
            using namespace mbgl;

            GeometryCollection geometry{
                {
                    // ring 1
                    {static_cast<int16_t>(util::EXTENT * 0.1f), static_cast<int16_t>(util::EXTENT * 0.2f)},
                    {static_cast<int16_t>(util::EXTENT * 0.5f), static_cast<int16_t>(util::EXTENT * 0.5f)},
                    {static_cast<int16_t>(util::EXTENT * 0.7f), static_cast<int16_t>(util::EXTENT * 0.5f)},
                    {static_cast<int16_t>(util::EXTENT * 0.5f), static_cast<int16_t>(util::EXTENT * 1.0f)},
                    {static_cast<int16_t>(util::EXTENT * 0.0f), static_cast<int16_t>(util::EXTENT * 0.5f)},
                    {static_cast<int16_t>(util::EXTENT * 0.1f), static_cast<int16_t>(util::EXTENT * 0.2f)},
                },
                {
                    // ring 2
                    {static_cast<int16_t>(util::EXTENT * 0.1f), static_cast<int16_t>(util::EXTENT * 0.25f)},
                    {static_cast<int16_t>(util::EXTENT * 0.15f), static_cast<int16_t>(util::EXTENT * 0.5f)},
                    {static_cast<int16_t>(util::EXTENT * 0.25f), static_cast<int16_t>(util::EXTENT * 0.45f)},
                    {static_cast<int16_t>(util::EXTENT * 0.1f), static_cast<int16_t>(util::EXTENT * 0.25f)},
                },
            };

            // set properties
            interface.setFillOptions({/*color=*/Color::green(), /*opacity=*/0.5f});

            // add fill
            interface.addFill(geometry);
        }

        // finish
        interface.finish();
    }

    void deinitialize() override {}
};

TEST(CustomDrawableLayer, Basic) {
    using namespace mbgl;
    using namespace mbgl::style;

    util::RunLoop loop;

    HeadlessFrontend frontend{1};
    Map map(frontend,
            MapObserver::nullObserver(),
            MapOptions().withMapMode(MapMode::Static).withSize(frontend.getSize()),
            ResourceOptions().withCachePath(":memory:").withAssetPath("test/fixtures/api/assets"));

    // load style
    map.getStyle().loadJSON(util::read_file("test/fixtures/api/water.json"));
    map.jumpTo(CameraOptions().withCenter(LatLng{37.8, -122.4426032}).withZoom(10.0));

    // add fill layer
    auto layer = std::make_unique<FillLayer>("landcover", "mapbox");
    layer->setSourceLayer("landcover");
    layer->setFillColor(Color{1.0, 1.0, 0.0, 1.0});
    map.getStyle().addLayer(std::move(layer));

    // add custom drawable layer
    map.getStyle().addLayer(
        std::make_unique<CustomDrawableLayer>("custom-drawable", std::make_unique<TestDrawableLayer>()));

    // render and test
    test::checkImage("test/fixtures/custom_drawable_layer/basic", frontend.render(map).image, 0.0006, 0.1);
}

#endif // MLN_DRAWABLE_RENDERER