DtInstanceManager.h

Go to the documentation of this file.

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


#pragma once

#include <vrvOsg/vrvOsg.h>

#include <vrvCore/DtDe.h>
#include <vrvCore/DtDisplay.h>
#include <vrvCore/DtUniqueID.h>

#include <vrvOsg/DtViewSpecificGroup.h>
#include <vrvOsg/DtInstanceState.h>
#include <vrvOsg/DtOsgCullVisitorParameters.h>

#include <vrvUtil/DtSignalConnectionManager.h>

#include <osgViewer/CompositeViewer>

#include <osgEarth/VirtualProgram>

#include <boost/unordered_map.hpp>
#include <boost/signals.hpp>

class DtVector;

namespace osg
{
   class Geometry;
   class Image;
   class Texture2D;
}

namespace makVrv
{
   class DtGLTextureBufferObject;
   class DtInstanceProcessor;
   class DtOsgIntersectorResult;
   class DtChannelManager;
   class DtChannel;
   class DtPerViewUniformCallback;
   class DtInstanceNode;
   class DtLight;
   
   class DT_DLL_VRVOSG DtInstanceManager : public DtVirtualBaseClass, public DtSignalConnectionManager
   {
   public:
      typedef std::vector<osg::ref_ptr<osg::Geometry> > GeometryVec;
      typedef std::vector<char> BoolVec; // Using vec of chars for threadsafety
      typedef std::vector<char> ByteVec;
      typedef std::map<osg::ref_ptr< osgEarth::VirtualProgram >, osg::ref_ptr<osgEarth::VirtualProgram> > VpMap;
      typedef std::vector<osg::ref_ptr<osgEarth::VirtualProgram> > VpRefList;
      typedef std::vector<ArrayRange> DrawableRangeVec;
      typedef std::vector<osg::Vec2f> TextureCoordinateVec;
      typedef std::vector<float> FloatVec;

      static const unsigned int NumVec4PerInstance = 8;

      struct InstancedLod
      {
         double minRange;

         double maxRange;

         ArrayRange drawableRange;
      };

      struct InstancedSwitch
      {
         DrawableRangeVec switchRanges;

         BoolVec switchStates;
      };

      struct InstancedSequence
      {
         enum SequenceMode { Play, Stop, Pause, Resume };

         DrawableRangeVec switchRanges;

         float timePerFrame;

         float startTime;

         float endTime;

         SequenceMode mode;

         float speed;

         int lastFrame;
      };

      struct InstancedLight
      {
         DtLight* light;
         osg::Vec3f position;
         osg::Vec3f direction;
         unsigned int drawableIx; // index to drawable the light is attached to
      };

      struct InstancedDof
      {
         InstanceMatrix transform;

         unsigned int numChildren;

         ArrayRange drawableRange;

         bool skipNextAnimationFrame;
      };

      struct ModelTypeDofData
      {
         InstanceMatrix putMatrix;

         InstanceMatrix putMatrixInverse;

         InstanceMatrix initialTransform;

         osg::Vec3f initialHpr;

         osg::Vec3f initialTranslate;

         unsigned int numChildren;

         ArrayRange drawableRange;
      };

      struct DofAnimation
      {
         osg::Vec3f trans;

         osg::Vec3f hpr;

         osg::Vec3f hprVel;

         float lastUpdate;
      };

      struct LightSource
      {
         osg::ref_ptr<osg::LightSource> lightSource;
         unsigned int attachDrawableIx;
      };

      typedef std::vector<LightSource> LightSourceVec;
      typedef std::vector<ModelTypeDofData> PerDofDataVec;
      typedef std::vector<InstancedSwitch> InstancedSwitchVec;
      typedef std::vector<InstancedSequence> InstancedSequenceVec;
      typedef std::vector<DofAnimation> DofAnimations;
      typedef std::vector<osg::Node*> NodeVec;
      typedef std::vector<InstancedSwitch*> SwitchVec;
      typedef std::vector<InstancedSequence*> SequenceVec;
      typedef std::vector<InstancedLod*> LodVec;

      struct InstanceTemplate
      {
         PerDofDataVec dofTemplate;

         LightSourceVec lightSources;

         InstancedSwitchVec switchTemplate;

         InstancedSequenceVec sequenceTemplate;

         ArrayRange drawableRange;

         ArrayRange lodRange;

         float boundingRadius;

         osg::Vec3f boundingCenter;

         unsigned int nodeMask;
      };


      boost::signal<void (const std::string&)> signal_modelTypeRegistered;


      bool intersect(const osg::Vec3d& start, const osg::Vec3d& end, DtOsgIntersectorResult& result);

      bool intersect(const osg::Vec3d& start, const osg::Vec3d& end, std::vector<DtOsgIntersectorResult>& results);

      void setSequenceMode(DtInstanceState& instance, unsigned int sequenceIndex, InstancedSequence::SequenceMode mode);

      void setSequenceSpeed(DtInstanceState& instance, unsigned int sequenceIndex, float speed);

      void setSequencePlayTime(DtInstanceState& instance, unsigned int sequenceIndex, float loopTime);

