MRDirectionWidget.h

Go to the documentation of this file.

#pragma once
 
#include "exports.h"
#include "MRViewerEventsListener.h"
#include "MRMesh/MRMeshFwd.h"
#include "MRMesh/MRVector3.h"
#include "MRMesh/MRChangeXfAction.h"
#include "MRMesh/MRColor.h"
#include "MRMesh/MRObjectMesh.h"
 
namespace MR
{
class MRVIEWER_CLASS DirectionWidget : public MultiListener<MouseDownListener, MouseMoveListener, MouseUpListener>
{
public:
    using OnDirectionChangedCallback = std::function<void( const Vector3f&, bool )>;
 
    class ChangeDirAction : public ChangeXfAction
    {
    public:
        ChangeDirAction( DirectionWidget& widget ) :
            ChangeXfAction( "Change Direction", static_pointer_cast<Object>( widget.directionObj_ ) ),
            widget_{ widget },
            dir_{ widget.dir_ }
        {}
        virtual void action( Type type ) override
        {
            ChangeXfAction::action( type ); 
            auto tempDir = widget_.dir_;
            widget_.updateDirection( dir_ );
            dir_ = tempDir;
        }
    private:
        DirectionWidget& widget_;
        Vector3f dir_;
    };
 
    class ChangeBaseAction : public ChangeXfAction
    {
    public:
        ChangeBaseAction( DirectionWidget& widget ) :
            ChangeXfAction( "Change Base", static_pointer_cast< Object >( widget.directionObj_ ) ),
            widget_{ widget },
            base_{ widget.base_ }
        {}
        virtual void action( Type type ) override
        {
            ChangeXfAction::action( type );
            std::swap( base_, widget_.base_ );
        }
    private:
        DirectionWidget& widget_;
        Vector3f base_;
    };
 
    class ChangeLengthAction : public ChangeXfAction
    {
    public:
        ChangeLengthAction( DirectionWidget& widget ) :
            ChangeXfAction( "Change Length", static_pointer_cast< Object >( widget.directionObj_ ) ),
            widget_{ widget },
            length_{ widget.length_ }
        {}
        virtual void action( Type ) override
        {
            auto len = widget_.length_;
            widget_.updateLength( length_ );
            length_ = len;
 
        }
    private:
        DirectionWidget& widget_;
        float length_;
    };
 
    class ChangeVisibleAction : public HistoryAction
    {
    public:
        ChangeVisibleAction( DirectionWidget& widget ) :
            widget_{ widget },
            visible_{ widget.directionObj_->visibilityMask() }
        {}
        virtual void action( Type ) override
        {
            auto oldVisible = widget_.directionObj_->visibilityMask();
            widget_.directionObj_->setVisibilityMask( visible_ );
            visible_ = oldVisible;
        }
        virtual std::string name() const override
        {
            return name_;
        }
        [[nodiscard]] virtual size_t heapBytes() const override
        {
            return name_.capacity();
        }
    private:
        DirectionWidget& widget_;
        ViewportMask visible_;
        std::string name_ = "Change Visible";
    };
 
private:
    std::shared_ptr<ObjectMesh> directionObj_;
    VisualObject* parent_;
 
    Vector3f dir_;
    Vector3f base_;
    float length_;
    bool mousePressed_ = false;
    // if blocked cannot be moved with mouse
    bool blockedMouse_{ false };
    Vector3f worldStartPoint_;
    float viewportStartPointZ_{ 0.0f };
    OnDirectionChangedCallback onDirectionChanged_;
    Color color_ = Color::red();
    bool needToSaveHistory_ = true;
    void clear_();
 
public:
    MRVIEWER_API void create( const Vector3f& dir, const Vector3f& base, float length, OnDirectionChangedCallback onDirectionChanged, VisualObject* parent = nullptr  );
 
    MRVIEWER_API void reset();
 
    MRVIEWER_API void setOnDirectionChangedCallback( OnDirectionChangedCallback cb );
 
    MRVIEWER_API void updateDirection( const Vector3f& dir );
 
    MRVIEWER_API void updateBase( const Vector3f& base );
 
    MRVIEWER_API void updateLength( float length );
 
    MRVIEWER_API void updateArrow( const Vector3f& base, float length );
    
    std::shared_ptr<ObjectMesh> obj() const { return directionObj_; }
 
    MRVIEWER_API void setVisible( bool visible );
 
    MRVIEWER_API bool isVisible() const;
 
    MRVIEWER_API void setColor( const Color& color );
 
    MRVIEWER_API const Color& getColor() const;
 
    MRVIEWER_API const Vector3f& getBase() const;
 
    MRVIEWER_API const Vector3f& getDirection() const;
 
    MRVIEWER_API const VisualObject* getParentPtr() const;
 
    bool isMouseBlocked() const { return blockedMouse_; }
 
    void setMouseBlocked( bool blocked ) { blockedMouse_ = blocked; }
private:
    MRVIEWER_API virtual bool onMouseDown_( MouseButton button, int modifier ) override;
    MRVIEWER_API virtual bool onMouseUp_( MouseButton button, int modifier ) override;
    MRVIEWER_API virtual bool onMouseMove_( int mouse_x, int mouse_y ) override;
};
 
}