T-Node/TNodeGraphViewImpl/Editor/GraphEditor.cs

using System;
using TNodeCore.Editor;
using TNodeCore.Editor.EditorPersistence;
using TNodeCore.Editor.NodeGraphView;
using TNodeCore.Models;
using TNodeGraphViewImpl.Editor.Cache;
using TNodeGraphViewImpl.Editor.NodeGraphView;
using UnityEditor;
using UnityEngine;
using UnityEngine.Serialization;
using UnityEngine.UIElements;

namespace TNodeGraphViewImpl.Editor{
    
    // public class SelectGraphWindow : EditorWindow{
    //     public EditorWindow parent;
    //     public Type graphType;
    //     public static void ShowWindow<T> (GraphEditor<T> parent) where T:GraphData{
    //         var window = GetWindow<SelectGraphWindow>();
    //         window.graphType = typeof(T);
    //         window.Show();
    //         window.parent = parent;
    //     }
    //     private void OnGUI(){
    //             
    //         if(GUILayout.Button("Create An Graph")){
    //             //Add a save file dialog to save the graph
    //             //Create the graph
    //             var graphAsset = ScriptableObject.CreateInstance(graphType);
    //             var path = EditorUtility.SaveFilePanel("Save Graph", "", "", "asset");
    //             //Save the graph
    //             AssetDatabase.CreateAsset(graphAsset, path); 
    //             AssetDatabase.SaveAssets();
    //             AssetDatabase.Refresh();
    //             //Load the graph
    //             var graph = AssetDatabase.LoadAssetAtPath<ScriptableObject>(path) as GraphData;
    //             var graphEditor = parent as IGraphEditor;
    //             if (graphEditor.GetGraphView() != null){
    //                 graphEditor.GetGraphView().SetGraphData(graph);
    //                 Debug.Log(graph);
    //             }
    //         }
    //         //Drag and drop a graph asset to load it
    //         if(Event.current.type == EventType.DragUpdated){
    //             DragAndDrop.visualMode = DragAndDropVisualMode.Copy;
    //             Event.current.Use();
    //         }
    //     }
    // }
    public abstract class GraphEditor<T> : EditorWindow,IGraphEditor where T:GraphData{ 
        protected BaseDataGraphView<T> GraphView;
        [SerializeField]
        private VisualTreeAsset mVisualTreeAsset = default;
        //Persist editor data ,such as node position,node size ,etc ,in this script object
        [FormerlySerializedAs("nodeEditorData")] public GraphEditorData graphEditorData;
        private bool _windowShowed=false;

        public void CreateGUI(){
            
            // Each editor window contains a root VisualElement object
            VisualElement root = rootVisualElement;
            
            // Instantiate UXML
            VisualElement labelFromUXML = mVisualTreeAsset.Instantiate();
            root.Add(labelFromUXML);
            
            BuildGraphView();
            DefineGraphEditorActions();
            GraphView.Owner = this;
            OnCreate();
        }
  
        public void Update(){
            if (GraphView == null) return;
            if (GraphView.Data != null) return;
            if (_windowShowed==false){
                _windowShowed = true;
            }
        }


        public void SetupNonRuntime(T graphData){
            GraphView.Data = graphData;
            GraphView.IsRuntimeGraph = false;
        }
        private void BuildGraphView(){
            GraphView = NodeEditorExtensions.CreateViewComponentFromBaseType<BaseDataGraphView<T>>();
            rootVisualElement.Add(GraphView);
            GraphView.StretchToParentSize();
        }

        private void DefineGraphEditorActions(){
            //Register a  event when user press ctrl + s
            rootVisualElement.RegisterCallback<KeyUpEvent>((evt) => {
                if (evt.keyCode == KeyCode.S && evt.ctrlKey)
                {
                    Save();
                }
            });
            
        }

        private void Save(){
            //if no graph is loaded ,create a file save dialogue
            if (GraphView.Data == null)
            {
                string path = EditorUtility.SaveFilePanel("Save Graph", "", "", "asset");
                if (path.Length != 0){
                    //Create a new asset file with type of GraphDataType
                    T asset = ScriptableObject.CreateInstance<T>();
                    AssetDatabase.CreateAsset(asset, path);
                    AssetDatabase.SaveAssets();
                    
                }
            }
            else{
                GraphView.SaveWithEditorData(graphEditorData);
                AssetDatabase.Refresh();
                
            }
  
        }
        
        protected virtual void OnCreate(){
            
        }

        public void SetGraphView(IBaseDataGraphView graphView){
            GraphView = graphView as BaseDataGraphView<T>;
        }

        public IBaseDataGraphView GetGraphView(){
            return GraphView;
        }
    }
}