PrusaSlicer-NonPlainar/xs/src/slic3r/GUI/GLGizmo.hpp

#ifndef slic3r_GLGizmo_hpp_
#define slic3r_GLGizmo_hpp_

#include "../../slic3r/GUI/GLTexture.hpp"
#include "../../libslic3r/Point.hpp"
#include "../../libslic3r/BoundingBox.hpp"

#include <vector>

#define ENABLE_GIZMOS_3D 1

namespace Slic3r {

class BoundingBoxf3;
class Pointf3;
class Linef3;

namespace GUI {

class GLGizmoBase
{
protected:
    struct Grabber
    {
        static const float HalfSize;
        static const float DraggingScaleFactor;

        Pointf3 center;
        float angle_x;
        float angle_y;
        float angle_z;
        float color[3];
        bool dragging;

        Grabber();

        void render(bool hover) const;
#if ENABLE_GIZMOS_3D
        void render_for_picking() const { render(color, false); }
#else
        void render_for_picking() const { render(color); }
#endif // ENABLE_GIZMOS_3D

    private:
#if ENABLE_GIZMOS_3D
        void render(const float* render_color, bool use_lighting) const;
#else
        void render(const float* render_color) const;
#endif // ENABLE_GIZMOS_3D
#if ENABLE_GIZMOS_3D
        void render_face(float half_size) const;
#endif // ENABLE_GIZMOS_3D
    };

public:
    enum EState
    {
        Off,
        Hover,
        On,
        Num_States
    };

protected:
    int m_group_id;
    EState m_state;
    // textures are assumed to be square and all with the same size in pixels, no internal check is done
    GLTexture m_textures[Num_States];
    int m_hover_id;
    float m_base_color[3];
    float m_drag_color[3];
    float m_highlight_color[3];
    mutable std::vector<Grabber> m_grabbers;
    bool m_is_container;

public:
    GLGizmoBase();
    virtual ~GLGizmoBase() {}

    bool init() { return on_init(); }

    int get_group_id() const { return m_group_id; }
    void set_group_id(int id) { m_group_id = id; }

    EState get_state() const { return m_state; }
    void set_state(EState state) { m_state = state; on_set_state(); }

    unsigned int get_texture_id() const { return m_textures[m_state].get_id(); }
    int get_textures_size() const { return m_textures[Off].get_width(); }

    int get_hover_id() const { return m_hover_id; }
    void set_hover_id(int id);

    void set_highlight_color(const float* color);

    void start_dragging();
    void stop_dragging();
    void update(const Linef3& mouse_ray);
    void refresh() { on_refresh(); }

    void render(const BoundingBoxf3& box) const { on_render(box); }
    void render_for_picking(const BoundingBoxf3& box) const { on_render_for_picking(box); }

protected:
    virtual bool on_init() = 0;
    virtual void on_set_state() {}
    virtual void on_set_hover_id() {}
    virtual void on_start_dragging() {}
    virtual void on_stop_dragging() {}
    virtual void on_update(const Linef3& mouse_ray) = 0;
    virtual void on_refresh() {}
    virtual void on_render(const BoundingBoxf3& box) const = 0;
    virtual void on_render_for_picking(const BoundingBoxf3& box) const = 0;

    float picking_color_component(unsigned int id) const;
    void render_grabbers() const;
    void render_grabbers_for_picking() const;
};

class GLGizmoRotate : public GLGizmoBase
{
    static const float Offset;
    static const unsigned int CircleResolution;
    static const unsigned int AngleResolution;
    static const unsigned int ScaleStepsCount;
    static const float ScaleStepRad;
    static const unsigned int ScaleLongEvery;
    static const float ScaleLongTooth;
    static const float ScaleShortTooth;
    static const unsigned int SnapRegionsCount;
    static const float GrabberOffset;

public:
    enum Axis : unsigned char
    {
        X,
        Y,
        Z
    };

private:
    Axis m_axis;
    float m_angle;

    mutable Pointf3 m_center;
    mutable float m_radius;
    mutable bool m_keep_initial_values;

public:
    explicit GLGizmoRotate(Axis axis);

    float get_angle() const { return m_angle; }
    void set_angle(float angle);

protected:
    virtual bool on_init();
    virtual void on_set_state() { m_keep_initial_values = (m_state == On) ? false : true; }
    virtual void on_update(const Linef3& mouse_ray);
    virtual void on_refresh() { m_keep_initial_values = false; }
    virtual void on_render(const BoundingBoxf3& box) const;
    virtual void on_render_for_picking(const BoundingBoxf3& box) const;

private:
    void render_circle() const;
    void render_scale() const;
    void render_snap_radii() const;
    void render_reference_radius() const;
    void render_angle() const;
    void render_grabber() const;

