DtElementData.h

Go to the documentation of this file.

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


#pragma once
#include <vrvCore/DtElementEntry.h>
#include <vrvCore/DtElementAttributes.h>
#include <vrvCore/DtModelSetEnums.h>

#include <vrvUtil/signalslib.h>
#include <unordered_map>
#include <unordered_set>

namespace makVrv
{
   class DtElementChanges;
   class DtElementEntry;
   class DtElementAttribute;
   class DtDe;
   class DtSceneObject;
   class DtElementAttributeManager;

   class DT_DLL_VRVCORE DtElementData
   {
   public:


      using SceneObjectIdList = std::vector<DtUniqueID>;

      using ElementEntryMap = std::unordered_map<DtElementID, DtElementEntry*> ;

      using ElementToSceneObjectMap = std::unordered_map<DtElementID, SceneObjectIdList> ;

      typedef std::vector<DtElementEntry*> ElementList;

      typedef std::vector<DtElementID> UniqueIdList; // deprecated (but still widely used)

      typedef std::vector<DtElementID> ElementIdList;

      DtElementData(DtDe& de);

      virtual ~DtElementData();

      void applyChanges(const DtElementChanges& changes);

      const ElementEntryMap& elements() const;

      const ElementToSceneObjectMap& elementToSceneObjectMap() const;

      DtElementEntry* findElement(DtElementID id );

      void findAll(DtElementEntry::ElementType type, 
         ElementList& outVector, bool sort = true);

      void findAll(const std::vector<DtElementEntry::ElementType>& types,  
         ElementList& outVector, bool sort = true);

      void findAll(ElementList& outVector, bool sort = true);


      virtual void addElementID( const DtElementID& elementID, 
         const DtUniqueID& sceneObjectID, DtModelSetId modelSet );

      virtual void addElementID( const DtElementID& elementID, 
         const DtUniqueID& sceneObjectID );

      virtual void removeElementID( const DtElementID& elementID, 
         const DtUniqueID& sceneObjectID, DtModelSetId modelSet );

      virtual void removeElementID(const DtElementID& elementID,
         const DtUniqueID& sceneObjectID);

      virtual void getSceneObjectIds( const DtElementID& elementID, 
         SceneObjectIdList& outputList, DtModelSetId modelSet = DtModelSetId::Unspecified /* all modelSets */ ) const;

      DtSceneObject* findSceneObjectFromElementId(DtElementID elementId, DtModelSetId modelSet = DtModelSetId::Models3D );

      DtSceneObject* findSceneObjectFromSceneObjectId(DtUniqueID sceneObjId);

      virtual DtElementID elementIdFromSceneObjectId( const DtUniqueID& sceneObjectID ) const;

      virtual const DtElementAttributeManager& attributeManager() const;

      virtual DtElementAttributeManager& attributeManager();

      boost::signalslib::signal< void (const ElementList&) > signal_elementsAdded;

      boost::signalslib::signal< void (const ElementIdList&) > signal_elementsInitialized;

      boost::signalslib::signal< void (const ElementIdList&) > signal_elementsToBeRemoved;

      boost::signalslib::signal< void(const ElementIdList&) > signal_sceneObjectsAdded;

   private:
      DtElementData();

      DtElementData(const DtElementData& orig);

      DtElementData& operator=(const DtElementData& rhs);

   protected:

      virtual void slot_preTick();

   protected:

      using ModelSetToElementSceneObjectMap = std::unordered_map<DtModelSetId, ElementToSceneObjectMap>;

      using SceneObjectToElementMap = std::unordered_map<DtUniqueID, DtElementID>;

      DtDe& myDe;
      ElementEntryMap myElements;

      // map to go from element ID to scene object IDs
      ElementToSceneObjectMap myElementToSceneObjectMap;

      // map to go from element ID and model set to scene object id(s)
      ModelSetToElementSceneObjectMap myModelSetToElementSceneObjectMap;

      // map to go from scene object id to element id
      SceneObjectToElementMap mySceneObjectToElementMap;     

      // List of elements that were added last frame OR that had scene objects added 
      // last frame.  
      std::unordered_set<DtElementID> myElementsAddedLastFrame;

      DtElementAttributeManager* myElementAttributeManager;
   };
}