com.skelril.nitro.droptable.roller.SlipperySingleHitDiceRoller.java Source code

Introduction

Here is the source code for com.skelril.nitro.droptable.roller.SlipperySingleHitDiceRoller.java

Source

/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */

package com.skelril.nitro.droptable.roller;

import com.google.common.collect.ImmutableList;
import com.skelril.nitro.droptable.DropTableEntry;
import com.skelril.nitro.probability.Probability;

import java.util.Collection;
import java.util.Collections;
import java.util.ListIterator;
import java.util.function.BiFunction;

public class SlipperySingleHitDiceRoller implements DiceRoller {

    private final BiFunction<Integer, Double, Integer> modiFunc;

    public SlipperySingleHitDiceRoller() {
        this((a, b) -> a);
    }

    public SlipperySingleHitDiceRoller(BiFunction<Integer, Double, Integer> modiFunc) {
        this.modiFunc = modiFunc;
    }

    @Override
    public <T extends DropTableEntry> Collection<T> getHits(ImmutableList<T> input, double modifier) {
        ListIterator<T> it = input.listIterator(input.size());

        T cur = null;
        while (it.hasPrevious()) {
            cur = it.previous();

            // Slip through until something which is >= the chance is hit, unless a modifier is applied
            // this is equivalent to a 1 / chance probability
            if (cur.getChance() <= Probability.getRandom(modiFunc.apply(cur.getChance(), modifier))) {
                break;
            }

            cur = null;
        }

        return cur == null ? Collections.emptySet() : Collections.singleton(cur);
    }
}