Fixed EgoComponents not being added to child GameObjects

andoowhy · Feb 3, 2017 · 9e78079 · 9e78079
1 parent 0dd59d2
commit 9e78079
Showing 1 changed file with 19 additions and 3 deletions.
diff --git a/Ego.cs b/Ego.cs
@@ -4,13 +4,29 @@ public static class Ego
 {
     public static EgoComponent AddGameObject( GameObject gameObject )
     {
-        var egoComponent = gameObject.GetComponent<EgoComponent>();
-		if( egoComponent == null ){ egoComponent = gameObject.AddComponent<EgoComponent>(); }
-        egoComponent.CreateMask();
+		var egoComponent = AddGameObjectToChildren( gameObject.transform );
         EgoEvents<AddedGameObject>.AddEvent( new AddedGameObject( gameObject, egoComponent ) );
         return egoComponent;
     }
 
+	private static EgoComponent AddGameObjectToChildren( Transform transform )
+	{
+		for (int i = 0; i < transform.childCount; i++)
+		{
+			AddGameObjectToChildren( transform.GetChild( i ) );
+		}
+
+		var egoComponent = transform.GetComponent<EgoComponent>();
+		if( egoComponent == null )
+		{
+			egoComponent = transform.gameObject.AddComponent<EgoComponent>();
+		}
+
+		egoComponent.CreateMask();
+
+		return egoComponent;
+	}
+
 	public static C AddComponent<C>( EgoComponent egoComponent ) where C : Component
     {
 		C component = null;