decouple that a bit more.

2009-12-22 20:03:53 +13:00
parent b9d231d500
commit d695344a05
3 changed files with 9 additions and 8 deletions
--- a/OpenRa.Game/Traits/Production.cs
+++ b/OpenRa.Game/Traits/Production.cs
@@ -46,8 +46,8 @@ namespace OpenRa.Game.Traits

 			Game.world.Add( newUnit );

-			if( self.traits.Contains<RenderWarFactory>() )
-				self.traits.Get<RenderWarFactory>().EjectUnit();
+			foreach (var t in self.traits.WithInterface<INotifyProduction>())
+				t.UnitProduced(self, newUnit);

 			return true;
 		}