48 lines
1.5 KiB
Java
48 lines
1.5 KiB
Java
|
package net.minecraft.world.entity.ai.goal;
|
||
|
|
||
|
import java.util.function.Predicate;
|
||
|
import javax.annotation.Nullable;
|
||
|
import net.minecraft.sounds.SoundEvent;
|
||
|
import net.minecraft.world.InteractionHand;
|
||
|
import net.minecraft.world.entity.EquipmentSlot;
|
||
|
import net.minecraft.world.entity.Mob;
|
||
|
import net.minecraft.world.item.ItemStack;
|
||
|
|
||
|
public class UseItemGoal<T extends Mob> extends Goal {
|
||
|
private final T mob;
|
||
|
private final ItemStack item;
|
||
|
private final Predicate<? super T> canUseSelector;
|
||
|
@Nullable
|
||
|
private final SoundEvent finishUsingSound;
|
||
|
|
||
|
public UseItemGoal(T p_25972_, ItemStack p_25973_, @Nullable SoundEvent p_25974_, Predicate<? super T> p_25975_) {
|
||
|
this.mob = p_25972_;
|
||
|
this.item = p_25973_;
|
||
|
this.finishUsingSound = p_25974_;
|
||
|
this.canUseSelector = p_25975_;
|
||
|
}
|
||
|
|
||
|
@Override
|
||
|
public boolean canUse() {
|
||
|
return this.canUseSelector.test(this.mob);
|
||
|
}
|
||
|
|
||
|
@Override
|
||
|
public boolean canContinueToUse() {
|
||
|
return this.mob.isUsingItem();
|
||
|
}
|
||
|
|
||
|
@Override
|
||
|
public void start() {
|
||
|
this.mob.setItemSlot(EquipmentSlot.MAINHAND, this.item.copy());
|
||
|
this.mob.startUsingItem(InteractionHand.MAIN_HAND);
|
||
|
}
|
||
|
|
||
|
@Override
|
||
|
public void stop() {
|
||
|
this.mob.setItemSlot(EquipmentSlot.MAINHAND, ItemStack.EMPTY);
|
||
|
if (this.finishUsingSound != null) {
|
||
|
this.mob.playSound(this.finishUsingSound, 1.0F, this.mob.getRandom().nextFloat() * 0.2F + 0.9F);
|
||
|
}
|
||
|
}
|
||
|
}
|