scriptMultiTaskController.h

Go to the documentation of this file.

/*******************************************************************************
** Copyright (c) 2013 MAK Technologies, Inc.
** All rights reserved.
*******************************************************************************/
#pragma once

#include <vrfmodel/vrfmodelDefines.h>

#include <map>
#include <vrfobjcore/multiTaskControllerComponent.h>
#include "vrfobjcore/scriptedTaskManager.h"
#include "vrftasks/radioMessageTypes.h"
#include <vrfobjcore/scriptInterface.h>
#include "vrfmsgs/vrfObjectMessage.h"
#include "vrfobjcore/scriptMultiTaskControllerPSR.h"
#include "vrfmsgs/simMessage.h"
#include "vrfutil/scriptedTaskMetaData.h"
#include "vrfobjcore/scriptTaskControllerInterface.h"

class DtSimMessage;
class DtScriptedTaskSet;

class DT_DLL_vrfmodel DtScriptMultiTaskController : public DtMultiTaskControllerComponent, public DtScriptTaskControllerInterface
{
public:
   DtScriptMultiTaskController(const DtString& name, 
      DtLocalObject* owner, DtSimulationServices* simManager, DtComponentDescriptor* desc,
      DtReaderWriterRegistry* parentRegistry);

   virtual ~DtScriptMultiTaskController() override;

   virtual const char* type() const override;

   virtual bool init() override;

   virtual void preFirstTickInit() override;

   virtual void tick() override;

   virtual bool tickWhileDestroyed() const override;

   virtual bool beginProcessingTask(const DtTaskMessage& task, DtSimTaskStatePtr taskState) override;

   virtual void clearTask(int taskId) override;

   virtual void taskComplete(int taskId, bool success) override;
   virtual void setComplete(DtSetMessageType setId);

   virtual void suspendTask(int taskId) override;

   virtual DtSimTaskStatePtr currentTaskState(int taskId) const override;

   virtual bool createPSR();
   
   virtual void registerScriptTaskCallbacks();
   virtual void unregisterScriptTaskCallbacks();

   virtual void registerTaskMsgCallbacks() override;
   virtual void unregisterTaskMsgCallbacks();

   virtual void registerScriptSetCallbacks();
   virtual void unregisterScriptSetCallbacks();

   virtual bool isTypeTaskable(const DtSimTaskMessageType& taskType, bool remap = true) const;
   virtual void addTypeTaskable(const DtSimTaskMessageType& taskType, const DtString& system = "");
   virtual void removeTypeTaskable(const DtSimTaskMessageType& taskType);

   virtual bool isTypeSetable(const DtSetMessageType& taskType, bool remap = true) const;
   virtual void addTypeSetable(const DtSetMessageType& taskType, const DtString& system = "");
   virtual void removeTypeSetable(const DtSetMessageType& taskType);

   virtual void addRunnableTaskType(const DtSimTaskMessageType& taskType, const DtString& system);
   virtual void removeRunnableTaskType(const DtSimTaskMessageType& taskType);

   virtual void addRunnableSetType(const DtSetMessageType& taskType, const DtString& system);
   virtual void removeRunnableSetType(const DtSetMessageType& taskType);
   
   virtual void sweepMarkedForDeletion(bool clearFromPsr = true);

   virtual void setScriptTickPeriod(int taskId, const DtString& scriptId, double period);

   static void scriptUpdatedCallback(const DtScriptedTaskManagerCallbackInfo& info, void* usr);
   static void scenarioSaveCallback(void* usr);
   static void generateGraphOutputCallback(const DtVrfObjectMessage * msg, void* usr);

   
   virtual void scriptSetTickPeriod(int taskId, const DtString& scriptId, double period) override;
   virtual void scriptSetTickWhilePaused(int taskId, const DtString& scriptId, bool enable) override;
   virtual void scriptSetRespondToNewSensorContacts(int taskId, const DtString& scriptId, bool respond) override;
   virtual void scriptTaskComplete(int taskId, bool success) override;
   virtual void scriptSetComplete(const DtString& scriptId) override;
   virtual int scriptSendSubtask(DtSimTask& task, int parentTaskId) override;
   virtual std::vector<int> scriptPendingSubstaskIds(int taskId) override;
   virtual DtSubtaskStatusEntry* scriptSubtaskStatus(int subtaskId, int parentTaskId) override;
   virtual void cleanupInactiveSubtaskStatus(int parentTaskId) override;
   virtual void scriptClearSubtask(int subtaskId, int parentTaskId) override;
   virtual void setCheckpointModeUseCheckpointState(int taskId, const DtString& scriptId, bool) override;
   virtual bool checkpointModeUseCheckpointState(int taskId, const DtString& scriptId) const override;

