Fixed Versioning

Editor/Scripts/Editor/Data Types/IPTPort.cs

@@ -10,6 +10,13 @@ namespace ImageProcessingGraph.Editor
     {
         public FieldInfo fieldInfo;
         private VisualElement _exposedPropertyContainer;
+
+        public delegate void OnPortConnectedEvent();
+        public OnPortConnectedEvent OnPortConnected;
+        
+        public delegate void OnPortDisconnectedEvent();
+        public OnPortDisconnectedEvent OnPortDisconnected;
+        
         public VisualElement ExposedPropertyContainer
         {
             set
@@ -24,9 +31,27 @@ namespace ImageProcessingGraph.Editor
         
         protected IPTPort(Orientation portOrientation, Direction portDirection, Capacity portCapacity, Type type) : base(portOrientation, portDirection, portCapacity, type)
         {
-
+            
         }
-        
+
+        public override void Connect(Edge edge)
+        {
+            base.Connect(edge);
+            OnPortConnected?.Invoke();
+        }
+
+        public override void Disconnect(Edge edge)
+        {
+            base.Disconnect(edge);
+            OnPortDisconnected?.Invoke();
+        }
+
+
+        private void PublicOnConnected(Port obj)
+        {
+            throw new NotImplementedException();
+        }
+
         public static IPTPort Create(IEdgeConnectorListener connectorListener, bool isInput, Type type)
         {
             var port = new IPTPort(Orientation.Horizontal, isInput ? Direction.Input : Direction.Output,

Editor/Scripts/Editor/ImageProcessingGraphAsset.cs

@@ -53,8 +53,7 @@ namespace ImageProcessingGraph.Editor
             Stopwatch stopwatch = new Stopwatch();
             stopwatch.Start();
 
-            if (runOrder == null)
-                runOrder = GetExecutionOrder(this.nodes, this.connections);
+            runOrder = GetExecutionOrder(this.nodes, this.connections);
 
             bool failed = false;
             
@@ -67,6 +66,12 @@ namespace ImageProcessingGraph.Editor
                 }
             }
 
+            foreach (var VARIABLE in runOrder)
+            {
+                if(!VARIABLE.RunCleanUp())
+                    failed = true;
+            }
+
             // Stop the stopwatch after running the nodes
             stopwatch.Stop();
 

Editor/Scripts/Editor/Nodes/Base/BaseImageNode.cs

@@ -124,6 +124,26 @@ namespace ImageProcessingGraph.Editor
         {
             Debug.Log("Uppies");
         }
+
+        public bool RunCleanUp()
+        {
+            try
+            {
+                this.CleanUp();
+                return true;
+            }
+            catch (Exception e)
+            {
+                onFailed?.Invoke();
+                Debug.LogError(e);
+                return false;
+            }
+        }
+
+        public virtual void CleanUp()
+        {
+            
+        }
         
         public void SetPosition(Rect position) => this.position = position;
     }

Editor/Scripts/Editor/Nodes/Base/RangeProperty.cs

@@ -1,5 +1,10 @@
 using System;
+#if UNITY_2022_1_OR_NEWER
 using UnityEngine.UIElements;
+#else
+using UnityEditor.UIElements;
+#endif
+
 
 namespace ImageProcessingGraph.Editor
 {
@@ -13,11 +18,12 @@ namespace ImageProcessingGraph.Editor
             
         }
     }
-
+    
     public class GreyscaleField : IntegerField
     {
         public (int, int) minMax = (0,255);
     }
     
     
+    
 }

Editor/Scripts/Editor/Nodes/NodeAttributes/NodeAttributes.cs

@@ -1,4 +1,5 @@
 using System;
+using ImageProcessingGraph.Editor.Unity_Image_Processing.Scripts.Editor.Windows;
 
 namespace ImageProcessingGraph.Editor.Nodes.NodeAttributes
 {
@@ -8,16 +9,19 @@ namespace ImageProcessingGraph.Editor.Nodes.NodeAttributes
         private string name;
         private string menuItem;
         private string ussPath;
-
+        private Type editorType;
+        
         public string Title => name;
         public string MenuItem => menuItem;
         public string UssPath => ussPath;
+        public Type EditorType => editorType;
 
-        public NodeInfoAttribute(string name, string menuItem = "", bool requiresImage = false, string ussPath = null)
+        public NodeInfoAttribute(string name, string menuItem = "", bool requiresImage = false, string ussPath = null, Type editorType = null)
         {
             this.name = name;
             this.menuItem = menuItem;
             this.ussPath = ussPath;
+            this.editorType = editorType;
         }
     }
 

Editor/Scripts/Editor/Nodes/Types/Image/Export/Texture2DOutput.cs

