DtSceneGraphTreeLogic.h

Go to the documentation of this file.

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

#pragma once

#include <vrvOsgQt/vrvOsgQt.h>

#if !defined(Q_MOC_RUN)
#include <vrvCore/DtSubjectObserver.h>
#endif

#include <osg/observer_ptr>
#include <osg/Node>
#include <osg/ShapeDrawable>

#include <QtWidgets/QAction>

#if !defined(Q_MOC_RUN)
#include <tbb/task.h>
#endif

#include <vector>

namespace osg
{
   class Node;
   class Drawable;
   class Geode;
}

class QIcon;
class QMenu;
class QTreeWidgetItem;

namespace makVrv
{

   class DtDe;
   class DtSceneGraphTreeWidgetInterface;
   class DtSceneGraphTreeLogic;

   class DT_DLL_VRVOSGQT ExpansionFunctor
   {
   public:
      virtual bool shouldExpandTo(const osg::Node* node);
   };

   //custom actions for context menu
   class DT_DLL_VRVOSGQT DtDrawBoundsAction : public QAction, public ExpansionFunctor
   {
      Q_OBJECT
   public:
      DtDrawBoundsAction(QObject* parent);
      virtual ~DtDrawBoundsAction();

   public:
      virtual void removeVolume();
      virtual void setNode(const osg::Node* node);
      virtual bool isNodeWithBV(const osg::Node* node);

      virtual void setAABoundingBoxMode();

   public slots:

      void slot_triggered(bool);

   private:
      osg::observer_ptr<const osg::Node> myNode;
      osg::observer_ptr<const osg::Node> myBVAttachedNode;
      osg::observer_ptr<osg::ShapeDrawable> myBoundingVolumeDrawable;
      osg::ref_ptr< osg::Image > myImage;
      osg::ref_ptr< osg::Texture > myTexture;
      bool myAABoundingBoxMode;
   };

   //custom actions for context menu
   class DT_DLL_VRVOSGQT DtRefreshNodeAction : public QAction
   {
      Q_OBJECT
   public:
      DtRefreshNodeAction(QObject* parent, DtSceneGraphTreeLogic* tree);
      virtual ~DtRefreshNodeAction();
      virtual void setNode(const osg::Node* node, QTreeWidgetItem* item);
      virtual bool shouldExpandTo(const osg::Node* node);

      public slots:
      void slot_triggered(bool);

   private:
      osg::observer_ptr<const osg::Node> myNode;
      DtSceneGraphTreeLogic* myTree;
      QTreeWidgetItem* myItem;

   };

   class DT_DLL_VRVOSGQT DtColorSubTreeAction : public QAction
   {
      Q_OBJECT
   public:
      DtColorSubTreeAction(QObject* parent);

      virtual ~DtColorSubTreeAction();

   public:
      
      virtual void setNode(osg::Node* node);
      
      virtual osg::Node* colorizedNode();

   public slots:

      virtual void slot_triggered(bool);

   protected:
      virtual void unColorizeNode();
      virtual void colorizeNode();

   protected:
      osg::observer_ptr<osg::Node> myContextMenuNode;
      osg::observer_ptr<osg::Node> myColorizedNode;
   };

   //custom actions for context menu
   class DT_DLL_VRVOSGQT DtExpandItemAction : public QAction, public ExpansionFunctor
   {
      Q_OBJECT
   public:
      DtExpandItemAction(QObject* parent, DtSceneGraphTreeLogic* tree);
      virtual ~DtExpandItemAction();
      virtual void setNode(const osg::Node* node, QTreeWidgetItem* item);
      virtual bool shouldExpandTo(const osg::Node* node);
      virtual void recursiveVisit(QTreeWidgetItem* item);

      public slots:
      void slot_triggered(bool);

   protected:
      osg::observer_ptr<const osg::Node> myNode;
      DtSceneGraphTreeLogic* myTree;
      QTreeWidgetItem* myItem;
   };

   class DT_DLL_VRVOSGQT DtExpandAllAction : public DtExpandItemAction
   {
      Q_OBJECT
   public:
      DtExpandAllAction(QObject* parent, DtSceneGraphTreeLogic* tree);
      virtual bool shouldExpandTo(const osg::Node* node) { return true; };

   };

   class DT_DLL_VRVOSGQT DtExpandToOQNAction : public DtExpandItemAction
   {
      Q_OBJECT
   public:
      DtExpandToOQNAction(QObject* parent, DtSceneGraphTreeLogic* tree);

      virtual bool shouldExpandTo(const osg::Node* node);

   };

   class DT_DLL_VRVOSGQT DtExpandToGeometryAction : public DtExpandItemAction
   {
   public:
      DtExpandToGeometryAction(QObject* parent, DtSceneGraphTreeLogic* tree) :DtExpandItemAction(parent, tree) 
      {
         //Do Nothing
      }
      virtual bool shouldExpandTo(const osg::Node* node);
   };

   class DT_DLL_VRVOSGQT DtExpandToLightsAction : public DtExpandItemAction
   {
   public:
      DtExpandToLightsAction(QObject* parent, DtSceneGraphTreeLogic* tree) :DtExpandItemAction(parent, tree) 
      {
         //Do Nothing
      }
      virtual bool shouldExpandTo(const osg::Node* node);
   };

