vrfStateViewCollectionManager.h

Go to the documentation of this file.

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



#pragma once

#include <vlutil/vlExceptions.h>

#include <vrfGuiCore/vrfGuiCore.h>
#include <vrfGuiCore/vrfObjectDataState.h>
#include <vrfGuiCore/spotReportState.h>
#include <vrfGuiCore/vrfTDLMessageDataState.h>
#include <vrfGuiCore/orbatState.h>
#include <vrfGuiCore/objectDataInterface.h>
#include <vrfutil/rwUUID.h>
#include <vrvUtil/DtVirtualBaseClass.h>
#include <vrvUtil/DtSignalConnectionManager.h>


#include <vrvCore/DtElementEntry.h>
#include <vrvCore/DtStateViewCollection.h>
#include <vrvCore/DtEntityState.h>
#include <vrvCore/DtSensorFieldOfViewState.h>
#include <vrvCore/DtElementAttributeManager.h>

#include <set>
#include <unordered_set>
#include <unordered_map>
#include <boost/function.hpp>

namespace makVrv
{
   class DtDe;
}

namespace makVrf
{
   class DtVrfSelectionFilter;
   class DtVrfGuiOrbatManager;

   class DtAggregateStateViewCollection;
   class DtPropulsionNoiseStateDataViewCollection;
   class DtIffStateDataViewCollection;
   class DtEmitterSystemStateViewCollection;
   class DtWeatherStateViewCollection;
   class DtEnvironmentStateViewCollection;
   class DtActiveSonarStateViewCollection;
   class DtLaserDesignatorStateViewCollection;
   class DtExtendedStateDataViewCollection;
   class DtAerodromeStateViewCollection;
   class DtAircraftLightingStateViewCollection;
   class DtGroundVehicleLightingStateViewCollection;
   class DtAisDataStateViewCollection;

   typedef boost::function<int (const DtEntityType&)> DtMaxMarkingTextLengthFunction;

   class DT_DLL_VRFGUICORE DtStateViewMarkingTextProvider : public DtUUIDMarkingTextProvider
   {
   public:
      DtStateViewMarkingTextProvider(makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>* sv) : myStateView(sv) {};
      DtStateViewMarkingTextProvider(const DtStateViewMarkingTextProvider& orig) : myStateView(orig.myStateView) {};
      virtual ~DtStateViewMarkingTextProvider() {};
      virtual DtString markingText() { return myStateView->state().myMarkingText.c_str(); };
      virtual DtUUIDMarkingTextProvider* clone() const { return new DtStateViewMarkingTextProvider(*this); };

      operator DtString () { return markingText(); };

   protected:
      makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>* myStateView;
   };

   class DT_DLL_VRFGUICORE DtVrfSimulatedEntityStateViewCollection
   : public makVrv::DtStateViewCollection<makVrv::DtVrlinkSimulatedEntityState>
   {
   public:
      DtVrfSimulatedEntityStateViewCollection(makVrv::DtDe& de) 
      : makVrv::DtStateViewCollection<makVrv::DtVrlinkSimulatedEntityState>(
         de, makVrv::DtSimEntryUpdater::NoUpdates)
      {
      }
   };

   class DT_DLL_VRFGUICORE DtVrlinkSimulatedBaseStateViewCollection
   : public makVrv::DtStateViewCollection<makVrv::DtVrlinkSimulatedBaseState>
   {
   public:
      DtVrlinkSimulatedBaseStateViewCollection(makVrv::DtDe& de) :
         makVrv::DtStateViewCollection<makVrv::DtVrlinkSimulatedBaseState>(de, makVrv::DtSimEntryUpdater::NoUpdates)
      {
      }

      bool isCompatible(makVrv::DtSimObjectEntry& entry)
      {
         return makVrv::DtStateViewCollection<makVrv::DtVrlinkSimulatedBaseState>::isCompatible(entry);
      }
   };
   
