#include <render_2D/plottable/Plottables.hpp>
#include <render_2D/scene/Render_Scene_2D.hpp>
#include <gtest/gtest.h>
#include <array>

namespace {
using namespace aethera;
using namespace aethera::render_2d;

template <typename Object, typename... Args>
std::unique_ptr<Object> build_object(Args&&... args) {
    typename Object::Builder builder(std::forward<Args>(args)...);
    auto result = builder.build();
    if (!result) std::terminate();
    return std::move(result).value();
}

template <typename Axis>
void configure_axis(Axis* axis, Axis_Orientation orientation, Point_F position, Axis_Pixel_Length length, Size canvas) {
    axis->template set<&Abs_Axis::Prop::orientation>(orientation);
    axis->template set<&Abs_Axis::Prop::position>(position);
    axis->template set<&Abs_Axis::Prop::pixel_length>(length);
    axis->template set<&Abs_Axis::Prop::canvas_size>(canvas);
}
}

TEST(plottable_migration, curve_plots_share_partitioned_rendering) {
    using Scene = Impl<Render_Scene_2D>;
    using Frequency = Impl<Frequency_Axis>;
    using Numeric = Impl<Numeric_Axis>;
    using Time = Impl<Time_Axis>;
    using Trace = Impl<Frequency_Trace>;
    using Sweep = Impl<Sweep_Spectrum>;
    initialize_runtime(2);
    const Size canvas{160, 120};
    auto scene = build_object<Scene>();
    auto frequency = build_object<Frequency>();
    auto power = build_object<Numeric>();
    auto time = build_object<Time>();
    auto value = build_object<Numeric>();
    configure_axis(frequency.get(), Axis_Orientation::horizontal, {20.0, 100.0}, 120.0, canvas);
    configure_axis(power.get(), Axis_Orientation::vertical, {20.0, 100.0}, -80.0, canvas);
    configure_axis(time.get(), Axis_Orientation::horizontal, {20.0, 100.0}, 120.0, canvas);
    configure_axis(value.get(), Axis_Orientation::vertical, {20.0, 100.0}, -80.0, canvas);
    frequency->set<&Numeric_Axis::Prop::coordinate_range>(Axis_Range{0.0, 100.0});
    power->set<&Numeric_Axis::Prop::coordinate_range>(Axis_Range{-100.0, 0.0});
    value->set<&Numeric_Axis::Prop::coordinate_range>(Axis_Range{0.0, 100.0});
    auto trace = build_object<Trace>(scene.get(), time.get(), value.get());
    auto sweep = build_object<Sweep>(scene.get(), frequency.get(), power.get());
    trace->set<&Frequency_Trace::Prop::partition_mode>(Plot_Partition_Mode::fixed);
    trace->set<&Frequency_Trace::Prop::partition_count>(2u);
    trace->append_sample(0, 10.0);
    trace->append_sample(1, 50.0);
    trace->append_sample(2, 90.0);
    sweep->set<&Sweep_Spectrum::Prop::frequency_range>(Axis_Range{0.0, 100.0});
    sweep->set<&Sweep_Spectrum::Prop::partition_mode>(Plot_Partition_Mode::fixed);
    sweep->set<&Sweep_Spectrum::Prop::partition_count>(2u);
    const std::array<Plot_Value, 4> block{-90.0, -60.0, -30.0, -10.0};
    sweep->append_block(block);
    scene->set<&Render_Scene_2D::Prop::viewport>(canvas);
    scene->activate_view();
    scene->render();
    EXPECT_EQ(trace->sample_count(), 3u);
    EXPECT_GT(trace->rendered_point_count(), 0u);
    EXPECT_EQ(sweep->stored_block_count(), 1u);
    EXPECT_GT(sweep->rendered_point_count(), 0u);
}

