404 lines
13 KiB
Java
404 lines
13 KiB
Java
package net.minecraft.world.entity;
|
|
|
|
import com.google.common.collect.Maps;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
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.network.syncher.EntityDataAccessor;
|
|
import net.minecraft.network.syncher.EntityDataSerializers;
|
|
import net.minecraft.network.syncher.SynchedEntityData;
|
|
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;
|
|
import net.minecraft.world.level.storage.ValueInput;
|
|
import net.minecraft.world.level.storage.ValueOutput;
|
|
import org.jetbrains.annotations.Nullable;
|
|
|
|
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);
|
|
@Nullable
|
|
private ParticleOptions customParticle;
|
|
private PotionContents potionContents = PotionContents.EMPTY;
|
|
private float potionDurationScale = 1.0F;
|
|
private final Map<Entity, Integer> victims = Maps.<Entity, Integer>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 EntityReference<LivingEntity> owner;
|
|
|
|
public AreaEffectCloud(EntityType<? extends AreaEffectCloud> entityType, Level level) {
|
|
super(entityType, level);
|
|
this.noPhysics = true;
|
|
}
|
|
|
|
public AreaEffectCloud(Level level, double x, double y, double z) {
|
|
this(EntityType.AREA_EFFECT_CLOUD, level);
|
|
this.setPos(x, y, z);
|
|
}
|
|
|
|
@Override
|
|
protected void defineSynchedData(SynchedEntityData.Builder builder) {
|
|
builder.define(DATA_RADIUS, 3.0F);
|
|
builder.define(DATA_WAITING, false);
|
|
builder.define(DATA_PARTICLE, DEFAULT_PARTICLE);
|
|
}
|
|
|
|
public void setRadius(float radius) {
|
|
if (!this.level().isClientSide) {
|
|
this.getEntityData().set(DATA_RADIUS, Mth.clamp(radius, 0.0F, 32.0F));
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void refreshDimensions() {
|
|
double d = this.getX();
|
|
double e = this.getY();
|
|
double f = this.getZ();
|
|
super.refreshDimensions();
|
|
this.setPos(d, e, f);
|
|
}
|
|
|
|
public float getRadius() {
|
|
return this.getEntityData().get(DATA_RADIUS);
|
|
}
|
|
|
|
public void setPotionContents(PotionContents potionContents) {
|
|
this.potionContents = potionContents;
|
|
this.updateParticle();
|
|
}
|
|
|
|
public void setCustomParticle(@Nullable ParticleOptions customParticle) {
|
|
this.customParticle = customParticle;
|
|
this.updateParticle();
|
|
}
|
|
|
|
public void setPotionDurationScale(float potionDurationScale) {
|
|
this.potionDurationScale = potionDurationScale;
|
|
}
|
|
|
|
private void updateParticle() {
|
|
if (this.customParticle != null) {
|
|
this.entityData.set(DATA_PARTICLE, this.customParticle);
|
|
} else {
|
|
int i = ARGB.opaque(this.potionContents.getColor());
|
|
this.entityData.set(DATA_PARTICLE, ColorParticleOption.create(DEFAULT_PARTICLE.getType(), i));
|
|
}
|
|
}
|
|
|
|
public void addEffect(MobEffectInstance effectInstance) {
|
|
this.setPotionContents(this.potionContents.withEffectAdded(effectInstance));
|
|
}
|
|
|
|
public ParticleOptions getParticle() {
|
|
return this.getEntityData().get(DATA_PARTICLE);
|
|
}
|
|
|
|
/**
|
|
* Sets if the cloud is waiting. While waiting, the radius is ignored and the cloud shows fewer particles in its area.
|
|
*/
|
|
protected void setWaiting(boolean waiting) {
|
|
this.getEntityData().set(DATA_WAITING, waiting);
|
|
}
|
|
|
|
/**
|
|
* Returns {@code true} if the cloud is waiting. While waiting, the radius is ignored and the cloud shows fewer particles in its area.
|
|
*/
|
|
public boolean isWaiting() {
|
|
return this.getEntityData().get(DATA_WAITING);
|
|
}
|
|
|
|
public int getDuration() {
|
|
return this.duration;
|
|
}
|
|
|
|
public void setDuration(int duration) {
|
|
this.duration = duration;
|
|
}
|
|
|
|
@Override
|
|
public void tick() {
|
|
super.tick();
|
|
if (this.level() instanceof ServerLevel serverLevel) {
|
|
this.serverTick(serverLevel);
|
|
} else {
|
|
this.clientTick();
|
|
}
|
|
}
|
|
|
|
private void clientTick() {
|
|
boolean bl = this.isWaiting();
|
|
float f = this.getRadius();
|
|
if (!bl || !this.random.nextBoolean()) {
|
|
ParticleOptions particleOptions = this.getParticle();
|
|
int i;
|
|
float g;
|
|
if (bl) {
|
|
i = 2;
|
|
g = 0.2F;
|
|
} else {
|
|
i = Mth.ceil((float) Math.PI * f * f);
|
|
g = f;
|
|
}
|
|
|
|
for (int j = 0; j < i; j++) {
|
|
float h = this.random.nextFloat() * (float) (Math.PI * 2);
|
|
float k = Mth.sqrt(this.random.nextFloat()) * g;
|
|
double d = this.getX() + Mth.cos(h) * k;
|
|
double e = this.getY();
|
|
double l = this.getZ() + Mth.sin(h) * k;
|
|
if (particleOptions.getType() == ParticleTypes.ENTITY_EFFECT) {
|
|
if (bl && this.random.nextBoolean()) {
|
|
this.level().addAlwaysVisibleParticle(DEFAULT_PARTICLE, d, e, l, 0.0, 0.0, 0.0);
|
|
} else {
|
|
this.level().addAlwaysVisibleParticle(particleOptions, d, e, l, 0.0, 0.0, 0.0);
|
|
}
|
|
} else if (bl) {
|
|
this.level().addAlwaysVisibleParticle(particleOptions, d, e, l, 0.0, 0.0, 0.0);
|
|
} else {
|
|
this.level().addAlwaysVisibleParticle(particleOptions, d, e, l, (0.5 - this.random.nextDouble()) * 0.15, 0.01F, (0.5 - this.random.nextDouble()) * 0.15);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private void serverTick(ServerLevel level) {
|
|
if (this.duration != -1 && this.tickCount - this.waitTime >= this.duration) {
|
|
this.discard();
|
|
} else {
|
|
boolean bl = this.isWaiting();
|
|
boolean bl2 = this.tickCount < this.waitTime;
|
|
if (bl != bl2) {
|
|
this.setWaiting(bl2);
|
|
}
|
|
|
|
if (!bl2) {
|
|
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(entry -> this.tickCount >= (Integer)entry.getValue());
|
|
if (!this.potionContents.hasEffects()) {
|
|
this.victims.clear();
|
|
} else {
|
|
List<MobEffectInstance> list = new ArrayList();
|
|
this.potionContents.forEachEffect(list::add, this.potionDurationScale);
|
|
List<LivingEntity> list2 = this.level().getEntitiesOfClass(LivingEntity.class, this.getBoundingBox());
|
|
if (!list2.isEmpty()) {
|
|
for (LivingEntity livingEntity : list2) {
|
|
if (!this.victims.containsKey(livingEntity) && livingEntity.isAffectedByPotions() && !list.stream().noneMatch(livingEntity::canBeAffected)) {
|
|
double d = livingEntity.getX() - this.getX();
|
|
double e = livingEntity.getZ() - this.getZ();
|
|
double g = d * d + e * e;
|
|
if (g <= f * f) {
|
|
this.victims.put(livingEntity, this.tickCount + this.reapplicationDelay);
|
|
|
|
for (MobEffectInstance mobEffectInstance : list) {
|
|
if (mobEffectInstance.getEffect().value().isInstantenous()) {
|
|
mobEffectInstance.getEffect().value().applyInstantenousEffect(level, 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 radiusOnUse) {
|
|
this.radiusOnUse = radiusOnUse;
|
|
}
|
|
|
|
public float getRadiusPerTick() {
|
|
return this.radiusPerTick;
|
|
}
|
|
|
|
public void setRadiusPerTick(float radiusPerTick) {
|
|
this.radiusPerTick = radiusPerTick;
|
|
}
|
|
|
|
public int getDurationOnUse() {
|
|
return this.durationOnUse;
|
|
}
|
|
|
|
public void setDurationOnUse(int durationOnUse) {
|
|
this.durationOnUse = durationOnUse;
|
|
}
|
|
|
|
public int getWaitTime() {
|
|
return this.waitTime;
|
|
}
|
|
|
|
public void setWaitTime(int waitTime) {
|
|
this.waitTime = waitTime;
|
|
}
|
|
|
|
public void setOwner(@Nullable LivingEntity owner) {
|
|
this.owner = owner != null ? new EntityReference<>(owner) : null;
|
|
}
|
|
|
|
@Nullable
|
|
public LivingEntity getOwner() {
|
|
return (LivingEntity)EntityReference.get(this.owner, this.level(), LivingEntity.class);
|
|
}
|
|
|
|
@Override
|
|
protected void readAdditionalSaveData(ValueInput input) {
|
|
this.tickCount = input.getIntOr("Age", 0);
|
|
this.duration = input.getIntOr("Duration", -1);
|
|
this.waitTime = input.getIntOr("WaitTime", 20);
|
|
this.reapplicationDelay = input.getIntOr("ReapplicationDelay", 20);
|
|
this.durationOnUse = input.getIntOr("DurationOnUse", 0);
|
|
this.radiusOnUse = input.getFloatOr("RadiusOnUse", 0.0F);
|
|
this.radiusPerTick = input.getFloatOr("RadiusPerTick", 0.0F);
|
|
this.setRadius(input.getFloatOr("Radius", 3.0F));
|
|
this.owner = EntityReference.read(input, "Owner");
|
|
this.setCustomParticle((ParticleOptions)input.read("custom_particle", ParticleTypes.CODEC).orElse(null));
|
|
this.setPotionContents((PotionContents)input.read("potion_contents", PotionContents.CODEC).orElse(PotionContents.EMPTY));
|
|
this.potionDurationScale = input.getFloatOr("potion_duration_scale", 1.0F);
|
|
}
|
|
|
|
@Override
|
|
protected void addAdditionalSaveData(ValueOutput output) {
|
|
output.putInt("Age", this.tickCount);
|
|
output.putInt("Duration", this.duration);
|
|
output.putInt("WaitTime", this.waitTime);
|
|
output.putInt("ReapplicationDelay", this.reapplicationDelay);
|
|
output.putInt("DurationOnUse", this.durationOnUse);
|
|
output.putFloat("RadiusOnUse", this.radiusOnUse);
|
|
output.putFloat("RadiusPerTick", this.radiusPerTick);
|
|
output.putFloat("Radius", this.getRadius());
|
|
output.storeNullable("custom_particle", ParticleTypes.CODEC, this.customParticle);
|
|
EntityReference.store(this.owner, output, "Owner");
|
|
if (!this.potionContents.equals(PotionContents.EMPTY)) {
|
|
output.store("potion_contents", PotionContents.CODEC, this.potionContents);
|
|
}
|
|
|
|
if (this.potionDurationScale != 1.0F) {
|
|
output.putFloat("potion_duration_scale", this.potionDurationScale);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void onSyncedDataUpdated(EntityDataAccessor<?> dataAccessor) {
|
|
if (DATA_RADIUS.equals(dataAccessor)) {
|
|
this.refreshDimensions();
|
|
}
|
|
|
|
super.onSyncedDataUpdated(dataAccessor);
|
|
}
|
|
|
|
@Override
|
|
public PushReaction getPistonPushReaction() {
|
|
return PushReaction.IGNORE;
|
|
}
|
|
|
|
@Override
|
|
public EntityDimensions getDimensions(Pose pose) {
|
|
return EntityDimensions.scalable(this.getRadius() * 2.0F, 0.5F);
|
|
}
|
|
|
|
@Override
|
|
public final boolean hurtServer(ServerLevel level, DamageSource damageSource, float amount) {
|
|
return false;
|
|
}
|
|
|
|
@Nullable
|
|
@Override
|
|
public <T> T get(DataComponentType<? extends T> component) {
|
|
if (component == DataComponents.POTION_CONTENTS) {
|
|
return castComponentValue((DataComponentType<T>)component, this.potionContents);
|
|
} else {
|
|
return component == DataComponents.POTION_DURATION_SCALE
|
|
? castComponentValue((DataComponentType<T>)component, this.potionDurationScale)
|
|
: super.get(component);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
protected void applyImplicitComponents(DataComponentGetter componentGetter) {
|
|
this.applyImplicitComponentIfPresent(componentGetter, DataComponents.POTION_CONTENTS);
|
|
this.applyImplicitComponentIfPresent(componentGetter, DataComponents.POTION_DURATION_SCALE);
|
|
super.applyImplicitComponents(componentGetter);
|
|
}
|
|
|
|
@Override
|
|
protected <T> boolean applyImplicitComponent(DataComponentType<T> component, T value) {
|
|
if (component == DataComponents.POTION_CONTENTS) {
|
|
this.setPotionContents(castComponentValue(DataComponents.POTION_CONTENTS, value));
|
|
return true;
|
|
} else if (component == DataComponents.POTION_DURATION_SCALE) {
|
|
this.setPotionDurationScale(castComponentValue(DataComponents.POTION_DURATION_SCALE, value));
|
|
return true;
|
|
} else {
|
|
return super.applyImplicitComponent(component, value);
|
|
}
|
|
}
|
|
}
|