summaryrefslogtreecommitdiff
path: root/dice-lang/src/bjc/dicelang/dice/CompoundingDie.java
blob: 28badf41f2ea54484e97f7c25692886ba0cbd5f5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
package bjc.dicelang.dice;

import java.util.function.Predicate;

/**
 * Implements a compounding die.
 *
 * This means that the source will be rolled, and then more single rolls will be
 * added while it meets a qualification.
 *
 * @author Ben Culkin
 */
public class CompoundingDie implements Die {
	private Die source;

	private Predicate<Long> compoundOn;
	private String compoundPattern;

	/**
	 * Create a new compounding die with no pattern.
	 *
	 * @param src
	 *                The die to compound from
	 * @param compound
	 *                The conditions to compound on
	 */
	public CompoundingDie(Die src, Predicate<Long> compound) {
		this(src, compound, null);
	}

	/**
	 * Create a new compounding die with a specified pattern.
	 *
	 * @param src
	 *                The die to compound from
	 * @param compound
	 *                The conditions to compound on
	 * @param patt
	 *                The string pattern the condition came from, for
	 *                printing
	 */
	public CompoundingDie(Die src, Predicate<Long> compound, String patt) {
		source = src;

		compoundOn = compound;
		compoundPattern = patt;
	}

	@Override
	public boolean canOptimize() {
		return source.canOptimize() && source.optimize() == 0;
	}

	@Override
	public long optimize() {
		return 0;
	}

	@Override
	public long roll() {
		long res = source.roll();
		long oldRes = res;

		while (compoundOn.test(oldRes)) {
			oldRes = source.rollSingle();

			res += oldRes;
		}

		return res;
	}

	@Override
	public long rollSingle() {
		/*
		 * Just compound on a single roll
		 */
		long res = source.rollSingle();
		long oldRes = res;

		while (compoundOn.test(oldRes)) {
			oldRes = source.rollSingle();

			res += oldRes;
		}

		return res;
	}

	@Override
	public String toString() {
		if (compoundPattern == null) {
			return source + "!!";
		} else {
			return source + "!!" + compoundPattern;
		}
	}
}