#include "viz/line_plot.h"
#include "viz/plot_static.h"
#include "implot.h"

namespace {

template <typename T>
void draw_line(const char* title, const T* xs, const T* ys, int count, float height) {
    if (count <= 0) return;

    T x_min = xs[0], x_max = xs[0];
    T y_min = ys[0], y_max = ys[0];
    for (int i = 1; i < count; i++) {
        if (xs[i] < x_min) x_min = xs[i];
        if (xs[i] > x_max) x_max = xs[i];
        if (ys[i] < y_min) y_min = ys[i];
        if (ys[i] > y_max) y_max = ys[i];
    }
    double dy = static_cast<double>(y_max) - static_cast<double>(y_min);
    if (dy < 1e-9) dy = 1.0;
    double y_lo = static_cast<double>(y_min) - dy * 0.05;
    double y_hi = static_cast<double>(y_max) + dy * 0.05;

    const ImVec2 plot_size(-1.0f, height > 0.0f ? height : 200.0f);

    if (ImPlot::BeginPlot(title, plot_size, plot_static::kPlotFlags)) {
        ImPlot::SetupAxes(nullptr, nullptr, plot_static::kAxisFlags, plot_static::kAxisFlags);
        ImPlot::SetupAxisLimits(ImAxis_X1, static_cast<double>(x_min),
                                static_cast<double>(x_max), ImPlotCond_Always);
        ImPlot::SetupAxisLimits(ImAxis_Y1, y_lo, y_hi, ImPlotCond_Always);
        ImPlot::PlotLine("##data", xs, ys, count);
        ImPlot::EndPlot();
    }
}

} // namespace

void line_plot(const char* title, const float* xs, const float* ys, int count, float height) {
    draw_line<float>(title, xs, ys, count, height);
}

void line_plot(const char* title, const double* xs, const double* ys, int count, float height) {
    draw_line<double>(title, xs, ys, count, height);
}