DtEntityElement.h

Go to the documentation of this file.

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


#pragma once

#include <vrvCore/vrvCore.h>

#include <vrvUtil/DtSignalConnectionManager.h>
#include <vrvCore/DtElement.h>
#include <vrvCore/DtStateListener.h>
#include <vrvCore/DtStateVisualizer.h>
#include <vrvCore/DtEntityVisualizerSet.h>
#include <vrvCore/DtEntityStateVisualizer.h>
#include <vrvCore/DtUniqueID.h>

#include <boost/unordered_map.hpp>

namespace makVrv
{
   class DtSceneObjectAgent;
   class DtEntityKinematicState; 
   class DtBaseConnection;

   class DT_DLL_VRVCORE DtEntityElement : public DtElement
   {
   public:

      struct DtEntityObjectStruct
      {
         int modelSet;
         DtSceneObjectAgent* sceneObject;
         DtEntityVisualizerSet* visualizerSet;
         DtEntityKinematicState* kinematicState;
         
         DtEntityObjectStruct() : modelSet(0)
            , sceneObject( nullptr )
            , visualizerSet( nullptr )
            , kinematicState( nullptr )
         {
         }

         DtEntityObjectStruct( const DtEntityObjectStruct& ) = delete;

         DtEntityObjectStruct& operator=( const DtEntityObjectStruct& ) = delete;
      };

      using ModelSetToEntityObjectMap = boost::unordered_map<int, DtEntityObjectStruct*>;

      DtEntityElement( DtBaseConnection& conn, DtStateListener& listener );

      virtual ~DtEntityElement();
      
      DtEntityElement() = delete;

      DtEntityElement(const DtEntityElement&) = delete;

      DtEntityElement& operator=(const DtEntityElement&) = delete;

   public:
      virtual void visualize( const DtSharedModelSetSettings::ModelSetVisualizedMap& sets ) override;

      virtual void setModelSetEnabled( int modelSet, bool enabled ) override;

      virtual void setVisible(bool visible);

      virtual const ModelSetToEntityObjectMap& entityObjectMap() const;

      template <typename VisualizerClass>
      void findVisualizers( std::vector<VisualizerClass*>& visualizers );

      template <typename VisualizerClass>
      void findVisualizersByDynamicCast( std::vector<VisualizerClass*>& visualizers );

      virtual DtEntityStateListener& entityStateListener();

      virtual const DtElementEntry::ElementType elementType() const override;
      static const DtElementEntry::ElementType theElementType();

      virtual void update2DEmbarkationVisibility();

      virtual void createVisualizersByModelSetAndType(int modelSet, int visualType);
      virtual void destroyVisualizersByModelSetAndType(int modelSet, int visualType);


      virtual DtEntityObjectStruct* findObjectStructForModelSet( int modelSet ) const;
      virtual DtSceneObjectAgent* findSceneObjectForModelSet( int modelSet ) const ;
      virtual DtEntityVisualizerSet* findVisualizerSetForModelSet( int modelSet ) const ;
      virtual DtEntityKinematicState* findKinematicStateForModelSet( int modelSet) const;


      virtual void setSmoothingEnabled( bool enabled );
      virtual void setSmoothingEnabled( bool enabled, int modelSet );

      virtual void setGroundClampingEnabled( bool enabled );
      virtual void setGroundClampingEnabled( bool enabled, int modelSet );

      virtual void setPickable( bool pickable );
      virtual void setPickable( bool pickable, int modelSet );


      virtual std::vector<DtUniqueID> sceneObjectUniqueIds();
      virtual DtUniqueID sceneObjectIdForModelSet( int modelSet );

   protected:
      virtual void slot_visibilityChanged(bool vis);

      virtual DtEntityKinematicState* createEntityKinState( DtSceneObjectAgent& sceneObject, int modelSet );
      
      virtual void processElementParameters();
   protected:

      DtEntityStateListener& myListener;

      ModelSetToEntityObjectMap myModelSetToEntityObjectMap;
      DtSignalConnectionManager  myConnections;

   };

   template<typename VisualizerClass>
   inline void DtEntityElement::findVisualizers( std::vector<VisualizerClass*>& visualizers )
   {
      ModelSetToEntityObjectMap::iterator i = myModelSetToEntityObjectMap.begin();
      ModelSetToEntityObjectMap::iterator e = myModelSetToEntityObjectMap.end();
      for( ; i != e; ++i )
      {
         i->second->visualizerSet->findVisualizers( visualizers );
      }
   }

   template<typename VisualizerClass>
   inline void DtEntityElement::findVisualizersByDynamicCast( std::vector<VisualizerClass*>& visualizers )
   {
      ModelSetToEntityObjectMap::iterator i = myModelSetToEntityObjectMap.begin();
      ModelSetToEntityObjectMap::iterator e = myModelSetToEntityObjectMap.end();
      for( ; i != e; ++i )
      {
         i->second->visualizerSet->findVisualizersByDynamicCast( visualizers );
      }
   }

   using DtEntityElementCreator = DtElementCreatorTemplate<DtEntityElement>;
}