36 lines
1.6 KiB
Java
36 lines
1.6 KiB
Java
|
package net.minecraft.world.entity.ai.behavior;
|
||
|
|
|
||
|
|
import net.minecraft.server.level.ServerLevel;
|
||
|
|
import net.minecraft.world.entity.LivingEntity;
|
||
|
|
import net.minecraft.world.entity.ai.behavior.declarative.BehaviorBuilder;
|
||
|
|
import net.minecraft.world.entity.ai.behavior.declarative.MemoryAccessor;
|
||
|
|
import net.minecraft.world.entity.ai.memory.MemoryModuleType;
|
||
|
|
|
||
|
|
public class VillagerCalmDown {
|
||
|
|
private static final int SAFE_DISTANCE_FROM_DANGER = 36;
|
||
|
|
|
||
|
|
public static BehaviorControl<LivingEntity> create() {
|
||
|
|
return BehaviorBuilder.create(
|
||
|
|
p_258884_ -> p_258884_.group(
|
||
|
|
p_258884_.registered(MemoryModuleType.HURT_BY),
|
||
|
|
p_258884_.registered(MemoryModuleType.HURT_BY_ENTITY),
|
||
|
|
p_258884_.registered(MemoryModuleType.NEAREST_HOSTILE)
|
||
|
|
)
|
||
|
|
.apply(
|
||
|
|
p_258884_,
|
||
|
|
(p_258886_, p_258887_, p_258888_) -> (p_390614_, p_390615_, p_390616_) -> {
|
||
|
|
boolean flag = p_258884_.tryGet(p_258886_).isPresent()
|
||
|
|
|| p_258884_.tryGet(p_258888_).isPresent()
|
||
|
|
|| p_258884_.<LivingEntity>tryGet(p_258887_).filter(p_326904_ -> p_326904_.distanceToSqr(p_390615_) <= 36.0).isPresent();
|
||
|
|
if (!flag) {
|
||
|
|
p_258886_.erase();
|
||
|
|
p_258887_.erase();
|
||
|
|
p_390615_.getBrain().updateActivityFromSchedule(p_390614_.getDayTime(), p_390614_.getGameTime());
|
||
|
|
}
|
||
|
|
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
)
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|