DroneBatteryDecorator.h

#pragma once
 
#include "DroneDecorator.h"
#include "SimulationModel.h"
 
class DroneBatteryDecorator : public DroneDecorator {
 public:
    DroneBatteryDecorator(Drone* drone, unsigned maxCharge_,
                          unsigned currentCharge_, unsigned lowCharge_,
                          double decreaseTime_);
 
    virtual ~DroneBatteryDecorator();
 
    void decreaseCharge(double dt);
 
    void charge(double dt);
 
    Vector3 findNearestRechargeStation();
 
    Vector3 findNearestRechargeStation(Vector3 startPosition);
 
    void update(double dt);
 
    bool isAtRechargeStation();
 
    void headToRechargeStation(Vector3 station);
 
    void addToDeadDronesList();
 
    void addToFunctionalDroneList();
 
    void removeFromFunctionalDroneList();
 
 private:
    void arriveAtRechargeStation();
 
    void lookAheadForRechargeStation();
 
    double batteryNeededForDistance(double distance);
    unsigned maxCharge;
    unsigned currentCharge;
    unsigned lowCharge;
    double movingDecreaseTime;
    double timeElapsed = 0;
    bool ismoving = false;
    bool notCharging = false;
    bool droneReady = false;
    double chargingRate = 2;
    IStrategy* toRechargeStation = nullptr;
    int idleFrames = 0;
    bool goingToPackage = false;
    bool goingToFinalDestination = false;
    double malfunctionedStationTime = 0;
};