DtSceneGraphTreeLogic.h

Go to the documentation of this file.

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

#pragma once

#include <vrvOsgQt/vrvOsgQt.h>

#include <vrvCore/DtSubjectObserver.h>

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

#include <QtWidgets/QAction>
#include <osg/ShapeDrawable>


namespace osg
{
   class Node;
   class Drawable;
}

class QTreeWidgetItem;
class QIcon;
class QMenu;

namespace makVrv
{

   class DtDe;
   class DtSceneGraphTreeWidgetInterface;
   
   //custom actions for context menu
   class DT_DLL_VRVOSGQT DtDrawBoundsAction : public QAction
   {
      Q_OBJECT
   public:
      DtDrawBoundsAction(QObject *parent, const osg::Node* node);
      virtual ~DtDrawBoundsAction();
      virtual void setNode(const osg::Node* node);
   public slots:
      void slot_triggered(bool);
   
   private:
      osg::observer_ptr<const osg::Node> myNode;
      osg::observer_ptr<osg::ShapeDrawable> myBoundingVolumeDrawable;

   };

   class DT_DLL_VRVOSGQT DtSceneGraphTreeLogic
   {
   public:
      static DtSceneGraphTreeLogic* create(DtDe& de, DtSceneGraphTreeWidgetInterface* iface);

      virtual ~DtSceneGraphTreeLogic();

      enum SceneGraphRoot
      {
         RenderSystem,
         Environment,
         Terrain,
         Props,
         ScaledTerrainRoot,
         ScaledPropsRoot,
         ModelSet0,
         ModelSet1,
         ModelSet2,
         ModelSet3
      };

      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);
   protected:

      DtSceneGraphTreeLogic(DtDe& de, DtSceneGraphTreeWidgetInterface* iface);

   private:

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

   protected:

      DtDe& myDe;
      DtSceneGraphTreeWidgetInterface* myWidgetInterface;

      typedef osg::observer_ptr<const osg::Node> NodeObsPtr;
      typedef std::map<QTreeWidgetItem*, NodeObsPtr> WidgetToNodeMap;
      WidgetToNodeMap myWidgetToNodeMap;
      DtDrawBoundsAction *myBoundsAction;
   };
   



}