embeddedEntitiesController.h

Go to the documentation of this file.

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


#pragma once
#include "vrfobjcore/embeddedEntitiesPSR.h"
#include "vrfobjcore/multiTaskControllerComponent.h"

#include "vrfmodel/vrfmodelDefines.h"

#include "vrfobjparam/embeddedEntitiesDescriptor.h"

#include <boost/signals2.hpp>
#include <tbb/mutex.h>

struct DtEmbeddedEntity;

class DT_DLL_vrfmodel DtEmbeddedEntitiesController :
   public DtMultiTaskControllerComponent
{
public:
   DtEmbeddedEntitiesController(const DtString& name,
      DtLocalObject* owner,
      DtSimulationServices* simManager,
      DtComponentDescriptor* desc = 0,
      DtReaderWriterRegistry* parentRegistry = 0);
   virtual ~DtEmbeddedEntitiesController() override;

   static DtSimComponent* creator(const DtString& name, 
      DtLocalObject* owner,
      DtSimulationServices* simManager, 
      DtComponentDescriptor* desc = 0,
      DtReaderWriterRegistry* parentRegistry = 0);
   virtual const char* type() const override;

   virtual bool init() override;
   virtual void preFirstTickInit() override;

   virtual void registerTaskMsgCallbacks() override;
   virtual DtEmbeddedEntitiesPSR* createPSR();

   virtual void clearTask(int tskId) override;

   virtual void tick() override;

   virtual void processEmbeddedEntityCurrentlyInContext(double deltaTime);
   virtual bool checkWithinRecoveryRange();
   virtual bool performDeployment();
   virtual bool performRecovery();
   virtual bool performSetForward();
   virtual bool performTaskForward();
   virtual DtVector calculateSpawnPosOfTarget();
   virtual bool commandEeToReturn(const DtString& entityName);

   virtual DtUUID getObjectNameOfEmbeddedEntity(std::string eeName, bool& found);

   virtual DtRwEmbeddedEntityStatus::DtEmbeddedEntityDeploymentState
      getStatusOfEmbeddedEntity(std::string eeName, bool& found);

   static void setForwardTaskCallback(DtSimMessage * msg, void * usrData);
   static void taskForwardTaskCallback(DtSimMessage * msg, void * usrData);
   virtual void processForwardTask(DtSimMessage * msg);

   static void deployTaskCallback(DtSimMessage * msg, void * usrData);
   static void recoverTaskCallback(DtSimMessage * msg, void * usrData);
   virtual void processTask(DtSimMessage * msg);

   virtual std::list<DtEmbeddedEntity> availableEmbeddedEntities();


   virtual void processDeletion(const DtSimObject* spawnedObject);


   static void hostRestoreCallback(DtSimMessage* message, void* usrData);
   virtual void processHostRestore(DtSimMessage* message);

protected:
   virtual void restoreEmbedded(DtRwEmbeddedEntityStatus* const eeStatus);

protected:
   DtEmbeddedEntitiesPSR* myProcessState;
   const DtEmbeddedEntitiesDescriptor* myComponentDescriptor;

   DtRwEmbeddedEntityTaskState* myEeCurrentlyProcessing;

   std::map<std::string, DtRwEmbeddedEntity*> myNamedEntityMap;

   std::list<int> myFaultedTasks;

   typedef std::map<DtUUID, boost::signals2::connection> RemovalSignalsMap;
   RemovalSignalsMap myRemovalSignalsMap;
   tbb::mutex myRemovalMutex;
};