   virtual int askUserQuestion(const DtString& scriptId, const DtTranslatableStream& title,
      const DtTranslatableStream& message, const std::vector<DtTranslatableStream>& options, int displayFlag) override;
   virtual int statusOfUserQuestion(int questionId) override;
   virtual void cancelUserQuestion(int questionId) override;

   virtual int scriptSendTask(DtSimTask& task, const DtUUID& entityName, int parentTaskId, bool actAsSubtask = true,
      bool useRadio = false) override;

   virtual void notifyScriptTaskError(const DtString& scriptId, int taskId) override;

   virtual makVrf::DtPointTaskVisualizationPtr pointTaskVisualization(int taskId, const DtString& visName) override;
   virtual makVrf::DtLineTaskVisualizationPtr lineTaskVisualization(int taskId, const DtString& visName) override;
   virtual makVrf::DtAreaTaskVisualizationPtr areaTaskVisualization(int taskId, const DtString& visName) override;
   virtual makVrf::DtTextTaskVisualizationPtr textTaskVisualization(int taskId, const DtString& visName) override;
   virtual void deleteTaskVisualization(int taskId, const DtString& visName) override;

   virtual void addAdditionalReactiveTask(const DtSimTaskMessageType&);


   virtual void processNewSensorContact();

   static DtSimComponent* creator(const DtString& name, 
      DtLocalObject* owner, DtSimulationServices* simManager, DtComponentDescriptor* desc,
      DtReaderWriterRegistry* parentRegistry);

   virtual int nextBackgroundProcessId() const;

   virtual void startScriptedSet(DtScriptedTaskSet*);

protected:
   DtScriptMultiTaskController(const DtScriptMultiTaskController& other);
   DtScriptMultiTaskController& operator=(const DtScriptMultiTaskController& other);

   virtual void processScriptUpdatedCallback(const DtScriptedTaskManagerCallbackInfo& info);
   virtual void processScenarioSaveCallback();
   virtual void processGenerateGraphOutputCallback(const DtVrfObjectMessage * msg);

   virtual void propagateScriptIdChange(const DtString& oldId, const DtString& newId);
   virtual bool isForMyEntityType(const DtScriptedTaskMetaData& metaData);
   virtual void processMetaDataChange(const DtScriptedTaskManagerCallbackInfo& info);
   virtual void restartScriptInterfaceByType(const DtSimTaskMessageType& taskType);
   virtual void restartScriptInterfaceByType(const DtSetMessageType& taskType);

   virtual void clearAndRestoreTasks(bool forceStop = false);

   virtual void clearAndRestoreSets();

   static void userQuestionOperationCallback(DtSimMessage* msg, void* usrData);
   virtual void processUserQuestionOperation(DtSimMessage*);

   virtual void sendOffOutstandingQuestionFor(const DtString& scriptId);

   virtual bool backgroundProcessAvailable() const;

   static void processScriptedSetCallback(DtSimMessage *msg, void * usrData);
   virtual void processScriptedSet(DtSimMessage *msg);

   virtual bool checkRemoveTaskId(int taskId, bool clearFromPsr = true);

   virtual bool markedForDeletion(int taskId) const;

protected:
   std::map<DtSimTaskMessageType, int> myAvailableTaskScripts;
   std::map<DtSetMessageType, int> myAvailableSetScripts;
   std::map<DtSimTaskMessageType, int> myBackgroundProcesses;

   std::map<int, std::shared_ptr<DtScriptInterface> > myCurrentScriptInterfaceMap;
   std::list<int> myMarkedForDeletionList;

   typedef std::pair<int, DtString> TaskAndScriptIds;
   std::list<TaskAndScriptIds> myRespondToNewSensorContactList;
   DtScriptMultiTaskControllerPSR* myProcessState;
   bool myAreTaskCallbacksRegistered;
   std::map<DtSimTaskMessageType, DtString> myAdditionalRunnableTaskTypes;
   std::map<DtSetMessageType, DtString> myAdditionalRunnableSetTypes;
   DtForceType myCurrentForceType;

   bool myIsDestroyed;

   bool myQueueTasksToBeAdded;
   bool myQueueSetsToBeAdded;
   std::map<DtSimTaskMessageType, DtString> myTasksToBeAdded;
   std::map<DtSetMessageType, DtString> mySetsToBeAdded;

   std::set<DtSimTaskMessageType> myAdditionalReactiveTasks;
   bool myTasksRegistered;
   std::set<int> myRemovedTasks;
   boost::signals2::connection myNewSensorContactConnection;
};