DtOsgFindNodesVisitor.h

Go to the documentation of this file.

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


#pragma once

#include <vrvOsg/vrvOsg.h>

#include <vrvOsg/DtDynamicTerrainSwitchNode.h>
#include <vrvOsg/DtDynamicTerrainSequenceNode.h>
#include <vrvOsg/DtOsgConvertUtils.h>
#include <vrvOsg/DtIndirectSwitchInfo.h>
#include <vrvOsg/DtIndirectLodManager.h>
#include <vrvOsg/DtIndirectSwitchManager.h>
#include <vrvOsg/DtIndirectDofManager.h>
#include <vrvOsg/DtIndirectPolygonOffset.h>

#include <vrvGraphicsUtils/DtAssert.h>

#include <osg/Switch>
#include <osg/LOD>
#include <osg/PositionAttitudeTransform>
#include <osg/NodeVisitor>
#include <osg/MatrixTransform>
#include <osgSim/MultiSwitch>
#include <osgSim/DOFTransform>
#include <osgSim/LightPointNode>

#include <vector>
#include <deque>
#include <osg/PolygonOffset>

namespace makVrv
{

   struct DtNodePlusTraversalInfo
   {
      DtNodePlusTraversalInfo()
         : cullFaceEnabled(true)
      {
      }

      osg::ref_ptr<osg::Node> node;
      osg::Matrixd xform;
      DtIndirectLodManager::DequeIndirectLodInfos lodStack;
      osg::NodePath nodePath;
      DtIndirectSwitchManager::DequeNodeSwitchInfoPairs switchStack;
      DtIndirectDofManager::DequeNodeDofInfoPairs dofStack;
      bool cullFaceEnabled;
      DtIndirectPolygonOffset polygonOffset;
   };

   template <typename T>
   struct DtOsgFindNodesVisitor : public osg::NodeVisitor
   {
      DtOsgFindNodesVisitor(osg::NodeVisitor::TraversalMode traversalMode);

      void apply(osg::Node& node);

      std::vector<T*> results;
   };

   template<typename T>
   struct DtOsgFindNodesVisitorRefPtr : public osg::NodeVisitor
   {
      DtOsgFindNodesVisitorRefPtr(osg::NodeVisitor::TraversalMode traversalMode);

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

      std::vector< osg::ref_ptr<T> > results;
   };

   template<typename T>
   struct DtOsgFindNodesVisitorRefPtrFlattenGraph : public osg::NodeVisitor
   {
      DtOsgFindNodesVisitorRefPtrFlattenGraph(osg::NodeVisitor::TraversalMode traversalMode
         , osg::Node::NodeMask gatherNodeMask = 0
         , const std::set< osg::Node* >& nodeIgnoreList = std::set<osg::Node*>()
         , bool adjustLodStart = false);

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

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

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

      // Avoid use of T to avoid shadowing class template name
      template<typename RESULT>
      bool checkThisNodesMask(osg::Node& node, RESULT* result);

      bool checkThisNodesMask(osg::Node& node, DtLightPointNode* result);

      bool checkThisNodesMask(osg::Node& node, osgSim::LightPointNode* result);

      std::vector< DtNodePlusTraversalInfo > results;
      std::set< osg::Node* > nodeIgnoreList;
      std::deque<osg::Matrixd> matrixStack;
      DtIndirectLodManager::DequeIndirectLodInfos lodStack;
      DtIndirectSwitchManager::DequeNodeSwitchInfoPairs switchStack;
      osg::Node::NodeMask myGatherNodeMask;

      std::stack<bool> myCullFaceEnabledStack;
      std::stack<DtIndirectPolygonOffset> myPolygonOffsetStack;
      bool myAdjustLodStart;

      // This stack (unlike the aggregator is a regular stack that
      // can use the stack size for finding the parent dof ids by pushing/popping
      DtIndirectDofManager::DequeNodeDofInfoPairs myDofStack;
   };
}

#include "DtOsgFindNodesVisitor.inl"