31 lines
921 B
Java
31 lines
921 B
Java
package net.minecraft.world.item.enchantment.effects;
|
|
|
|
import com.mojang.serialization.MapCodec;
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
|
import net.minecraft.util.RandomSource;
|
|
import net.minecraft.world.item.enchantment.LevelBasedValue;
|
|
|
|
public record RemoveBinomial(LevelBasedValue chance) implements EnchantmentValueEffect {
|
|
public static final MapCodec<RemoveBinomial> CODEC = RecordCodecBuilder.mapCodec(
|
|
instance -> instance.group(LevelBasedValue.CODEC.fieldOf("chance").forGetter(RemoveBinomial::chance)).apply(instance, RemoveBinomial::new)
|
|
);
|
|
|
|
@Override
|
|
public float process(int enchantmentLevel, RandomSource random, float value) {
|
|
float f = this.chance.calculate(enchantmentLevel);
|
|
int i = 0;
|
|
|
|
for (int j = 0; j < value; j++) {
|
|
if (random.nextFloat() < f) {
|
|
i++;
|
|
}
|
|
}
|
|
|
|
return value - i;
|
|
}
|
|
|
|
@Override
|
|
public MapCodec<RemoveBinomial> codec() {
|
|
return CODEC;
|
|
}
|
|
}
|