   class DT_DLL_VRFGUICORE DtVrfStateViewCollectionManager : public makVrv::DtVirtualBaseClass
   {
   public:
      typedef boost::signals2::signal<void (const makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>*)> StateViewAddedOrRemovedSignal;

   public:

      DtVrfStateViewCollectionManager(makVrv::DtDe&);
      virtual ~DtVrfStateViewCollectionManager();

      static DtVrfStateViewCollectionManager& instance(makVrv::DtDe& de);
      


      virtual DtBoost::shared_ptr<DtObjectDataInterface> lookupObjectData(const makVrv::DtUniqueID& id);

      virtual DtBoost::shared_ptr<DtObjectDataInterface> lookupSpotReportData(const makVrv::DtUniqueID& id);

      virtual DtBoost::shared_ptr<DtObjectDataInterface> lookupVrfTDLData(const makVrv::DtUniqueID& id);

      virtual DtBoost::shared_ptr<DtObjectDataInterface> lookupSpotReportData(const std::string& name);

      virtual DtBoost::shared_ptr<DtObjectDataInterface> lookupVrfTDLData(const std::string& uniqueIdentifier);

      virtual DtBoost::shared_ptr<DtObjectDataInterface> lookupOrbatData(const makVrv::DtUniqueID& id);

      virtual DtBoost::shared_ptr<DtObjectDataInterface> lookupOrbatData(const std::string& name);

      virtual DtBoost::shared_ptr<DtObjectDataInterface> lookupObjectData(const DtUUID&);

      virtual DtVector locationOf(makVrv::DtUniqueID id);

      virtual DtVector centerOfMass(const std::set<DtUUID>& uuids,
         bool ignoreEmbarked = true);

      template<typename T_ObjectDataInterfaceType>
      DtBoost::shared_ptr<T_ObjectDataInterfaceType> lookupAndCastObjectData(const makVrv::DtUniqueID& id)
      {
         return DtObjectDataInterface::cast<T_ObjectDataInterfaceType>(lookupObjectData(id));
      }

      template<typename T_ObjectDataInterfaceType>
      DtBoost::shared_ptr<T_ObjectDataInterfaceType> lookupAndCastSpotReportData(const makVrv::DtUniqueID& id)
      {
         return DtObjectDataInterface::cast<T_ObjectDataInterfaceType>(lookupSpotReportData(id));
      }

      template<typename T_ObjectDataInterfaceType>
      DtBoost::shared_ptr<T_ObjectDataInterfaceType> lookupAndCastVrfTDLData(const makVrv::DtUniqueID& id)
      {
         return DtObjectDataInterface::cast<T_ObjectDataInterfaceType>(lookupVrfTDLData(id));
      }

      template<typename T_ObjectDataInterfaceType>
      DtBoost::shared_ptr<T_ObjectDataInterfaceType> lookupAndCastOrvatData(const makVrv::DtUniqueID& id)
      {
         return DtObjectDataInterface::cast<T_ObjectDataInterfaceType>(lookupOrbatData(id));
      }

      template<typename T_ObjectDataInterfaceType>
      DtBoost::shared_ptr<T_ObjectDataInterfaceType> lookupAndCastObjectData(const DtUUID& uuid)
      {
         return DtObjectDataInterface::cast<T_ObjectDataInterfaceType>(lookupObjectData(uuid));
      }

      template<typename T_ObjectDataInterfaceType>
      DtBoost::shared_ptr<T_ObjectDataInterfaceType> lookupAndCastSpotReportData(const std::string& name)
      {
         return DtObjectDataInterface::cast<T_ObjectDataInterfaceType>(lookupSpotReportData(name));
      }

      template<typename T_ObjectDataInterfaceType>
      DtBoost::shared_ptr<T_ObjectDataInterfaceType> lookupAndCastVrfTDLData(const std::string& id)
      {
         return DtObjectDataInterface::cast<T_ObjectDataInterfaceType>(lookupVrfTDLData(id));
      }

