infinipaint/src/DrawComponents/DrawEllipse.cpp

#include "DrawEllipse.hpp"
#include "Helpers/ConvertVec.hpp"
#include "Helpers/SCollision.hpp"

#ifndef IS_SERVER
#include "../DrawCollision.hpp"
#endif

DrawComponentType DrawEllipse::get_type() const {
    return DRAWCOMPONENT_ELLIPSE;
}

void DrawEllipse::save(cereal::PortableBinaryOutputArchive& a) const {
    a(d.strokeColor, d.fillColor, d.strokeWidth, d.p1, d.p2, d.fillStrokeMode);
}

void DrawEllipse::load(cereal::PortableBinaryInputArchive& a) {
    a(d.strokeColor, d.fillColor, d.strokeWidth, d.p1, d.p2, d.fillStrokeMode);
}

#ifndef IS_SERVER
std::shared_ptr<DrawComponent> DrawEllipse::copy() const {
    auto a = std::make_shared<DrawEllipse>();
    a->d = d;
    a->coords = coords;
    return a;
}

void DrawEllipse::update_from_delayed_ptr() {
    std::shared_ptr<DrawEllipse> newPtr = std::static_pointer_cast<DrawEllipse>(delayedUpdatePtr);
    d = newPtr->d;
}

void DrawEllipse::draw(SkCanvas* canvas, const DrawData& drawData) {
    if(drawSetupData.shouldDraw) {
        canvas->save();
        canvas_do_calculated_transform(canvas);
        if(d.fillStrokeMode == 0 || d.fillStrokeMode == 2) {
            SkPaint p;
            p.setStyle(SkPaint::kFill_Style);
            p.setColor4f(convert_vec4<SkColor4f>(d.fillColor));
            canvas->drawPath(ellipsePath, p);
        }
        if(d.fillStrokeMode == 1 || d.fillStrokeMode == 2) {
            SkPaint p;
            p.setStyle(SkPaint::kStroke_Style);
            p.setColor4f(convert_vec4<SkColor4f>(d.strokeColor));
            p.setStrokeWidth(d.strokeWidth);
            canvas->drawPath(ellipsePath, p);
        }
        canvas->restore();
    }
}

void DrawEllipse::create_draw_data() {
    ellipsePath = SkPath();
    SkRect newRect = SkRect::MakeLTRB(d.p1.x(), d.p1.y(), d.p2.x(), d.p2.y());
    ellipsePath.addOval(newRect);
}

void DrawEllipse::initialize_draw_data(DrawingProgram& drawP) {
    create_draw_data();
}

void DrawEllipse::update(DrawingProgram& drawP) {
}

bool DrawEllipse::collides_within_coords(const SCollision::ColliderCollection<float>& checkAgainst) {
    return collisionTree.is_collide(checkAgainst);
}

void DrawEllipse::create_collider() {
    using namespace SCollision;
    ColliderCollection<float> strokeObjects;

    if(d.fillStrokeMode == 0) {
        Vector2f ellipseCenter = (d.p1 + d.p2) * 0.5;
        float a = ellipseCenter.x() - d.p1.x();
        float b = ellipseCenter.y() - d.p1.y();
        Vector2f prevPoint = ellipseCenter + Vector2f{a * std::cos(0), b * std::sin(0)};
        unsigned numOfSegments = 20;
        float segmentStep = 1.0 / numOfSegments;
        float t = 0.0;
        for(unsigned i = 0; i <= numOfSegments; i++) {
            Vector2f nextPoint = ellipseCenter + Vector2f{a * std::cos(t), b * std::sin(t)};
            strokeObjects.triangle.emplace_back(ellipseCenter, prevPoint, nextPoint);
            prevPoint = nextPoint;
            t += segmentStep * std::numbers::pi * 2.0;
        }
    }
    else if(d.fillStrokeMode == 1) {
        std::vector<Vector2f> points;

        Vector2f ellipseCenter = (d.p1 + d.p2) * 0.5;
        float a = ellipseCenter.x() - d.p1.x();
        float b = ellipseCenter.y() - d.p1.y();
        unsigned numOfSegments = 40;
        float segmentStep = 1.0 / numOfSegments;
        float t = 0.0;
        for(unsigned i = 0; i < numOfSegments; i++) {
            points.emplace_back(ellipseCenter + Vector2f{a * std::cos(t), b * std::sin(t)});
            t += segmentStep * std::numbers::pi * 2.0;
        }
        generate_polyline(strokeObjects, points, d.strokeWidth, true);

    }
    else if(d.fillStrokeMode == 2) {
        float strokeRadius = d.strokeWidth * 0.5f;
        Vector2f wNewP1 = d.p1 - Vector2f{strokeRadius, strokeRadius};
        Vector2f wNewP2 = d.p2 + Vector2f{strokeRadius, strokeRadius};
        Vector2f ellipseCenter = (wNewP1 + wNewP2) * 0.5;
        float a = ellipseCenter.x() - wNewP1.x();
        float b = ellipseCenter.y() - wNewP1.y();
        Vector2f prevPoint = ellipseCenter + Vector2f{a * std::cos(0), b * std::sin(0)};
        unsigned numOfSegments = 20;
        float segmentStep = 1.0 / numOfSegments;
        float t = 0.0;
        for(unsigned i = 0; i <= numOfSegments; i++) {
            Vector2f nextPoint = ellipseCenter + Vector2f{a * std::cos(t), b * std::sin(t)};
            strokeObjects.triangle.emplace_back(ellipseCenter, prevPoint, nextPoint);
            prevPoint = nextPoint;
            t += segmentStep * std::numbers::pi * 2.0;
        }
    }

    collisionTree.clear();
    collisionTree.calculate_bvh_recursive(strokeObjects);
    calculate_world_bounds();
}

SCollision::AABB<float> DrawEllipse::get_obj_coord_bounds() const {
    return collisionTree.objects.bounds;
}

#endif