33 lines
995 B
Java
33 lines
995 B
Java
package net.minecraft.world.entity.ai.goal;
|
|
|
|
import net.minecraft.world.entity.EquipmentSlot;
|
|
import net.minecraft.world.entity.PathfinderMob;
|
|
import net.minecraft.world.entity.ai.navigation.GroundPathNavigation;
|
|
import net.minecraft.world.entity.ai.util.GoalUtils;
|
|
|
|
public class RestrictSunGoal extends Goal {
|
|
private final PathfinderMob mob;
|
|
|
|
public RestrictSunGoal(PathfinderMob mob) {
|
|
this.mob = mob;
|
|
}
|
|
|
|
@Override
|
|
public boolean canUse() {
|
|
return this.mob.level().isBrightOutside() && this.mob.getItemBySlot(EquipmentSlot.HEAD).isEmpty() && GoalUtils.hasGroundPathNavigation(this.mob);
|
|
}
|
|
|
|
@Override
|
|
public void start() {
|
|
if (this.mob.getNavigation() instanceof GroundPathNavigation groundPathNavigation) {
|
|
groundPathNavigation.setAvoidSun(true);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void stop() {
|
|
if (GoalUtils.hasGroundPathNavigation(this.mob) && this.mob.getNavigation() instanceof GroundPathNavigation groundPathNavigation) {
|
|
groundPathNavigation.setAvoidSun(false);
|
|
}
|
|
}
|
|
}
|