DtSimObjectEntryCollection.h

Go to the documentation of this file.

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


#pragma once

#include <vrvCore/vrvCore.h>
#include <vrvCore/DtDe.h>
#include <vrvCore/DtUniqueID.h>
#include <vrvCore/DtSimObjectEntry.h>
#include <vrvCore/DtDataBank.h>
#include <vrvCore/DtStateView.h>
#include <boost/bind.hpp>

namespace makVrv
{
   template<class T>
   class DtSimObjectEntryCollection
   {

   public:

      typedef std::map<DtUniqueID, DtSimObjectEntry*> SimObjectEntryMap;

      DtSimObjectEntryCollection(DtDe& de);

      virtual ~DtSimObjectEntryCollection();

      size_t size() const;

      const SimObjectEntryMap& simObjectEntryMap() const;

      const DtSimObjectEntry* findSimObjectEntry(DtUniqueID id) const;

      boost::signalslib::signal<void (DtSimObjectEntry&)> signal_simObjectEntryAdded;

      boost::signalslib::signal<void (DtSimObjectEntry&)> signal_simObjectEntryAboutToBeRemoved;

   protected:

      void onSimObjectEntryAdded(DtSimObjectEntry& entry);

      void onSimObjectEntryAboutToBeRemoved(DtSimObjectEntry& entry);

      void onSimObjectStateEntryAdded(DtSimObjectEntry& entry, DtSimState&);

      void onSimObjectStateEntryAboutToBeRemoved(DtSimObjectEntry& entry, DtSimState&);

      bool isCompatible(DtSimObjectEntry& entry);

      bool isCompatible(DtSimState& entry);

      void populateFromExistingEntries();

   private:
      DtSimObjectEntryCollection();
      DtSimObjectEntryCollection(const DtSimObjectEntryCollection&);
      DtSimObjectEntryCollection& operator=(const DtSimObjectEntryCollection&);

   protected:

      DtDe& myDe;
      SimObjectEntryMap mySimObjectEntryMap;

   };

   template<class T>
   DtSimObjectEntryCollection<T>::DtSimObjectEntryCollection(DtDe& de)
   : myDe(de)
   , mySimObjectEntryMap()
   {
      DtSimData& simDataObj = myDe.dataBank().simData();

      simDataObj.signal_simObjectEntryAdded.connect(boost::bind(
         &DtSimObjectEntryCollection<T>::onSimObjectEntryAdded, this, _1));

      simDataObj.signal_simObjectEntryToBeRemoved.connect(boost::bind(
         &DtSimObjectEntryCollection<T>::onSimObjectEntryAboutToBeRemoved, this, _1));

      simDataObj.signal_simStateAdded.connect(boost::bind(
         &DtSimObjectEntryCollection<T>::onSimObjectStateEntryAdded, this, _1, _2));

      simDataObj.signal_simStateToBeRemoved.connect(boost::bind(
         &DtSimObjectEntryCollection<T>::onSimObjectStateEntryAboutToBeRemoved, this, _1, _2));

      populateFromExistingEntries();
   }

   template<class T>
   DtSimObjectEntryCollection<T>::~DtSimObjectEntryCollection()
   {
      DtSimData& simDataObj = myDe.dataBank().simData();

      simDataObj.signal_simObjectEntryAdded.disconnect(boost::bind(
         &DtSimObjectEntryCollection<T>::onSimObjectEntryAdded, this, _1));

      simDataObj.signal_simObjectEntryToBeRemoved.disconnect(boost::bind(
         &DtSimObjectEntryCollection<T>::onSimObjectEntryAboutToBeRemoved, this, _1));

      simDataObj.signal_simStateAdded.disconnect(boost::bind(
         &DtSimObjectEntryCollection<T>::onSimObjectStateEntryAdded, this, _1, _2));

      simDataObj.signal_simStateToBeRemoved.disconnect(boost::bind(
         &DtSimObjectEntryCollection<T>::onSimObjectStateEntryAboutToBeRemoved, this, _1, _2));
   }

   template<class T>
   void DtSimObjectEntryCollection<T>::onSimObjectEntryAdded(DtSimObjectEntry& entry)
   {
      if(isCompatible(entry))
      {
         onSimObjectEntryAboutToBeRemoved(entry);

         mySimObjectEntryMap[ entry.id() ] = &entry;
         signal_simObjectEntryAdded(entry);
      }
   }

   template<class T>
   void DtSimObjectEntryCollection<T>::onSimObjectStateEntryAdded(DtSimObjectEntry& entry, DtSimState& state)
   {
      if(isCompatible(state))
      {
         onSimObjectStateEntryAboutToBeRemoved(entry, state);

         mySimObjectEntryMap[ entry.id() ] = &entry;
         signal_simObjectEntryAdded(entry);
      }
   }

   template<class T>
   void DtSimObjectEntryCollection<T>::onSimObjectEntryAboutToBeRemoved(DtSimObjectEntry& entry)
   {
      if(isCompatible(entry))
      {
         SimObjectEntryMap::iterator iter = mySimObjectEntryMap.find(entry.id());
         if(iter != mySimObjectEntryMap.end())
         {
            signal_simObjectEntryAboutToBeRemoved(entry);
            mySimObjectEntryMap.erase(iter);
         }
      }
   } 

   template<class T>
   void DtSimObjectEntryCollection<T>::onSimObjectStateEntryAboutToBeRemoved(DtSimObjectEntry& entry, DtSimState& state)
   {
      if(isCompatible(state))
      {
         SimObjectEntryMap::iterator iter = mySimObjectEntryMap.find(entry.id());
         if(iter != mySimObjectEntryMap.end())
         {
            signal_simObjectEntryAboutToBeRemoved(entry);
            mySimObjectEntryMap.erase(iter);
         }
      }
   } 

   template<class T>
   bool DtSimObjectEntryCollection<T>::isCompatible(DtSimObjectEntry& entry)
   {
      return DtStateView<T>::isCompatible(entry);
   }

   template<class T>
   bool DtSimObjectEntryCollection<T>::isCompatible(DtSimState& state)
   {
      return DtStateView<T>::isCompatible(state);
   }

   template<class T>
   size_t DtSimObjectEntryCollection<T>::size() const
   {
      return mySimObjectEntryMap.size();
   }

   template<class T>
   void DtSimObjectEntryCollection<T>::populateFromExistingEntries()
   {
      DtSimData::SimEntryMap::const_iterator pos;
      for(pos=myDe.dataBank().simData().simEntryMap().begin(); pos != 
         myDe.dataBank().simData().simEntryMap().end(); ++pos )
      {
         DtSimObjectEntry* pEntry = dynamic_cast<DtSimObjectEntry*>(pos->second);
         if(pEntry)
         {
            onSimObjectEntryAdded(*pEntry);
         }         
      }
   }

   template<class T>
   const typename DtSimObjectEntryCollection<T>::SimObjectEntryMap& 
      DtSimObjectEntryCollection<T>::simObjectEntryMap() const
   {
      return mySimObjectEntryMap;
   }

   template<class T>
   const DtSimObjectEntry* DtSimObjectEntryCollection<T>::findSimObjectEntry(
      DtUniqueID id) const
   {
      DtSimObjectEntry* entry = 0;
      SimObjectEntryMap::const_iterator iter = mySimObjectEntryMap.find(id);
      if(iter != mySimObjectEntryMap.end())
      {
         entry = iter->second;
      }
      return entry;
   }
}