DtOsgRibbon.h

Go to the documentation of this file.

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


#pragma once

#include <vrvOsg/vrvOsg.h>

#include <vrvCore/DtModelSetEnums.h>

#include <osg/MatrixTransform>
#include <osg/Group>
#include <osg/Geode>
#include <osg/Vec3>
#include <osg/Vec4>

namespace makVrv
{

   class DT_DLL_VRVOSG DtOsgRibbon : public osg::Group
   {
   public:

      DtOsgRibbon(
         DtDe& de,
         float segmentSpacing = 1.0,
         const osg::Vec4& color = osg::Vec4d(0.5, 0.5, 0.5, 1.0),
         float maxLength = 2000,
         float timeout = 3600.0);

      virtual ~DtOsgRibbon();
      
   private:
      DtOsgRibbon() = delete;
      DtOsgRibbon(const DtOsgRibbon&) = delete;
      DtOsgRibbon &operator=(const DtOsgRibbon&) = delete;
      
   public:
      virtual void traverse(osg::NodeVisitor& nv);

      virtual osg::Node* node();

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

      virtual const osg::Vec4& color() const { return myColor; }

      virtual void setDroppedPoints(const osg::Vec3d& pointA, const osg::Vec3d& pointB);

      virtual void setDroppedPointsPositionOffset(const osg::Vec3d& posOffset);

      virtual void update(double dt, double simTime);

     virtual void update();

      virtual void setMaxSegments(long segments);

      inline virtual long maxSegments() const { return myMaxSegments; }

      virtual void setSegmentSpacing(float spacing);

      virtual void setSegmentTimeout(float duration);

      inline virtual float segmentSpacing() const { return mySegmentSpacing; }

      // Set ribbon to be stippled or alpha-blended.
      virtual void setStippled(bool stippled);

      inline virtual bool isStippled() const { return myStippled; }

      virtual void setLightingEnabled(bool enable);

      virtual bool lightingEnabled();

      virtual void setHeightLinesEnabled(bool enabled);

      virtual void setHeightLinesColor(float r, float g, float b, float a);

      virtual void setHeightLinesSpacing(float spacing);

      virtual void setHeightLinesModelSetId(DtModelSetId modelSetId);

   protected:

      virtual void addParticle(osg::Vec3d const& left, osg::Vec3d const& right, 
         double simTime) = 0;

      virtual float currentSegmentLength() = 0;
      virtual void setCurrentSegmentLength( float len );

      virtual void cleanupOldSegments(double simTime) = 0;

      virtual void extendCurrentSegment(osg::Vec3d const& left, 
         osg::Vec3d const& right, double simTime) = 0;

      virtual osg::Drawable* getDrawable() = 0;

      virtual osg::Geode* createGeode();

      virtual void setDrawable();

   protected:

      osg::ref_ptr<osg::MatrixTransform> myPositionNode;
      osg::ref_ptr<osg::Geode> myGeode;
      osg::Matrixd myWorldTransformInv;
      osg::Matrixd myLastLocalToWork;

      int myLastUpdatedFrame;
      double myLastUpdateTime;
      double myRibbonLengthSquared;

      long myMaxSegments;
      int myActiveSegmentCount;
      float mySegmentSpacingSquared;
      float mySegmentSpacing;
      float mySegmentTimeout;

      osg::Vec3d myLocalPointA;
      osg::Vec3d myLocalPointB;
      osg::Vec3d myLocalPosOffset;

      osg::Vec4 myColor;
      bool myFirstUpdate;
      bool myStippled;

      DtDe& myDe;
      bool myLightingEnabled;
      bool myShaderSet;
      bool myHeightLinesEnabled;
      float myHeightLinesColor[4];
      float myHeightLinesSpacing;
      DtModelSetId myHeightLinesModelSetId;
      
   };

}