DtIndirectGeometry.h

Go to the documentation of this file.

/******************************************************************************
** Copyright (c) 2018 VT-MAK Technologies, Inc.
** All rights reserved.
******************************************************************************/


#pragma once

#include <vrvOsg/vrvOsg.h>

#include <vrvOsg/DtIndirectGeometryRegistrar.h>

#include <vrvIndirectRendering/DtInstanceContainer.h>
#include <vrvIndirectRendering/DtGpuCullingAndLodDefines.h>

#include <vrvUtil/DtVirtualBaseClass.h>

#include <osg/Drawable>
#include <osg/Uniform>

#include <boost/unordered_map.hpp>
#include <tbb/atomic.h>
#include <tbb/concurrent_queue.h>

#include <string>

#define INDIRECTGEOMETRY_CONSTRUCT_DESTRUCT_DEBUG 0

namespace osg
{
   class NodeVisitor;
   class State;
}

namespace osgUtil
{
   class CullVisitor;
}

namespace makVrv
{
   class DtIndirectGeometryRegistrar;
   class DtGpuCullingAndLod;
   class DtGlBuffer;
   class DtOsgShaderFilesCache;

   class DT_DLL_VRVOSG DtIndirectGeometryStateManager
   {
   public:
      DtIndirectGeometryStateManager();
      
      ~DtIndirectGeometryStateManager();

   public:

      void insertMode(int index, bool enable, int mode);

      void applyMode(int index) const;

      void restoreAllModes(osg::State &state) const;

      int numberOfStates() const;

   protected:

      typedef std::pair<bool, int> GlModeSetting;
      typedef std::vector<GlModeSetting> GlModeVec;
      GlModeVec myModesVec;

      typedef std::set<int> GlModeSet;
      GlModeSet myModeSet;
   };

   class DT_DLL_VRVOSG DtIndirectGeometry : public osg::Drawable
   {
   public:


      enum IndirectStates
      {
         CullFaceOnState = 0,
         CullFaceOffState = 1,
         NumStates = 2
      };

   public:
      DtIndirectGeometry(DtIndirectGeometryRegistrar* registrar, bool viewFrustumCulling, bool lodChecking, bool useInstanceCount);

      virtual void drawImplementation(osg::RenderInfo& renderInfo) const;

      virtual void addInstance(const DtInstance_64& instance);

      virtual void removeAllInstances(void);

      virtual void accept(osg::NodeVisitor& nv);

      virtual void setParent(const osg::Group* parent);

      virtual void setDebugDrawString(const std::string& drawString);

      virtual void setRegistrarInvalid(void);

      virtual void releaseGLObjects(osg::State* state) const;

   protected:
      DtIndirectGeometry();

      virtual ~DtIndirectGeometry();


      virtual void addInstanceActual(const DtInstance_64& instance);
      virtual bool actuallyAddQueuedUpInstances(void);

      virtual void setVisibility(bool visibility);

      virtual GLuint getOrCreateAndRecordVAO(const osg::RenderInfo& renderInfo) const;
      virtual void clearVAO(const osg::RenderInfo& renderInfo);
      virtual void recordVAO(GLuint vao) const;

      virtual void slot_rebuild(const DtIndirectGeometryRegistrar::RebuildResult& result);

      virtual void setUpShadersIfNecessary(osgUtil::CullVisitor* cullVisitor);

      virtual void installIndirectRenderingProgram(DtOsgShaderFilesCache* shaderFilesCache, const std::string& shaderFolderPath, osg::StateSet* ss) const;

      virtual void installDefaultMaterial(osg::StateSet* ss);

   protected:
      DtIndirectGeometryRegistrar* myRegistrar;

      DtInstanceContainer myInstanceContainer;

      tbb::concurrent_queue<DtInstance_64> myQueueInstancesToAdd;

      typedef boost::unordered_map<unsigned int, GLuint> MapContextToVAO;
      mutable MapContextToVAO myMutableMapContextToVAO;

      DtGpuCullingAndLod* myCullingAndLOD;

      mutable DtIndirectRenderingParameters myMutableIrParams;

      MapIntToInt myModelIDsToInstances;

      osg::ref_ptr<osg::Uniform> myViewMatrix;
      osg::ref_ptr<osg::Uniform> myProjectionMatrix;

      std::string myDrawString;

      mutable bool myMutableShadersSetup;

      DtIndirectGeometryStateManager myStateManager;

#if INDIRECTGEOMETRY_CONSTRUCT_DESTRUCT_DEBUG
      static tbb::atomic<int> myRefCount;
#endif
   };

   class DT_DLL_VRVOSG DtIndirectGeometryCreator : public DtVirtualBaseClass
   {
   public:
      DtIndirectGeometryCreator(DtDe &de);

      virtual ~DtIndirectGeometryCreator();

      virtual DtIndirectGeometry* create(DtIndirectGeometryRegistrar* registrar, bool viewFrustumCulling, bool lodChecking, bool useInstanceCount) const;

      static DtIndirectGeometryCreator& instance(DtDe& de);

      static const std::string& theClassName();
   protected:
      DtDe& myDe;
   };

}