DtIndirectGeometryAggregator.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/DtIndirectGeometryTextures.h>
#include <vrvOsg/DtIndirectEffectTexturesTemplate.h>
#include <vrvOsg/DtIndirectSwitchManager.h>
#include <vrvOsg/DtIndirectPolygonOffset.h>
#include <vrvOsg/DtIndirectBlendFunction.h>
#include <vrvOsg/DtIndirectBlendEquation.h>
#include <vrvOsg/DtIndirectLodManager.h>
#include <vrvOsg/DtIndirectDofManager.h>
#include <vrvOsg/DtIndirectUvOffsetManager.h>
#include <vrvOsg/DtIndirectIndexedMultiSwitch.h>

#include <vrvIndirectRendering/DtMaterial.h>
#include <vrvMath/DtAxisAlignedBoundingBox.h>

#include <vrvUtil/DtVirtualBaseClassCreator.h>

#include <osg/NodeVisitor>
#include <osg/Geometry>
#include <osg/ref_ptr>

#include <stack>


namespace osg
{
   class Texture;
   class LOD;
}

namespace osgSim
{
   class MultiSwitch;
   class DOFTransform;
}

namespace makVrv
{
   class DtIndirectTextureCollectionPolicy;
   class DtIndirectLimitTextureStrategy;
   class DtIndirectGeometryStateManager;

   class DtDynamicTerrainSwitchNode;
   class DtDynamicTerrainSequenceNode;
   class DtIndirectAggregateHelper;
   class DtIndirectModelInfoContainer;
   class DtLightPointNode;
   class DtIndirectModelInfo;
   class DtIndirectDefaultTexturesContainer;
   class DtIndirectTexturePrototypeManager;
   class DtIndirectModelRegistrationInfo;

   class DtDe;

   struct DtCopyModelDataStructure;

   class DtIndirectGeometryAggregator;

   typedef DtVirtualBaseClassCreator<DtIndirectGeometryAggregator
      , DtIndirectModelInfoContainer&
      , DtIndirectEffectTexturesTemplate&
      , DtIndirectGeometryStateManager& 
      , const DtIndirectTextureCollectionPolicy&
      , const DtIndirectLimitTextureStrategy& > DtIndirectGeometryAggregatorCreator;

   template <>
   inline const char* creatorTypeName<DtIndirectGeometryAggregator>(void) { return "DtIndirectGeometryAggregatorCreator"; }

   class DT_DLL_VRVOSG DtIndirectGeometryAggregator : public osg::NodeVisitor
   {
   public:
      friend DtIndirectGeometryAggregatorCreator;
      
      virtual ~DtIndirectGeometryAggregator();

   protected:
      DtIndirectGeometryAggregator(DtDe& de
         , DtIndirectModelInfoContainer& aggregatedGeometry
         , DtIndirectEffectTexturesTemplate& aggregatedEffectTexturesState
         , DtIndirectGeometryStateManager& stateManager
         , const DtIndirectTextureCollectionPolicy& textureCollectionPolicy
         , const DtIndirectLimitTextureStrategy& limitTextureStrategy
      );


   private:
      //Default constructor not implemented
      DtIndirectGeometryAggregator();
      DtIndirectGeometryAggregator(const DtIndirectGeometryAggregator &other);
      DtIndirectGeometryAggregator &operator=(const DtIndirectGeometryAggregator &other);

   public:

      virtual void aggregate(const DtIndirectModelRegistrationInfo& info);

      virtual void apply(osg::Geometry& geometry);

      virtual void apply(DtDynamicTerrainSwitchNode& node);

      virtual void apply(DtIndirectIndexedMultiSwitch& node);

      virtual void apply(osg::Geode& geode);

      virtual void apply(osg::Node& node);

      virtual void apply(osg::Switch& node);

      virtual void apply(osgSim::MultiSwitch& node);

      virtual void apply(osg::LOD& node);

      virtual void apply(osg::Sequence& node);

      virtual void apply(osg::Billboard& node);

      virtual void apply(DtLightPointNode& node);

      virtual void augmentModelInfo(osg::Geometry* geometry
      , const AxisAlignedBoundingBox_32& bbox32
      , const int state
      , const DtIndirectGeometryTextures& texturesForThisGeometry
      , Material& materialForThisGeometry
      , const bool isLight);
      
      virtual void apply(osg::MatrixTransform& matTrans);

