serein-flow/Workbench/Node/NodeControlBase.cs

using Serein.Library;
using Serein.Workbench.Api;
using Serein.Workbench.Node.ViewModel;
using Serein.Workbench.Themes;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Data;
using System.Windows.Media;

namespace Serein.Workbench.Node.View
{

    /// <summary>
    /// 节点控件基类（控件）
    /// </summary>
    public abstract class NodeControlBase : UserControl //, IDynamicFlowNode
    {
        /// <summary>
        /// 节点所在的画布（以后需要将画布封装出来，实现多画布的功能）
        /// </summary>
        public IFlowCanvas FlowCanvas { get; set; }

        /// <summary>
        /// 如果该节点放置在了某个容器节点，就会记录这个容器节点
        /// </summary>

        private INodeContainerControl nodeContainerControl;


        /// <summary>
        /// 记录与该节点控件有关的所有连接
        /// </summary>
        private readonly List<ConnectionControl> connectionControls = new List<ConnectionControl>();

        public NodeControlViewModelBase ViewModel { get; set; }


        protected NodeControlBase()
        {
            this.Background = Brushes.Transparent;
        }

        protected NodeControlBase(NodeControlViewModelBase viewModelBase)
        {
            ViewModel = viewModelBase;
            this.Background = Brushes.Transparent;
            this.DataContext = viewModelBase;
            SetBinding();
        }

        /// <summary>
        /// 放置在某个节点容器中
        /// </summary>
        public void PlaceToContainer(INodeContainerControl nodeContainerControl)
        {
            this.nodeContainerControl = nodeContainerControl;
            FlowCanvas.Remove(this); // 临时从画布上移除
            var result = nodeContainerControl.PlaceNode(this);
            if (!result) // 检查是否放置成功，如果不成功，需要重新添加回来
            {
                FlowCanvas.Add(this); 

            }
        }

        /// <summary>
        /// 从某个节点容器取出
        /// </summary>
        public void TakeOutContainer()
        {
            var result = nodeContainerControl.TakeOutNode(this); // 从控件取出
            if (result) // 移除成功时才添加到画布上
            {
                FlowCanvas.Add(this); // 重新添加到画布上
                if (nodeContainerControl is NodeControlBase containerControl)
                {
                    this.ViewModel.NodeModel.Position.X = containerControl.ViewModel.NodeModel.Position.X + containerControl.Width + 10;
                    this.ViewModel.NodeModel.Position.Y = containerControl.ViewModel.NodeModel.Position.Y;
                }
            }
            
        }

        /// <summary>
        /// 添加与该节点有关的连接后，记录下来
        /// </summary>
        /// <param name="connection"></param>
        public void AddCnnection(ConnectionControl connection)
        {
            connectionControls.Add(connection);
        }

        /// <summary>
        /// 删除了连接之后，还需要从节点中的记录移除
        /// </summary>
        /// <param name="connection"></param>
        public void RemoveConnection(ConnectionControl connection)
        {
            connectionControls.Remove(connection);
            connection.Remove(); // 主动删除连接
        }

        /// <summary>
        /// 删除所有连接
        /// </summary>
        public void RemoveAllConection()
        {
            foreach (var connection in this.connectionControls)
            {
                connection.Remove();  // 主动删除连接
            }
        }

        /// <summary>
        /// 更新与该节点有关的数据
        /// </summary>
        public void UpdateLocationConnections()
        {
            foreach (var connection in this.connectionControls)
            {
                connection.RefreshLine(); // 主动更新连线位置
            }
        }


        /// <summary>
        /// 设置绑定：
        /// Canvas.X and Y ： 画布位置
        /// </summary>
        public void SetBinding()
        {
            var p = ViewModel.NodeModel.Position;

            Binding leftBinding = new(nameof(p.X)) { Source = p, Mode = BindingMode.TwoWay };
            BindingOperations.SetBinding(this, Canvas.LeftProperty, leftBinding);

            Binding topBinding = new(nameof(p.Y)) { Source = p, Mode = BindingMode.TwoWay };
            BindingOperations.SetBinding(this, Canvas.TopProperty, topBinding);
        }

        /// <summary>
        /// 穿透视觉树获取指定类型的第一个元素
        /// </summary>
        /// <typeparam name="T"></typeparam>
        /// <param name="parent"></param>
        /// <returns></returns>
        protected static T FindVisualChild<T>(DependencyObject parent) where T : DependencyObject
        {
            for (int i = 0; i < VisualTreeHelper.GetChildrenCount(parent); i++)
            {
                var child = VisualTreeHelper.GetChild(parent, i);
                if (child is T typedChild)
                {
                    return typedChild;
                }

                var childOfChild = FindVisualChild<T>(child);
                if (childOfChild != null)
                {
                    return childOfChild;
                }
            }
            return null;
        }

        protected static JunctionControlBase[] GetArgJunction(NodeControlBase nodeControl, MethodDetailsControl methodDetailsControl)
        {
            // 获取 MethodDetailsControl 实例
            try
            {
                var itemsControl = FindVisualChild<ItemsControl>(methodDetailsControl); // 查找 ItemsControl
                if (itemsControl != null)
                {
                    var md = nodeControl.ViewModel.NodeModel.MethodDetails;
                    if (md is null)
                    {
                        return [];
                    }
                    if(md.ParameterDetailss is null)
                    {
                        return [];
                    }
                    var argDataJunction = new JunctionControlBase[md.ParameterDetailss.Length];
                    var controls = new List<JunctionControlBase>();

                    for (int i = 0; i < itemsControl.Items.Count; i++)
                    {
                        var container = itemsControl.ItemContainerGenerator.ContainerFromIndex(i) as FrameworkElement;
                        if (container != null)
                        {
                            var argControl = FindVisualChild<ArgJunctionControl>(container);
                            if (argControl != null)
                            {
                                controls.Add(argControl); // 收集 ArgJunctionControl 实例
                            }
                        }
                    }
                    return argDataJunction = controls.ToArray();
                }
                return [];
            }
            catch (Exception ex)
            {

                SereinEnv.WriteLine(InfoType.ERROR,$"节点获取入参控制点时发生异常{Environment.NewLine}节点：{nodeControl.ViewModel.NodeModel.Guid}");
                SereinEnv.WriteLine(ex);
                return [];
            }
        }



    }




    //public class FLowNodeObObservableCollection<T> : ObservableCollection<T>
    //{

    //    public void AddRange(IEnumerable<T> items)
    //    {
    //        foreach (var item in items)
    //        {
    //            this.Items.Add(item);
    //        }
    //        OnCollectionChanged(new NotifyCollectionChangedEventArgs(NotifyCollectionChangedAction.Add));
    //    }
    //}
}