#include "SetsquareController.h"

#include "control/Control.h"
#include "control/layer/LayerController.h"
#include "gui/XournalView.h"
#include "model/GeometryTool.h"
#include "model/Setsquare.h"
#include "model/Stroke.h"
#include "model/XojPage.h"

using xoj::util::Rectangle;
SetsquareController::SetsquareController(XojPageView* view, Setsquare* setsquare):
        GeometryToolController(view, setsquare) {}

SetsquareController::~SetsquareController() = default;

auto SetsquareController::getType() const -> GeometryToolType { return GeometryToolType::SETSQUARE; }

auto SetsquareController::posRelToSide(Leg leg, double x, double y) const -> xoj::util::Point<double> {
    cairo_matrix_t matrix = geometryTool->getMatrix();
    cairo_matrix_invert(&matrix);
    cairo_matrix_transform_point(&matrix, &x, &y);
    switch (leg) {
        case HYPOTENUSE:
            return xoj::util::Point<double>(x, -y);
        case LEFT_LEG:
            return xoj::util::Point<double>((y + x) / std::sqrt(2.),
                                            (y - x - geometryTool->getHeight()) / std::sqrt(2.));
        case RIGHT_LEG:
            return xoj::util::Point<double>((y - x) / std::sqrt(2.),
                                            (y + x - geometryTool->getHeight()) / std::sqrt(2.));
        default:
            g_error("Invalid enum value: %d", leg);
    }
}

auto SetsquareController::isInsideGeometryTool(double x, double y, double border) const -> bool {
    return posRelToSide(HYPOTENUSE, x, y).y < border && posRelToSide(LEFT_LEG, x, y).y < border &&
           posRelToSide(RIGHT_LEG, x, y).y < border;
}

auto SetsquareController::getPointForPos(double xCoord) const -> xoj::util::Point<double> {
    double x = xCoord;
    double y = 0.0;
    cairo_matrix_t matrix = geometryTool->getMatrix();
    cairo_matrix_transform_point(&matrix, &x, &y);

    return xoj::util::Point<double>(x, y);
}

void SetsquareController::createEdgeStroke(double x) {
    if (!std::isnan(x)) {
        hypotenuseMax = x;
        hypotenuseMin = x;

        const auto p = this->getPointForPos(x);
        initializeStroke();
        stroke->addPoint(Point(p.x, p.y));
        stroke->addPoint(Point(p.x, p.y));  // doubled point
        geometryTool->notify();
    } else {
        g_warning("No valid stroke from setsquare!");
    }
}

void SetsquareController::createRadialStroke(double x, double y) {
    const auto p = posRelToSide(HYPOTENUSE, x, y);
    this->strokeAngle = std::atan2(p.y, p.x);
    initializeStroke();
    updateRadialStroke(x, y);
}

void SetsquareController::updateEdgeStroke(double x) {
    hypotenuseMax = std::max(this->hypotenuseMax, x);
    hypotenuseMin = std::min(this->hypotenuseMin, x);
    stroke->deletePointsFrom(0);
    const auto p1 = getPointForPos(hypotenuseMin);
    const auto p2 = getPointForPos(hypotenuseMax);

    stroke->addPoint(Point(p1.x, p1.y));
    stroke->addPoint(Point(p2.x, p2.y));
    geometryTool->notify();
}

void SetsquareController::updateRadialStroke(double x, double y) {
    stroke->deletePointsFrom(0);
    const auto c = getPointForPos(0);
    stroke->addPoint(Point(c.x, c.y));

    const auto p = posRelToSide(HYPOTENUSE, x, y);
    const auto rad = std::hypot(p.x, p.y);

    if (rad >= Setsquare::radiusFromHeight(geometryTool->getHeight()) || p.y > 0) {
        this->strokeAngle = std::atan2(p.y, p.x);
        stroke->addPoint(Point(x, y));
    } else {
        cairo_matrix_t matrix = geometryTool->getMatrix();
        auto qx = rad * std::cos(this->strokeAngle);
        auto qy = -rad * std::sin(this->strokeAngle);
        cairo_matrix_transform_point(&matrix, &qx, &qy);

        stroke->addPoint(Point(qx, qy));
    }
    geometryTool->notify();
}

void SetsquareController::finalizeEdgeStroke() {
    hypotenuseMax = std::numeric_limits<double>::lowest();
    hypotenuseMin = std::numeric_limits<double>::max();
    addStrokeToLayer();
}

void SetsquareController::finalizeRadialStroke() {
    strokeAngle = NAN;
    addStrokeToLayer();
}

auto SetsquareController::existsEdgeStroke() -> bool { return hypotenuseMax != std::numeric_limits<double>::lowest(); }

auto SetsquareController::existsRadialStroke() -> bool { return !std::isnan(strokeAngle); }