      InstanceMatrix dofState(DtInstanceState& instanceState, unsigned int partIndex);

      void setDofState(InstanceMatrix& m, DtInstanceState& state, unsigned int partIndex);

      void setDofAnimation(osg::Vec3f& hpr, osg::Vec3f hprVel, DtInstanceState& state, unsigned int partIndex);

      void setDofTranslation(osg::Vec3f& trans, DtInstanceState& state, unsigned int partIndex);

      void setSwitchState(DtInstanceState& instance, unsigned int partIndex, unsigned int childIndex, bool state);

      void setUVOffset(DtUniqueID id, unsigned int drawableIx, osg::Vec2f uvOffset);

      void setUserDefined(DtUniqueID id, float userDefined);

      static DtInstanceManager& instance(DtDe& de);

      DtInstanceManager(DtDe& de);

      virtual ~DtInstanceManager();

      virtual void reset();

      virtual osg::Node* instanceRoot();

      bool isRegistered(const std::string& modelType);

      void registerModel(const std::string& modelType, GeometryVec& drawables, LodVec& lods,
         SwitchVec& switches, SequenceVec& sequences, PerDofDataVec& perDofData,
         LightSourceVec& lightSources, float boundingRadius, const DtVector& boundingCenter,
         unsigned int nodeMask, osg::ref_ptr<osg::StateSet> rootNodeSS);

      void setupModelGeode(const std::string& modelType, unsigned int nodeMask);

      void allocateImage( osg::View* view );

      void freeInstance(DtUniqueID id);

      bool newInstance(const std::string& modelType, DtUniqueID id, DtElementID elementId);

      DtInstanceState* findInstanceState(DtUniqueID id);

      InstanceCullData* findInstanceCullData(DtUniqueID id);

      typedef std::vector<DtInstanceState> InstanceStateVec;
      typedef std::vector<InstancedLod> InstancedLodVec;
      typedef std::vector<InstancedDof> InstancedDofVec;
      typedef std::vector<InstanceCullData> InstanceCullDataVec;
      typedef std::vector<InstanceMatrix> InstanceMatrixVec;
      typedef std::vector<float> InstanceScaleVec;
      typedef std::vector<DtElementID> InstanceElementIdVec;
      typedef std::vector<InstancedLight> InstancedLightVec;

      inline InstanceElementIdVec& elementIds() { return myElementIds; }

      inline InstanceScaleVec& instanceScales() { return myInstanceScales; }

      inline ByteVec& instanceVisibleSet() { return myInstanceVisibleSet; }

      inline InstanceMatrixVec& drawableInstanceMatrix() { return myDrawableInstanceMatrix; }

      inline InstanceCullDataVec& instanceCullData() { return myInstanceCullData; }

      inline InstanceStateVec& instances() { return myInstances; }

      inline TextureCoordinateVec& uvOffsets() { return myUVOffsets; }

      inline InstancedLodVec& instancedLods() { return myInstancedLods; }

      inline InstancedSwitchVec& instancedSwitches() { return myInstancedSwitches; }

      inline InstancedSequenceVec& instancedSequences() { return myInstancedSequences; }

      inline InstancedDofVec& instancedDofs() { return myInstancedDofs; }

      typedef std::vector<InstanceTemplate> InstanceTemplateVec;

      inline InstanceTemplateVec& instanceTemplates() { return myInstanceTemplates; }

      inline std::vector<float>& distanceToEye() { return myDistanceToEye; }

      inline DofAnimations& dofAnimations() { return myDofAnimations; }

      inline double& lastUpdateTime() { return myLastUpdateTime; }

      inline DtOsgCullVisitorParameters* lastCullParameters() { return myLastCullParameters.get(); }

      inline unsigned int visibleSetStride() { return myVisibleSetStride; }

      inline BoolVec& visibleSet() { return myVisibleSet; }

      inline osg::Vec3d& cameraPosition() { return myCameraPosition; }

      inline osg::View* lastView() { return myLastView; }

      inline unsigned int drawableIndex(unsigned int instanceIx);

      void reloadShaders();

      inline GeometryVec &drawables(){ return myDrawables;};

   protected:
      BoolVec myVisibleSet;

      ByteVec myInstanceVisibleSet;

      TextureCoordinateVec myUVOffsets;

      InstanceMatrixVec myDrawableInstanceMatrix;

      InstanceCullDataVec myInstanceCullData;

      InstanceScaleVec myInstanceScales;

      InstanceElementIdVec myElementIds;

      FloatVec myInstanceUserDefined;

      InstanceStateVec myInstances;
      InstancedLodVec myInstancedLods;
      InstancedSwitchVec myInstancedSwitches;
      InstancedSequenceVec myInstancedSequences;
      InstancedDofVec myInstancedDofs;
      InstancedLightVec myInstancedLights;

      InstanceTemplateVec myInstanceTemplates;

      std::vector<float> myDistanceToEye;

      DofAnimations myDofAnimations;

      double myLastUpdateTime;

      unsigned int myLastNodeMask;

      unsigned int myVisibleSetStride;

      osg::Vec3d myCameraPosition;

      osg::View* myLastView;

   protected:

      friend class DtInstanceNode;

