001/*
002 * PlotSquared, a land and world management plugin for Minecraft.
003 * Copyright (C) IntellectualSites <https://intellectualsites.com>
004 * Copyright (C) IntellectualSites team and contributors
005 *
006 * This program is free software: you can redistribute it and/or modify
007 * it under the terms of the GNU General Public License as published by
008 * the Free Software Foundation, either version 3 of the License, or
009 * (at your option) any later version.
010 *
011 * This program is distributed in the hope that it will be useful,
012 * but WITHOUT ANY WARRANTY; without even the implied warranty of
013 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
014 * GNU General Public License for more details.
015 *
016 * You should have received a copy of the GNU General Public License
017 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
018 */
019package com.plotsquared.core.collection;
020
021import java.util.Map;
022import java.util.Random;
023
024import static com.google.common.base.Preconditions.checkNotNull;
025
026public abstract class RandomCollection<T> {
027
028    protected Random random;
029
030    public RandomCollection(Map<T, Double> weights, Random random) {
031        this.random = random;
032    }
033
034    public static <T> RandomCollection<T> of(Map<T, Double> weights, Random random) {
035        try {
036            return new FlatRandomCollection<>(weights, random);
037        } catch (IllegalArgumentException ignore) {
038            return new SimpleRandomCollection<>(weights, random);
039        }
040    }
041
042    public Random getRandom() {
043        return random;
044    }
045
046    public void setRandom(Random random) {
047        checkNotNull(random);
048        this.random = random;
049    }
050
051    public abstract T next();
052
053}