@@ -6,12 +6,12 @@ using UnityEngine;
 
 namespace ImageProcessingGraph.Editor.Nodes.Output
 {
-    [NodeInfo("Texture Export", "Export/Export Texture", true)]
+    [NodeInfo("Texture Export", "Export/Texture Export", true)]
     public class Texture2DOutput : BaseImageNode
     {
         [NodeAttributes.Input("")] public ImageData inputPixels;
         [NodeAttributes.Input("File Name")] public string fileName;
-        [NodeAttributes.Input("File Path")] public string fileDirectory;
+        [NodeAttributes.Input("File Directory")] public string fileDirectory;
 
         public enum ExportType
         {
@@ -20,16 +20,36 @@ namespace ImageProcessingGraph.Editor.Nodes.Output
         }
         
         [NodeAttributes.Input("Export Type")] public ExportType exportType;
+        [NodeAttributes.Input("Texture Type")] public TextureImporterType textureType;
+        
+        [NodeAttributes.Output("Output Texture")] public Texture2D textureOutput;
 
         public override void Process()
         {
+            
             switch (exportType)
             {
                 case ExportType.Texture2D:
-                    AssetDatabase.CreateAsset(inputPixels.ToTexture2D(), $"{fileDirectory}/{fileName}.asset");
+                    string pathT2D = $"{fileDirectory}/{fileName}.asset";
+                    AssetDatabase.CreateAsset(inputPixels.ToTexture2D(), pathT2D);
+                    textureOutput = AssetDatabase.LoadAssetAtPath<Texture2D>(pathT2D);
                     break;
                 case ExportType.PNG:
-                    inputPixels.ExportPNG($"{fileDirectory}/{fileName}.png");
+                    string pathPNG = $"{fileDirectory}/{fileName}.png";
+                    inputPixels.ExportPNG(pathPNG);
+                    
+                    AssetDatabase.ImportAsset(pathPNG);
+                    
+                    TextureImporter textureImporter = AssetImporter.GetAtPath(pathPNG) as TextureImporter;
+                    if (textureImporter != null)
+                    {
+                        textureImporter.textureType = textureType;
+                        EditorUtility.SetDirty(AssetDatabase.LoadAssetAtPath<Texture2D>(pathPNG));
+                        textureImporter.SaveAndReimport();
+                    }
+                    AssetDatabase.ImportAsset(pathPNG);
+                    
+                    textureOutput = AssetDatabase.LoadAssetAtPath<Texture2D>(pathPNG);
                     break;
             }
         }

Editor/Scripts/Editor/Nodes/Types/Image/Import/Texture2DImport.cs

@@ -6,7 +6,7 @@ using UnityEngine;
 
 namespace ImageProcessingGraph.Editor.Nodes.Import_Nodes
 {
-    [NodeInfo("Texture Import", "Imports/Import Texture", true)]
+    [NodeInfo("Texture Import", "Imports/Import Texture")]
     public class Texture2DImport : BaseImageNode
     {
         [NodeAttributes.Input("")]
@@ -16,7 +16,7 @@ namespace ImageProcessingGraph.Editor.Nodes.Import_Nodes
         public ImageData textureOutput;
         [NodeAttributes.Output("File Name")]
         public string fileName;
-        [NodeAttributes.Output("File Path")]
+        [NodeAttributes.Output("File Directory")]
         public string filePath;
 
         public override void Process()
@@ -42,8 +42,8 @@ namespace ImageProcessingGraph.Editor.Nodes.Import_Nodes
                 this.fileName = textureImport.name;
                 this.filePath = Path.GetDirectoryName(AssetDatabase.GetAssetPath(textureImport));
             }
-            else
-                Debug.LogError("UH!");
+            // else
+            //     Debug.LogError("UH!");
         }
     }
 }

Editor/Scripts/Editor/Nodes/Types/Image/Utilities/ViewNode.meta

@@ -0,0 +1,3 @@
+fileFormatVersion: 2
+guid: b462f37a291b49a696e3b4b554d26868
+timeCreated: 1745899248

Editor/Scripts/Editor/Nodes/Types/Image/Utilities/ViewNode/ViewImageNode.cs

