From b5c2fb1ed923d43412694729b4445a66fa9f47fc Mon Sep 17 00:00:00 2001 From: Ben Culkin Date: Sat, 21 Nov 2020 18:10:30 -0500 Subject: Restructure new die implementation --- .../dicelang/neodice/diepool/KeepFirstDiePool.java | 58 ++++++++++++++++++++++ 1 file changed, 58 insertions(+) create mode 100644 dice/src/main/java/bjc/dicelang/neodice/diepool/KeepFirstDiePool.java (limited to 'dice/src/main/java/bjc/dicelang/neodice/diepool/KeepFirstDiePool.java') diff --git a/dice/src/main/java/bjc/dicelang/neodice/diepool/KeepFirstDiePool.java b/dice/src/main/java/bjc/dicelang/neodice/diepool/KeepFirstDiePool.java new file mode 100644 index 0000000..d31f104 --- /dev/null +++ b/dice/src/main/java/bjc/dicelang/neodice/diepool/KeepFirstDiePool.java @@ -0,0 +1,58 @@ +package bjc.dicelang.neodice.diepool; + +import java.util.*; + +import bjc.dicelang.neodice.*; + +public class KeepFirstDiePool implements DiePool { + private final int number; + private final DiePool pool; + + public KeepFirstDiePool(DiePool pool, int number) { + this.pool = pool; + this.number = number; + } + + @Override + public int[] roll(Random rng) { + int[] rolls = pool.roll(rng); + + if (rolls.length >= number) { + return rolls; + } else { + int[] newRolls = new int[number]; + + for (int index = 0; index < number; index++) { + newRolls[index] = rolls[index]; + } + + return newRolls; + } + } + + @Override + public Die[] contained() { + return pool.contained(); + } + + @Override + public String toString() { + return String.format("%skF%d", pool, number); + } + + @Override + public int hashCode() { + return Objects.hash(number, pool); + } + + @Override + public boolean equals(Object obj) { + if (this == obj) return true; + if (obj == null) return false; + if (getClass() != obj.getClass()) return false; + + KeepFirstDiePool other = (KeepFirstDiePool) obj; + + return number == other.number && Objects.equals(pool, other.pool); + } +} \ No newline at end of file -- cgit v1.2.3