DtCellIndirectGeo.h

Go to the documentation of this file.

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


#pragma once

#include <vrvOsg/vrvOsg.h>

#include <vrvOsg/DtOsgThreadingUtils.h>
#include <vrvOsg/DtIndirectSwitchManager.h>
#include <vrvOsg/DtIndirectLodManager.h>
#include <vrvOsg/DtIndirectDofManager.h>

#include <osg/Matrixd>
#include <osg/Group>

#include <atomic>
#include <unordered_map>
#include <functional>

#define DTCELLINDIRECTGEO_CONSTRUCT_DESTRUCT_DEBUG 0

namespace makVrv
{
   class DtDe;
   class DtCullShaderConfig;
   class DtIndirectModelsContainer;
   class DtIndirectGeometry;
   class DtIndirectFadeParameters;
   class DtCullShaderConfig;
   enum class DtIndirectGeometrySubType;

   class DT_DLL_VRVOSG DtCellIndirectGeoInstance
   {
   public:
      DtCellIndirectGeoInstance(const osg::Matrixd& instanceXForm);

      virtual ~DtCellIndirectGeoInstance();

   private:
      DtCellIndirectGeoInstance() = delete;
      DtCellIndirectGeoInstance(const DtCellIndirectGeoInstance& other) = delete;
      DtCellIndirectGeoInstance& operator-(const DtCellIndirectGeoInstance& other) = delete;
         
   public:
      int myModelId;
      const osg::Matrixd& myInstanceXForm;
      const osg::Node* myModelNode;

      osg::NodePath myNodePath;
      bool myAddToIntersectionGroup;
      bool myAddTopLevelSwitch;

      DtIndirectLodManager::DequeIndirectLodInfos myLodStack;
      DtIndirectSwitchManager::DequeNodeSwitchInfoPairs mySwitchStack;
      DtIndirectDofManager::DequeNodeDofInfoPairs myDofStack;

      std::function<void(const osg::ref_ptr<osg::Group>& origionalNode, osg::ref_ptr<osg::Group>& newRootNode, osg::NodePath& newNodePath)> myNodeInsertionCallback;
   };

   class DT_DLL_VRVOSG DtCellIndirectGeo
   {
   public:
      DtCellIndirectGeo(DtDe& de
         , const DtCullShaderConfig& cullShaderConfig
         , bool includeIntersectionGeoInNodePath
         , const DtIndirectGeometrySubType& geometrySubType
         , const DtIndirectFadeParameters& fadeParameters);

      virtual ~DtCellIndirectGeo();

   private:
      DtCellIndirectGeo() = delete;
      DtCellIndirectGeo(const DtCellIndirectGeo &other) = delete;
      DtCellIndirectGeo &operator=(const DtCellIndirectGeo &other) = delete;

   public:

      virtual void setModelsContainer(DtIndirectModelsContainer* modelsContainer, bool ownsModelsContainer);

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

      virtual const osg::BoundingBoxd& boundingBox(void) const;

      virtual const DtIndirectGeometry* indirectGeometry(void) const;
      virtual DtIndirectGeometry* indirectGeometry(void);

      virtual osg::Group*  indirectIntersectionGroup();

      virtual int addInstance(const DtCellIndirectGeoInstance& cellIndirectGeoInstance);

      virtual const std::unordered_map<int, int>& modelIdToNumInstances(void) const;
      
      virtual const std::unordered_map<int, osg::ref_ptr<osg::Node>>& modelIdToModelNodeMap(void) const;

   protected:
      virtual void expandBoundingBox(const osg::Node* modelNode, const osg::Matrixd& instanceXForm);

   protected:
      DtIndirectModelsContainer* myModelsContainer;
      bool myOwnsModelsContainer;

#if DTCELLINDIRECTGEO_CONSTRUCT_DESTRUCT_DEBUG
      static std::atomic<int> myBaseRefCount;
#endif

      osg::ref_ptr<DtIndirectGeometry> myIndirectGeometry;
      osg::ref_ptr<osg::Group> myIndirectGeoIntersectionGroup;

      osg::BoundingBoxd myBoundingBox;
      mutable Threading::Mutex myMutableBoundingBoxMutex;

      DtDe& myDe;
      bool myFirstInstance;

      bool myIncludeIntersectionGeoInNodePath;

      std::unordered_map<int, int> myModelIdToNumInstances;
      std::unordered_map<int, osg::ref_ptr<osg::Node>> myModelIdToModelNode;
   };
}