      virtual void apply(osgSim::DOFTransform& node);

      virtual void apply(osg::LightSource& node) override;

      virtual int numGeodes() const;

      virtual int numGeometries() const;

   protected:
      virtual Material gatherMaterial(osg::StateSet* ss) const;

      virtual bool isCullFaceEnabled(const osg::StateSet* ss) const;

      virtual bool isBlendingEnabled(const osg::StateSet* ss) const;

      virtual bool isAlphaToCoverageEnabled(const osg::StateSet* ss) const;

      virtual DtIndirectBlendFunction gatherBlendFunction(const osg::StateSet* ss) const;

      virtual DtIndirectBlendEquation gatherBlendEquation(const osg::StateSet* ss) const;

      virtual DtIndirectPolygonOffset gatherPolygonOffset(const osg::StateSet* ss) const;

      virtual bool isDepthOnly(const osg::StateSet* ss) const;

      virtual void gatherTextures(DtIndirectGeometryTextures& outTextures, const osg::StateSet* ss, const std::string& modelFileName);

      virtual osg::Geometry* createDebugLightPointCube(DtLightPointNode& node);
      
      virtual osg::Geometry* createDebugLightSourceCube(osg::LightSource& node);
     
      virtual void applyToSwitchStack(osg::Group* node, osg::NodePath& nodePath);

      virtual void computeIndicesOffetAndCount(osg::Geometry* geomPtr, int& indicesOffset, int& indicesCount);

   protected:
      int myNumGeodes;
      int myNumGeometries;
      int myModelId;

      int myLastIndexOffset;
      
      unsigned int myMaxUvOffsetId;
      DtIndirectUvOffsetManager::AnnotationToUvOffsetMap myAnnotationToIdMap;
      DtIndirectUvOffsetManager::ScrollableUvOffsetIdsAndMultipliers myScrollableUvOffsetIdsAndMultipliers;
      
      std::stack<unsigned int> myLastUvOffsetIdStack;

      DtIndirectEffectTexturesTemplate& myAggregatedEffectTexturesState;
      DtIndirectGeometryStateManager& myStateManager;
      DtIndirectModelInfoContainer& myModelInfoContainer;
      
      struct AggregatorData
      {
         DtIndirectGeometryTextures myTextures;
         Material myMaterial;
         bool myCullFaceEnabled;

         bool myBlendingEnabled;

         bool myAlphaToCoverageEnabled;

         DtIndirectBlendFunction myBlendFunction;

         DtIndirectBlendEquation myBlendEquation;
         
         DtIndirectPolygonOffset myPolygonOffset;

         bool myDepthOnly;
      };

      std::stack<AggregatorData> myDataStack;
      std::stack<osg::Matrix> myMatrixStack;

      std::stack<int> myBillboardModeStack;

      DtIndirectLodManager::DequeIndirectLodInfos myLodStack;

      DtIndirectSwitchManager::DequeNodeSwitchInfoPairs mySwitchStack;

      // This stack (unlike the terrain/find nodes visitor is not a regular stack 
      // and needs to maintain the full hierarchical paths as it traverses down
      DtIndirectDofManager::DequeNodeDofInfoPairs myDofStack;
      int myMaxDofId;
      int myCurrentParentId;

      const DtIndirectTextureCollectionPolicy& myTextureCollectionPolicy;

      // used only for outputting debug messages
      std::string myModelFileName;

      std::string myModelDefName;

      Material myDefaultMaterial;

      DtIndirectGeometryTextures myDefaultTextures;

      friend class ScopedStackOperator;
      friend class ScopedAnnotationParser;

      DtDe& myDe;

      bool myLightSourcesDaylightControlled;

      DtIndirectDefaultTexturesContainer& myDefaultTexturesContainer;

      const DtIndirectLimitTextureStrategy& myLimitTextureStrategy;

      bool myModelIsOcean;

      bool myModelIsShadowCastOff;

      DtIndirectTexturePrototypeManager& myTexturePrototypeManager;

      DtCopyModelDataStructure* myCopyModelDataStructure;

      DtIndirectIndexedMultiSwitch* myCurrentIndexedMultiSwitch = nullptr;

      const DtIndirectModelSubGeometry::VertexRange* myVerticesRange = nullptr;

      const DtIndirectModelSubGeometry::IndexRange* myIndicesRange = nullptr;
   };
}