      struct Eyepoint 
      {
         osg::Vec3d center;
         osg::Vec3d eye;
         osg::Vec3d up;
      };

      struct Frustum
      {
         double l, r, b, t, n, f;
      };
      
      struct InstancingHit
      {
         unsigned int instanceIx;
         unsigned int drawableIx;
         osg::Vec3f pos;
         osgUtil::LineSegmentIntersector::Intersection result;
      };

   protected:
      void update(double simTime, osg::View* view, DtOsgCullVisitorParameters* params, unsigned int nodeMask);
      
      void registerGeometry(const std::string& modelType, osg::Geometry* geometry, VpMap& vpMap);

      float computeDistanceToEye(const Eyepoint& eye, const osg::Vec3d& pos) const;

      void allocateRenderList();

      unsigned int updateBuffers( osg::View* view, DtGLTextureBufferObject* tbo, const ArrayRange& drawableRange, unsigned int bufferOffset, osg::Vec4* bufferStorage );

      unsigned int updateRenderList( const Eyepoint& eye, const Frustum& frustum, const osg::Matrixd& viewMat, unsigned int nodeMask);

      void allocateInstanceData();

      unsigned int fillRenderList(unsigned visibleSetStride, unsigned int instanceOffset, unsigned int numInstances );

      void cullRange(const Eyepoint& eye, unsigned int instanceOffset, unsigned int instanceCount,
         std::vector<osg::Vec4d>& planes, bool useScale, unsigned int nodeMask );

      void slot_channelAdded(DtChannelManager* mgr, DtChannel* channel);

      void slot_channelRemoved(DtChannelManager* mgr, DtChannel* channel);

      void createHitList(const osg::Vec3d& end, const osg::Vec3d& start, std::vector<InstancingHit>& hitList);

      void loadInstanceShaders(osg::ref_ptr<osgEarth::VirtualProgram> vp);


   protected:

      osg::Vec4* myDynamicBufferData;
      unsigned int myDynamicBufferDataSz;
      
      struct RenderState
      {
         InstanceMatrix l2w;
         float scale;
         float userDefined;
         osg::Vec2f uvOffset;
         osg::Vec3f filterColor;
      };

      typedef std::vector<RenderState> RenderStateVec;

      struct GeometryRenderList
      {
         unsigned int count;

         RenderStateVec states;
      };

      typedef std::vector<GeometryRenderList> RenderList;

      RenderList myRenderList;

      bool myRenderListDirty;

      typedef boost::unordered_map<DtUniqueID, unsigned int> InstanceStateMap;

      InstanceStateMap myIdToInstances;

      struct InstanceBuffer
      {
         osg::View* view;

         DtGLTextureBufferObject* image;
      };

      typedef std::vector<InstanceBuffer> InstanceBufferVec;

      InstanceBufferVec myInstanceBuffers;

      typedef boost::unordered_map<std::string, int> modelTypeToTemplateMap;

      modelTypeToTemplateMap myModelTypeToTemplate;

      DtDe& myDe;

      GeometryVec myDrawables;

      unsigned int myRequiredUvOffsets;

      unsigned int myNumThreads;

      typedef boost::unordered_map<unsigned int, unsigned int> PerTemplateCount;

      PerTemplateCount myTemplateMaxInstances;

      PerTemplateCount myTemplateActiveInstances;

      VpMap myVpMap;
      VpRefList myVpRefs;

      typedef boost::unordered_map<std::string, osg::ref_ptr<osg::Geode> > modelTypeToGroup;

      modelTypeToGroup myModelTypeToGeometryGroup;

      class DT_DLL_VRVOSG AlwaysCull : public osg::Drawable::CullCallback
      {
         virtual bool cull(osg::NodeVisitor* nv, osg::Drawable* drawable, osg::RenderInfo* renderInfo) const
         {
            return true;
         }
      };

      osg::ref_ptr<AlwaysCull> myAlwaysCullCallback;

      osg::ref_ptr<DtInstanceNode> myInstanceRoot;

      typedef std::vector<DtInstanceProcessor*> InstanceProcessorVec;

      InstanceProcessorVec myPostCullInstanceProcessors;

      InstanceProcessorVec myPreCullInstanceProcessors;

      typedef std::vector<osg::View*> ViewVec;
      ViewVec myKnownViews;

      typedef boost::unordered_map<osg::View*, osg::ref_ptr<DtViewSpecificGroup> > ViewToGroup;
      ViewToGroup myViewToGroup;

      typedef boost::unordered_map<osg::Geometry*, osg::ref_ptr<DtPerViewUniformCallback> > GeometryToCallback;
      GeometryToCallback myGeometryToCallback;

      osg::ref_ptr<DtOsgCullVisitorParameters> myLastCullParameters;
   };

   class DT_DLL_VRVOSG DtInstanceManagerCreator : public DtVirtualBaseClass
   {
   public:
      virtual ~DtInstanceManagerCreator();

      static DtInstanceManagerCreator& instance(DtDe& de);

      virtual DtInstanceManager* create();

   protected:
      DtInstanceManagerCreator(DtDe& de);

   protected:
      DtDe& myDe;
   };
}