ScriptCore: Show Custom Editor

Implemented basic Dictionary editor
2026-09-05 13:01:52 +00:00 · 2024-01-05 15:46:54 +01:00
parent 9da675e4bb
commit 13baf2f4bc
3 changed files with 209 additions and 14 deletions
@@ -0,0 +1,124 @@
+using GlitchyEngine.Extensions;
+using ImGuiNET;
+using System;
+using System.Collections;
+using System.Collections.Generic;
+using System.Diagnostics;
+using System.Runtime.CompilerServices;
+
+namespace GlitchyEngine.Editor;
+
+[CustomEditor(typeof(Dictionary<,>))]
+public class DictionaryEditor
+{
+    private static object? dictionaryForNewValue = null;
+
+    public static object? ShowEditor(object? reference, Type fieldType, string fieldName)
+    {
+        object newDictionary = EntityEditor.DidNotChange;
+        IDictionary? dictionary = reference as IDictionary;
+        
+        Type keyType = fieldType.GetGenericArguments()[0];
+        Type valueType = fieldType.GetGenericArguments()[1];
+
+        // The buttons should always be visible -> save whether the node is open
+        // If we have no instance, the user shouldn't be able to open the list
+        bool listOpen = ImGui.TreeNodeEx(fieldName,  ImGuiTreeNodeFlags.AllowOverlap | ImGuiTreeNodeFlags.SpanFullWidth |
+            (dictionary == null ? ImGuiTreeNodeFlags.Leaf | ImGuiTreeNodeFlags.NoTreePushOnOpen : ImGuiTreeNodeFlags.Framed));
+
+        if (dictionary == null)
+            listOpen = false;
+
+        var addButtonWidth = ImGui.CalcTextSize("+").X + 2 * ImGui.GetStyle().FramePadding.X;
+        var removeButtonWidth = ImGui.CalcTextSize("-").X + 2 * ImGui.GetStyle().FramePadding.X;
+
+        ImGui.SameLine(ImGui.GetWindowContentRegionMax().X - addButtonWidth - ImGui.GetStyle().FramePadding.X);
+
+        if (ImGui.SmallButton("+"))
+        {
+            if (dictionary == null)
+            {
+                newDictionary = Activator.CreateInstance(fieldType);
+                dictionary = newDictionary as IDictionary;
+
+                Debug.Assert(dictionary != null);
+            }
+
+            dictionaryForNewValue = dictionary;
+
+            // TODO: show fake entry!
+
+            //object? newKey = ActivatorExtension.CreateInstanceSafe(keyType);
+            //object? newValue = ActivatorExtension.CreateInstanceSafe(valueType);
+
+            //if (newKey != null)
+            //    dictionary!.Add(newKey, newValue);
+        }
+
+        ImGuiExtension.AttachTooltip("Add a new Element to the dictionary.");
+        
+        if (listOpen)
+        {
+            Debug.Assert(dictionary != null, "Opened tree node even though it should have been a leaf!");
+
+            int id = 0;
+
+            List<DictionaryEntry> newEntries = new();
+            List<object> keysToDelete = new();
+
+            foreach (DictionaryEntry entry in dictionary!)
+            {
+                id++;
+                ImGui.PushID(id);
+
+                bool isEntryOpen = ImGui.TreeNode("");
+
+                ImGui.SameLine(ImGui.GetWindowContentRegionMax().X - removeButtonWidth);
+
+                if (ImGui.SmallButton("-"))
+                {
+                    keysToDelete.Add(entry.Key!);
+                }
+
+                ImGuiExtension.AttachTooltip("Remove the last Element from the list.");
+
+                if (isEntryOpen)
+                {
+                    object newKey = EntityEditor.ShowFieldEditor(entry.Key, entry.Key?.GetType() ?? keyType, "Key");
+
+                    if (newKey != EntityEditor.DidNotChange)
+                    {
+                        keysToDelete.Add(entry.Key!);
+                        newEntries.Add(new DictionaryEntry(newKey, entry.Value));
+                    }
+
+                    object newValue = EntityEditor.ShowFieldEditor(entry.Value, entry.Value?.GetType() ?? valueType, "Value");
+
+                    if (newValue != EntityEditor.DidNotChange)
+                    {
+                        keysToDelete.Add(entry.Key!);
+                        newEntries.Add(new DictionaryEntry(entry.Key!, newValue));
+                    }
+
+                    ImGui.TreePop();
+                }
+                
+                ImGui.PopID();
+            }
+
+            foreach (object key in keysToDelete)
+            {
+                dictionary.Remove(key);
+            }
+
+            foreach (DictionaryEntry newEntry in newEntries)
+            {
+                dictionary.Add(newEntry.Key, newEntry.Value);
+            }
+
+            ImGui.TreePop();
+        }
+
+        return newDictionary;
+    }
+}