53 lines
1.5 KiB
Java
53 lines
1.5 KiB
Java
package net.minecraft.util.valueproviders;
|
|
|
|
import com.mojang.serialization.MapCodec;
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
|
import net.minecraft.util.RandomSource;
|
|
import net.minecraft.util.random.Weighted;
|
|
import net.minecraft.util.random.WeightedList;
|
|
|
|
public class WeightedListInt extends IntProvider {
|
|
public static final MapCodec<WeightedListInt> CODEC = RecordCodecBuilder.mapCodec(
|
|
instance -> instance.group(WeightedList.nonEmptyCodec(IntProvider.CODEC).fieldOf("distribution").forGetter(weightedListInt -> weightedListInt.distribution))
|
|
.apply(instance, WeightedListInt::new)
|
|
);
|
|
private final WeightedList<IntProvider> distribution;
|
|
private final int minValue;
|
|
private final int maxValue;
|
|
|
|
public WeightedListInt(WeightedList<IntProvider> distribution) {
|
|
this.distribution = distribution;
|
|
int i = Integer.MAX_VALUE;
|
|
int j = Integer.MIN_VALUE;
|
|
|
|
for (Weighted<IntProvider> weighted : distribution.unwrap()) {
|
|
int k = weighted.value().getMinValue();
|
|
int l = weighted.value().getMaxValue();
|
|
i = Math.min(i, k);
|
|
j = Math.max(j, l);
|
|
}
|
|
|
|
this.minValue = i;
|
|
this.maxValue = j;
|
|
}
|
|
|
|
@Override
|
|
public int sample(RandomSource random) {
|
|
return this.distribution.getRandomOrThrow(random).sample(random);
|
|
}
|
|
|
|
@Override
|
|
public int getMinValue() {
|
|
return this.minValue;
|
|
}
|
|
|
|
@Override
|
|
public int getMaxValue() {
|
|
return this.maxValue;
|
|
}
|
|
|
|
@Override
|
|
public IntProviderType<?> getType() {
|
|
return IntProviderType.WEIGHTED_LIST;
|
|
}
|
|
}
|