@@ -0,0 +1,135 @@
+using ImageProcessingGraph.Editor.Nodes.NodeAttributes;
+using UnityEngine;
+using Unity.Collections;
+using Unity.Jobs;
+using Unity.Burst;
+
+namespace ImageProcessingGraph.Editor.Nodes.Types.Image.Utilities.ViewNode
+{
+    [NodeInfo("View Texture", "Utility/View Texture", false, null , editorType: typeof(ViewTextureNodeEditor))]
+    public partial class ViewTextureNode : BaseImageNode
+    {
+        [NodeAttributes.Input("Texture")] public Texture2D texture;
+        [NodeAttributes.Input("Image Data")] public ImageData imageData;
+
+        public delegate void OnImageUpdated();
+        public OnImageUpdated onImageUpdated;
+
+        public Texture2D cachedRGB;
+        public Texture2D cachedR;
+        public Texture2D cachedG;
+        public Texture2D cachedB;
+        public Texture2D cachedA;
+
+        public override void Process()
+        {
+            if (texture == null)
+            {
+                texture = imageData.ToTexture2D();
+            }
+
+            GenerateChannelTextures(texture);
+
+            onImageUpdated?.Invoke();
+        }
+
+        public override void CleanUp()
+        {
+            texture = null;
+            imageData = default;
+        }
+
+        private void GenerateChannelTextures(Texture2D source)
+        {
+            var pixels = source.GetPixels();
+            int length = pixels.Length;
+
+            NativeArray<Color> inputPixels = new NativeArray<Color>(pixels, Allocator.TempJob);
+
+            NativeArray<Color> outputR = new NativeArray<Color>(length, Allocator.TempJob);
+            NativeArray<Color> outputG = new NativeArray<Color>(length, Allocator.TempJob);
+            NativeArray<Color> outputB = new NativeArray<Color>(length, Allocator.TempJob);
+            NativeArray<Color> outputA = new NativeArray<Color>(length, Allocator.TempJob);
+
+            var jobR = new ChannelJob(ChannelType.R, inputPixels, outputR);
+            var jobG = new ChannelJob(ChannelType.G, inputPixels, outputG);
+            var jobB = new ChannelJob(ChannelType.B, inputPixels, outputB);
+            var jobA = new ChannelJob(ChannelType.A, inputPixels, outputA);
+
+            JobHandle handleR = jobR.Schedule(length, 64);
+            JobHandle handleG = jobG.Schedule(length, 64);
+            JobHandle handleB = jobB.Schedule(length, 64);
+            JobHandle handleA = jobA.Schedule(length, 64);
+
+            var handles = new NativeArray<JobHandle>(4, Allocator.Temp);
+            handles[0] = handleR;
+            handles[1] = handleG;
+            handles[2] = handleB;
+            handles[3] = handleA;
+
+            JobHandle.CompleteAll(handles);
+            handles.Dispose(); 
+
+
+            cachedRGB = source;
+            cachedR = CreateTexture(source.width, source.height, outputR);
+            cachedG = CreateTexture(source.width, source.height, outputG);
+            cachedB = CreateTexture(source.width, source.height, outputB);
+            cachedA = CreateTexture(source.width, source.height, outputA);
+
+            inputPixels.Dispose();
+            outputR.Dispose();
+            outputG.Dispose();
+            outputB.Dispose();
+            outputA.Dispose();
+        }
+
+        private Texture2D CreateTexture(int width, int height, NativeArray<Color> colors)
+        {
+            Texture2D tex = new Texture2D(width, height, TextureFormat.RGBA32, false);
+            tex.SetPixels(colors.ToArray());
+            tex.Apply();
+            return tex;
+        }
+
+        [BurstCompile]
+        private struct ChannelJob : IJobParallelFor
+        {
+            public ChannelType channelType;
+            [ReadOnly] public NativeArray<Color> input;
+            public NativeArray<Color> output;
+
+            public ChannelJob(ChannelType type, NativeArray<Color> input, NativeArray<Color> output)
+            {
+                this.channelType = type;
+                this.input = input;
+                this.output = output;
+            }
+
+            public void Execute(int index)
+            {
+                Color c = input[index];
+                switch (channelType)
+                {
+                    case ChannelType.R:
+                        output[index] = new Color(c.r, c.r, c.r, 1f);
+                        break;
+                    case ChannelType.G:
+                        output[index] = new Color(c.g, c.g, c.g, 1f);
+                        break;
+                    case ChannelType.B:
+                        output[index] = new Color(c.b, c.b, c.b, 1f);
+                        break;
+                    case ChannelType.A:
+                        output[index] = new Color(c.a, c.a, c.a, 1f);
+                        break;
+                }
+            }
+        }
+
+        private enum ChannelType
+        {
+            R, G, B, A
+        }
+    }
+}

Editor/Scripts/Editor/Nodes/Types/Image/Utilities/ViewNode/ViewImageNode.cs.meta

@@ -0,0 +1,3 @@
+fileFormatVersion: 2
+guid: 461687d0e0d346ac9dd5960b76db7538
+timeCreated: 1745899335

Editor/Scripts/Editor/Nodes/Types/Image/Utilities/ViewNode/ViewImageNodeEditor.cs

