53 lines
2.9 KiB
Java
53 lines
2.9 KiB
Java
|
package net.minecraft.world.entity.ai.behavior;
|
||
|
|
|
||
|
|
import java.util.Optional;
|
||
|
|
import net.minecraft.core.GlobalPos;
|
||
|
|
import net.minecraft.core.Holder;
|
||
|
|
import net.minecraft.core.registries.BuiltInRegistries;
|
||
|
|
import net.minecraft.server.MinecraftServer;
|
||
|
|
import net.minecraft.server.level.ServerLevel;
|
||
|
|
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;
|
||
|
|
import net.minecraft.world.entity.ai.village.poi.PoiType;
|
||
|
|
import net.minecraft.world.entity.npc.Villager;
|
||
|
|
import net.minecraft.world.entity.npc.VillagerProfession;
|
||
|
|
|
||
|
|
public class AssignProfessionFromJobSite {
|
||
|
|
public static BehaviorControl<Villager> create() {
|
||
|
|
return BehaviorBuilder.create(
|
||
|
|
p_258312_ -> p_258312_.group(p_258312_.present(MemoryModuleType.POTENTIAL_JOB_SITE), p_258312_.registered(MemoryModuleType.JOB_SITE))
|
||
|
|
.apply(
|
||
|
|
p_258312_,
|
||
|
|
(p_258304_, p_258305_) -> (p_258309_, p_258310_, p_258311_) -> {
|
||
|
|
GlobalPos globalpos = p_258312_.get(p_258304_);
|
||
|
|
if (!globalpos.pos().closerToCenterThan(p_258310_.position(), 2.0) && !p_258310_.assignProfessionWhenSpawned()) {
|
||
|
|
return false;
|
||
|
|
} else {
|
||
|
|
p_258304_.erase();
|
||
|
|
p_258305_.set(globalpos);
|
||
|
|
p_258309_.broadcastEntityEvent(p_258310_, (byte)14);
|
||
|
|
if (!p_258310_.getVillagerData().profession().is(VillagerProfession.NONE)) {
|
||
|
|
return true;
|
||
|
|
} else {
|
||
|
|
MinecraftServer minecraftserver = p_258309_.getServer();
|
||
|
|
Optional.ofNullable(minecraftserver.getLevel(globalpos.dimension()))
|
||
|
|
.flatMap(p_22467_ -> p_22467_.getPoiManager().getType(globalpos.pos()))
|
||
|
|
.flatMap(
|
||
|
|
p_390545_ -> BuiltInRegistries.VILLAGER_PROFESSION
|
||
|
|
.listElements()
|
||
|
|
.filter(p_390547_ -> p_390547_.value().heldJobSite().test((Holder<PoiType>)p_390545_))
|
||
|
|
.findFirst()
|
||
|
|
)
|
||
|
|
.ifPresent(p_390550_ -> {
|
||
|
|
p_258310_.setVillagerData(p_258310_.getVillagerData().withProfession(p_390550_));
|
||
|
|
p_258310_.refreshBrain(p_258309_);
|
||
|
|
});
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
)
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|