blockandseek/src/main/java/hdvtdev/blockandseek/roulette/RouletteGenerator.java

package hdvtdev.blockandseek.roulette;


import com.lewdev.probabilitylib.ProbabilityCollection;

import hdvtdev.blockandseek.objects.PropBlock;

import org.bukkit.inventory.ItemStack;

import java.util.ArrayList;
import java.util.List;

public class RouletteGenerator {

    private final ProbabilityCollection<ItemStack> probabilityCollection = new ProbabilityCollection<>();

    public RouletteGenerator(List<PropBlock> blocks) {
        for (PropBlock block : blocks) {
            probabilityCollection.add(block.getBlock(), block.getRarity().getChance());
        }
    }

    public List<ItemStack> getRandomRow(int count) {
        List<ItemStack> items = new ArrayList<>(count);
        for (; count > 0; --count) items.add(probabilityCollection.get());
        return items;
    }

}