@@ -0,0 +1,104 @@
+using ImageProcessingGraph.Editor.Unity_Image_Processing.Scripts.Editor.Windows;
+using UnityEngine;
+using UnityEngine.UIElements;
+
+namespace ImageProcessingGraph.Editor.Nodes.Types.Image.Utilities.ViewNode
+{
+    public class ViewTextureNodeEditor : ImageProcessingGraphNodeVisual
+    {
+        private UnityEngine.UIElements.Image viewableImage;
+        private Foldout foldout;
+        private VisualElement buttonRow;
+
+        public ViewTextureNodeEditor(BaseImageNode node, ImageProcessingGraphViewWindow window) : base(node, window)
+        {
+            //Port 0 is Texture2D
+            //Port 1 is ImageData
+
+            IPTPort tex2DPort = InputPorts[0] as IPTPort;
+            IPTPort imageDataPort = InputPorts[1] as IPTPort;
+
+            tex2DPort.OnPortConnected += () => { imageDataPort.style.display = DisplayStyle.None; };
+
+            tex2DPort.OnPortDisconnected += () => { imageDataPort.style.display = DisplayStyle.Flex; };
+
+            imageDataPort.OnPortConnected += () => { tex2DPort.style.display = DisplayStyle.None; };
+
+            imageDataPort.OnPortDisconnected += () => { tex2DPort.style.display = DisplayStyle.Flex; };
+
+            this.Q("output").style.display = DisplayStyle.None;
+
+            ViewTextureNode viewImageNode = node as ViewTextureNode;
+
+            foldout = new Foldout
+            {
+                text = "Texture Viewer",
+                value = true
+            };
+            foldout.style.backgroundColor = new Color(63f / 255f, 63f / 255f, 63f / 255f, 205f / 255f);
+            foldout.contentContainer.style.justifyContent = Justify.Center;
+            Add(foldout);
+
+            viewableImage = new UnityEngine.UIElements.Image
+            {
+                scaleMode = ScaleMode.ScaleToFit
+            };
+            viewableImage.style.width = 256;
+            viewableImage.style.height = 256;
+            viewableImage.style.backgroundColor = Color.black;
+            foldout.Add(viewableImage);
+
+            viewImageNode.onImageUpdated += () => { viewableImage.image = viewImageNode.texture; };
+
+            buttonRow = new VisualElement
+            {
+                style =
+                {
+                    flexDirection = FlexDirection.Row,
+                    justifyContent = Justify.SpaceBetween,
+                    alignItems = Align.Center,
+                    marginTop = 8,
+                    marginBottom = 8,
+                }
+            };
+
+            string[] channels = { "RGB", "R", "G", "B", "A" };
+            foreach (var channel in channels)
+            {
+                var btn = new Button(() => OnChannelClicked(viewImageNode, channel))
+                {
+                    text = channel
+                };
+                btn.style.flexGrow = 1;
+                buttonRow.Add(btn);
+            }
+
+            foldout.Add(buttonRow);
+        }
+
+        private void OnChannelClicked(ViewTextureNode viewNode, string channel)
+        {
+            switch (channel)
+            {
+                case "RGB":
+                    viewableImage.image = viewNode.cachedRGB;
+                    break;
+                case "R":
+                    viewableImage.image = viewNode.cachedR;
+                    break;
+                case "G":
+                    viewableImage.image = viewNode.cachedG;
+                    break;
+                case "B":
+                    viewableImage.image = viewNode.cachedB;
+                    break;
+                case "A":
+                    viewableImage.image = viewNode.cachedA;
+                    break;
+                default:
+                    Debug.LogError("[ViewTextureNodeEditor] Unknown channel clicked! Did you hack reality again?");
+                    break;
+            }
+        }
+    }
+}

Editor/Scripts/Editor/Nodes/Types/Image/Utilities/ViewNode/ViewImageNodeEditor.cs.meta

@@ -0,0 +1,3 @@
+fileFormatVersion: 2
+guid: 83fd847c22794f0a92efbecd18c528c0
+timeCreated: 1745899443

Editor/Scripts/Editor/Windows/ImageProcessingGraphEdgeConnectorListener.cs

@@ -26,28 +26,27 @@ namespace ImageProcessingGraph.Editor.Unity_Image_Processing.Scripts.Editor.Wind
         public void OnDropOutsidePort(Edge edge, Vector2 position)
         {
             window.searchProvider.target = (VisualElement)window.focusController.focusedElement;
-            SearchWindow.Open(new SearchWindowContext(position), window.searchProvider);
-            
+
+            // ⚡ Override the search window open with a customized search tree
+            SearchWindow.Open(new SearchWindowContext(position), new ImageProcessingGraphSearchProvider.CustomSearchProviderForEdge(window.searchProvider, edge, (IPTPort)edge.input, (IPTPort)edge.output));
+    
+            // Remove connections as you did
             List<GraphConnection> connections = new List<GraphConnection>();
-            
             foreach (var conn in window.asset.Connections)
             {
                 if (conn.internalEdge == edge)
-                {
                     connections.Add(conn);
-                }
             }
-
             foreach (var VARIABLE in connections)
             {
                 window.asset.Connections.Remove(VARIABLE);
             }
-            
-            if(edge.input != null)
-                if(edge.input.node != null)
+
+            if (edge.input != null)
+                if (edge.input.node != null)
                     ((ImageProcessingGraphNodeVisual)edge.input.node).ToggleExposedVariable((IPTPort)edge.input, true);
         }
-        
+
 
 
         public void OnDrop(UnityEditor.Experimental.GraphView.GraphView graphView, Edge edge)

Editor/Scripts/Editor/Windows/ImageProcessingGraphEditorWindow.cs

