Dt3DLineModelInstanceSegment.h

Go to the documentation of this file.

/******************************************************************************
** Copyright (c) 2012 MAK Technologies, Inc.
** All rights reserved.
******************************************************************************/


#ifndef DtAttachableLineModelInstanceSegment_H_
#define DtAttachableLineModelInstanceSegment_H_

#include <vrvOsg/vrvOsg.h>
#include <vrvOsg/Dt3DLineModelInstanceSegmentInterface.h>

#include <osg/Billboard>
#include <osg/MatrixTransform>

namespace makVrv
{
   class DT_DLL_VRVOSG Dt3DLineModelInstanceSegment 
      : public Dt3DLineModelInstanceSegmentInterface
   {
   public:
      Dt3DLineModelInstanceSegment();

      virtual ~Dt3DLineModelInstanceSegment();

      virtual void setStartPoint(const osg::Vec3d& startPoint);

      virtual void setEndPoint(const osg::Vec3d& endPoint);

      const osg::Vec3d& startPoint() const
      {
         return myState.startPoint;
      }

      const osg::Vec3d& endPoint() const
      {
         return myState.endPoint;
      }

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

      virtual void setVertexColor( unsigned int vertexIndex, const osg::Vec4& color );

      virtual void setStartArrow(bool arrowEnabled);

      virtual void setEndArrow(bool arrowEnabled);

      virtual void setLightningStyle(bool lightningStyle);

      virtual void setWidth(double width);

      virtual double width() const;

      virtual void setStipplePattern(unsigned int stipplePattern, int repeatFactor);

      virtual osg::Node* rootNode();

      const std::vector<osg::Vec3>& pickableVertices() const;

      virtual void update();

      virtual void updateGeometry();

   protected:

      virtual void build();

      virtual void calculateArrowVertices(
         const osg::Vec3d& segmentStart, const osg::Vec3d& segmentEnd,
         osg::Vec3d& arrowStartOut, osg::Vec3d& arrowEndOut);

      virtual void buildVertices();

      virtual void buildLinePrimitive();

      virtual void buildPickableVertices();

      virtual void buildColorArray();

      virtual unsigned int startArrowVertexIndex();

      virtual unsigned int endArrowVertexIndex();

      virtual unsigned int lightningVertexIndex();

   protected:

      struct LineSegmentState
      {
         osg::Vec3d startPoint;
         osg::Vec3d endPoint;
         osg::Vec4 color[2];
         bool startArrow;
         bool endArrow;
         bool lightning;
         double width;
         unsigned int stipplePattern;
      };

      LineSegmentState myState;

      bool myGeometryDirty;
      bool myIndexBufferDirty;

      osg::ref_ptr<osg::Billboard> myGeode;
      osg::ref_ptr<osg::Geometry> myGeometry;
      osg::ref_ptr<osg::Vec3Array> myPickableVertices;
      osg::ref_ptr<osg::Vec3Array> myVertexArray;
      osg::ref_ptr<osg::Vec4Array> myColorArray;
      osg::ref_ptr<osg::DrawElementsUInt> myLinePrimitive;
   };
}

#endif