#include <algorithm>
#include <vector>
#include "svg.h"
#include <iostream>
#include <string>

using namespace std;

void svg_begin(double width, double height) {
    cout << "<?xml version='1.0' encoding='UTF-8'?>\n";
    cout << "<svg ";
    cout << "width='" << width << "' ";
    cout << "height='" << height << "' ";
    cout << "viewBox='0 0 " << width << " " << height << "' ";
    cout << "xmlns='http://www.w3.org/2000/svg'>\n";
}

void svg_end() {
    cout << "</svg>\n";
}

void svg_text(double left, double baseline, const string &text) {
    cout << "<text x='" << left << "' y='" << baseline
         << "' font-family='Verdana' font-size='16'>" << text << "</text>\n";
}

void svg_rect(double x, double y, double width, double height,
              string stroke = "black", string fill = "black") {
    cout << "<rect x='" << x << "' y='" << y
         << "' width='" << width << "' height='" << height
         << "' stroke='" << stroke << "' fill='" << fill << "' />\n";
}

void show_histogram_svg(const vector<size_t>& bins) {
    const auto IMAGE_WIDTH = 800;
    const auto IMAGE_HEIGHT = 600;
    const auto TEXT_LEFT = 30;
    const auto TEXT_BASELINE = 20;
    const auto TEXT_WIDTH = 50;
    const auto BIN_HEIGHT = 40;

    svg_begin(IMAGE_WIDTH, IMAGE_HEIGHT);

    size_t max_bin = 0;
    if (!bins.empty()) {
        max_bin = *max_element(bins.begin(), bins.end());
    }


    cout << "// Debug: max_bin = " << max_bin << endl;

    double max_bar_width = IMAGE_WIDTH - TEXT_WIDTH ;
    double scaling_factor = 1.0;
    if (max_bin > 0) {
        scaling_factor = max_bar_width / static_cast<double>(max_bin);

        cout << "// Debug: scaling_factor = " << scaling_factor << endl;
    }

    double top = 0;
    for (size_t bin : bins) {
        double bar_width = bin * scaling_factor;
        svg_text(TEXT_LEFT, top + TEXT_BASELINE, std::to_string(bin));
        svg_rect(TEXT_WIDTH, top, bar_width, BIN_HEIGHT, "pink", "000000");
        top += BIN_HEIGHT;
    }

    svg_end();
}