@@ -14,23 +14,25 @@ namespace ImageProcessingGraph.Editor.Windows
 
         public static void Open(ImageProcessingGraphAsset asset)
         {
-            ImageProcessingGraphEditorWindow[] windows = Resources.FindObjectsOfTypeAll<ImageProcessingGraphEditorWindow>();
-
-            foreach (var w in windows)
+            var existingWindows = Resources.FindObjectsOfTypeAll<ImageProcessingGraphEditorWindow>();
+            foreach (var w in existingWindows)
             {
-                w.Focus();
-                return;
+                if (w.CurrentGraph == asset)
+                {
+                    w.Focus(); // 👁 focus the OG window
+                    return;
+                }
             }
 
-            ImageProcessingGraphEditorWindow window =
-                CreateWindow<ImageProcessingGraphEditorWindow>(typeof(ImageProcessingGraphEditorWindow),
-                    typeof(SceneView));
-            
-            window.titleContent = new GUIContent($"{asset.name}", EditorGUIUtility.ObjectContent(null, typeof(ImageProcessingGraphAsset)).image
-            );
+            var window = CreateWindow<ImageProcessingGraphEditorWindow>(typeof(SceneView));
+            window.titleContent = new GUIContent($"{asset.name}",
+                EditorGUIUtility.ObjectContent(null, typeof(ImageProcessingGraphAsset)).image);
             window.Load(asset);
+            window.Focus();
         }
 
+
+
         void OnEnable()
         {
             if(currentGraph != null)

Editor/Scripts/Editor/Windows/ImageProcessingGraphNodeVisual.cs

@@ -29,12 +29,13 @@ namespace ImageProcessingGraph.Editor.Unity_Image_Processing.Scripts.Editor.Wind
         {
             this.AddToClassList("image-node-visual");
             this.window = window;
-
             graphNode = node;
+            
+            
             Type typeInfo = node.GetType();
-
             NodeInfoAttribute info = typeInfo.GetCustomAttribute<NodeInfoAttribute>();
             title = info.Title;
+            this.name = typeInfo.Name;
 
             string[] depths = info.MenuItem.Split('/');
             foreach (var depth in depths)
@@ -69,16 +70,6 @@ namespace ImageProcessingGraph.Editor.Unity_Image_Processing.Scripts.Editor.Wind
             
             CreateInputPorts(inputFieldInfo);
             CreateOutputPorts(outputFieldInfo);
-
-            foreach (Port input in InputPorts)
-            {
-                
-            }
-            
-            foreach (Port input in OutputPorts)
-            {
-                
-            }
             
             defaaStyleSheet = AssetDatabase.LoadAssetAtPath<StyleSheet>("Assets/Unity Image Processing/Node.uss");
             if (defaaStyleSheet == null)
@@ -103,9 +94,6 @@ namespace ImageProcessingGraph.Editor.Unity_Image_Processing.Scripts.Editor.Wind
                 if (styleSheets.Contains(errorStyleSheet))
                     styleSheets.Remove(errorStyleSheet);
             };
-            
-
-            this.name = typeInfo.Name;
         }
 
         private void CreateInputPorts(List<FieldInfo> fields)
@@ -119,6 +107,7 @@ namespace ImageProcessingGraph.Editor.Unity_Image_Processing.Scripts.Editor.Wind
                 if (label != "")
                     port.portName = label; 
                 InputPorts.Add(port);
+                
 
                 inputContainer.Add(port);
                 ExposeVariableToPort(port, field);
@@ -324,6 +313,17 @@ namespace ImageProcessingGraph.Editor.Unity_Image_Processing.Scripts.Editor.Wind
             return null;
         }
 
+        public override void BuildContextualMenu(ContextualMenuPopulateEvent evt)
+        {
+            evt.menu.InsertAction(evt.menu.MenuItems().Count, "Open Documentation", OpenDocumentation);
+            evt.menu.AppendSeparator();
+            base.BuildContextualMenu(evt);
+        }
+
+        private void OpenDocumentation(DropdownMenuAction obj)
+        {
+            
+        }
 
         public void SavePosition() => graphNode.SetPosition(GetPosition());
     }

Editor/Scripts/Editor/Windows/ImageProcessingGraphSearchProvider.cs

@@ -8,6 +8,7 @@ using UnityEditor;
 using UnityEditor.Experimental.GraphView;
 using UnityEngine;
 using UnityEngine.UIElements;
+using Input = ImageProcessingGraph.Editor.Nodes.NodeAttributes.Input;
 
 namespace ImageProcessingGraph.Editor.Unity_Image_Processing.Scripts.Editor.Windows
 {
@@ -16,13 +17,21 @@ namespace ImageProcessingGraph.Editor.Unity_Image_Processing.Scripts.Editor.Wind
         public object target { get; private set; }
         public string title { get; private set; }
 
-        public SearchContextElement(object target, string title)
+        public Dictionary<int, Type> ImportPortTypes { get; private set; }
+        public Dictionary<int, Type> ExportPortTypes { get; private set; }
+        public int portID;
+
+        public SearchContextElement(object target, string title, Dictionary<int, Type> importPortTypes,
+            Dictionary<int, Type> exportPortTypes, int portID)
         {
             this.target = target;
             this.title = title;
+            this.ImportPortTypes = importPortTypes;
+            this.ExportPortTypes = exportPortTypes;
+            this.portID = portID;
         }
     }