    void transform_to_local() const;
    Pointf mouse_position_in_local_plane(const Linef3& mouse_ray) const;
};

class GLGizmoRotate3D : public GLGizmoBase
{
    GLGizmoRotate m_x;
    GLGizmoRotate m_y;
    GLGizmoRotate m_z;

public:
    GLGizmoRotate3D();

    float get_angle_x() const { return m_x.get_angle(); }
    void set_angle_x(float angle) { m_x.set_angle(angle); }

    float get_angle_y() const { return m_y.get_angle(); }
    void set_angle_y(float angle) { m_y.set_angle(angle); }

    float get_angle_z() const { return m_z.get_angle(); }
    void set_angle_z(float angle) { m_z.set_angle(angle); }

protected:
    virtual bool on_init();
    virtual void on_set_state()
    {
        m_x.set_state(m_state);
        m_y.set_state(m_state);
        m_z.set_state(m_state);
    }
    virtual void on_set_hover_id()
    {
        m_x.set_hover_id(m_hover_id == 0 ? 0 : -1);
        m_y.set_hover_id(m_hover_id == 1 ? 0 : -1);
        m_z.set_hover_id(m_hover_id == 2 ? 0 : -1);
    }
    virtual void on_start_dragging();
    virtual void on_stop_dragging();
    virtual void on_update(const Linef3& mouse_ray)
    {
        m_x.update(mouse_ray);
        m_y.update(mouse_ray);
        m_z.update(mouse_ray);
    }
    virtual void on_refresh()
    {
        m_x.refresh();
        m_y.refresh();
        m_z.refresh();
    }
    virtual void on_render(const BoundingBoxf3& box) const;
    virtual void on_render_for_picking(const BoundingBoxf3& box) const
    {
        m_x.render_for_picking(box);
        m_y.render_for_picking(box);
        m_z.render_for_picking(box);
    }
};

class GLGizmoScale : public GLGizmoBase
{
    static const float Offset;

    float m_scale;
    float m_starting_scale;

    Pointf m_starting_drag_position;

public:
    GLGizmoScale();

    float get_scale() const { return m_scale; }
    void set_scale(float scale) { m_starting_scale = scale; }

protected:
    virtual bool on_init();
    virtual void on_start_dragging();
    virtual void on_update(const Linef3& mouse_ray);
    virtual void on_render(const BoundingBoxf3& box) const;
    virtual void on_render_for_picking(const BoundingBoxf3& box) const;
};

class GLGizmoScale3D : public GLGizmoBase
{
    static const float Offset;

    mutable BoundingBoxf3 m_box;

    float m_scale_x;
    float m_scale_y;
    float m_scale_z;

    float m_starting_scale_x;
    float m_starting_scale_y;
    float m_starting_scale_z;

    Pointf3 m_starting_drag_position;
    Pointf3 m_starting_center;

public:
    GLGizmoScale3D();

    float get_scale_x() const { return m_scale_x; }
    void set_scale_x(float scale) { m_starting_scale_x = scale; }

    float get_scale_y() const { return m_scale_y; }
    void set_scale_y(float scale) { m_starting_scale_y = scale; }

    float get_scale_z() const { return m_scale_z; }
    void set_scale_z(float scale) { m_starting_scale_z = scale; }

    void set_scale(float scale)
    {
        m_starting_scale_x = scale;
        m_starting_scale_y = scale;
        m_starting_scale_z = scale;
    }

protected:
    virtual bool on_init();
    virtual void on_start_dragging();
    virtual void on_update(const Linef3& mouse_ray);
    virtual void on_render(const BoundingBoxf3& box) const;
    virtual void on_render_for_picking(const BoundingBoxf3& box) const;

private:
    void render_box() const;
    void render_grabbers_connection(unsigned int id_1, unsigned int id_2) const;

    void do_scale_x(const Linef3& mouse_ray);
    void do_scale_y(const Linef3& mouse_ray);
    void do_scale_z(const Linef3& mouse_ray);
    void do_scale_uniform(const Linef3& mouse_ray);

    double calc_ratio(unsigned int preferred_plane_id, const Linef3& mouse_ray, const Pointf3& center) const;
};

} // namespace GUI
} // namespace Slic3r

#endif // slic3r_GLGizmo_hpp_