31 lines
868 B
Java
31 lines
868 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().isDay() && this.mob.getItemBySlot(EquipmentSlot.HEAD).isEmpty() && GoalUtils.hasGroundPathNavigation(this.mob);
|
|
}
|
|
|
|
@Override
|
|
public void start() {
|
|
((GroundPathNavigation)this.mob.getNavigation()).setAvoidSun(true);
|
|
}
|
|
|
|
@Override
|
|
public void stop() {
|
|
if (GoalUtils.hasGroundPathNavigation(this.mob)) {
|
|
((GroundPathNavigation)this.mob.getNavigation()).setAvoidSun(false);
|
|
}
|
|
}
|
|
}
|