X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=src%2Fedu%2Fberkeley%2Ffleet%2Floops%2FShipPool.java;h=e829c7cf5176c3ff8b518eacbc72230d80285b25;hb=cf8fdbe291499018cbb4edab0447172af20f752d;hp=673872a9180cf0a32422f34051ecf015adcd218c;hpb=c8c6fdab27c15623b579fd6d6a24d41e915e90d2;p=fleet.git

diff --git a/src/edu/berkeley/fleet/loops/ShipPool.java b/src/edu/berkeley/fleet/loops/ShipPool.java
index 673872a..e829c7c 100644
--- a/src/edu/berkeley/fleet/loops/ShipPool.java
+++ b/src/edu/berkeley/fleet/loops/ShipPool.java
@@ -23,6 +23,12 @@ public class ShipPool implements Iterable<Ship> {
 
     public Iterator<Ship> iterator() { return allocatedShips.iterator(); }
 
+    public void allocateShip(Ship ship) {
+        if (allocatedShips.contains(ship))
+            throw new RuntimeException("already allocated!");
+        allocatedShips.add(ship);
+    }
+
     /** allocate a ship */
     public Ship allocateShip(String type) {
         Ship ship = null;