From 775394cf6b4a3c6a7e7345affeb45bb34f6e4517 Mon Sep 17 00:00:00 2001
From: "Benjamin J. Culkin" <bjculkin@mix.wvu.edu>
Date: Sat, 2 Jun 2018 17:32:24 -0300
Subject: Finish die rewrite

This adds all of the remaining missing die modifiers, as well as a few
base die types that were missing.
---
 .../main/java/bjc/dicelang/dicev2/CountDieMod.java | 52 ++++++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100644 dice/src/main/java/bjc/dicelang/dicev2/CountDieMod.java

(limited to 'dice/src/main/java/bjc/dicelang/dicev2/CountDieMod.java')

diff --git a/dice/src/main/java/bjc/dicelang/dicev2/CountDieMod.java b/dice/src/main/java/bjc/dicelang/dicev2/CountDieMod.java
new file mode 100644
index 0000000..0b5ec5b
--- /dev/null
+++ b/dice/src/main/java/bjc/dicelang/dicev2/CountDieMod.java
@@ -0,0 +1,52 @@
+package bjc.dicelang.dicev2;
+
+import java.util.function.LongPredicate;
+
+public class CountDieMod extends Die {
+	public final Die[] dice;
+
+	public final LongPredicate success;
+
+	public LongPredicate failure;
+
+	public CountDieMod(LongPredicate success, Die... dice) {
+		this(success, null, dice);
+	}
+
+	public CountDieMod(LongPredicate success, LongPredicate failure, Die... dice) {
+		super();
+
+		this.success = success;
+		this.failure = failure;
+
+		this.dice    = dice;
+	}
+
+	public long[] roll() {
+		return new long[] { rollSingle() };
+	}
+
+	public long rollSingle() {
+		long count = 0;
+
+		for(Die die : dice) {
+			for(long val : die.roll()) {
+				if(success.test(val)) count += 1;
+
+				if(failure != null && failure.test(val)) count -= 1;
+			}
+		}
+
+		return count;
+	}
+
+	/* :UnoptimizableDice */
+
+	public boolean canOptimize() {
+		return false;
+	}
+
+	public long optimize() {
+		throw new UnsupportedOperationException("Counted dice can't be optimized");
+	}
+}
-- 
cgit v1.2.3