-    
+
     public class ImageProcessingGraphSearchProvider : ScriptableObject, ISearchWindowProvider
     {
         public ImageProcessingGraphViewWindow graph;
@@ -30,44 +39,34 @@ namespace ImageProcessingGraph.Editor.Unity_Image_Processing.Scripts.Editor.Wind
 
         public static List<SearchContextElement> elements;
         private Assembly[] assemblies;
+
         public List<SearchTreeEntry> CreateSearchTree(SearchWindowContext context)
         {
             List<SearchTreeEntry> tree = new List<SearchTreeEntry>();
             tree.Add(new SearchTreeGroupEntry(new GUIContent("Nodes"), 0));
-            
-            elements = new List<SearchContextElement>();
-            
-            /*
-            assemblies = AppDomain.CurrentDomain.GetAssemblies();
-            */
 
-            /*foreach (var assembly in assemblies)
-            {
-                foreach (Type type in assembly.GetTypes())
-                {
-                    if (type.CustomAttributes.ToList() != null)
-                    {
-                        var attr = type.GetCustomAttribute(typeof(NodeInfoAttribute));
-                        if (attr != null)
-                        {
-                            NodeInfoAttribute info = attr as NodeInfoAttribute;
-                            var node = Activator.CreateInstance(type);
-                            if(string.IsNullOrEmpty(info.MenuItem)) continue;
-                            
-                            elements.Add(new SearchContextElement(node, info.MenuItem));
-                        }
-                    }
-                }
-            }*/
-            
+            elements = new List<SearchContextElement>();
+
             foreach (var type in TypeCache.GetTypesWithAttribute<NodeInfoAttribute>())
             {
                 var attr = type.GetCustomAttribute<NodeInfoAttribute>();
                 NodeInfoAttribute info = attr as NodeInfoAttribute;
                 var node = Activator.CreateInstance(type);
-                if(string.IsNullOrEmpty(info.MenuItem)) continue;
-                            
-                elements.Add(new SearchContextElement(node, info.MenuItem));
+                if (string.IsNullOrEmpty(info.MenuItem)) continue;
+
+                var Fields = type.GetFields();
+                Dictionary<int, Type> ImportPortTypes = new Dictionary<int, Type>();
+                Dictionary<int, Type> OutputPortTypes = new Dictionary<int, Type>();
+
+                foreach (var field in Fields)
+                {
+                    if (field.GetCustomAttribute<Input>() != null)
+                        ImportPortTypes.Add(ImportPortTypes.Count, field.FieldType);
+                    if (field.GetCustomAttribute<Output>() != null)
+                        OutputPortTypes.Add(OutputPortTypes.Count, field.FieldType);
+                }
+
+                elements.Add(new SearchContextElement(node, info.MenuItem, ImportPortTypes, OutputPortTypes, 0));
             }
 
             elements.Sort((entry1, entry2) =>
@@ -78,20 +77,20 @@ namespace ImageProcessingGraph.Editor.Unity_Image_Processing.Scripts.Editor.Wind
                 for (int i = 0; i < splits1.Length; i++)
                 {
                     if (i >= splits2.Length) return 1;
-                    
+
                     int value = splits1[i].CompareTo(splits2[i]);
                     if (value != 0)
                     {
-                        if(splits1.Length != splits2.Length && (i == splits1.Length - 1 || i == splits2.Length - 1))
-                           return splits1.Length < splits2.Length ? 1 : -1;
-                           
+                        if (splits1.Length != splits2.Length && (i == splits1.Length - 1 || i == splits2.Length - 1))
+                            return splits1.Length < splits2.Length ? 1 : -1;
+
                         return value;
                     }
                 }
 
                 return 0;
             });
-            
+
             List<string> groups = new List<string>();
 
             foreach (var element in elements)
@@ -108,12 +107,13 @@ namespace ImageProcessingGraph.Editor.Unity_Image_Processing.Scripts.Editor.Wind
                         tree.Add(new SearchTreeGroupEntry(new GUIContent(groupName), i + 1));
                         groups.Add(groupName);
                     }
+
                     groupName += '/';
                 }
-                
+
                 SearchTreeEntry entry = new SearchTreeEntry(new GUIContent(entryTitle.Last()));
                 entry.level = entryTitle.Length;
-                entry.userData = new SearchContextElement(element.target, element.title);
+                entry.userData = element;
                 tree.Add(entry);
             }
 
@@ -122,11 +122,12 @@ namespace ImageProcessingGraph.Editor.Unity_Image_Processing.Scripts.Editor.Wind
 
         public bool OnSelectEntry(SearchTreeEntry SearchTreeEntry, SearchWindowContext context)
         {
-            var mousePos = graph.ChangeCoordinatesTo(graph, context.screenMousePosition - graph.Window.position.position);
+            var mousePos =
+                graph.ChangeCoordinatesTo(graph, context.screenMousePosition - graph.Window.position.position);
             var graphMousePosition = graph.contentViewContainer.WorldToLocal(mousePos);
-            
+
             SearchContextElement element = (SearchContextElement)SearchTreeEntry.userData;
-            
+
             BaseImageNode node = (BaseImageNode)element.target;
             node.SetPosition(new Rect(graphMousePosition, new Vector2()));
             graph.Add(node);
@@ -134,5 +135,148 @@ namespace ImageProcessingGraph.Editor.Unity_Image_Processing.Scripts.Editor.Wind
 
             return true;
         }
