24 lines
824 B
Java
24 lines
824 B
Java
|
package net.minecraft.world.entity.ai.sensing;
|
||
|
|
|
||
|
|
import com.google.common.collect.ImmutableSet;
|
||
|
|
import java.util.Set;
|
||
|
|
import net.minecraft.server.level.ServerLevel;
|
||
|
|
import net.minecraft.util.Unit;
|
||
|
|
import net.minecraft.world.entity.LivingEntity;
|
||
|
|
import net.minecraft.world.entity.ai.memory.MemoryModuleType;
|
||
|
|
|
||
|
|
public class IsInWaterSensor extends Sensor<LivingEntity> {
|
||
|
|
@Override
|
||
|
|
public Set<MemoryModuleType<?>> requires() {
|
||
|
|
return ImmutableSet.of(MemoryModuleType.IS_IN_WATER);
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
protected void doTick(ServerLevel p_217816_, LivingEntity p_217817_) {
|
||
|
|
if (p_217817_.isInWater()) {
|
||
|
|
p_217817_.getBrain().setMemory(MemoryModuleType.IS_IN_WATER, Unit.INSTANCE);
|
||
|
|
} else {
|
||
|
|
p_217817_.getBrain().eraseMemory(MemoryModuleType.IS_IN_WATER);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|