Code/net/minecraft/world/entity/AreaEffectCloud.java

413 lines
17 KiB
Java
Raw Normal View History

2025-07-01 06:20:03 +00:00
package net.minecraft.world.entity;
import com.google.common.collect.Maps;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.UUID;
import java.util.Map.Entry;
import javax.annotation.Nullable;
import net.minecraft.core.UUIDUtil;
import net.minecraft.core.component.DataComponentGetter;
import net.minecraft.core.component.DataComponentType;
import net.minecraft.core.component.DataComponents;
import net.minecraft.core.particles.ColorParticleOption;
import net.minecraft.core.particles.ParticleOptions;
import net.minecraft.core.particles.ParticleTypes;
import net.minecraft.nbt.CompoundTag;
import net.minecraft.nbt.NbtOps;
import net.minecraft.nbt.Tag;
import net.minecraft.network.syncher.EntityDataAccessor;
import net.minecraft.network.syncher.EntityDataSerializers;
import net.minecraft.network.syncher.SynchedEntityData;
import net.minecraft.resources.RegistryOps;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.util.ARGB;
import net.minecraft.util.Mth;
import net.minecraft.world.damagesource.DamageSource;
import net.minecraft.world.effect.MobEffectInstance;
import net.minecraft.world.item.alchemy.PotionContents;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.material.PushReaction;
public class AreaEffectCloud extends Entity implements TraceableEntity {
private static final int TIME_BETWEEN_APPLICATIONS = 5;
private static final EntityDataAccessor<Float> DATA_RADIUS = SynchedEntityData.defineId(AreaEffectCloud.class, EntityDataSerializers.FLOAT);
private static final EntityDataAccessor<Boolean> DATA_WAITING = SynchedEntityData.defineId(AreaEffectCloud.class, EntityDataSerializers.BOOLEAN);
private static final EntityDataAccessor<ParticleOptions> DATA_PARTICLE = SynchedEntityData.defineId(AreaEffectCloud.class, EntityDataSerializers.PARTICLE);
private static final float MAX_RADIUS = 32.0F;
private static final int DEFAULT_AGE = 0;
private static final int DEFAULT_DURATION_ON_USE = 0;
private static final float DEFAULT_RADIUS_ON_USE = 0.0F;
private static final float DEFAULT_RADIUS_PER_TICK = 0.0F;
private static final float DEFAULT_POTION_DURATION_SCALE = 1.0F;
private static final float MINIMAL_RADIUS = 0.5F;
private static final float DEFAULT_RADIUS = 3.0F;
public static final float DEFAULT_WIDTH = 6.0F;
public static final float HEIGHT = 0.5F;
public static final int INFINITE_DURATION = -1;
public static final int DEFAULT_LINGERING_DURATION = 600;
private static final int DEFAULT_WAIT_TIME = 20;
private static final int DEFAULT_REAPPLICATION_DELAY = 20;
private static final ColorParticleOption DEFAULT_PARTICLE = ColorParticleOption.create(ParticleTypes.ENTITY_EFFECT, -1);
private PotionContents potionContents = PotionContents.EMPTY;
private float potionDurationScale = 1.0F;
private final Map<Entity, Integer> victims = Maps.newHashMap();
private int duration = -1;
private int waitTime = 20;
private int reapplicationDelay = 20;
private int durationOnUse = 0;
private float radiusOnUse = 0.0F;
private float radiusPerTick = 0.0F;
@Nullable
private LivingEntity owner;
@Nullable
private UUID ownerUUID;
public AreaEffectCloud(EntityType<? extends AreaEffectCloud> p_19704_, Level p_19705_) {
super(p_19704_, p_19705_);
this.noPhysics = true;
}
public AreaEffectCloud(Level p_19707_, double p_19708_, double p_19709_, double p_19710_) {
this(EntityType.AREA_EFFECT_CLOUD, p_19707_);
this.setPos(p_19708_, p_19709_, p_19710_);
}
@Override
protected void defineSynchedData(SynchedEntityData.Builder p_330412_) {
p_330412_.define(DATA_RADIUS, 3.0F);
p_330412_.define(DATA_WAITING, false);
p_330412_.define(DATA_PARTICLE, DEFAULT_PARTICLE);
}
public void setRadius(float p_19713_) {
if (!this.level().isClientSide) {
this.getEntityData().set(DATA_RADIUS, Mth.clamp(p_19713_, 0.0F, 32.0F));
}
}
@Override
public void refreshDimensions() {
double d0 = this.getX();
double d1 = this.getY();
double d2 = this.getZ();
super.refreshDimensions();
this.setPos(d0, d1, d2);
}
public float getRadius() {
return this.getEntityData().get(DATA_RADIUS);
}
public void setPotionContents(PotionContents p_332440_) {
this.potionContents = p_332440_;
this.updateColor();
}
public void setPotionDurationScale(float p_396093_) {
this.potionDurationScale = p_396093_;
}
private void updateColor() {
ParticleOptions particleoptions = this.entityData.get(DATA_PARTICLE);
if (particleoptions instanceof ColorParticleOption colorparticleoption) {
int i = this.potionContents.equals(PotionContents.EMPTY) ? 0 : this.potionContents.getColor();
this.entityData.set(DATA_PARTICLE, ColorParticleOption.create(colorparticleoption.getType(), ARGB.opaque(i)));
}
}
public void addEffect(MobEffectInstance p_19717_) {
this.setPotionContents(this.potionContents.withEffectAdded(p_19717_));
}
public ParticleOptions getParticle() {
return this.getEntityData().get(DATA_PARTICLE);
}
public void setParticle(ParticleOptions p_19725_) {
this.getEntityData().set(DATA_PARTICLE, p_19725_);
}
protected void setWaiting(boolean p_19731_) {
this.getEntityData().set(DATA_WAITING, p_19731_);
}
public boolean isWaiting() {
return this.getEntityData().get(DATA_WAITING);
}
public int getDuration() {
return this.duration;
}
public void setDuration(int p_19735_) {
this.duration = p_19735_;
}
@Override
public void tick() {
super.tick();
if (this.level() instanceof ServerLevel serverlevel) {
this.serverTick(serverlevel);
} else {
this.clientTick();
}
}
private void clientTick() {
boolean flag = this.isWaiting();
float f = this.getRadius();
if (!flag || !this.random.nextBoolean()) {
ParticleOptions particleoptions = this.getParticle();
int i;
float f1;
if (flag) {
i = 2;
f1 = 0.2F;
} else {
i = Mth.ceil((float) Math.PI * f * f);
f1 = f;
}
for (int j = 0; j < i; j++) {
float f2 = this.random.nextFloat() * (float) (Math.PI * 2);
float f3 = Mth.sqrt(this.random.nextFloat()) * f1;
double d0 = this.getX() + Mth.cos(f2) * f3;
double d1 = this.getY();
double d2 = this.getZ() + Mth.sin(f2) * f3;
if (particleoptions.getType() == ParticleTypes.ENTITY_EFFECT) {
if (flag && this.random.nextBoolean()) {
this.level().addAlwaysVisibleParticle(ColorParticleOption.create(ParticleTypes.ENTITY_EFFECT, -1), d0, d1, d2, 0.0, 0.0, 0.0);
} else {
this.level().addAlwaysVisibleParticle(particleoptions, d0, d1, d2, 0.0, 0.0, 0.0);
}
} else if (flag) {
this.level().addAlwaysVisibleParticle(particleoptions, d0, d1, d2, 0.0, 0.0, 0.0);
} else {
this.level()
.addAlwaysVisibleParticle(particleoptions, d0, d1, d2, (0.5 - this.random.nextDouble()) * 0.15, 0.01F, (0.5 - this.random.nextDouble()) * 0.15);
}
}
}
}
private void serverTick(ServerLevel p_361199_) {
if (this.duration != -1 && this.tickCount >= this.waitTime + this.duration) {
this.discard();
} else {
boolean flag = this.isWaiting();
boolean flag1 = this.tickCount < this.waitTime;
if (flag != flag1) {
this.setWaiting(flag1);
}
if (!flag1) {
float f = this.getRadius();
if (this.radiusPerTick != 0.0F) {
f += this.radiusPerTick;
if (f < 0.5F) {
this.discard();
return;
}
this.setRadius(f);
}
if (this.tickCount % 5 == 0) {
this.victims.entrySet().removeIf(p_287380_ -> this.tickCount >= p_287380_.getValue());
if (!this.potionContents.hasEffects()) {
this.victims.clear();
} else {
List<MobEffectInstance> list = new ArrayList<>();
this.potionContents.forEachEffect(list::add, this.potionDurationScale);
List<LivingEntity> list1 = this.level().getEntitiesOfClass(LivingEntity.class, this.getBoundingBox());
if (!list1.isEmpty()) {
for (LivingEntity livingentity : list1) {
if (!this.victims.containsKey(livingentity) && livingentity.isAffectedByPotions() && !list.stream().noneMatch(livingentity::canBeAffected)) {
double d0 = livingentity.getX() - this.getX();
double d1 = livingentity.getZ() - this.getZ();
double d2 = d0 * d0 + d1 * d1;
if (d2 <= f * f) {
this.victims.put(livingentity, this.tickCount + this.reapplicationDelay);
for (MobEffectInstance mobeffectinstance : list) {
if (mobeffectinstance.getEffect().value().isInstantenous()) {
mobeffectinstance.getEffect()
.value()
.applyInstantenousEffect(p_361199_, this, this.getOwner(), livingentity, mobeffectinstance.getAmplifier(), 0.5);
} else {
livingentity.addEffect(new MobEffectInstance(mobeffectinstance), this);
}
}
if (this.radiusOnUse != 0.0F) {
f += this.radiusOnUse;
if (f < 0.5F) {
this.discard();
return;
}
this.setRadius(f);
}
if (this.durationOnUse != 0 && this.duration != -1) {
this.duration = this.duration + this.durationOnUse;
if (this.duration <= 0) {
this.discard();
return;
}
}
}
}
}
}
}
}
}
}
}
public float getRadiusOnUse() {
return this.radiusOnUse;
}
public void setRadiusOnUse(float p_19733_) {
this.radiusOnUse = p_19733_;
}
public float getRadiusPerTick() {
return this.radiusPerTick;
}
public void setRadiusPerTick(float p_19739_) {
this.radiusPerTick = p_19739_;
}
public int getDurationOnUse() {
return this.durationOnUse;
}
public void setDurationOnUse(int p_146786_) {
this.durationOnUse = p_146786_;
}
public int getWaitTime() {
return this.waitTime;
}
public void setWaitTime(int p_19741_) {
this.waitTime = p_19741_;
}
public void setOwner(@Nullable LivingEntity p_19719_) {
this.owner = p_19719_;
this.ownerUUID = p_19719_ == null ? null : p_19719_.getUUID();
}
@Nullable
public LivingEntity getOwner() {
if (this.owner != null && !this.owner.isRemoved()) {
return this.owner;
} else {
if (this.ownerUUID != null && this.level() instanceof ServerLevel serverlevel) {
this.owner = serverlevel.getEntity(this.ownerUUID) instanceof LivingEntity livingentity ? livingentity : null;
}
return this.owner;
}
}
@Override
protected void readAdditionalSaveData(CompoundTag p_19727_) {
this.tickCount = p_19727_.getIntOr("Age", 0);
this.duration = p_19727_.getIntOr("Duration", -1);
this.waitTime = p_19727_.getIntOr("WaitTime", 20);
this.reapplicationDelay = p_19727_.getIntOr("ReapplicationDelay", 20);
this.durationOnUse = p_19727_.getIntOr("DurationOnUse", 0);
this.radiusOnUse = p_19727_.getFloatOr("RadiusOnUse", 0.0F);
this.radiusPerTick = p_19727_.getFloatOr("RadiusPerTick", 0.0F);
this.setRadius(p_19727_.getFloatOr("Radius", 3.0F));
this.ownerUUID = p_19727_.read("Owner", UUIDUtil.CODEC).orElse(null);
RegistryOps<Tag> registryops = this.registryAccess().createSerializationContext(NbtOps.INSTANCE);
this.setParticle(p_19727_.read("Particle", ParticleTypes.CODEC, registryops).orElse(DEFAULT_PARTICLE));
this.setPotionContents(p_19727_.read("potion_contents", PotionContents.CODEC, registryops).orElse(PotionContents.EMPTY));
this.potionDurationScale = p_19727_.getFloatOr("potion_duration_scale", 1.0F);
}
@Override
protected void addAdditionalSaveData(CompoundTag p_19737_) {
p_19737_.putInt("Age", this.tickCount);
p_19737_.putInt("Duration", this.duration);
p_19737_.putInt("WaitTime", this.waitTime);
p_19737_.putInt("ReapplicationDelay", this.reapplicationDelay);
p_19737_.putInt("DurationOnUse", this.durationOnUse);
p_19737_.putFloat("RadiusOnUse", this.radiusOnUse);
p_19737_.putFloat("RadiusPerTick", this.radiusPerTick);
p_19737_.putFloat("Radius", this.getRadius());
RegistryOps<Tag> registryops = this.registryAccess().createSerializationContext(NbtOps.INSTANCE);
p_19737_.store("Particle", ParticleTypes.CODEC, registryops, this.getParticle());
p_19737_.storeNullable("Owner", UUIDUtil.CODEC, this.ownerUUID);
if (!this.potionContents.equals(PotionContents.EMPTY)) {
p_19737_.store("potion_contents", PotionContents.CODEC, registryops, this.potionContents);
}
if (this.potionDurationScale != 1.0F) {
p_19737_.putFloat("potion_duration_scale", this.potionDurationScale);
}
}
@Override
public void onSyncedDataUpdated(EntityDataAccessor<?> p_19729_) {
if (DATA_RADIUS.equals(p_19729_)) {
this.refreshDimensions();
}
super.onSyncedDataUpdated(p_19729_);
}
@Override
public PushReaction getPistonPushReaction() {
return PushReaction.IGNORE;
}
@Override
public EntityDimensions getDimensions(Pose p_19721_) {
return EntityDimensions.scalable(this.getRadius() * 2.0F, 0.5F);
}
@Override
public final boolean hurtServer(ServerLevel p_360854_, DamageSource p_364045_, float p_363449_) {
return false;
}
@Nullable
@Override
public <T> T get(DataComponentType<? extends T> p_394344_) {
if (p_394344_ == DataComponents.POTION_CONTENTS) {
return castComponentValue((DataComponentType<T>)p_394344_, this.potionContents);
} else {
return p_394344_ == DataComponents.POTION_DURATION_SCALE ? castComponentValue((DataComponentType<T>)p_394344_, this.potionDurationScale) : super.get(p_394344_);
}
}
@Override
protected void applyImplicitComponents(DataComponentGetter p_396444_) {
this.applyImplicitComponentIfPresent(p_396444_, DataComponents.POTION_CONTENTS);
this.applyImplicitComponentIfPresent(p_396444_, DataComponents.POTION_DURATION_SCALE);
super.applyImplicitComponents(p_396444_);
}
@Override
protected <T> boolean applyImplicitComponent(DataComponentType<T> p_392513_, T p_395434_) {
if (p_392513_ == DataComponents.POTION_CONTENTS) {
this.setPotionContents(castComponentValue(DataComponents.POTION_CONTENTS, p_395434_));
return true;
} else if (p_392513_ == DataComponents.POTION_DURATION_SCALE) {
this.setPotionDurationScale(castComponentValue(DataComponents.POTION_DURATION_SCALE, p_395434_));
return true;
} else {
return super.applyImplicitComponent(p_392513_, p_395434_);
}
}
}