69 lines
2.5 KiB
Java
69 lines
2.5 KiB
Java
package net.minecraft.world.entity.ai.behavior;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.Optional;
|
|
import java.util.function.BiPredicate;
|
|
import java.util.function.Function;
|
|
import net.minecraft.core.BlockPos;
|
|
import net.minecraft.core.Direction;
|
|
import net.minecraft.server.level.ServerLevel;
|
|
import net.minecraft.sounds.SoundEvent;
|
|
import net.minecraft.tags.TagKey;
|
|
import net.minecraft.util.valueproviders.UniformInt;
|
|
import net.minecraft.world.entity.Mob;
|
|
import net.minecraft.world.level.block.Block;
|
|
|
|
public class LongJumpToPreferredBlock<E extends Mob> extends LongJumpToRandomPos<E> {
|
|
private final TagKey<Block> preferredBlockTag;
|
|
private final float preferredBlocksChance;
|
|
private final List<LongJumpToRandomPos.PossibleJump> notPrefferedJumpCandidates = new ArrayList();
|
|
private boolean currentlyWantingPreferredOnes;
|
|
|
|
public LongJumpToPreferredBlock(
|
|
UniformInt timeBetweenLongJumps,
|
|
int maxLongJumpHeight,
|
|
int maxLongJumpWidth,
|
|
float maxJumpVelocity,
|
|
Function<E, SoundEvent> getJumpSound,
|
|
TagKey<Block> preferredBlockTag,
|
|
float preferredBlocksChance,
|
|
BiPredicate<E, BlockPos> acceptableLandingSpot
|
|
) {
|
|
super(timeBetweenLongJumps, maxLongJumpHeight, maxLongJumpWidth, maxJumpVelocity, getJumpSound, acceptableLandingSpot);
|
|
this.preferredBlockTag = preferredBlockTag;
|
|
this.preferredBlocksChance = preferredBlocksChance;
|
|
}
|
|
|
|
@Override
|
|
protected void start(ServerLevel serverLevel, E mob, long l) {
|
|
super.start(serverLevel, mob, l);
|
|
this.notPrefferedJumpCandidates.clear();
|
|
this.currentlyWantingPreferredOnes = mob.getRandom().nextFloat() < this.preferredBlocksChance;
|
|
}
|
|
|
|
@Override
|
|
protected Optional<LongJumpToRandomPos.PossibleJump> getJumpCandidate(ServerLevel level) {
|
|
if (!this.currentlyWantingPreferredOnes) {
|
|
return super.getJumpCandidate(level);
|
|
} else {
|
|
BlockPos.MutableBlockPos mutableBlockPos = new BlockPos.MutableBlockPos();
|
|
|
|
while (!this.jumpCandidates.isEmpty()) {
|
|
Optional<LongJumpToRandomPos.PossibleJump> optional = super.getJumpCandidate(level);
|
|
if (optional.isPresent()) {
|
|
LongJumpToRandomPos.PossibleJump possibleJump = (LongJumpToRandomPos.PossibleJump)optional.get();
|
|
if (level.getBlockState(mutableBlockPos.setWithOffset(possibleJump.targetPos(), Direction.DOWN)).is(this.preferredBlockTag)) {
|
|
return optional;
|
|
}
|
|
|
|
this.notPrefferedJumpCandidates.add(possibleJump);
|
|
}
|
|
}
|
|
|
|
return !this.notPrefferedJumpCandidates.isEmpty()
|
|
? Optional.of((LongJumpToRandomPos.PossibleJump)this.notPrefferedJumpCandidates.remove(0))
|
|
: Optional.empty();
|
|
}
|
|
}
|
|
}
|