4jcraft/Minecraft.World/AI/Goals/FollowOwnerGoal.h

#pragma once

#include "Goal.h"

class PathNavigation;
class TamableAnimal;

class FollowOwnerGoal : public Goal {
public:
    static const int TeleportDistance = 12;

private:
    TamableAnimal* tamable;  // Owner of this goal
    std::weak_ptr<Mob> owner;
    Level* level;
    float speed;
    PathNavigation* navigation;
    int timeToRecalcPath;
    float stopDistance, startDistance;
    bool oldAvoidWater;

public:
    FollowOwnerGoal(TamableAnimal* tamable, float speed, float startDistance,
                    float stopDistance);

    virtual bool canUse();
    virtual bool canContinueToUse();
    virtual void start();
    virtual void stop();
    virtual void tick();

    // 4J Added override to update ai elements when loading entity from
    // schematics
    virtual void setLevel(Level* level) { this->level = level; }
};