      template<typename T_ObjectDataInterfaceType>
      DtBoost::shared_ptr<T_ObjectDataInterfaceType> lookupAndCastOrbatData(const std::string& name)
      {
         return DtObjectDataInterface::cast<T_ObjectDataInterfaceType>(lookupOrbatData(name));
      }

      virtual DtBoost::shared_ptr<DtObjectDataInterface> lookupObjectData(makVrv::DtStateView<makVrv::DtSimState>* stateView);

      virtual DtEntityType entityTypeFromID(makVrv::DtUniqueID id);

      virtual makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>* addMappingBasedOnMarkingTextAndUUID(const std::string& markingText, const DtUUID& uuid);

      virtual makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>* findStateView(const DtUUID&, bool fullSearchIfMissing = false);

      virtual makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>* findStealthStateView(const std::string& address,
         const std::string& markingText);

      virtual makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>* findStateViewByMarkingText(const std::string& markingText);

      virtual makVrv::DtStateView<DtSpotReportState>* findSpotReportStateView(const std::string& markingText);

      virtual makVrv::DtStateView<DtVrfTDLMessageDataState>* findVrfTDLDataStateView(const std::string& id);

      virtual makVrv::DtStateView<DtOrbatState>* findOrbatStateView(const std::string& markingText);

      virtual makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>* findStateView(const makVrv::DtUniqueID& id);

      makVrv::DtStateView<DtSpotReportState>* findSpotReportStateView(const makVrv::DtUniqueID& id)
      {
         if (!mySpotReportStateViewCollection->hasStateView(id))
         {
            return 0;
         }

         return const_cast<makVrv::DtStateView<DtSpotReportState>*>
            (mySpotReportStateViewCollection->findStateView(id));
      }

      makVrv::DtStateView<DtVrfTDLMessageDataState>* findVrfTDLDataStateView(const makVrv::DtUniqueID& id)
      {
         if (!myVrfTDLDataStateViewCollection->hasStateView(id))
         {
            return 0;
         }

         return const_cast<makVrv::DtStateView<DtVrfTDLMessageDataState>*>
            (myVrfTDLDataStateViewCollection->findStateView(id));
      }

      makVrv::DtStateView<DtExternalDataState>* findExternalDataStateView(const makVrv::DtUniqueID& id)
      {
         if (!myExternalDataStateViewCollection->hasStateView(id))
         {
            return 0;
         }

         return const_cast<makVrv::DtStateView<DtExternalDataState>*>
            (myExternalDataStateViewCollection->findStateView(id));
      }

      makVrv::DtStateView<DtOrbatState>* findOrbatStateView(const makVrv::DtUniqueID& id)
      {
         if (!myOrbatStateViewCollection->hasStateView(id))
         {
            return 0;
         }

         return const_cast<makVrv::DtStateView<DtOrbatState>*>
            (myOrbatStateViewCollection->findStateView(id));
      }

      virtual void clearUUIDInformation();

      DtVrfSimulatedEntityStateViewCollection* simulatedEntityStateViewCollection()
      {
         return mySimulatedEntityStateView;
      }

      DtVrlinkSimulatedBaseStateViewCollection* simulatedBaseStateViewCollection()
      {
         return mySimulatedBaseStateViewCollection;
      }

      DtVrfObjectDataStateViewCollection* objectDataStateViewCollection()
      {
         return myObjectDataStateViewCollection;
      }
      
      DtEnvironmentStateViewCollection* environmentStateViewCollection()
      {
         return myEnvironmentStateViewCollection;
      }
      
      DtExtendedStateDataViewCollection* extendedStateDataViewCollection()
      {
         return myExtendedStateDataViewCollection;
      }

      DtWeatherStateViewCollection* weatherStateViewCollection()
      {
         return myWeatherStateViewCollection;
      }

      DtIffStateDataViewCollection* iffStateDataViewCollection()
      {
         return myIffStateDataViewCollection;
      }
      
      DtPropulsionNoiseStateDataViewCollection* propulsionNoiseStateDataViewCollection()
      {
         return myPropulsionNoiseStateDataViewCollection;
      }
      