+
+        public List<SearchTreeEntry> CreateSearchTreeForEdge(SearchWindowContext context, Edge edge)
+        {
+            var tree = new List<SearchTreeEntry>();
+            tree.Add(new SearchTreeGroupEntry(new GUIContent("Compatible Nodes"), 0));
+
+            elements = new List<SearchContextElement>();
+            var groups = new List<string>();
+
+            var sourcePort = edge.output ?? edge.input;
+            bool isSourceOutput = edge.output != null;
+            Type targetType = (sourcePort as IPTPort)?.portType;
+
+            if (targetType == null)
+            {
+                Debug.LogWarning("Could not determine port type from edge!");
+                return tree;
+            }
+
+            foreach (var type in TypeCache.GetTypesWithAttribute<NodeInfoAttribute>())
+            {
+                var attr = type.GetCustomAttribute<NodeInfoAttribute>();
+                if (string.IsNullOrEmpty(attr.MenuItem)) continue;
+
+                var node = Activator.CreateInstance(type);
+                var fields = type.GetFields();
+
+                // Get all ports
+                var inputPorts = fields.Where(f => f.GetCustomAttribute<Input>() != null).ToList();
+                var outputPorts = fields.Where(f => f.GetCustomAttribute<Output>() != null).ToList();
+
+                // REVERSED LOGIC:
+                // If dragging from output, we want compatible INPUT ports
+                // If dragging from input, we want compatible OUTPUT ports
+                var compatiblePorts = isSourceOutput
+                    ? inputPorts.Where(f => f.FieldType.IsAssignableFrom(targetType)).ToList()
+                    : outputPorts.Where(f => targetType.IsAssignableFrom(f.FieldType)).ToList();
+
+                if (compatiblePorts.Count == 0) continue;
+
+                // Build group hierarchy
+                var menuPath = attr.MenuItem.Split('/');
+                var currentGroupPath = "";
+
+                for (int i = 0; i < menuPath.Length - 1; i++)
+                {
+                    currentGroupPath += menuPath[i];
+                    if (!groups.Contains(currentGroupPath))
+                    {
+                        tree.Add(new SearchTreeGroupEntry(new GUIContent(menuPath[i]), i + 1));
+                        groups.Add(currentGroupPath);
+                    }
+
+                    currentGroupPath += "/";
+                }
+
+                // // Add node entry
+                // var nodeTitle = $"{type.Name}: {menuPath.Last()}";
+                // tree.Add(new SearchTreeEntry(new GUIContent(nodeTitle))
+                // {
+                //     level = menuPath.Length,
+                //     userData = new SearchContextElement(
+                //         node,
+                //         nodeTitle,
+                //         inputPorts.ToDictionary(f => inputPorts.IndexOf(f), f => f.FieldType),
+                //         outputPorts.ToDictionary(f => outputPorts.IndexOf(f), f => f.FieldType))
+                // });
+
+                // Add compatible port entries
+                foreach (var portField in compatiblePorts)
+                {
+                    var portList = portField.GetCustomAttribute<Input>() != null ? inputPorts : outputPorts;
+                    int portIndex = portList.IndexOf(portField);
+
+                    var portTitle = $"{attr.Title}: {portField.Name}";
+                    tree.Add(new SearchTreeEntry(new GUIContent(portTitle))
+                    {
+                        level = menuPath.Length,
+                        userData = new SearchContextElement(
+                            node,
+                            portTitle,
+                            inputPorts.ToDictionary(f => inputPorts.IndexOf(f), f => f.FieldType),
+                            outputPorts.ToDictionary(f => outputPorts.IndexOf(f), f => f.FieldType),
+                            portIndex)
+                    });
+                }
+            }
+
+            return tree;
+        }
+
+        public class CustomSearchProviderForEdge : ScriptableObject, ISearchWindowProvider
+        {
+            private readonly ImageProcessingGraphSearchProvider original;
+            private readonly Edge edge;
+            private readonly IPTPort inputPort;
+            private readonly IPTPort outputPort;
+            
+            public CustomSearchProviderForEdge(ImageProcessingGraphSearchProvider original, Edge edge, IPTPort inputPort, IPTPort outputPort)
+            {
+                this.original = original;
+                this.edge = edge;
+                this.inputPort = inputPort;
+                this.outputPort = outputPort;
+            }
+
+            public List<SearchTreeEntry> CreateSearchTree(SearchWindowContext context)
+            {
+                return original.CreateSearchTreeForEdge(context, edge);
+            }
+
+            public bool OnSelectEntry(SearchTreeEntry selectedEntry, SearchWindowContext context)
+            {
+                var mousePos =
+                    original.graph.ChangeCoordinatesTo(original.graph, context.screenMousePosition - original.graph.Window.position.position);
+                var graphMousePosition = original.graph.contentViewContainer.WorldToLocal(mousePos);
+
+                SearchContextElement element = (SearchContextElement)selectedEntry.userData;
+
+                BaseImageNode node = (BaseImageNode)element.target;
+                node.SetPosition(new Rect(graphMousePosition, new Vector2()));
+                original.graph.Add(node);
+                node.asset = original.graph.asset;
+
+
+                Edge edge = new Edge();
+                
+                if (inputPort != null)
+                {
+                    edge = inputPort.ConnectTo(original.graph.nodeDictionary[node.ID].OutputPorts[element.portID]);
+                }
+                else if (outputPort != null)
+                {
+                    edge = outputPort.ConnectTo(original.graph.nodeDictionary[node.ID].InputPorts[element.portID]);
+                }
+                
+                original.graph.CreateEdge(edge);
+                original.graph.AddElement((GraphElement)edge);
+                
+                return true;
+            }
+        }
+
     }
 }

