ToolGlobal.h

#ifndef TOOLGLOBAL_H
#define TOOLGLOBAL_H

#include <vector>
#include <iostream>
#include <stdio.h>

#include <osg/Group>
#include <osg/Switch>
#include <osg/AutoTransform>
#include <osg/Geode>
#include <osg/Geometry>
#include <osg/Camera>

#include "Settings.h"
#include "ProtectedGroup.h"

namespace entity {

class ToolGlobal : public osg::ProtectedGroup
{
public:
    ToolGlobal(int nVerts, osg::Array::Binding colorBind,
         osg::PrimitiveSet* primitiveSet, float linewidth=3.f);

protected:
    virtual void initializeSG() = 0;

public:
    virtual void setVertices(const std::vector<osg::Vec3f>& source);

    virtual const osg::Vec3Array* getVertices() const;

    virtual void setColor(const osg::Vec4f& color);

    virtual const osg::Vec4f& getColor() const;

    virtual void setVisibility(bool on) = 0;

    virtual bool getVisibility() const = 0;

    osg::Geode* getGeodeWire() const;

    osg::Vec3f getCenterLocal() const;

protected:
    void updateGeometry();
    void updateGeometry(osg::Geometry* geom);

    osg::Switch* m_switch;  
    osg::Geode* m_geodeWire;  
    osg::Geometry* m_geomWire; 
};

class BookmarkTool : public ToolGlobal
{
public:
    BookmarkTool(const osg::Vec3d &eye, const osg::Vec3d &center, const osg::Vec3d &up);

protected:
    void initializeSG();

public:
    void setPose(const osg::Vec3d &eye, const osg::Vec3d &center, const osg::Vec3d &up);

    void getPose(osg::Vec3d& eye, osg::Vec3d& center, osg::Vec3d& up);

    void setColorDefault();

    void setColorSelected();

protected:
    void updatePosition();

public:
    void setVisibility(bool on);

    bool getVisibility() const;

private:
    osg::AutoTransform* m_AT;

    osg::Vec3f m_eye, m_center, m_up;
};

class AxisGlobalTool : public ToolGlobal
{
public:
    AxisGlobalTool();

protected:
    void initializeSG();

public:
    void setVisibility(bool on);

    bool getVisibility() const;

    virtual void setColor(const osg::Vec4f c1, const osg::Vec4f c2, const osg::Vec4f c3);

private:
    osg::AutoTransform* m_AT;
    osg::Camera* m_camera;
};

class ATGeode : public osg::AutoTransform
{
public:
    ATGeode();

protected:
    friend class FrameTool;

    osg::Geode* geode;
    osg::Geometry* geometry;
};

class ATRGeode : public osg::AutoTransform
{
public:
    ATRGeode();

protected:
    friend class FrameTool;

    osg::Geode* geode;
    osg::Geometry* geometry;
    osg::Camera* camera;

};

class FrameTool : public ToolGlobal
{
public:
    FrameTool();

protected:
    void initializeSG();

public:
    void setVisibility(bool on);

    bool getVisibility() const;

    void setVertices(const osg::Vec3f &center, float szX, float szY, float szCr, float szAx,
                     const osg::Vec3f& centerCustom = cher::CENTER ,
                     double theta = 0,
                     bool selectionIsEmpty = true);

    void setColor(const osg::Vec4f& color, const osg::Vec4f& colorIntersect = cher::CANVAS_CLR_PREVIOUS);

    void setIntersection(const osg::Vec3f &P1, const osg::Vec3f &P2, const osg::Vec3f &P3, const osg::Vec3f &P4);

protected:
    void setColorIntersection(const osg::Vec4f& colorIntersect);

public:
    void setEditable(bool editable);

    const osg::Geometry* getPickable() const;

    bool isSelected() const;

    virtual void moveDelta(double du, double dv);

    virtual void scale(double scaleX, double scaleY, osg::Vec3f center);

    virtual void scale(double scale, osg::Vec3f center);

    virtual void rotate(double theta, osg::Vec3f center);

protected:
    void initQuadGeometry(osg::Geometry* geom, const std::string& name = "");
    void initLineGeometry(osg::Geometry* geom, float lineWidth, const std::string& name = "");

    void setQuadGeometry(osg::Geometry* geom, const osg::Vec3f& P, float szX, float szY, float theta = 0, const osg::Vec3f& center = cher::CENTER);

    void setScaleGeometry(osg::Geometry* geom, const osg::Vec3f& P0, const osg::Vec3f& P1, const osg::Vec3f& P2, const osg::Vec3f& P3);

    void setLineGeometry(osg::Geometry* geom, const osg::Vec3f& P1, const osg::Vec3f& P2);

    void setColorGeometry(osg::Geometry* geom, const osg::Vec4f& color);

    void moveDeltaWireGeometry(osg::Geometry*geometry, double du, double dv);
    void scaleWireGeometry(osg::Geometry* geometry, double scaleX, double scaleY, osg::Vec3f center);
    void scaleWireGeometry(osg::Geometry* geometry, double scale, osg::Vec3f center);
    void rotateWireGeometry(osg::Geometry* geometry, double theta, osg::Vec3f center);

private:
    osg::Geode* m_geodeIntersect, * m_geodeNormal, * m_geodeRotation;
    osg::Geometry* m_geomIntersect;

    /* canvas offset and 3d rotation drawables */
    osg::Geometry * m_geomNormal1, * m_geomNormal2; 
    osg::Geometry * m_geomRotateX1, * m_geomRotateX2, * m_geomRotateY1, * m_geomRotateY2; 
    ATGeode* m_AT_pick; 
    ATRGeode* m_AT_scaleUV1, * m_AT_scaleUV2, * m_AT_scaleUV3, * m_AT_scaleUV4; 
    ATRGeode* m_AT_center; 
    ATRGeode* m_AT_axisU, * m_AT_axisV; 
    bool m_selected; 
    bool m_editable; 
    bool m_visible; 
    std::vector<bool> m_visibilityState;
};

}


#endif // TOOLGlobal_H