33 lines
1.2 KiB
Java
33 lines
1.2 KiB
Java
package net.minecraft.world.entity.ai.sensing;
|
|
|
|
import com.google.common.collect.ImmutableSet;
|
|
import com.google.common.collect.Iterables;
|
|
import java.util.Collection;
|
|
import java.util.List;
|
|
import java.util.Set;
|
|
import net.minecraft.server.level.ServerLevel;
|
|
import net.minecraft.world.entity.EntitySelector;
|
|
import net.minecraft.world.entity.ai.memory.MemoryModuleType;
|
|
import net.minecraft.world.entity.monster.breeze.Breeze;
|
|
|
|
public class BreezeAttackEntitySensor extends NearestLivingEntitySensor<Breeze> {
|
|
@Override
|
|
public Set<MemoryModuleType<?>> requires() {
|
|
return ImmutableSet.copyOf(Iterables.concat(super.requires(), List.of(MemoryModuleType.NEAREST_ATTACKABLE)));
|
|
}
|
|
|
|
protected void doTick(ServerLevel serverLevel, Breeze breeze) {
|
|
super.doTick(serverLevel, breeze);
|
|
breeze.getBrain()
|
|
.getMemory(MemoryModuleType.NEAREST_LIVING_ENTITIES)
|
|
.stream()
|
|
.flatMap(Collection::stream)
|
|
.filter(EntitySelector.NO_CREATIVE_OR_SPECTATOR)
|
|
.filter(livingEntity -> Sensor.isEntityAttackable(serverLevel, breeze, livingEntity))
|
|
.findFirst()
|
|
.ifPresentOrElse(
|
|
livingEntity -> breeze.getBrain().setMemory(MemoryModuleType.NEAREST_ATTACKABLE, livingEntity),
|
|
() -> breeze.getBrain().eraseMemory(MemoryModuleType.NEAREST_ATTACKABLE)
|
|
);
|
|
}
|
|
}
|