438 lines
14 KiB
Java
438 lines
14 KiB
Java
package net.minecraft.world.entity.animal.frog;
|
|
|
|
import com.google.common.collect.ImmutableList;
|
|
import com.mojang.serialization.Dynamic;
|
|
import java.util.Objects;
|
|
import java.util.Optional;
|
|
import java.util.OptionalInt;
|
|
import net.minecraft.core.BlockPos;
|
|
import net.minecraft.core.Holder;
|
|
import net.minecraft.core.Registry;
|
|
import net.minecraft.core.component.DataComponentGetter;
|
|
import net.minecraft.core.component.DataComponentType;
|
|
import net.minecraft.core.component.DataComponents;
|
|
import net.minecraft.core.registries.Registries;
|
|
import net.minecraft.nbt.CompoundTag;
|
|
import net.minecraft.network.protocol.game.DebugPackets;
|
|
import net.minecraft.network.syncher.EntityDataAccessor;
|
|
import net.minecraft.network.syncher.EntityDataSerializers;
|
|
import net.minecraft.network.syncher.SynchedEntityData;
|
|
import net.minecraft.network.syncher.SynchedEntityData.Builder;
|
|
import net.minecraft.resources.ResourceKey;
|
|
import net.minecraft.server.level.ServerLevel;
|
|
import net.minecraft.sounds.SoundEvent;
|
|
import net.minecraft.sounds.SoundEvents;
|
|
import net.minecraft.sounds.SoundSource;
|
|
import net.minecraft.tags.BlockTags;
|
|
import net.minecraft.tags.EntityTypeTags;
|
|
import net.minecraft.tags.ItemTags;
|
|
import net.minecraft.util.Mth;
|
|
import net.minecraft.util.RandomSource;
|
|
import net.minecraft.util.Unit;
|
|
import net.minecraft.util.profiling.Profiler;
|
|
import net.minecraft.util.profiling.ProfilerFiller;
|
|
import net.minecraft.world.DifficultyInstance;
|
|
import net.minecraft.world.damagesource.DamageSource;
|
|
import net.minecraft.world.entity.AgeableMob;
|
|
import net.minecraft.world.entity.AnimationState;
|
|
import net.minecraft.world.entity.Entity;
|
|
import net.minecraft.world.entity.EntitySpawnReason;
|
|
import net.minecraft.world.entity.EntityType;
|
|
import net.minecraft.world.entity.LivingEntity;
|
|
import net.minecraft.world.entity.Mob;
|
|
import net.minecraft.world.entity.MoverType;
|
|
import net.minecraft.world.entity.Pose;
|
|
import net.minecraft.world.entity.SpawnGroupData;
|
|
import net.minecraft.world.entity.ai.Brain;
|
|
import net.minecraft.world.entity.ai.attributes.Attributes;
|
|
import net.minecraft.world.entity.ai.control.LookControl;
|
|
import net.minecraft.world.entity.ai.control.SmoothSwimmingMoveControl;
|
|
import net.minecraft.world.entity.ai.memory.MemoryModuleType;
|
|
import net.minecraft.world.entity.ai.navigation.AmphibiousPathNavigation;
|
|
import net.minecraft.world.entity.ai.navigation.PathNavigation;
|
|
import net.minecraft.world.entity.ai.sensing.Sensor;
|
|
import net.minecraft.world.entity.ai.sensing.SensorType;
|
|
import net.minecraft.world.entity.animal.Animal;
|
|
import net.minecraft.world.entity.monster.Slime;
|
|
import net.minecraft.world.entity.variant.SpawnContext;
|
|
import net.minecraft.world.entity.variant.VariantUtils;
|
|
import net.minecraft.world.item.ItemStack;
|
|
import net.minecraft.world.level.Level;
|
|
import net.minecraft.world.level.LevelAccessor;
|
|
import net.minecraft.world.level.ServerLevelAccessor;
|
|
import net.minecraft.world.level.block.state.BlockState;
|
|
import net.minecraft.world.level.pathfinder.AmphibiousNodeEvaluator;
|
|
import net.minecraft.world.level.pathfinder.Node;
|
|
import net.minecraft.world.level.pathfinder.PathFinder;
|
|
import net.minecraft.world.level.pathfinder.PathType;
|
|
import net.minecraft.world.level.pathfinder.PathfindingContext;
|
|
import net.minecraft.world.phys.Vec3;
|
|
import org.jetbrains.annotations.Nullable;
|
|
|
|
public class Frog extends Animal {
|
|
protected static final ImmutableList<SensorType<? extends Sensor<? super Frog>>> SENSOR_TYPES = ImmutableList.of(
|
|
SensorType.NEAREST_LIVING_ENTITIES, SensorType.HURT_BY, SensorType.FROG_ATTACKABLES, SensorType.FROG_TEMPTATIONS, SensorType.IS_IN_WATER
|
|
);
|
|
protected static final ImmutableList<MemoryModuleType<?>> MEMORY_TYPES = ImmutableList.of(
|
|
MemoryModuleType.LOOK_TARGET,
|
|
MemoryModuleType.NEAREST_LIVING_ENTITIES,
|
|
MemoryModuleType.NEAREST_VISIBLE_LIVING_ENTITIES,
|
|
MemoryModuleType.WALK_TARGET,
|
|
MemoryModuleType.CANT_REACH_WALK_TARGET_SINCE,
|
|
MemoryModuleType.PATH,
|
|
MemoryModuleType.BREED_TARGET,
|
|
MemoryModuleType.LONG_JUMP_COOLDOWN_TICKS,
|
|
MemoryModuleType.LONG_JUMP_MID_JUMP,
|
|
MemoryModuleType.ATTACK_TARGET,
|
|
MemoryModuleType.TEMPTING_PLAYER,
|
|
MemoryModuleType.TEMPTATION_COOLDOWN_TICKS,
|
|
MemoryModuleType.IS_TEMPTED,
|
|
MemoryModuleType.HURT_BY,
|
|
MemoryModuleType.HURT_BY_ENTITY,
|
|
MemoryModuleType.NEAREST_ATTACKABLE,
|
|
MemoryModuleType.IS_IN_WATER,
|
|
MemoryModuleType.IS_PREGNANT,
|
|
MemoryModuleType.IS_PANICKING,
|
|
MemoryModuleType.UNREACHABLE_TONGUE_TARGETS
|
|
);
|
|
private static final EntityDataAccessor<Holder<FrogVariant>> DATA_VARIANT_ID = SynchedEntityData.defineId(Frog.class, EntityDataSerializers.FROG_VARIANT);
|
|
private static final EntityDataAccessor<OptionalInt> DATA_TONGUE_TARGET_ID = SynchedEntityData.defineId(
|
|
Frog.class, EntityDataSerializers.OPTIONAL_UNSIGNED_INT
|
|
);
|
|
private static final int FROG_FALL_DAMAGE_REDUCTION = 5;
|
|
private static final ResourceKey<FrogVariant> DEFAULT_VARIANT = FrogVariants.TEMPERATE;
|
|
public final AnimationState jumpAnimationState = new AnimationState();
|
|
public final AnimationState croakAnimationState = new AnimationState();
|
|
public final AnimationState tongueAnimationState = new AnimationState();
|
|
public final AnimationState swimIdleAnimationState = new AnimationState();
|
|
|
|
public Frog(EntityType<? extends Animal> entityType, Level level) {
|
|
super(entityType, level);
|
|
this.lookControl = new Frog.FrogLookControl(this);
|
|
this.setPathfindingMalus(PathType.WATER, 4.0F);
|
|
this.setPathfindingMalus(PathType.TRAPDOOR, -1.0F);
|
|
this.moveControl = new SmoothSwimmingMoveControl(this, 85, 10, 0.02F, 0.1F, true);
|
|
}
|
|
|
|
@Override
|
|
protected Brain.Provider<Frog> brainProvider() {
|
|
return Brain.provider(MEMORY_TYPES, SENSOR_TYPES);
|
|
}
|
|
|
|
@Override
|
|
protected Brain<?> makeBrain(Dynamic<?> dynamic) {
|
|
return FrogAi.makeBrain(this.brainProvider().makeBrain(dynamic));
|
|
}
|
|
|
|
@Override
|
|
public Brain<Frog> getBrain() {
|
|
return (Brain<Frog>)super.getBrain();
|
|
}
|
|
|
|
@Override
|
|
protected void defineSynchedData(Builder builder) {
|
|
super.defineSynchedData(builder);
|
|
Registry<FrogVariant> registry = this.registryAccess().lookupOrThrow(Registries.FROG_VARIANT);
|
|
builder.define(DATA_VARIANT_ID, VariantUtils.getDefaultOrAny(this.registryAccess(), DEFAULT_VARIANT));
|
|
builder.define(DATA_TONGUE_TARGET_ID, OptionalInt.empty());
|
|
}
|
|
|
|
public void eraseTongueTarget() {
|
|
this.entityData.set(DATA_TONGUE_TARGET_ID, OptionalInt.empty());
|
|
}
|
|
|
|
public Optional<Entity> getTongueTarget() {
|
|
return this.entityData.get(DATA_TONGUE_TARGET_ID).stream().mapToObj(this.level()::getEntity).filter(Objects::nonNull).findFirst();
|
|
}
|
|
|
|
public void setTongueTarget(Entity tongueTarget) {
|
|
this.entityData.set(DATA_TONGUE_TARGET_ID, OptionalInt.of(tongueTarget.getId()));
|
|
}
|
|
|
|
@Override
|
|
public int getHeadRotSpeed() {
|
|
return 35;
|
|
}
|
|
|
|
@Override
|
|
public int getMaxHeadYRot() {
|
|
return 5;
|
|
}
|
|
|
|
public Holder<FrogVariant> getVariant() {
|
|
return this.entityData.get(DATA_VARIANT_ID);
|
|
}
|
|
|
|
private void setVariant(Holder<FrogVariant> variant) {
|
|
this.entityData.set(DATA_VARIANT_ID, variant);
|
|
}
|
|
|
|
@Nullable
|
|
@Override
|
|
public <T> T get(DataComponentType<? extends T> component) {
|
|
return component == DataComponents.FROG_VARIANT ? castComponentValue((DataComponentType<T>)component, this.getVariant()) : super.get(component);
|
|
}
|
|
|
|
@Override
|
|
protected void applyImplicitComponents(DataComponentGetter componentGetter) {
|
|
this.applyImplicitComponentIfPresent(componentGetter, DataComponents.FROG_VARIANT);
|
|
super.applyImplicitComponents(componentGetter);
|
|
}
|
|
|
|
@Override
|
|
protected <T> boolean applyImplicitComponent(DataComponentType<T> component, T value) {
|
|
if (component == DataComponents.FROG_VARIANT) {
|
|
this.setVariant(castComponentValue(DataComponents.FROG_VARIANT, value));
|
|
return true;
|
|
} else {
|
|
return super.applyImplicitComponent(component, value);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void addAdditionalSaveData(CompoundTag tag) {
|
|
super.addAdditionalSaveData(tag);
|
|
VariantUtils.writeVariant(tag, this.getVariant());
|
|
}
|
|
|
|
@Override
|
|
public void readAdditionalSaveData(CompoundTag tag) {
|
|
super.readAdditionalSaveData(tag);
|
|
VariantUtils.readVariant(tag, this.registryAccess(), Registries.FROG_VARIANT).ifPresent(this::setVariant);
|
|
}
|
|
|
|
@Override
|
|
protected void customServerAiStep(ServerLevel level) {
|
|
ProfilerFiller profilerFiller = Profiler.get();
|
|
profilerFiller.push("frogBrain");
|
|
this.getBrain().tick(level, this);
|
|
profilerFiller.pop();
|
|
profilerFiller.push("frogActivityUpdate");
|
|
FrogAi.updateActivity(this);
|
|
profilerFiller.pop();
|
|
super.customServerAiStep(level);
|
|
}
|
|
|
|
@Override
|
|
public void tick() {
|
|
if (this.level().isClientSide()) {
|
|
this.swimIdleAnimationState.animateWhen(this.isInWater() && !this.walkAnimation.isMoving(), this.tickCount);
|
|
}
|
|
|
|
super.tick();
|
|
}
|
|
|
|
@Override
|
|
public void onSyncedDataUpdated(EntityDataAccessor<?> dataAccessor) {
|
|
if (DATA_POSE.equals(dataAccessor)) {
|
|
Pose pose = this.getPose();
|
|
if (pose == Pose.LONG_JUMPING) {
|
|
this.jumpAnimationState.start(this.tickCount);
|
|
} else {
|
|
this.jumpAnimationState.stop();
|
|
}
|
|
|
|
if (pose == Pose.CROAKING) {
|
|
this.croakAnimationState.start(this.tickCount);
|
|
} else {
|
|
this.croakAnimationState.stop();
|
|
}
|
|
|
|
if (pose == Pose.USING_TONGUE) {
|
|
this.tongueAnimationState.start(this.tickCount);
|
|
} else {
|
|
this.tongueAnimationState.stop();
|
|
}
|
|
}
|
|
|
|
super.onSyncedDataUpdated(dataAccessor);
|
|
}
|
|
|
|
@Override
|
|
protected void updateWalkAnimation(float partialTick) {
|
|
float f;
|
|
if (this.jumpAnimationState.isStarted()) {
|
|
f = 0.0F;
|
|
} else {
|
|
f = Math.min(partialTick * 25.0F, 1.0F);
|
|
}
|
|
|
|
this.walkAnimation.update(f, 0.4F, this.isBaby() ? 3.0F : 1.0F);
|
|
}
|
|
|
|
@Override
|
|
public void playEatingSound() {
|
|
this.level().playSound(null, this, SoundEvents.FROG_EAT, SoundSource.NEUTRAL, 2.0F, 1.0F);
|
|
}
|
|
|
|
@Nullable
|
|
@Override
|
|
public AgeableMob getBreedOffspring(ServerLevel level, AgeableMob otherParent) {
|
|
Frog frog = EntityType.FROG.create(level, EntitySpawnReason.BREEDING);
|
|
if (frog != null) {
|
|
FrogAi.initMemories(frog, level.getRandom());
|
|
}
|
|
|
|
return frog;
|
|
}
|
|
|
|
@Override
|
|
public boolean isBaby() {
|
|
return false;
|
|
}
|
|
|
|
@Override
|
|
public void setBaby(boolean baby) {
|
|
}
|
|
|
|
@Override
|
|
public void spawnChildFromBreeding(ServerLevel level, Animal mate) {
|
|
this.finalizeSpawnChildFromBreeding(level, mate, null);
|
|
this.getBrain().setMemory(MemoryModuleType.IS_PREGNANT, Unit.INSTANCE);
|
|
}
|
|
|
|
@Override
|
|
public SpawnGroupData finalizeSpawn(
|
|
ServerLevelAccessor level, DifficultyInstance difficulty, EntitySpawnReason spawnReason, @Nullable SpawnGroupData spawnGroupData
|
|
) {
|
|
FrogVariants.selectVariantToSpawn(this.random, this.registryAccess(), SpawnContext.create(level, this.blockPosition())).ifPresent(this::setVariant);
|
|
FrogAi.initMemories(this, level.getRandom());
|
|
return super.finalizeSpawn(level, difficulty, spawnReason, spawnGroupData);
|
|
}
|
|
|
|
public static net.minecraft.world.entity.ai.attributes.AttributeSupplier.Builder createAttributes() {
|
|
return Animal.createAnimalAttributes()
|
|
.add(Attributes.MOVEMENT_SPEED, 1.0)
|
|
.add(Attributes.MAX_HEALTH, 10.0)
|
|
.add(Attributes.ATTACK_DAMAGE, 10.0)
|
|
.add(Attributes.STEP_HEIGHT, 1.0);
|
|
}
|
|
|
|
@Nullable
|
|
@Override
|
|
protected SoundEvent getAmbientSound() {
|
|
return SoundEvents.FROG_AMBIENT;
|
|
}
|
|
|
|
@Nullable
|
|
@Override
|
|
protected SoundEvent getHurtSound(DamageSource damageSource) {
|
|
return SoundEvents.FROG_HURT;
|
|
}
|
|
|
|
@Nullable
|
|
@Override
|
|
protected SoundEvent getDeathSound() {
|
|
return SoundEvents.FROG_DEATH;
|
|
}
|
|
|
|
@Override
|
|
protected void playStepSound(BlockPos pos, BlockState state) {
|
|
this.playSound(SoundEvents.FROG_STEP, 0.15F, 1.0F);
|
|
}
|
|
|
|
@Override
|
|
public boolean isPushedByFluid() {
|
|
return false;
|
|
}
|
|
|
|
@Override
|
|
protected void sendDebugPackets() {
|
|
super.sendDebugPackets();
|
|
DebugPackets.sendEntityBrain(this);
|
|
}
|
|
|
|
@Override
|
|
protected int calculateFallDamage(double fallDistance, float damageMultiplier) {
|
|
return super.calculateFallDamage(fallDistance, damageMultiplier) - 5;
|
|
}
|
|
|
|
@Override
|
|
public void travel(Vec3 travelVector) {
|
|
if (this.isInWater()) {
|
|
this.moveRelative(this.getSpeed(), travelVector);
|
|
this.move(MoverType.SELF, this.getDeltaMovement());
|
|
this.setDeltaMovement(this.getDeltaMovement().scale(0.9));
|
|
} else {
|
|
super.travel(travelVector);
|
|
}
|
|
}
|
|
|
|
public static boolean canEat(LivingEntity entity) {
|
|
return entity instanceof Slime slime && slime.getSize() != 1 ? false : entity.getType().is(EntityTypeTags.FROG_FOOD);
|
|
}
|
|
|
|
@Override
|
|
protected PathNavigation createNavigation(Level level) {
|
|
return new Frog.FrogPathNavigation(this, level);
|
|
}
|
|
|
|
@Nullable
|
|
@Override
|
|
public LivingEntity getTarget() {
|
|
return this.getTargetFromBrain();
|
|
}
|
|
|
|
@Override
|
|
public boolean isFood(ItemStack stack) {
|
|
return stack.is(ItemTags.FROG_FOOD);
|
|
}
|
|
|
|
public static boolean checkFrogSpawnRules(
|
|
EntityType<? extends Animal> entityType, LevelAccessor level, EntitySpawnReason spawnReason, BlockPos pos, RandomSource random
|
|
) {
|
|
return level.getBlockState(pos.below()).is(BlockTags.FROGS_SPAWNABLE_ON) && isBrightEnoughToSpawn(level, pos);
|
|
}
|
|
|
|
class FrogLookControl extends LookControl {
|
|
FrogLookControl(final Mob mob) {
|
|
super(mob);
|
|
}
|
|
|
|
@Override
|
|
protected boolean resetXRotOnTick() {
|
|
return Frog.this.getTongueTarget().isEmpty();
|
|
}
|
|
}
|
|
|
|
static class FrogNodeEvaluator extends AmphibiousNodeEvaluator {
|
|
private final BlockPos.MutableBlockPos belowPos = new BlockPos.MutableBlockPos();
|
|
|
|
public FrogNodeEvaluator(boolean bl) {
|
|
super(bl);
|
|
}
|
|
|
|
@Override
|
|
public Node getStart() {
|
|
return !this.mob.isInWater()
|
|
? super.getStart()
|
|
: this.getStartNode(
|
|
new BlockPos(Mth.floor(this.mob.getBoundingBox().minX), Mth.floor(this.mob.getBoundingBox().minY), Mth.floor(this.mob.getBoundingBox().minZ))
|
|
);
|
|
}
|
|
|
|
@Override
|
|
public PathType getPathType(PathfindingContext context, int x, int y, int z) {
|
|
this.belowPos.set(x, y - 1, z);
|
|
BlockState blockState = context.getBlockState(this.belowPos);
|
|
return blockState.is(BlockTags.FROG_PREFER_JUMP_TO) ? PathType.OPEN : super.getPathType(context, x, y, z);
|
|
}
|
|
}
|
|
|
|
static class FrogPathNavigation extends AmphibiousPathNavigation {
|
|
FrogPathNavigation(Frog mob, Level level) {
|
|
super(mob, level);
|
|
}
|
|
|
|
@Override
|
|
public boolean canCutCorner(PathType pathType) {
|
|
return pathType != PathType.WATER_BORDER && super.canCutCorner(pathType);
|
|
}
|
|
|
|
@Override
|
|
protected PathFinder createPathFinder(int maxVisitedNodes) {
|
|
this.nodeEvaluator = new Frog.FrogNodeEvaluator(true);
|
|
return new PathFinder(this.nodeEvaluator, maxVisitedNodes);
|
|
}
|
|
}
|
|
}
|