56 lines
2.5 KiB
Java
56 lines
2.5 KiB
Java
|
package net.minecraft.world.entity.ai.behavior;
|
||
|
|
|
||
|
|
import com.google.common.collect.ImmutableMap;
|
||
|
|
import java.util.Optional;
|
||
|
|
import net.minecraft.core.BlockPos;
|
||
|
|
import net.minecraft.core.GlobalPos;
|
||
|
|
import net.minecraft.core.Holder;
|
||
|
|
import net.minecraft.network.protocol.game.DebugPackets;
|
||
|
|
import net.minecraft.server.level.ServerLevel;
|
||
|
|
import net.minecraft.world.entity.ai.memory.MemoryModuleType;
|
||
|
|
import net.minecraft.world.entity.ai.memory.MemoryStatus;
|
||
|
|
import net.minecraft.world.entity.ai.village.poi.PoiManager;
|
||
|
|
import net.minecraft.world.entity.npc.Villager;
|
||
|
|
import net.minecraft.world.entity.schedule.Activity;
|
||
|
|
|
||
|
|
public class GoToPotentialJobSite extends Behavior<Villager> {
|
||
|
|
private static final int TICKS_UNTIL_TIMEOUT = 1200;
|
||
|
|
final float speedModifier;
|
||
|
|
|
||
|
|
public GoToPotentialJobSite(float p_23098_) {
|
||
|
|
super(ImmutableMap.of(MemoryModuleType.POTENTIAL_JOB_SITE, MemoryStatus.VALUE_PRESENT), 1200);
|
||
|
|
this.speedModifier = p_23098_;
|
||
|
|
}
|
||
|
|
|
||
|
|
protected boolean checkExtraStartConditions(ServerLevel p_23103_, Villager p_23104_) {
|
||
|
|
return p_23104_.getBrain()
|
||
|
|
.getActiveNonCoreActivity()
|
||
|
|
.map(p_23115_ -> p_23115_ == Activity.IDLE || p_23115_ == Activity.WORK || p_23115_ == Activity.PLAY)
|
||
|
|
.orElse(true);
|
||
|
|
}
|
||
|
|
|
||
|
|
protected boolean canStillUse(ServerLevel p_23106_, Villager p_23107_, long p_23108_) {
|
||
|
|
return p_23107_.getBrain().hasMemoryValue(MemoryModuleType.POTENTIAL_JOB_SITE);
|
||
|
|
}
|
||
|
|
|
||
|
|
protected void tick(ServerLevel p_23121_, Villager p_23122_, long p_23123_) {
|
||
|
|
BehaviorUtils.setWalkAndLookTargetMemories(p_23122_, p_23122_.getBrain().getMemory(MemoryModuleType.POTENTIAL_JOB_SITE).get().pos(), this.speedModifier, 1);
|
||
|
|
}
|
||
|
|
|
||
|
|
protected void stop(ServerLevel p_23129_, Villager p_23130_, long p_23131_) {
|
||
|
|
Optional<GlobalPos> optional = p_23130_.getBrain().getMemory(MemoryModuleType.POTENTIAL_JOB_SITE);
|
||
|
|
optional.ifPresent(p_23111_ -> {
|
||
|
|
BlockPos blockpos = p_23111_.pos();
|
||
|
|
ServerLevel serverlevel = p_23129_.getServer().getLevel(p_23111_.dimension());
|
||
|
|
if (serverlevel != null) {
|
||
|
|
PoiManager poimanager = serverlevel.getPoiManager();
|
||
|
|
if (poimanager.exists(blockpos, p_217230_ -> true)) {
|
||
|
|
poimanager.release(blockpos);
|
||
|
|
}
|
||
|
|
|
||
|
|
DebugPackets.sendPoiTicketCountPacket(p_23129_, blockpos);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
p_23130_.getBrain().eraseMemory(MemoryModuleType.POTENTIAL_JOB_SITE);
|
||
|
|
}
|
||
|
|
}
|