Editor/Scripts/Editor/Windows/ImageProcessingGraphViewWindow.cs

@@ -1,5 +1,8 @@
+using System;
 using System.Collections.Generic;
 using System.Linq;
+using System.Reflection;
+using ImageProcessingGraph.Editor.Nodes.NodeAttributes;
 using ImageProcessingGraph.Editor.Unity_Image_Processing.Scripts.Editor.Windows;
 using ImageProcessingGraph.Editor.Windows;
 using UnityEditor;
@@ -69,20 +72,16 @@ namespace ImageProcessingGraph.Editor
             
             DrawNodes();
             DrawConnections();
-
-            foreach (var conn in asset.Connections)
-            {
-                if (conn.internalEdge == null)
-                {
-                    
-                }
-                
-                //GetNode(conn.inputPort.nodeID).ToggleExposedVariable(conn.internalEdge.input, true);
-                
-            }
-
+            
             graphViewChanged += OnGraphViewChanged;
-            Undo.undoRedoEvent += UndoEvent;
+            
+            Undo.undoRedoPerformed += UndoRedoPerformed;
+        }
+        
+        private void UndoRedoPerformed()
+        {
+            DrawNodes();
+            DrawConnections();
         }
 
         private void CreateButtons()
@@ -234,12 +233,6 @@ namespace ImageProcessingGraph.Editor
             return compatiblePorts;
         }
 
-        private void UndoEvent(in UndoRedoInfo undo)
-        {
-            DrawNodes();
-            DrawConnections();
-        }
-
         private GraphViewChange OnGraphViewChanged(GraphViewChange graphviewchange)
         {
             
@@ -299,7 +292,7 @@ namespace ImageProcessingGraph.Editor
 
         #region Edges
     
-        void CreateEdge(Edge edge)
+        public void CreateEdge(Edge edge)
         {
             ImageProcessingGraphNodeVisual outputNode = (ImageProcessingGraphNodeVisual)edge.output.node;
             ImageProcessingGraphNodeVisual inputNode = (ImageProcessingGraphNodeVisual)edge.input.node;
@@ -443,7 +436,18 @@ namespace ImageProcessingGraph.Editor
         {
             node.typeName = node.GetType().AssemblyQualifiedName;
 
-            ImageProcessingGraphNodeVisual editorNode = new ImageProcessingGraphNodeVisual(node, this);
+            var infoAttr = node.GetType().GetCustomAttribute<NodeInfoAttribute>();
+
+            ImageProcessingGraphNodeVisual editorNode = null;
+            if (typeof(ImageProcessingGraphNodeVisual).IsAssignableFrom(infoAttr.EditorType))
+            {
+                editorNode = (ImageProcessingGraphNodeVisual)Activator.CreateInstance(infoAttr.EditorType, node, this);
+            }
+            else
+            {
+                editorNode = new ImageProcessingGraphNodeVisual(node, this);
+            }
+            
             editorNode.SetPosition(node.Position);
 
             graphNodes.Add(editorNode);

README.md

@@ -6,9 +6,11 @@ A tool for applying image transformations and effects using a graph-based node s
 
 - **Tested on:**
     - Unity 2022.3.50f1
+    - Unity 2021.3.16f1
     - Unity 6000.0.19f1 (Unity 6)
 
 - **Supported versions:**
+    - Unity 2021.3
     - Unity 2022.0
     - Unity 2023.0
     - Unity 6000.0

package.json

@@ -1,9 +1,9 @@
 {
   "name": "com.chromium.imageprocessingrah",
-  "version": "0.0.0",
+  "version": "0.0.1",
   "displayName": "Image Processing Graph",
   "description": "Stuffs",
-  "unity": "2022.3",
+  "unity": "2021.3",
   "dependencies": {
     "com.unity.burst": "1.8.17"
   },