      DtEmitterSystemStateViewCollection* emitterSystemStateViewCollection()
      {
         return myEmitterSystemStateViewCollection;
      }
      
      DtActiveSonarStateViewCollection* activeSonarStateViewCollection()
      {
         return myActiveSonarStateViewCollection;
      }
      
      DtLaserDesignatorStateViewCollection* laserDesignatorStateViewCollection()
      {
         return myLaserDesignatorStateViewCollection;
      }
      
      DtSpotReportStateViewCollection* spotReportStateViewCollection()
      {
         return mySpotReportStateViewCollection;
      }
      
      DtExternalDataStateViewCollection* externalDataStateViewCollection()
      {
         return myExternalDataStateViewCollection;
      }
      
      DtVrfTDLMessageDataStateViewCollection* vrfTDLDataStateViewCollection()
      {
         return myVrfTDLDataStateViewCollection;
      }
      
      DtOrbatStateViewCollection* orbatStateViewCollection()
      {
         return myOrbatStateViewCollection;
      }
      
      DtAggregateStateViewCollection* aggregateStateViewCollection()
      {
         return myAggregateStateViewCollection;
      }
      
      DtAerodromeStateViewCollection* aerodromeStateViewCollection()
      {
         return myAerodromeStateViewCollection;
      }

      makVrv::DtVrlinkSensorFieldsOfViewStateViewCollection* sensorFieldOfViewStateViewCollection()
      {
         return mySensorFieldOfViewStateViewCollection;
      }

      DtAircraftLightingStateViewCollection* aircraftLightingStateViewCollection()
      {
         return myAircraftLightingStateViewCollection;
      }

      DtGroundVehicleLightingStateViewCollection* groundVehicleLightingStateViewCollection()
      {
         return myGroundVehicleLightingStateViewCollection;
      }

      DtAisDataStateViewCollection* aisDataStateViewCollection()
      {
         return myAisDataStateViewCollection;
      }

      makVrv::DtUniqueID parentId(const makVrv::DtUniqueID&) const;

      makVrv::DtUniqueID displayIdFromSimId(const makVrv::DtUniqueID& id, makVrv::DtModelSetId modelSet = makVrv::DtModelSetId::Unspecified) const;
      makVrv::DtElementData::SceneObjectIdList displayIdsFromSimId(const makVrv::DtUniqueID& id, makVrv::DtModelSetId modelSet = makVrv::DtModelSetId::Unspecified) const;

      makVrv::DtUniqueID simIdFromDisplayId(const makVrv::DtUniqueID& id) const;

      const std::map<makVrv::DtUniqueID, const makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>*>&
         stealthStates() const
      {
         return myStealthStates;
      }

      virtual std::string entityName(const makVrv::DtUniqueID simId) const;

      virtual DtUUID objectUUID(makVrv::DtUniqueID simId) const;

      virtual std::string overlayName(const makVrv::DtUniqueID simId) const;

      virtual makVrv::DtElementEntry::ElementType elementType(const makVrv::DtElementID&) const;

      virtual makVrv::DtSimEntryUpdater::UpdateRate updateRateFor(const makVrv::DtBaseStateView* state) const;

      typedef std::vector<makVrv::DtSimEntryUpdater::UpdateRate> UpdateRates;
      virtual makVrv::DtSimEntryUpdater::UpdateRate minimumRate(const UpdateRates&) const;
      virtual makVrv::DtSimEntryUpdater::UpdateRate maximumRate(const UpdateRates&) const;

      virtual void setUpdateRate(makVrv::DtSimEntry*, makVrv::DtSimEntryUpdater::UpdateRate, 
         makVrv::DtBaseStateView* state);

      virtual void forceUpdateForRelevantInformation(makVrv::DtUniqueID id);

      virtual StateViewAddedOrRemovedSignal& signalWhenObjectAdded(const DtVrfSelectionFilter&);
      virtual void processObjectAddedCallbacks(const makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>*);
      virtual void removeAddedSignalReference(const DtVrfSelectionFilter&);
      virtual void waiting();

