From: Adam Megacz <adam@megacz.com>
Date: Mon, 27 Jul 2009 03:23:13 +0000 (-0700)
Subject: add ShipPool.setParent()
X-Git-Url: http://git.megacz.com/?p=fleet.git;a=commitdiff_plain;h=7cb7ef29daa0d2a0e530ee452700911e13b9cb9f

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
--- 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;
+    }
 }