FnGLTranslateAxisHandle.h

// Copyright (c) 2017 The Foundry Visionmongers Ltd. All Rights Reserved.

#ifndef GLTRANSLATEAXISHANDLE_H_
#define GLTRANSLATEAXISHANDLE_H_

#include <stdint.h>
#include <string>

#include "FnGLTransformManipulator.h"

#include <FnViewer/plugin/FnManipulatorHandle.h>
#include <FnViewer/plugin/FnMathTypes.h>
#include <FnViewer/utils/FnDrawable.h>

using namespace Foundry::Katana::ViewerAPI;
using namespace Foundry::Katana::ViewerUtils;

class GLTranslateAxisHandle : public GLTransformManipulatorHandle
{
public:
    GLTranslateAxisHandle();

    virtual ~GLTranslateAxisHandle();

    static Foundry::Katana::ViewerAPI::ManipulatorHandle* create()
    {
        return new GLTranslateAxisHandle();
    }

    static void flush(){}

    void draw();

    void pickerDraw(int64_t pickerId);

    void setup(const IMATH_NAMESPACE::V3d& axis);

protected:
    virtual std::string getComponentName() { return "translate"; }

    bool getDraggingPlane(Vec3d& origin, Vec3d& normal);

    void startDrag(const Vec3d& initialPointOnPlane,
        const Vec2i& initialMousePosition);

    void endDrag();

    void drag(const Vec3d& initialPointOnPlane,
              const Vec3d& previousPointOnPlane,
              const Vec3d& currentPointOnPlane,
              const Vec2i& initialMousePosition,
              const Vec2i& previousMousePosition,
              const Vec2i& currentMousePosition,
              bool isFinal);

    void applyXformToLocation(const std::string& locationPath,
                              const IMATH_NAMESPACE::M44d& xform,
                              bool isFinal);

    virtual IMATH_NAMESPACE::M44d getOrientationXform(
        const std::string& locationPath);

    virtual void calculateAndSetLocalXform(const std::string& locationPath);

private:
    void initMesh();

    IMATH_NAMESPACE::V3d m_axis;

    IMATH_NAMESPACE::V3d m_axisWorldSpace;

    Drawable m_coneMesh;
    Drawable m_trunkMesh;

    // Drawable for the infinite line that represents the axis where the
    // dragging is occurring.
    Drawable m_guideLineMesh;

    Vec3d m_originWorldSpace;
    Vec3d m_normalWorldSpace;

    IMATH_NAMESPACE::M44d m_meshXform;

    Vec4f m_axisColor;
};

#endif  // GLTranslateManipulator_H_