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

#pragma once


class Goal;

class GoalSelector
{
private:
	class InternalGoal
	{
	public:
		// 4J Added canDelete param
		InternalGoal(int prio, Goal *goal, bool canDeletePointer);

		Goal *goal;
		int prio;
		bool canDeletePointer;
	};

private:
	std::vector<InternalGoal *> goals;
	std::vector<InternalGoal *> usingGoals;
	int tickCount;
	int newGoalRate;

public:
	GoalSelector();
	~GoalSelector();

	// 4J Added canDelete param
	void addGoal(int prio, Goal *goal, bool canDeletePointer = true);
	void tick();
	std::vector<InternalGoal *> *getRunningGoals();

private:
	bool canContinueToUse(InternalGoal *ig);
	bool canUseInSystem(InternalGoal *goal);
	bool canCoExist(InternalGoal *goalA, InternalGoal *goalB);

public:
	void setNewGoalRate(int newGoalRate);

	// 4J Added override to update ai elements when loading entity from schematics
	void setLevel(Level *level);
};