destroying buildings no longer crashes the game

2009-12-19 23:48:31 +13:00
parent 8c2194c903
commit 301873f88e
1 changed files with 3 additions and 2 deletions
--- a/OpenRa.Game/UnitInfluenceMap.cs
+++ b/OpenRa.Game/UnitInfluenceMap.cs
@@ -57,8 +57,9 @@ namespace OpenRa.Game

 		public void Remove( Actor self, IOccupySpace unit )
 		{
-			foreach (var c in unit.OccupiedCells())
-				influence[c.X, c.Y] = null;
+			if (unit != null)
+				foreach (var c in unit.OccupiedCells())
+					influence[c.X, c.Y] = null;
 		}

 		public void Update(Actor self, IOccupySpace unit)