   //custom actions for context menu
   class DT_DLL_VRVOSGQT DtAddOcclusionNodeAction : public QAction
   {
      Q_OBJECT

   public:
      DtAddOcclusionNodeAction(QObject* parent, DtSceneGraphTreeLogic* tree);
      virtual ~DtAddOcclusionNodeAction();

   public:
      virtual void setNode(const osg::Node* node, QTreeWidgetItem* item);
   
   public slots:
      virtual void slot_triggered(bool);

   private:
      osg::observer_ptr<const osg::Node> myNode;
      DtSceneGraphTreeLogic* myTree;
      QTreeWidgetItem* myItem;

   };

   class DT_DLL_VRVOSGQT DtSearchSceneGraphAction
   {
      public:
         DtSearchSceneGraphAction(DtSceneGraphTreeLogic* tree);
         virtual ~DtSearchSceneGraphAction();

         virtual QTreeWidgetItem* getPreviousNode(const std::string& query);

         virtual QTreeWidgetItem* getNextNode(const std::string& query);

         virtual void setSearchQuery(const std::string& query);

         virtual void search(QTreeWidgetItem* node);

         virtual void reset();

         virtual size_t getCurrentIndex();

         virtual bool hasResults(const std::string& query);

         virtual size_t getTotalResults(const std::string& query);

      private:
         size_t myCurrentIndex;
         std::string mySearchQuery;
         std::vector<QTreeWidgetItem*> mySearchResults;

         DtSceneGraphTreeLogic* myTree;
   };

   class DT_DLL_VRVOSGQT DtSceneGraphTreeLogic
   {
   public:
      typedef osg::observer_ptr<const osg::Node> NodeObsPtr;
      typedef std::map<QTreeWidgetItem*, NodeObsPtr> WidgetToNodeMap;
   
   public:
   
      enum SceneGraphRoot
      {
         RenderSystem,
         WindowChannelConfig,
         Environment,
         Terrain,
         Props,
         ScaledTerrainRoot,
         ScaledPropsRoot,
         ModelSet0,
         ModelSet1,
         ModelSet2,
         ModelSet3,
         Overlays
      };

   protected:

      DtSceneGraphTreeLogic(DtDe& de, DtSceneGraphTreeWidgetInterface* iface);

      virtual void init();

      friend class DtSceneGraphTreeLogicCreator;

   public:
      virtual ~DtSceneGraphTreeLogic();

   private:
      DtSceneGraphTreeLogic(const DtSceneGraphTreeLogic &other);
      DtSceneGraphTreeLogic &operator=(const DtSceneGraphTreeLogic &other);

   public:

      virtual void clearAndTraverse();

      virtual void clearAndTraverse(int domain);

      virtual bool iterateNode(QTreeWidgetItem* pParent, const osg::Node* pNode, bool bDrawIcon = true);

      virtual void buildContextMenu(QMenu* menu, QTreeWidgetItem* item);

      NodeObsPtr  getNode(QTreeWidgetItem* item);

      virtual void displayPrevious(const std::string& query);
      virtual void displayNext(const std::string& query);
      virtual std::string getSearchStatusText(const std::string& query);

      virtual void resetSearchAction();

   private:

      DtSceneGraphTreeLogic();
      const QIcon & getNodeIcon(const osg::Node* pNode);
      const QIcon & getDrawableIcon(const osg::Drawable* drawable);

   protected:

      DtDe& myDe;
      DtSceneGraphTreeWidgetInterface* myWidgetInterface;

      WidgetToNodeMap myWidgetToNodeMap;
      DtDrawBoundsAction* myShowBoundsAction;
      DtDrawBoundsAction* myShowAABBoxAction;
      DtColorSubTreeAction* myColorSubTreeAction;
      DtRefreshNodeAction* myRefreshAction;
      DtExpandAllAction* myExpandAllAction;
      DtExpandToGeometryAction* myExpandGeometryAction;
      DtExpandToOQNAction* myExpandOcclusionNodesAction;
      DtExpandToLightsAction* myExpandLightNodesAction;
      DtAddOcclusionNodeAction* myOcclusionNodeAction;
      DtSearchSceneGraphAction* mySearchAction;
      QIcon mySunIcon;
      QIcon mySkyIcon;
      QIcon myFolderIcon;
      QIcon myGeodeIcon;
      QIcon myGeometryIcon;
      QIcon myEmptyIcon;
      QIcon myCloudsIcon;

   };

   class DT_DLL_VRVOSGQT DtSceneGraphTreeLogicCreator : public DtVirtualBaseClass
   {
   public:
      // CTOR
      DtSceneGraphTreeLogicCreator(DtDe& de);

      virtual ~DtSceneGraphTreeLogicCreator();

   private:
      DtSceneGraphTreeLogicCreator(const DtSceneGraphTreeLogicCreator &other);
      DtSceneGraphTreeLogicCreator &operator=(const DtSceneGraphTreeLogicCreator &other);

   public:
      static DtSceneGraphTreeLogicCreator& instance(DtDe& de);

      static void registerInstance(DtDe& de,
         DtSceneGraphTreeLogicCreator* instance);

      virtual DtSceneGraphTreeLogic* create(DtSceneGraphTreeWidgetInterface* iface);
   protected:
      DtDe& myDe;
   };
}