51 lines
2.7 KiB
Java
51 lines
2.7 KiB
Java
|
package net.minecraft.world.entity.ai.behavior;
|
||
|
|
|
||
|
|
import com.mojang.datafixers.kinds.K1;
|
||
|
|
import java.util.function.Predicate;
|
||
|
|
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;
|
||
|
|
import net.minecraft.world.entity.ai.memory.WalkTarget;
|
||
|
|
import net.minecraft.world.entity.item.ItemEntity;
|
||
|
|
|
||
|
|
public class GoToWantedItem {
|
||
|
|
public static BehaviorControl<LivingEntity> create(float p_260027_, boolean p_259769_, int p_259671_) {
|
||
|
|
return create(p_23158_ -> true, p_260027_, p_259769_, p_259671_);
|
||
|
|
}
|
||
|
|
|
||
|
|
public static <E extends LivingEntity> BehaviorControl<E> create(Predicate<E> p_259490_, float p_260346_, boolean p_259637_, int p_259054_) {
|
||
|
|
return BehaviorBuilder.create(
|
||
|
|
p_258371_ -> {
|
||
|
|
BehaviorBuilder<E, ? extends MemoryAccessor<? extends K1, WalkTarget>> behaviorbuilder = p_259637_
|
||
|
|
? p_258371_.registered(MemoryModuleType.WALK_TARGET)
|
||
|
|
: p_258371_.absent(MemoryModuleType.WALK_TARGET);
|
||
|
|
return p_258371_.group(
|
||
|
|
p_258371_.registered(MemoryModuleType.LOOK_TARGET),
|
||
|
|
behaviorbuilder,
|
||
|
|
p_258371_.present(MemoryModuleType.NEAREST_VISIBLE_WANTED_ITEM),
|
||
|
|
p_258371_.registered(MemoryModuleType.ITEM_PICKUP_COOLDOWN_TICKS)
|
||
|
|
)
|
||
|
|
.apply(
|
||
|
|
p_258371_,
|
||
|
|
(p_258387_, p_258388_, p_258389_, p_258390_) -> (p_358949_, p_358950_, p_358951_) -> {
|
||
|
|
ItemEntity itementity = p_258371_.get(p_258389_);
|
||
|
|
if (p_258371_.tryGet(p_258390_).isEmpty()
|
||
|
|
&& p_259490_.test(p_358950_)
|
||
|
|
&& itementity.closerThan(p_358950_, p_259054_)
|
||
|
|
&& p_358950_.level().getWorldBorder().isWithinBounds(itementity.blockPosition())
|
||
|
|
&& p_358950_.canPickUpLoot()) {
|
||
|
|
WalkTarget walktarget = new WalkTarget(new EntityTracker(itementity, false), p_260346_, 0);
|
||
|
|
p_258387_.set(new EntityTracker(itementity, true));
|
||
|
|
p_258388_.set(walktarget);
|
||
|
|
return true;
|
||
|
|
} else {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
);
|
||
|
|
}
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|