      virtual StateViewAddedOrRemovedSignal& signalWhenObjectRemoved(const DtVrfSelectionFilter&);
      virtual void removeRemovalSignalReference(const DtVrfSelectionFilter&);

      virtual void processObjectRemovedCallbacks(const makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>*);

      virtual void slot_simStateAdded(makVrv::DtSimObjectEntry& e);
      virtual void slot_simStateToBeRemoved(makVrv::DtSimObjectEntry& e, makVrv::DtSimState&);

      virtual std::string lookupNameById(makVrv::DtUniqueID id);
      virtual makVrv::DtUniqueID lookupIdByUUID(const DtUUID&);

      virtual void uniqueEntityTypes(makVrv::DtUniqueID, std::vector<DtEntityType>& entityTypes);

      virtual bool getSupportsWeightVariations(makVrv::DtUniqueID id);

      virtual const std::set<makVrv::DtUniqueID>& dIGuyIds();

      virtual void signalWhenAllObjectsDiscovered(const std::map<short, int>&);


      virtual DtUUID uuidFromState(const makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>*) const;
      virtual DtUUID uuidFromState(const makVrv::DtVrlinkSimulatedBaseState*) const;
      virtual makVrv::DtUniqueID idFromState(const makVrv::DtVrlinkSimulatedBaseState*);

      boost::signals2::signal <void(const makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>*)> signal_stateAdded;
      boost::signals2::signal <void(const makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>*)> signal_simEntryAdded;
      boost::signals2::signal <void(const makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>*)> signal_stateRemoved;

      boost::signals2::signal<void (makVrv::DtUniqueID, const DtUUID& oldUUID, const DtUUID& newUUID)> signal_uuidChanged;

      boost::signals2::signal <void ()> signal_allObjectsDiscovered;

      boost::signals2::signal <void (const makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>*, const std::string& newName)> signal_nameChanged;

      const std::set<makVrv::DtUniqueID>& tacticalGraphicElements() const;

      virtual int maxMarkingTextLengthFor(const DtEntityType&) const;
      virtual void setMaxMarkingTextLengthFunction(DtMaxMarkingTextLengthFunction);
      virtual DtMaxMarkingTextLengthFunction maxMarkingTextLengthFunction() const;

      virtual int defaultMaxMarkingTextLengthFunction(const DtEntityType&);

   protected:
      virtual void uniqueEntityTypes(makVrv::DtUniqueID, std::vector<DtEntityType>& entityTypes, std::set<std::string>& typesChecked);
      virtual void slot_onAttributesChanged( const makVrv::DtElementAttributeManager::EntryAttributesList& signalList );

      DtVrfGuiOrbatManager* vrfGuiOrbatManager() const;

      virtual bool shouldProcessState(const makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>*) const;

      virtual void slot_attributesChanged(const makVrv::DtElementAttributeManager::EntryAttributesList& l);

   protected:
      DtVrfSimulatedEntityStateViewCollection*  mySimulatedEntityStateView;
      DtVrlinkSimulatedBaseStateViewCollection* mySimulatedBaseStateViewCollection;
      DtVrfObjectDataStateViewCollection*       myObjectDataStateViewCollection;
      DtEnvironmentStateViewCollection*         myEnvironmentStateViewCollection;
      DtSpotReportStateViewCollection*          mySpotReportStateViewCollection;
      DtExternalDataStateViewCollection*        myExternalDataStateViewCollection;
      DtVrfTDLMessageDataStateViewCollection*   myVrfTDLDataStateViewCollection;
      DtOrbatStateViewCollection*               myOrbatStateViewCollection;
      DtAggregateStateViewCollection*           myAggregateStateViewCollection;
      DtAerodromeStateViewCollection*           myAerodromeStateViewCollection;
      DtExtendedStateDataViewCollection*        myExtendedStateDataViewCollection;
      DtIffStateDataViewCollection*             myIffStateDataViewCollection;
      DtWeatherStateViewCollection*             myWeatherStateViewCollection;
      DtPropulsionNoiseStateDataViewCollection* myPropulsionNoiseStateDataViewCollection;
      DtEmitterSystemStateViewCollection*       myEmitterSystemStateViewCollection;
      DtActiveSonarStateViewCollection*         myActiveSonarStateViewCollection;
      DtLaserDesignatorStateViewCollection*     myLaserDesignatorStateViewCollection;
      makVrv::DtVrlinkSensorFieldsOfViewStateViewCollection* mySensorFieldOfViewStateViewCollection;
      DtAircraftLightingStateViewCollection*                 myAircraftLightingStateViewCollection;
      DtGroundVehicleLightingStateViewCollection*            myGroundVehicleLightingStateViewCollection;
      DtAisDataStateViewCollection*                          myAisDataStateViewCollection;
      
