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

package hdvtdev.blockAndSeek.roulette;

import com.lewdev.probabilitylib.ProbabilityCollection;
import hdvtdev.blockAndSeek.BlockAndSeekMap;
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<BlockAndSeekMap.Block> blocks) {
        for (BlockAndSeekMap.Block block : blocks) {
            probabilityCollection.add(block.block(), block.chance());
        }
    }

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


}