X-Git-Url: https://troll.desast.re/troll.git/blobdiff_plain/2a9c53c2715aae687335b7c842e4142973f28ad7..b41b9823fbe2eed146db478fd5a1353bb558215c:/src/main/java/com/codingame/game/Model.java

diff --git a/src/main/java/com/codingame/game/Model.java b/src/main/java/com/codingame/game/Model.java
index e8cdc57..405db66 100644
--- a/src/main/java/com/codingame/game/Model.java
+++ b/src/main/java/com/codingame/game/Model.java
@@ -25,16 +25,37 @@ class Model {
 
         private int multiplier;
         public int getMultiplier() { return multiplier; }
-        public void setMultiplier(int m){ multiplier = m; }
+        public void setMultiplier(int m) { multiplier = m; }
+
+        class FailedToThrowStonesAndShouldHave extends Exception {}
+        class ThrewMoreStonesThanHad extends Exception {}
 
         private int stones;
         public int getStones() { return stones; }
-        public void consumeStones(int n) throws InvalidAction {
+        public void consumeStones(int n)
+            throws ThrewMoreStonesThanHad,
+                   FailedToThrowStonesAndShouldHave
+        {
             if (n > stones) {
-                throw new InvalidAction("attempted to throw more stones than they had.");
+                throw new ThrewMoreStonesThanHad();
+            }
+            if (n == 0 && stones > 0) {
+                throw new FailedToThrowStonesAndShouldHave();
             }
             setStones(stones - n);
         }
+        public int consumeMaxStones() {
+            int r = stones;
+            stones = 0;
+            return r;
+        }
+        public int consumeMinStones() {
+            if (stones < 1) {
+                throw new Error("Internal error: tried to consume min stones on an empty heap.");
+            }
+            stones--;
+            return 1;
+        }
         public void setStones(int n) {
             stones = n;
         }