Immediately track parent components when adding them on child.

isadorasophia · Jul 22, 2024 · 478c70f · 478c70f
1 parent 3366c95
commit 478c70f
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/Bang/Entities/Entity.cs b/src/Bang/Entities/Entity.cs
@@ -579,9 +579,16 @@ private void NotifyAndSubscribeOnComponentAdded(int index, IComponent c)
 
             OnComponentAdded?.Invoke(this, index);
 
-            if (_lookup.IsRelative(index))
+            if (_lookup.IsRelative(index) && _parent is not null)
             {
-                _parent?.TrackComponent(index, OnParentModified);
+                _parent.TrackComponent(index, OnParentModified);
+
+                if (c is IParentRelativeComponent relativeComponent &&
+                    _parent.TryGetComponent(index, out IComponent? parentComponent))
+                {
+                    // Immediately place the parent component reference.
+                    relativeComponent.OnParentModified(parentComponent, this);
+                }
             }
         }