resupplySupplierController.h

Go to the documentation of this file.

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



#pragma once

#include <vrfMsgTransport/events.h>

#include "vrfobjcore/singleTaskControllerComponent.h"
#include <vrfMsgTransport/resupplyOfferEvent.h>
#include <vrfMsgTransport/resupplyCancelEvent.h>
#include "vrfmodel/resupplySupplierStateData.h"
#include "vrfmodel/vrfmodelDefines.h"
#include <boost/signals2.hpp>

class DtResupplySupplierControllerDescriptor;

namespace makVrfEvents
{
   class DtServiceRequestEvent;
   class DtResupplyReceivedEvent;
   class DtEvent;
}

class DT_DLL_vrfmodel DtResupplySupplierController : public DtSingleTaskControllerComponent
{
public:

   DtResupplySupplierController(const DtString& name, DtLocalObject* owner,
      DtSimulationServices* simManager, 
      DtComponentDescriptor* compDesc = 0,
      DtReaderWriterRegistry* parentRegistry = 0);

   virtual ~DtResupplySupplierController();

   virtual const char* type() const;

   virtual bool postAddComponentsInit();

   virtual bool init();

   static void setSupplyingCallback(DtSimMessage * msg, void * usr);

   virtual void processSetSupplying(DtSimMessage * msg);

   virtual void addSetDataCallbacks();

   void removeSetDataCallbacks();

   virtual void tick();

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

protected:

   DtResupplySupplierController();

   DtResupplySupplierController(const DtResupplySupplierController &
      orig);

   const DtResupplySupplierController & operator=(const
      DtResupplySupplierController & orig);

   virtual const DtResupplySupplierControllerDescriptor* getDescriptor();


   virtual void serviceRequestEvent(const makVrfEvents::DtEvent&);
   virtual void processServiceRequest(const makVrfEvents::DtServiceRequestEvent& serviceRequest);
   virtual void resupplyReceivedEvent(const makVrfEvents::DtEvent&);
   virtual void processResupplyReceived(const makVrfEvents::DtResupplyReceivedEvent& resupplyReceived);
   virtual void resupplyCancelEvent(const makVrfEvents::DtEvent&);
   virtual void processResupplyCancel(const makVrfEvents::DtResupplyCancelEvent& resupplyCancel);


   virtual makVrfEvents::DtResupplyOfferEvent createResupplyOfferInteraction(const DtUUID& receiver);
   virtual void sendResupplyOfferInteraction(const makVrfEvents::DtResupplyOfferEvent& resupplyOffer);

   virtual makVrfEvents::DtResupplyCancelEvent createResupplyCancelInteraction(const DtUUID& receiver);
   virtual void sendResupplyCancelInteraction(const  makVrfEvents::DtResupplyCancelEvent& resupplyCancel);

protected:
   void transferSupplies();
   virtual void processReadyState(const DtUUID& receiver);
   void processOfferingState(const DtUUID& receiver);

   virtual void updateRequestedSupplies(const makVrfEvents::DtServiceRequestEvent& serviceRequest);
   virtual void updateOfferedSupplies(const DtUUID& receiver);
   virtual void updateSentSupplies(const makVrfEvents::DtResupplyReceivedEvent& resupplyReceived);
   virtual void removeSentSupplies(const DtUUID& receiver);
   virtual bool checkSuppliesRequested(const DtUUID& receiver);
   virtual bool checkResourcesAvailable(const DtUUID& receiver);
   virtual bool checkResourceAvailable(const DtEntityType& type);
   virtual bool checkResourceAvailable(const DtString& supplyName);
   virtual double resourceAvailable(const DtEntityType& type);
   virtual double resourceAvailable(const DtString& supplyName);
   virtual void updateResource(const DtEntityType& type, double quantity);
   virtual void updateResource(const DtString& supplyName, double quantity);

   virtual const DtString lookUpSupplyType(const DtEntityType& type) const;

   virtual DtTime currentSimTime();

protected:
   const DtResupplySupplierControllerDescriptor* myResupplySupplierDesc;

   std::map<DtUUID, DtResupplySupplierStateData> mySupplierStateData;

   boost::signals2::scoped_connection myServiceRequestEventConnection;
   boost::signals2::scoped_connection myResupplyReceivedEventConnection;
   boost::signals2::scoped_connection myResupplyCancelEventConnection;

protected:
   const DtTime theOfferingStateDefaultTimeout = 120.0;     
};