summaryrefslogtreecommitdiffstats
path: root/Scripts/Editor/PipelineExecutorEditor.cs
blob: 857fd8479ad18095c8bbf7db4cc2cb20e9478d6f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#if UNITY_EDITOR

using UnityEngine;
using UnityEditor;
using System.Linq;

[CustomEditor(typeof(PipelineExecutor))]
public class PipelineExecutorEditor : Editor
{
    public override void OnInspectorGUI()
    {
        DrawDefaultInspector();

        PipelineExecutor executor = (PipelineExecutor)target;

        EditorGUILayout.Space();

        if (GUILayout.Button("Auto-Add All Pipelines"))
        {
            AutoAddPipelines(executor);
        }
    }

    private void AutoAddPipelines(PipelineExecutor executor)
    {
        // Find all LinearPipeline components in the current scene
        LinearPipeline[] allPipelines = Object.FindObjectsOfType<LinearPipeline>();

        // Sort by hierarchy order
        System.Array.Sort(allPipelines, (a, b) =>
        {
            // Get hierarchy paths and compare
            string pathA = GetHierarchyPath(a.transform);
            string pathB = GetHierarchyPath(b.transform);
            return pathA.CompareTo(pathB);
        });

        LinearPipeline[] existingPipelines = executor.pipelines ?? new LinearPipeline[0];
        LinearPipeline[] newPipelines = allPipelines.Where(p => p != null && !existingPipelines.Contains(p)).ToArray();

        if (newPipelines.Length > 0)
        {
            Undo.RecordObject(executor, "Auto-Add Pipelines");
            executor.pipelines = existingPipelines.Concat(newPipelines).ToArray();
            EditorUtility.SetDirty(executor);
            Debug.Log($"[PipelineExecutor] Added {newPipelines.Length} new pipelines. Total: {executor.pipelines.Length}");
        }
        else
        {
            Debug.Log("[PipelineExecutor] No new pipelines to add.");
        }
    }

    private string GetHierarchyPath(Transform transform)
    {
        System.Collections.Generic.List<int> indices = new System.Collections.Generic.List<int>();
        Transform current = transform;

        while (current != null)
        {
            indices.Add(current.GetSiblingIndex());
            current = current.parent;
        }

        indices.Reverse();
        return string.Join(".", indices.Select(i => i.ToString("D4")));
    }
}

#endif