      makVrv::DtDe&                                          myDe;

      std::map<short, int>                                   myWaitingForDiscovery;

      typedef boost::unordered_map<makVrv::DtBaseStateView*, UpdateRates> UpdateRateMap;

      UpdateRateMap myUpdateRateMap;

      int myNextToken;
      struct StateViewAddedOrRemovedStruct
      {
         DtVrfSelectionFilter* filter;
         StateViewAddedOrRemovedSignal* added;
         int addedReference;
         StateViewAddedOrRemovedSignal* removed;
         int removedReference;
      };

      typedef std::vector<StateViewAddedOrRemovedStruct> StateViewAddedOrRemovedCallbacks;
      StateViewAddedOrRemovedCallbacks myStateViewAddedOrRemovedCallbacks;

      makVrv::DtSignalConnectionManager myConnectionManager;
      std::map<makVrv::DtElementID, bool> myWaiting;

      typedef std::unordered_map<DtUUID, makVrv::DtUniqueID> UUIDToIdMap;
      typedef std::unordered_map<makVrv::DtUniqueID, DtUUID> IdToUUIDMap;

      mutable UUIDToIdMap myUUIDToIdMap;
      mutable IdToUUIDMap myIdToUUIDMap;

      std::unordered_set<makVrv::DtUniqueID> myAddedObjects;

      mutable DtVrfGuiOrbatManager* myVrfGuiOrbatManager;

      std::map<makVrv::DtUniqueID, const makVrv::DtStateView<makVrv::DtVrlinkSimulatedBaseState>*> myStealthStates;
      std::string myStealthType;

      std::set<makVrv::DtUniqueID> myTacticalGraphicElements;

      std::set<makVrv::DtUniqueID> myDIGuyIds;
      boost::unordered_set<makVrv::DtUniqueID> myIdsToCheckForDIGuy;

      boost::unordered_set<DtUUID> myDidNotFindStateViewFor;

      DtMaxMarkingTextLengthFunction myMaxMarkingTextLengthFunction;
   };

   template<typename T> T findStateFor(makVrv::DtSimEntry* entry)
   {
      makVrv::DtSimEntry::SimStateVector::const_iterator pos    = entry->simStates().begin();
      makVrv::DtSimEntry::SimStateVector::const_iterator endPos = entry->simStates().end();
      for(; pos != endPos; ++pos)
      {
         T tState = dynamic_cast<T>(*pos);
         if(tState)
         {
            return tState;
         }
      }

      return 0;
   }

   DT_DLL_VRFGUICORE std::vector<DtUUID> convertMarkingTextVectorToUUIDVector(const std::vector<DtString>& v);
   DT_DLL_VRFGUICORE std::vector<DtUUID> convertMarkingTextVectorToUUIDVector(const std::vector<std::string>& v);

   DT_DLL_VRFGUICORE std::vector<std::string> convertUUIDVectorToUUIDStringVector(const std::vector<DtUUID>& v);

   DT_DLL_VRFGUICORE std::vector<std::string> convertUUIDVectorToMarkingTextVector(const std::vector<DtUUID>& v);
}