TEST(plottable_migration, raster_plots_share_partitioned_color_blocks) {
    using Scene = Impl<Render_Scene_2D>;
    using Frequency = Impl<Frequency_Axis>;
    using Numeric = Impl<Numeric_Axis>;
    using Time = Impl<Time_Axis>;
    using Glow = Impl<Afterglow>;
    using Fall = Impl<Waterfall>;
    initialize_runtime(2);
    const Size canvas{160, 120};
    auto scene = build_object<Scene>();
    auto frequency = build_object<Frequency>();
    auto power = build_object<Numeric>();
    auto time = build_object<Time>();
    configure_axis(frequency.get(), Axis_Orientation::horizontal, {20.0, 100.0}, 120.0, canvas);
    configure_axis(power.get(), Axis_Orientation::vertical, {20.0, 100.0}, -80.0, canvas);
    configure_axis(time.get(), Axis_Orientation::vertical, {20.0, 100.0}, -80.0, canvas);
    frequency->set<&Numeric_Axis::Prop::coordinate_range>(Axis_Range{0.0, 100.0});
    power->set<&Numeric_Axis::Prop::coordinate_range>(Axis_Range{-100.0, 0.0});
    auto glow = build_object<Glow>(scene.get(), frequency.get(), power.get());
    auto waterfall = build_object<Fall>(scene.get(), frequency.get(), time.get());
    glow->set<&Afterglow::Prop::frequency_range>(Axis_Range{0.0, 100.0});
    glow->set<&Afterglow::Prop::power_range>(Axis_Range{-100.0, 0.0});
    glow->set<&Afterglow::Prop::power_point_size>(16u);
    waterfall->set<&Waterfall::Prop::frequency_range>(Axis_Range{0.0, 100.0});
    waterfall->set<&Waterfall::Prop::power_range>(Axis_Range{-100.0, 0.0});
    const std::array<Plot_Value, 4> row{-90.0, -60.0, -30.0, -10.0};
    glow->append_spectrum(row);
    waterfall->append_row(0, row);
    waterfall->append_row(1, row);
    scene->set<&Render_Scene_2D::Prop::viewport>(canvas);
    scene->activate_view();
    scene->render();
    EXPECT_EQ(glow->history_count(), 1u);
    EXPECT_GT(glow->rendered_cell_count(), 0u);
    EXPECT_EQ(waterfall->row_count(), 2u);
    EXPECT_GT(waterfall->rendered_cell_count(), 0u);
}

TEST(plottable_migration, direct_overlay_and_constellation_build_and_render) {
    using Scene = Impl<Render_Scene_2D>;
    using Numeric = Impl<Numeric_Axis>;
    using Diagram = Impl<Constellation_Diagram>;
    using Overlay = Impl<Selection_Rectangle_Overlay>;
    initialize_runtime(2);
    const Size canvas{160, 120};
    auto scene = build_object<Scene>();
    auto horizontal = build_object<Numeric>();
    auto vertical = build_object<Numeric>();
    configure_axis(horizontal.get(), Axis_Orientation::horizontal, {20.0, 100.0}, 120.0, canvas);
    configure_axis(vertical.get(), Axis_Orientation::vertical, {20.0, 100.0}, -80.0, canvas);
    horizontal->set<&Numeric_Axis::Prop::coordinate_range>(Axis_Range{-1.0, 1.0});
    vertical->set<&Numeric_Axis::Prop::coordinate_range>(Axis_Range{-1.0, 1.0});
    auto diagram = build_object<Diagram>(scene.get(), horizontal.get(), vertical.get());
    auto overlay = build_object<Overlay>(scene.get(), horizontal.get(), vertical.get());
    diagram->set<&Constellation_Diagram::Prop::i_range>(Axis_Range{-1.0, 1.0});
    diagram->set<&Constellation_Diagram::Prop::q_range>(Axis_Range{-1.0, 1.0});
    diagram->append_point({0.25, -0.25});
    scene->set<&Render_Scene_2D::Prop::viewport>(canvas);
    scene->activate_view();
    scene->render();
    EXPECT_EQ(diagram->point_count(), 1u);
    EXPECT_EQ(overlay->read_state<Renderable::Base_Tag>().prepare_task_count, 0u);
    EXPECT_EQ(overlay->read_state<Renderable::Base_Tag>().paint_task_count, 1u);
}