de.janno.evaluator.dice.RandomElementsBuilder 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;


import com.google.common.collect.ImmutableList;
import lombok.EqualsAndHashCode;
import lombok.Getter;
import lombok.NonNull;

import java.util.*;

/**
 * Combines random elements of multiple sources. Random elements with the will overwrite elements with the diceId that where added bevor
 */
@EqualsAndHashCode
@Getter
public class RandomElementsBuilder {
    private final List randomElements = new ArrayList<>();
    private final RollContext rollContext;

    private RandomElementsBuilder(@NonNull RollContext rollContext) {
        this.rollContext = rollContext;
    }

    public static ImmutableList fromRolls(@NonNull Collection rolls, @NonNull RollContext rollContext) {
        return ofRolls(rolls, rollContext).build();
    }

    public static RandomElementsBuilder empty(@NonNull RollContext rollContext) {
        return new RandomElementsBuilder(rollContext);
    }

    public static RandomElementsBuilder ofRoll(@NonNull Roll roll, @NonNull RollContext rollContext) {
        return ofRolls(List.of(roll), rollContext);
    }

    public static RandomElementsBuilder ofRolls(@NonNull Collection rolls, @NonNull RollContext rollContext) {
        RandomElementsBuilder builder = new RandomElementsBuilder(rollContext);
        rolls.forEach(r -> builder.addRandomElements(r.getRandomElementsInRoll().stream().toList()));
        return builder;
    }

    public RandomElementsBuilder addRandomElements(@NonNull Collection randomElements) {
        this.randomElements.addAll(randomElements);
        return this;
    }

    public RandomElementsBuilder addRoll(@NonNull Roll roll) {
        this.randomElements.addAll(roll.getRandomElementsInRoll().stream().toList());
        return this;
    }

    public RandomElementsBuilder addWithColor(@NonNull Roll roll, @NonNull String color) {
        this.randomElements.addAll(roll.getRandomElementsInRoll().stream()
                .map(r -> r.copyWithTagAndColor(color))
                .toList());
        return this;
    }

    public ImmutableList build() {

        //This ensures that the last color application will overwrite the randomElements color
        Map uniqueMap = new HashMap<>(randomElements.size());
        randomElements.forEach(r -> uniqueMap.put(r.getDieId(), r));
        ImmutableList result = uniqueMap.values().stream()
                .sorted(Comparator.comparing(RandomElement::getDieId))
                .collect(ImmutableList.toImmutableList());
        rollContext.addRandomElements(result);

        return result;
    }


}