mirror of
https://gitee.com/akwkevin/aistudio.-wpf.-diagram
synced 2026-03-03 00:00:57 +08:00
492 lines
21 KiB
C#
492 lines
21 KiB
C#
using System;
|
||
using System.Collections.Generic;
|
||
using System.Linq;
|
||
using System.Text;
|
||
using AIStudio.Wpf.DiagramDesigner.Geometrys;
|
||
|
||
namespace AIStudio.Wpf.DiagramDesigner
|
||
{
|
||
public static class DiagramViewModelHelper
|
||
{
|
||
public static DesignerItemViewModelBase GetConnectorDataItem(IEnumerable<SelectableDesignerItemViewModelBase> items, Guid conectorDataItemId, Type connectorDataItemType)
|
||
{
|
||
DesignerItemViewModelBase dataItem = items.OfType<DesignerItemViewModelBase>().Single(x => x.Id == conectorDataItemId);
|
||
return dataItem;
|
||
}
|
||
|
||
public static RectangleBase GetBoundingRectangle(IEnumerable<DesignerItemViewModelBase> items)
|
||
{
|
||
double x1 = Double.MaxValue;
|
||
double y1 = Double.MaxValue;
|
||
double x2 = Double.MinValue;
|
||
double y2 = Double.MinValue;
|
||
|
||
foreach (DesignerItemViewModelBase item in items)
|
||
{
|
||
x1 = Math.Min(item.Left, x1);
|
||
y1 = Math.Min(item.Top, y1);
|
||
|
||
x2 = Math.Max(item.Left + item.GetItemWidth(), x2);
|
||
y2 = Math.Max(item.Top + item.GetItemHeight(), y2);
|
||
}
|
||
|
||
return new RectangleBase(new PointBase(x1, y1), new PointBase(x2, y2));
|
||
}
|
||
|
||
public static RectangleBase GetBoundingRectangle(DesignerItemViewModelBase item)
|
||
{
|
||
return GetBoundingRectangle(new DesignerItemViewModelBase[] { item });
|
||
}
|
||
|
||
#region 自动依附节点
|
||
public static FullyCreatedConnectorInfo FindNearPortToAttachTo(this IDiagramViewModel diagramViewModel, ConnectionViewModel partialConnection, ConnectorVertexType connectorVertexType)
|
||
{
|
||
if (partialConnection == null)
|
||
return null;
|
||
|
||
diagramViewModel.ClearNearPort();
|
||
foreach (var port in diagramViewModel.Items.OfType<DesignerItemViewModelBase>().ToList().SelectMany(n => n.Connectors))
|
||
{
|
||
if (connectorVertexType == ConnectorVertexType.Start)
|
||
{
|
||
if (partialConnection.SourceConnectorInfo.Position.DistanceTo(port.Position) < diagramViewModel.DiagramOption.SnappingOption.SnappingRadius)
|
||
{
|
||
if (partialConnection.SinkConnectorInfo?.CanAttachTo(port) == true)
|
||
{
|
||
port.DataItem.ShowConnectors = true;
|
||
port.BeAttachTo = true;
|
||
return port;
|
||
}
|
||
else
|
||
{
|
||
port.DataItem.ShowConnectors = true;
|
||
port.DisableAttachTo = true;
|
||
return null;
|
||
}
|
||
}
|
||
}
|
||
else if (connectorVertexType == ConnectorVertexType.End)
|
||
{
|
||
if (partialConnection.SinkConnectorInfo.Position.DistanceTo(port.Position) < diagramViewModel.DiagramOption.SnappingOption.SnappingRadius)
|
||
{
|
||
if (partialConnection.SourceConnectorInfo?.CanAttachTo(port) == true)
|
||
{
|
||
port.DataItem.ShowConnectors = true;
|
||
port.BeAttachTo = true;
|
||
return port;
|
||
}
|
||
else
|
||
{
|
||
port.DataItem.ShowConnectors = true;
|
||
port.DisableAttachTo = true;
|
||
return null;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
return null;
|
||
}
|
||
|
||
public static FullyCreatedConnectorInfo FindNearPortToAttachTo(this IDiagramViewModel diagramViewModel, ConnectionViewModel partialConnection)
|
||
{
|
||
if (partialConnection == null)
|
||
return null;
|
||
|
||
diagramViewModel.ClearNearPort();
|
||
foreach (var port in diagramViewModel.Items.OfType<DesignerItemViewModelBase>().ToList().SelectMany(n => n.Connectors))
|
||
{
|
||
if (partialConnection.OnGoingPosition.DistanceTo(port.Position) < diagramViewModel.DiagramOption.SnappingOption.SnappingRadius)
|
||
{
|
||
port.DataItem.ShowConnectors = true;
|
||
if (partialConnection.SourceConnectorInfo?.CanAttachTo(port) == true)
|
||
{
|
||
port.BeAttachTo = true;
|
||
return port;
|
||
}
|
||
else
|
||
{
|
||
port.DisableAttachTo = true;
|
||
return null;
|
||
}
|
||
|
||
}
|
||
}
|
||
|
||
return null;
|
||
}
|
||
|
||
/// <summary>
|
||
/// type=0最近且没有依附；=1
|
||
/// </summary>
|
||
/// <param name="blockDesignerItemViewModel"></param>
|
||
/// <param name="type"></param>
|
||
/// <returns></returns>
|
||
public static Tuple<FullyCreatedConnectorInfo, FullyCreatedConnectorInfo> FindNearPortToAttachTo(this IDiagramViewModel diagramViewModel, BlockDesignerItemViewModel blockDesignerItemViewModel, bool isExist)
|
||
{
|
||
if (blockDesignerItemViewModel == null)
|
||
return new Tuple<FullyCreatedConnectorInfo, FullyCreatedConnectorInfo>(null, null);
|
||
|
||
List<BlockDesignerItemViewModel> items;
|
||
if (isExist == false)
|
||
{
|
||
items = diagramViewModel.Items.OfType<BlockDesignerItemViewModel>().
|
||
Where(p => p != blockDesignerItemViewModel && p != blockDesignerItemViewModel.Parent && p != blockDesignerItemViewModel.Next)
|
||
.ToList();
|
||
|
||
}
|
||
else
|
||
{
|
||
items = diagramViewModel.Items.OfType<BlockDesignerItemViewModel>().
|
||
Where(p => p != blockDesignerItemViewModel)
|
||
.ToList();
|
||
}
|
||
|
||
FullyCreatedConnectorInfo parent = null;
|
||
FullyCreatedConnectorInfo next = null;
|
||
|
||
foreach (var port in items.SelectMany(n => n.Connectors))
|
||
{
|
||
foreach (var port2 in blockDesignerItemViewModel.Connectors)
|
||
{
|
||
//parent
|
||
if (parent == null)
|
||
{
|
||
if ((port.Orientation == ConnectorOrientation.Right && port2.Orientation == ConnectorOrientation.Left)
|
||
|| (port.Orientation == ConnectorOrientation.Bottom && port2.Orientation == ConnectorOrientation.Top))
|
||
{
|
||
|
||
if (port.Position.DistanceTo(port2.Position) < diagramViewModel.DiagramOption.SnappingOption.SnappingRadius)
|
||
{
|
||
port.DataItem.ShowConnectors = true;
|
||
if (port2.CanAttachTo(port) == true)
|
||
{
|
||
port.BeAttachTo = true;
|
||
parent = port;
|
||
continue;
|
||
}
|
||
else
|
||
{
|
||
port.DisableAttachTo = true;
|
||
}
|
||
|
||
}
|
||
}
|
||
}
|
||
|
||
//next
|
||
if (next == null)
|
||
{
|
||
if ((port.Orientation == ConnectorOrientation.Left && port2.Orientation == ConnectorOrientation.Right)
|
||
|| (port.Orientation == ConnectorOrientation.Top && port2.Orientation == ConnectorOrientation.Bottom))
|
||
{
|
||
|
||
if (port.Position.DistanceTo(port2.Position) < diagramViewModel.DiagramOption.SnappingOption.SnappingRadius)
|
||
{
|
||
port.DataItem.ShowConnectors = true;
|
||
if (port2.CanAttachTo(port) == true)
|
||
{
|
||
port.BeAttachTo = true;
|
||
next = port;
|
||
continue;
|
||
}
|
||
else
|
||
{
|
||
port.DisableAttachTo = true;
|
||
}
|
||
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
return new Tuple<FullyCreatedConnectorInfo, FullyCreatedConnectorInfo>(parent, next);
|
||
}
|
||
|
||
public static BlockItemsContainerInfo FindNearContainerToAttachTo(this IDiagramViewModel diagramViewModel, BlockDesignerItemViewModel blockDesignerItemViewModel)
|
||
{
|
||
if (blockDesignerItemViewModel == null)
|
||
return null;
|
||
|
||
List<BlockDesignerItemViewModel> items;
|
||
|
||
items = diagramViewModel.Items.OfType<BlockDesignerItemViewModel>().
|
||
Where(p => p != blockDesignerItemViewModel)
|
||
.ToList();
|
||
|
||
foreach (var port in items.SelectMany(n => n.Containers))
|
||
{
|
||
if (port.GetBounds().IntersectsWith(blockDesignerItemViewModel.GetBounds())) //如果两个位置相交
|
||
{
|
||
var innerport = port.GetAllContainers(port.Children, false).Where(p => p.GetBounds().IntersectsWith(blockDesignerItemViewModel.GetBounds())).OrderByDescending(p => p.ContainerLevel).FirstOrDefault();
|
||
if (innerport != null)
|
||
{
|
||
innerport.DataItem.ShowConnectors = true;
|
||
if (innerport.CanAttachTo(blockDesignerItemViewModel) == true)
|
||
{
|
||
innerport.BeAttachTo = true;
|
||
}
|
||
else
|
||
{
|
||
innerport.DisableAttachTo = true;
|
||
}
|
||
return innerport;
|
||
}
|
||
else
|
||
{
|
||
port.DataItem.ShowConnectors = true;
|
||
if (port.CanAttachTo(blockDesignerItemViewModel) == true)
|
||
{
|
||
port.BeAttachTo = true;
|
||
}
|
||
else
|
||
{
|
||
port.DisableAttachTo = true;
|
||
}
|
||
return port;
|
||
}
|
||
}
|
||
}
|
||
|
||
return null;
|
||
}
|
||
|
||
/// <summary>
|
||
/// type=0最近且没有依附；=1
|
||
/// </summary>
|
||
/// <param name="blockDesignerItemViewModel"></param>
|
||
/// <param name="type"></param>
|
||
/// <returns></returns>
|
||
public static Tuple<FullyCreatedConnectorInfo, FullyCreatedConnectorInfo> FindNearPortToAttachTo(this IDiagramViewModel diagramViewModel, BlockDesignerItemTempLink blockDesignerItemTempLink, bool isExist)
|
||
{
|
||
if (blockDesignerItemTempLink == null || blockDesignerItemTempLink.Items == null || blockDesignerItemTempLink.Items.Count == 0)
|
||
return new Tuple<FullyCreatedConnectorInfo, FullyCreatedConnectorInfo>(null, null);
|
||
|
||
List<BlockDesignerItemViewModel> items;
|
||
if (isExist == false)
|
||
{
|
||
items = diagramViewModel.Items.OfType<BlockDesignerItemViewModel>().Where(p => !blockDesignerItemTempLink.Items.Contains(p)).ToList();
|
||
}
|
||
else
|
||
{
|
||
items = diagramViewModel.Items.OfType<BlockDesignerItemViewModel>().Where(p => !blockDesignerItemTempLink.Items.Contains(p)).ToList();
|
||
}
|
||
|
||
FullyCreatedConnectorInfo parent = null;
|
||
FullyCreatedConnectorInfo next = null;
|
||
|
||
foreach (var port in items.SelectMany(n => n.Connectors).OfType<BlockConnectorInfo>())
|
||
{
|
||
//已经被连接的不允许在顶部吸附了
|
||
if ((port.Orientation == ConnectorOrientation.Top || port.Orientation == ConnectorOrientation.Left) && port.DataItem.Prev != null)
|
||
{
|
||
continue;
|
||
}
|
||
|
||
foreach (var port2 in blockDesignerItemTempLink.Connectors)
|
||
{
|
||
//parent
|
||
if (parent == null)
|
||
{
|
||
if ((port.Orientation == ConnectorOrientation.Right && port2.Orientation == ConnectorOrientation.Left)
|
||
|| (port.Orientation == ConnectorOrientation.Bottom && port2.Orientation == ConnectorOrientation.Top))
|
||
{
|
||
|
||
if (port.Position.DistanceTo(port2.Position) < diagramViewModel.DiagramOption.SnappingOption.SnappingRadius)
|
||
{
|
||
port.DataItem.ShowConnectors = true;
|
||
if (port2.CanAttachTo(port) == true)
|
||
{
|
||
port.BeAttachTo = true;
|
||
parent = port;
|
||
continue;
|
||
}
|
||
else
|
||
{
|
||
port.DisableAttachTo = true;
|
||
}
|
||
|
||
}
|
||
}
|
||
}
|
||
|
||
//next
|
||
if (next == null)
|
||
{
|
||
if ((port.Orientation == ConnectorOrientation.Left && port2.Orientation == ConnectorOrientation.Right)
|
||
|| (port.Orientation == ConnectorOrientation.Top && port2.Orientation == ConnectorOrientation.Bottom))
|
||
{
|
||
|
||
if (port.Position.DistanceTo(port2.Position) < diagramViewModel.DiagramOption.SnappingOption.SnappingRadius)
|
||
{
|
||
port.DataItem.ShowConnectors = true;
|
||
if (port2.CanAttachTo(port) == true)
|
||
{
|
||
port.BeAttachTo = true;
|
||
next = port;
|
||
continue;
|
||
}
|
||
else
|
||
{
|
||
port.DisableAttachTo = true;
|
||
}
|
||
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
return new Tuple<FullyCreatedConnectorInfo, FullyCreatedConnectorInfo>(parent, next);
|
||
}
|
||
|
||
public static BlockItemsContainerInfo FindNearContainerToAttachTo(this IDiagramViewModel diagramViewModel, BlockDesignerItemTempLink blockDesignerItemTempLink)
|
||
{
|
||
if (blockDesignerItemTempLink == null || blockDesignerItemTempLink.Items == null || blockDesignerItemTempLink.Items.Count == 0)
|
||
return null;
|
||
|
||
List<BlockDesignerItemViewModel> items;
|
||
|
||
items = diagramViewModel.Items.OfType<BlockDesignerItemViewModel>().Where(p => !blockDesignerItemTempLink.Items.Contains(p)).ToList();
|
||
|
||
foreach (var port in items.SelectMany(n => n.Containers))
|
||
{
|
||
if (port.GetBounds().IntersectsWith(blockDesignerItemTempLink.GetBounds())) //如果两个位置相交
|
||
{
|
||
var innerport = port.GetAllContainers(port.Children, false).Where(p => p.GetBounds().IntersectsWith(blockDesignerItemTempLink.GetBounds())).OrderByDescending(p => p.ContainerLevel).FirstOrDefault();
|
||
if (innerport != null)
|
||
{
|
||
innerport.DataItem.ShowConnectors = true;
|
||
if (innerport.CanAttachTo(blockDesignerItemTempLink.Items.FirstOrDefault()) == true)
|
||
{
|
||
innerport.BeAttachTo = true;
|
||
return innerport;
|
||
}
|
||
//else
|
||
//{
|
||
// innerport.DisableAttachTo = true;
|
||
// return null;
|
||
//}
|
||
|
||
}
|
||
else
|
||
{
|
||
port.DataItem.ShowConnectors = true;
|
||
if (port.CanAttachTo(blockDesignerItemTempLink.Items.FirstOrDefault()) == true)
|
||
{
|
||
port.BeAttachTo = true;
|
||
return port;
|
||
}
|
||
//else
|
||
//{
|
||
// port.DisableAttachTo = true;
|
||
// return null;
|
||
//}
|
||
}
|
||
}
|
||
}
|
||
|
||
return null;
|
||
}
|
||
|
||
public static void ClearNearPort(this IDiagramViewModel diagramViewModel)
|
||
{
|
||
diagramViewModel.Items.OfType<DesignerItemViewModelBase>().ToList().SelectMany(n => n.Connectors).Where(p => p.BeAttachTo == true || p.DisableAttachTo == true).ToList()
|
||
.ForEach(p => {
|
||
p.DisableAttachTo = false;
|
||
p.BeAttachTo = false;
|
||
});
|
||
}
|
||
|
||
public static void ClearNearContain(this IDiagramViewModel diagramViewModel)
|
||
{
|
||
diagramViewModel.Items.OfType<BlockDesignerItemViewModel>().ToList().SelectMany(n => n.GetAllContainers()).Where(p => p.BeAttachTo == true || p.DisableAttachTo == true).ToList()
|
||
.ForEach(p => {
|
||
p.DisableAttachTo = false;
|
||
p.BeAttachTo = false;
|
||
});
|
||
}
|
||
#endregion
|
||
|
||
#region Block拖拽预览-拖拽完成
|
||
public static void PreviewNearBlock(this IDiagramViewModel diagramViewModel, List<BlockDesignerItemViewModel> blocks)
|
||
{
|
||
if (blocks.Any())
|
||
{
|
||
diagramViewModel.ClearNearPort();
|
||
diagramViewModel.ClearNearContain();
|
||
var links = BlockDesignerItemTempLink.Build(blocks);
|
||
foreach (BlockDesignerItemTempLink item in links)
|
||
{
|
||
var container = diagramViewModel.FindNearContainerToAttachTo(item);
|
||
if (container != null)
|
||
{
|
||
continue;
|
||
}
|
||
diagramViewModel.FindNearPortToAttachTo(item, false);
|
||
}
|
||
}
|
||
}
|
||
|
||
|
||
|
||
public static void FinishNearBlock(this IDiagramViewModel diagramViewModel, List<BlockDesignerItemViewModel> blocks)
|
||
{
|
||
if (blocks.Any())
|
||
{
|
||
diagramViewModel.ClearNearPort();
|
||
diagramViewModel.ClearNearContain();
|
||
var links = BlockDesignerItemTempLink.Build(blocks);
|
||
|
||
foreach (BlockDesignerItemTempLink item in links)
|
||
{
|
||
var container = diagramViewModel.FindNearContainerToAttachTo(item);
|
||
if (container != null)
|
||
{
|
||
container.DataItem.AddChild(item.Items.FirstOrDefault(), container);//待完善
|
||
container.BeAttachTo = false;
|
||
container.DisableAttachTo = false;
|
||
continue;
|
||
}
|
||
|
||
var portTuple = diagramViewModel.FindNearPortToAttachTo(item, true);
|
||
var portParent = portTuple.Item1;
|
||
var portNext = portTuple.Item2;
|
||
|
||
if (portParent != null)
|
||
{
|
||
(portParent.DataItem as BlockDesignerItemViewModel).AddNext(item.Items.FirstOrDefault());
|
||
portParent.BeAttachTo = false;
|
||
portParent.DisableAttachTo = false;
|
||
}
|
||
else
|
||
{
|
||
if (item.Items.FirstOrDefault().Parent != null)
|
||
{
|
||
(item.Items.FirstOrDefault().Parent as BlockDesignerItemViewModel).RemoveNext();
|
||
}
|
||
}
|
||
|
||
if (portNext != null)
|
||
{
|
||
item.Items.LastOrDefault().AddNext(portNext.DataItem as BlockDesignerItemViewModel);
|
||
portNext.BeAttachTo = false;
|
||
portNext.DisableAttachTo = false;
|
||
}
|
||
else
|
||
{
|
||
if (item.Items.LastOrDefault().Next != null)
|
||
{
|
||
item.Items.LastOrDefault().RemoveNext();
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
|
||
#endregion
|
||
}
|
||
}
|
