de.janno.evaluator.dice.random.GivenDiceNumberSupplier Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dice-evaluator Show documentation

Dice infix notation (aka calculator notation) expression parser and evaluator

The newest version!

package de.janno.evaluator.dice.random;

import de.janno.evaluator.dice.DieId;
import de.janno.evaluator.dice.DieIdAndValue;
import de.janno.evaluator.dice.ExpressionException;
import lombok.NonNull;

import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

public class GivenDiceNumberSupplier implements NumberSupplier {
    private final NumberSupplier numberSupplier;
    private final Map givenDiceNumbers;

    public GivenDiceNumberSupplier(@NonNull List givenDiceNumbers) {
        this(new RandomNumberSupplier(), givenDiceNumbers);
    }

    public GivenDiceNumberSupplier(@NonNull NumberSupplier numberSupplier, @NonNull List givenDiceNumbers) {
        this.numberSupplier = numberSupplier;
        Set allDiceIds = new HashSet<>();
        Set duplicatedDiceIds = new HashSet<>();
        for (DieIdAndValue dieIdAndValue : givenDiceNumbers) {
            if (allDiceIds.contains(dieIdAndValue.getDieId())) {
                duplicatedDiceIds.add(dieIdAndValue.getDieId());
            }
            allDiceIds.add(dieIdAndValue.getDieId());
        }
        if (!duplicatedDiceIds.isEmpty()) {
            throw new IllegalStateException("Duplicated dice ids: " + duplicatedDiceIds);
        }
        this.givenDiceNumbers = givenDiceNumbers.stream().collect(Collectors.toMap(DieIdAndValue::getDieId, DieIdAndValue::getNumberSupplierValue));
    }

    @Override
    public int get(int minExcl, int maxIncl, @NonNull DieId dieId) throws ExpressionException {
        if (givenDiceNumbers.containsKey(dieId)) {
            return givenDiceNumbers.get(dieId);
        }

        return numberSupplier.get(minExcl, maxIncl, dieId);
    }

}