summaryrefslogtreecommitdiff
path: root/dice-lang/src/bjc/dicelang/dice/CompoundingDie.java
blob: 023282efbd558178bf0f6fd3a39eccecceae3e48 (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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
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 {
	/* The source die to compound. */
	private final Die source;

	/* The predicate that marks when to compound. */
	private final Predicate<Long>   compoundOn;
	/* The string version of the predicate, if one exists. */
	private final 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(final Die src, final 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(final Die src, final Predicate<Long> compound, final String patt) {
		source = src;

		compoundOn = compound;
		compoundPattern = patt;
	}

	@Override
	public boolean canOptimize() {
		if (source.canOptimize()) {
			/* We can only be optimized for a result of zero. */
			return source.optimize() == 0;
		}

		return false;
	}

	@Override
	public long optimize() {
		/* If we can be optimized, its to zero. */
		return 0;
	}

	@Override
	public long roll() {
		/* The current result. */
		long res = source.roll();
		/* The last result. */
		long oldRes = res;

		while (compoundOn.test(oldRes)) {
			/* Compound while the result should be compounded. */
			oldRes = source.rollSingle();

			/* Accumulate. */
			res += oldRes;
		}

		return res;
	}

	@Override
	public long rollSingle() {
		/* Just compound on an initial single role. */
		long res = source.rollSingle();
		/* The last result. */
		long oldRes = res;

		while (compoundOn.test(oldRes)) {
			/* Compound while the result should be compounded. */
			oldRes = source.rollSingle();

			/* Accumulate. */
			res += oldRes;
		}

		return res;
	}

	@Override
	public String toString() {
		String sourceString = source.toString();

		/* Can't print a parseable version. */
		if (compoundPattern == null) {
			return String.format("%s!!<complex-pattern>", sourceString);
		}

		return String.format("%s!!%s", sourceString, compoundPattern);
	}
}