add ShipPool.setParent()

diff --git a/src/edu/berkeley/fleet/loops/ShipPool.java b/src/edu/berkeley/fleet/loops/ShipPool.java
index f120139..1d35e1d 100644 (file)
--- a/src/edu/berkeley/fleet/loops/ShipPool.java
+++ b/src/edu/berkeley/fleet/loops/ShipPool.java
@@ -63,4 +63,17 @@ public class ShipPool implements Iterable<Ship> {
         for (Ship ship : toRelease)
             releaseShip(ship);
     }
+
+    public void setParent(ShipPool parent) {
+        if (this.parent!=null) {
+            for (Ship ship : allocatedShips)
+                this.parent.releaseShip(ship);
+            this.parent = null;
+        }
+        if (parent != null) {
+            for (Ship ship : allocatedShips)
+                parent.allocateShip(ship);
+        }
+        this.parent = parent;
+    }
 }