GhostEngine/Ghost.Editor.Core/SceneGraph/SceneGraphHelpers.cs

using Ghost.Engine.Components;
using Ghost.Entities;

namespace Ghost.Editor.Core.SceneGraph;

public class SceneGraphHelpers
{
    /// <summary>
    /// Creates a new <see cref="EntityNode"/> entity with default components.
    /// </summary>
    /// <param name="world">The world context where the entity will be created.</param>
    /// <param name="entity">The entity to be wrapped in the <see cref="EntityNode"/>.</param>
    public static EntityNode CreateEntityNode(SceneNode owner, Entity entity, string name)
    {
        owner.Scene.World.EntityManager.AddComponent(entity, new LocalToWorld { matrix = Misaki.HighPerformance.Mathematics.float4x4.identity });
        owner.Scene.World.EntityManager.AddComponent(entity, Hierarchy.Root);
        return new EntityNode(owner, entity, name);
    }

    /// <summary>
    /// Creates a new <see cref="Entity"/> and <see cref="EntityNode"/> entity with default components.
    /// </summary>
    /// <param name="owner">The world context where the entity will be created.</param>
    public static EntityNode CreateEntityNode(SceneNode owner, string name)
    {
        var entity = owner.Scene.World.EntityManager.CreateEntity();
        return CreateEntityNode(owner, entity, name);
    }

    /// <summary>
    /// Attaches childEntity to parentEntity in the scene graph.
    /// </summary>
    /// <param name="world">The world context where the entities exist.</param>
    /// <param name="parentNode">The parent entity to which the child will be attached.</param>
    /// <param name="childNode">The child entity to be attached.</param>
    public static void AttachChild(SceneNode scene, EntityNode parentNode, EntityNode childNode)
    {
        // 1) If the child already has a parent, detach it first
        var childHierarchy = scene.Scene.World.EntityManager.GetComponent<Hierarchy>(childNode.Entity);
        if (childHierarchy.parent != Entity.Invalid)
        {
            DetachFromParent(scene, childNode);
        }

        // 2) Link child to new parent
        childHierarchy.parent = parentNode.Entity;

        // 3) Insert child at the head of parent's child list
        var parentHierarchy = scene.Scene.World.EntityManager.GetComponent<Hierarchy>(parentNode.Entity);

        childHierarchy.nextSibling = parentHierarchy.firstChild;
        parentHierarchy.firstChild = childNode.Entity;

        // 4) Write back
        scene.Scene.World.EntityManager.SetComponent(parentNode.Entity, parentHierarchy);
        scene.Scene.World.EntityManager.SetComponent(childNode.Entity, childHierarchy);

        // 5) Update children list in parent node
        parentNode.AddChild(childNode);
    }

    /// <summary>
    /// Detaches the specified entity from its parent in the scene graph.
    /// </summary>
    /// <param name="world">The world context where the entities exist.</param>
    /// <param name="node">The entity to detach from its parent.</param>
    public static void DetachFromParent(SceneNode scene, EntityNode node)
    {
        var hierarchy = scene.Scene.World.EntityManager.GetComponent<Hierarchy>(node.Entity);
        var parent = hierarchy.parent;
        if (parent == Entity.Invalid)
        {
            return; // already root
        }

        var parentHierarchy = scene.Scene.World.EntityManager.GetComponent<Hierarchy>(parent);

        // If entity is the first child, simply move head
        if (parentHierarchy.firstChild == node.Entity)
        {
            parentHierarchy.firstChild = hierarchy.nextSibling;
        }
        else
        {
            // Otherwise, find the previous sibling in the linked list
            var prevSibling = parentHierarchy.firstChild;
            while (prevSibling != Entity.Invalid)
            {
                var prevHierarchy = scene.Scene.World.EntityManager.GetComponent<Hierarchy>(prevSibling);
                if (prevHierarchy.nextSibling == node.Entity)
                {
                    prevHierarchy.nextSibling = hierarchy.nextSibling;
                    scene.Scene.World.EntityManager.SetComponent(prevSibling, prevHierarchy);
                    break;
                }

                prevSibling = prevHierarchy.nextSibling;
            }
        }

        // Clear child's references
        hierarchy.parent = Entity.Invalid;
        hierarchy.nextSibling = Entity.Invalid;

        // Write back
        scene.Scene.World.EntityManager.SetComponent(parent, parentHierarchy);
        scene.Scene.World.EntityManager.SetComponent(node.Entity, hierarchy);

        // Remove from parent's children list
        scene.EntityNodeLookup[parent].RemoveChild(node);
    }
}