1 files changed, 20 insertions, 0 deletions
diff --git a/Scripts/Fold/Editor/FoldGraphView.cs b/Scripts/Fold/Editor/FoldGraphView.cs
new file mode 100644
index 0000000..7b592cd
--- /dev/null
+++ b/Scripts/Fold/Editor/FoldGraphView.cs
@@ -0,0 +1,20 @@
+using UnityEngine;
+using UnityEditor;
+using GraphProcessor;
+using System;
+using System.Collections.Generic;
+
+public class FoldGraphView : BaseGraphView
+{
+    public FoldGraphView(EditorWindow window) : base(window) {}
+
+    public override IEnumerable<(string path, Type type)> FilterCreateNodeMenuEntries()
+    {
+        // Only return our subset of nodes
+        foreach (var nodeMenuItem in NodeProvider.GetNodeMenuEntries(graph))
+        {
+            if (typeof(BaseFoldNode).IsAssignableFrom(nodeMenuItem.type))
+                yield return nodeMenuItem;
+        }
+    }
+}