SimulationModel.h

#ifndef SIMULATION_MODEL_H_
#define SIMULATION_MODEL_H_
 
#include <deque>
#include <map>
#include <set>
 
#include "CompositeFactory.h"
#include "Drone.h"
#include "MultiDeliveryDecorator.h"
#include "IController.h"
#include "IEntity.h"
#include "IObserver.h"
#include "POI.h"
#include "Robot.h"
#include "Graph.h"
 
class POI;
class MultiDeliveryDecorator;
 
//--------------------  Model ----------------------------
 
 
class SimulationModel : public IObserver {
 public:
  SimulationModel(IController& controller);
 
  ~SimulationModel();
 
  void setGraph(const routing::Graph* graph);
 
  IEntity* createEntity(const JsonObject& entity);
 
  void removeEntity(int id);
 
  void scheduleTrip(const JsonObject& details);
 
  void update(double dt);
 
  void stop();
 
  const routing::Graph* getGraph() const;
 
  void notify(const std::string& message) const;
 
  std::vector<Vector3> getRechargeStations() const;
 
  void addRechargeStation(Vector3 station);
 
  void removeRechargeStation(Vector3 station);
 
  std::deque<Package*> scheduledDeliveries;
 
  std::deque<Drone*> deadDrones;
 
  std::deque<Drone*> functionalDrones;
 
  std::deque<Drone*> chargingDrones;
 
  IController& controller;
 
  std::vector<POI*> pois;
 
  std::vector<MultiDeliveryDecorator*> drones;
 
 protected:
  // Keeps track of all pois and drones in the simulation
  std::map<int, IEntity*> entities;
  std::set<int> removed;
  void removeFromSim(int id);
  const routing::Graph* graph = nullptr;
  CompositeFactory entityFactory;
  std::vector<Vector3> rechargeStations;
};
 
#endif