64 lines
2.2 KiB
Java
64 lines
2.2 KiB
Java
package net.minecraft.world.item.consume_effects;
|
|
|
|
import com.mojang.serialization.Codec;
|
|
import com.mojang.serialization.MapCodec;
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
|
import java.util.List;
|
|
import net.minecraft.network.RegistryFriendlyByteBuf;
|
|
import net.minecraft.network.codec.ByteBufCodecs;
|
|
import net.minecraft.network.codec.StreamCodec;
|
|
import net.minecraft.world.effect.MobEffectInstance;
|
|
import net.minecraft.world.entity.LivingEntity;
|
|
import net.minecraft.world.item.ItemStack;
|
|
import net.minecraft.world.level.Level;
|
|
|
|
public record ApplyStatusEffectsConsumeEffect(List<MobEffectInstance> effects, float probability) implements ConsumeEffect {
|
|
public static final MapCodec<ApplyStatusEffectsConsumeEffect> CODEC = RecordCodecBuilder.mapCodec(
|
|
instance -> instance.group(
|
|
MobEffectInstance.CODEC.listOf().fieldOf("effects").forGetter(ApplyStatusEffectsConsumeEffect::effects),
|
|
Codec.floatRange(0.0F, 1.0F).optionalFieldOf("probability", 1.0F).forGetter(ApplyStatusEffectsConsumeEffect::probability)
|
|
)
|
|
.apply(instance, ApplyStatusEffectsConsumeEffect::new)
|
|
);
|
|
public static final StreamCodec<RegistryFriendlyByteBuf, ApplyStatusEffectsConsumeEffect> STREAM_CODEC = StreamCodec.composite(
|
|
MobEffectInstance.STREAM_CODEC.apply(ByteBufCodecs.list()),
|
|
ApplyStatusEffectsConsumeEffect::effects,
|
|
ByteBufCodecs.FLOAT,
|
|
ApplyStatusEffectsConsumeEffect::probability,
|
|
ApplyStatusEffectsConsumeEffect::new
|
|
);
|
|
|
|
public ApplyStatusEffectsConsumeEffect(MobEffectInstance effect, float probability) {
|
|
this(List.of(effect), probability);
|
|
}
|
|
|
|
public ApplyStatusEffectsConsumeEffect(List<MobEffectInstance> effects) {
|
|
this(effects, 1.0F);
|
|
}
|
|
|
|
public ApplyStatusEffectsConsumeEffect(MobEffectInstance effect) {
|
|
this(effect, 1.0F);
|
|
}
|
|
|
|
@Override
|
|
public ConsumeEffect.Type<ApplyStatusEffectsConsumeEffect> getType() {
|
|
return ConsumeEffect.Type.APPLY_EFFECTS;
|
|
}
|
|
|
|
@Override
|
|
public boolean apply(Level level, ItemStack stack, LivingEntity entity) {
|
|
if (entity.getRandom().nextFloat() >= this.probability) {
|
|
return false;
|
|
} else {
|
|
boolean bl = false;
|
|
|
|
for (MobEffectInstance mobEffectInstance : this.effects) {
|
|
if (entity.addEffect(new MobEffectInstance(mobEffectInstance))) {
|
|
bl = true;
|
|
}
|
|
}
|
|
|
|
return bl;
|
|
}
|
|
}
|
|
}
|