vrfVrlinkGuiOrbatManager.h

Go to the documentation of this file.

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


#pragma once

#include <vrfVrlSharedSource/vrfProtocol.h>
#include <vrvUtil/DtVirtualBaseClass.h>
#include <vrfutil/rwUUID.h>
#include <vrfutil/rwOrbatData.h>
#include <vlpi/entityIdentifier.h>
#include <vl/globalObjectDesignator.h>
#include <vrfGuiCore/vrfGuiOrbatManager.h>
#include <vrvUtil/DtSignalConnectionManager.h>

#include <boost/unordered_map.hpp>

class DtVrfRemoteController;
class DtSimMessage;

namespace makVrv
{
   class DtStateListener;

   namespace DT_PROTOCOL_NAMESPACE
   {
      class DtVrlinkConnection;
      class DtVrlinkAggregateStateListener;
      class DtVrlinkEntityStateListener;
   }
}

namespace makVrf 
{
   namespace VRF_PROTOCOL_NAMESPACE
   {
      class VRF_PROTOCOL_EXPORT DtVrfVrlinkGuiOrbatManager : public makVrv::DtVirtualBaseClass
      {
      public:
         struct OrbatRecord;

         typedef std::shared_ptr<OrbatRecord> OrbatRecordHandle;
         typedef std::map<makVrv::DtElementID, OrbatRecordHandle> OrbatMapByUniqueID;
         typedef std::map<DtUUID, OrbatRecordHandle> OrbatMapByUUID;
         typedef std::map<DtUUID, std::set<DtUUID> > ParentToChildrenMap;

         struct OrbatRecord
         {
            OrbatRecord(makVrv::DT_PROTOCOL_NAMESPACE::DtVrlinkConnection& sim, const DtRwOrbatData&, DtStateData*);

            virtual ~OrbatRecord();

            makVrv::DtElementID elementID() const
            {
               return myElementID;
            }

            const DtRwOrbatData& data() const;

            virtual const DtUUID& uuid() const;
            virtual const DtUUID& orbatUUID() const;
            virtual void setMarkingText(const DtString&);
            virtual const DtString& markingText() const;

            virtual void setEntityType(const DtEntityType&);
            virtual DtEntityType entityType() const;

            virtual void setForceType(int);
            virtual int forceType() const;

            virtual void setParent(const DtUUID&);
            virtual const DtUUID& parent() const;

            virtual void setOrbatOrder(int);
            virtual int orbatOrder() const;

            virtual void slot_aboutToBeDestroyed();

            makVrv::DtUniqueID myElementID = 0;
            DtStateData* myStateData = nullptr;
            makVrv::DT_PROTOCOL_NAMESPACE::DtVrlinkConnection& mySimulation;
            boost::signals2::connection myDeletedConnection;
         };

         static DtVrfVrlinkGuiOrbatManager* findInstance(makVrv::DT_PROTOCOL_NAMESPACE::DtVrlinkConnection& sim);

         static void registerInstance(makVrv::DT_PROTOCOL_NAMESPACE::DtVrlinkConnection& sim, DtVrfVrlinkGuiOrbatManager*);

         DtVrfVrlinkGuiOrbatManager(makVrv::DT_PROTOCOL_NAMESPACE::DtVrlinkConnection& sim, DtVrfRemoteController*);

         virtual ~DtVrfVrlinkGuiOrbatManager() override;

         virtual void requestOrbatList();

         virtual bool isOrbatNode(makVrv::DtUniqueID) const;

         virtual void clearOrbatList();

         virtual void connectionCleanup();

         virtual void rediscoverObjects();

      protected:
         static void receivedOrbatReponseMessage(DtSimMessage*, void* usr);
         virtual void processReceivedOrbatResponse(DtSimMessage*);

         virtual void addToOrbat(const DtRwOrbatData&);

         virtual void removeFromOrbat(const DtUUID&);

         virtual void modifyInOrbat(const DtRwOrbatData&);

         virtual bool addToParent(const DtUUID& parent, const DtUUID& child);
         virtual void removeFromParent(const DtUUID& parent, const DtUUID& child);
         virtual void aggregateDiscovered(makVrv::DT_PROTOCOL_NAMESPACE::DtVrlinkAggregateStateListener& listener);
         virtual void aggregateDestroyed(makVrv::DT_PROTOCOL_NAMESPACE::DtVrlinkAggregateStateListener& listener);

         virtual void entityDiscovered(makVrv::DT_PROTOCOL_NAMESPACE::DtVrlinkEntityStateListener& listener);
         virtual void entityDestroyed(makVrv::DT_PROTOCOL_NAMESPACE::DtVrlinkEntityStateListener& listener);
         virtual void slot_postConnectionTick();
         virtual void slot_uuidChanged(const DtUUID& oldUUID, const DtUUID& newUUID);

      protected:
         DtVrfRemoteController* myRemoteController;
         makVrv::DT_PROTOCOL_NAMESPACE::DtVrlinkConnection& mySimulation;
         int myRequestId;

         OrbatMapByUniqueID myOrbatMapByUniqueID;
         OrbatMapByUUID myOrbatMapByUUID;
         std::vector<DtRwOrbatData> myQueuedSubordinateChanges;

         typedef std::map<makVrv::DtUniqueID, makVrv::DtUniqueID> PublishedObjectsMap;

         PublishedObjectsMap myOrbatNodesToPublishedObjects;
         PublishedObjectsMap myPublishedObjectsToOrbatNodes;
         std::unordered_map<DtUUID, DtRwOrbatData> myOrbatData;
         makVrv::DtSignalConnectionManager myConnections;

         boost::signals2::connection myAwaitingUUIDConnection;
         std::set<makVrv::DtElementID> myAwaitingElementID;

         tbb::spin_mutex myUUIDChangedMutex;
         boost::signals2::connection myPendingUUIDChangesConnection;
         std::map<DtUUID, DtRwOrbatData> myPendingUUIDChanges;
      };

      class VRF_PROTOCOL_EXPORT DtVrfVrlinkGuiOrbatManagerCreator : public makVrv::DtVirtualBaseClass
      {
      public:
         DtVrfVrlinkGuiOrbatManagerCreator();

         static DtVrfVrlinkGuiOrbatManagerCreator& instance(makVrv::DT_PROTOCOL_NAMESPACE::DtVrlinkConnection& conn);

         static void registerInstance(makVrv::DT_PROTOCOL_NAMESPACE::DtVrlinkConnection& conn, 
            DtVrfVrlinkGuiOrbatManagerCreator* instance);

         virtual DtVrfVrlinkGuiOrbatManager* create(makVrv::DT_PROTOCOL_NAMESPACE::DtVrlinkConnection& conn, DtVrfRemoteController*);
      };
   }
}