using System; using System.Collections.Generic; using System.Collections.ObjectModel; using System.ComponentModel; using System.Linq; using System.Windows; using System.Windows.Input; //using System.Windows; using System.Windows.Media; using AIStudio.Wpf.DiagramDesigner.Geometrys; using AIStudio.Wpf.DiagramDesigner.Helpers; namespace AIStudio.Wpf.DiagramDesigner { public class ConnectorViewModel : SelectableDesignerItemViewModelBase { public ConnectorViewModel(IDiagramViewModel parent, FullyCreatedConnectorInfo sourceConnectorInfo, ConnectorInfoBase sinkConnectorInfo, DrawMode drawMode, RouterMode routerMode) { Parent = parent; PathMode = drawMode.ToString(); RouterMode = routerMode.ToString(); Init(sourceConnectorInfo, sinkConnectorInfo); } public ConnectorViewModel(IDiagramViewModel parent, FullyCreatedConnectorInfo sourceConnectorInfo, FullyCreatedConnectorInfo sinkConnectorInfo, ConnectionItem designer) : base(parent, designer) { PathMode = designer.PathMode; RouterMode = designer.RouterMode; Init(sourceConnectorInfo, sinkConnectorInfo); } public ConnectorViewModel(FullyCreatedConnectorInfo sourceConnectorInfo, ConnectorInfoBase sinkConnectorInfo, DrawMode drawMode, RouterMode routerMode) : this(null, sourceConnectorInfo, sinkConnectorInfo, drawMode, routerMode) { } private void Init(FullyCreatedConnectorInfo sourceConnectorInfo, ConnectorInfoBase sinkConnectorInfo) { this.Parent = sourceConnectorInfo.DataItem.Parent; if (Parent != null && Parent.ColorViewModel != null) { this.ColorViewModel = CopyHelper.Mapper(Parent.ColorViewModel); } if (sinkConnectorInfo is FullyCreatedConnectorInfo sink && sink.DataItem.ShowArrow == false) { this.ColorViewModel.RightArrowPathStyle = ArrowPathStyle.None; } ColorViewModel.PropertyChanged += ConnectorViewModel_PropertyChanged; this.PropertyChanged += ConnectorViewModel_PropertyChanged; var routetype = GlobalType.AllTypes.Where(p => typeof(IRouter).IsAssignableFrom(p)).FirstOrDefault(p => p.Name == RouterMode); Router = routetype != null ? (System.Activator.CreateInstance(routetype) as IRouter) : new RouterNormal(); var pathGeneratortype = GlobalType.AllTypes.Where(p => typeof(IPathGenerator).IsAssignableFrom(p)).FirstOrDefault(p => p.Name == PathMode); PathGenerator = pathGeneratortype != null ? (System.Activator.CreateInstance(pathGeneratortype) as IPathGenerator) : new ConnectingLineSmooth(); this.SourceConnectorInfo = sourceConnectorInfo; this.SinkConnectorInfo = sinkConnectorInfo; DeleteConnectionCommand = new SimpleCommand(DeleteConnection); AddVertexCommand = new SimpleCommand(AddVertex); } private void ConnectorViewModel_PropertyChanged1(object sender, PropertyChangedEventArgs e) { throw new NotImplementedException(); } public override SelectableDesignerItemBase ToXmlObject() { if (IsFullConnection) { ConnectionItem connection = new ConnectionItem(this); return connection; } else { return null; } } public override Type ToXmlType() { return typeof(ConnectionItem); } #region 属性 private PointBase _sourceA; public PointBase SourceA { get { return _sourceA; } set { SetProperty(ref _sourceA, value); } } private PointBase _sourceB; public PointBase SourceB { get { return _sourceB; } set { SetProperty(ref _sourceB, value); } } private PointBase _startPoint; public PointBase StartPoint { get { return _startPoint; } private set { SetProperty(ref _startPoint, value); } } private PointBase _endPoint; public PointBase EndPoint { get { return _endPoint; } private set { SetProperty(ref _endPoint, value); } } private double _startAngle; public double StartAngle { get { return _startAngle; } private set { SetProperty(ref _startAngle, value); } } private double _endAngle; public double EndAngle { get { return _endAngle; } private set { SetProperty(ref _endAngle, value); } } private RectangleBase _area; public RectangleBase Area { get { return _area; } private set { SetProperty(ref _area, value); } } public string PathMode { get; set; } public string RouterMode { get; set; } public IRouter Router { get; set; } public IPathGenerator PathGenerator { get; set; } private PathGeneratorResult _pathGeneratorResult; public PathGeneratorResult PathGeneratorResult { get { return _pathGeneratorResult; } private set { SetProperty(ref _pathGeneratorResult, value); } } private bool _shouldInsertAnchor; public bool ShouldInsertAnchor { get { return _shouldInsertAnchor; } set { SetProperty(ref _shouldInsertAnchor, value); } } //待完善这两处 private ObservableCollection _vertices = new ObservableCollection(); public ObservableCollection Vertices { get { return _vertices; } private set { SetProperty(ref _vertices, value); } } public List Labels { get; set; } = new List(); public virtual Dictionary PropertiesSetting { get { return new Dictionary() { { "Text","文本" }, }; } } private FullyCreatedConnectorInfo _sourceConnectorInfo; public FullyCreatedConnectorInfo SourceConnectorInfo { get { return _sourceConnectorInfo; } set { SetProperty(ref _sourceConnectorInfo, value); } } private ConnectorInfoBase _sinkConnectorInfo; public ConnectorInfoBase SinkConnectorInfo { get { return _sinkConnectorInfo; } set { SetProperty(ref _sinkConnectorInfo, value); } } public FullyCreatedConnectorInfo SinkConnectorInfoFully { get { return SinkConnectorInfo as FullyCreatedConnectorInfo; } } public PartCreatedConnectionInfo SinkConnectorInfoPart { get { return SinkConnectorInfo as PartCreatedConnectionInfo; } } public PointBase OnGoingPosition { get { return SinkConnectorInfoPart?.MiddlePosition ?? PointBase.Zero; } } public bool IsFullConnection { get { return SinkConnectorInfoFully != null; } } public bool IsPortless => SourceConnectorInfo?.DataItem?.Connectors?.Count() == 0; #endregion #region 方法 public SimpleCommand DeleteConnectionCommand { get; set; } public SimpleCommand AddVertexCommand { get; set; } #endregion private void ConnectorViewModel_PropertyChanged(object sender, PropertyChangedEventArgs e) { if (sender is ConnectorViewModel) { switch (e.PropertyName) { case nameof(SourceA): case nameof(SourceB): UpdateArea(); break; case nameof(Area): UpdatePathGeneratorResult(); if (e is ValuePropertyChangedEventArgs valuePropertyChangedEventArgs) { OutTextItemLocation((RectangleBase)valuePropertyChangedEventArgs.OldValue, (RectangleBase)valuePropertyChangedEventArgs.NewValue); } break; case nameof(Vertices): foreach (var vertice in Vertices) { vertice.PropertyChanged += new WeakINPCEventHandler(ConnectorViewModel_PropertyChanged).Handler; } break; case nameof(SourceConnectorInfo): SourceA = PointHelper.GetPointForConnector(SourceConnectorInfo); (SourceConnectorInfo.DataItem as INotifyPropertyChanged).PropertyChanged += new WeakINPCEventHandler(ConnectorViewModel_PropertyChanged).Handler; break; case nameof(SinkConnectorInfo): SourceB = SinkConnectorInfo.Position; if (SinkConnectorInfo is FullyCreatedConnectorInfo) { (((FullyCreatedConnectorInfo)SinkConnectorInfo).DataItem as INotifyPropertyChanged).PropertyChanged += new WeakINPCEventHandler(ConnectorViewModel_PropertyChanged).Handler; } break; } } else if (sender is ColorViewModel) { if (e.PropertyName == nameof(ColorViewModel.LeftArrowPathStyle) || e.PropertyName == nameof(ColorViewModel.LeftArrowSizeStyle) || e.PropertyName == nameof(ColorViewModel.RightArrowPathStyle) || e.PropertyName == nameof(ColorViewModel.RightArrowSizeStyle)) { UpdatePathGeneratorResult(); } } else if (sender is DesignerItemViewModelBase) { switch (e.PropertyName) { case nameof(DesignerItemViewModelBase.ItemHeight): case nameof(DesignerItemViewModelBase.ItemWidth): case nameof(DesignerItemViewModelBase.Left): case nameof(DesignerItemViewModelBase.Top): SourceA = PointHelper.GetPointForConnector(this.SourceConnectorInfo); if (IsFullConnection) { SourceB = PointHelper.GetPointForConnector(this.SinkConnectorInfoFully); } break; } } else if (sender is LinkVertexModel) { switch (e.PropertyName) { case nameof(ConnectorPoint.Left): case nameof(ConnectorPoint.Top): UpdatePathGeneratorResult(); break; } } } private void UpdateArea() { Area = new RectangleBase(SourceA, SourceB); } private void UpdatePathGeneratorResult() { if (SourceConnectorInfo == null || SinkConnectorInfo == null) return; var route = Router.Get(Parent, this); (var source, var target) = FindConnectionPoints(route); if (source == null || target == null) return; PathGeneratorResult = PathGenerator.Get(Parent, this, route, source.Value, target.Value); //修正旋转 switch (SourceConnectorInfo.Orientation) { case ConnectorOrientation.Left: { StartPoint = new PointBase(PathGeneratorResult.SourceMarkerPosition.X, PathGeneratorResult.SourceMarkerPosition.Y - ColorViewModel.LeftArrowSize / 2); break; } case ConnectorOrientation.Top: { StartPoint = new PointBase(PathGeneratorResult.SourceMarkerPosition.X - ColorViewModel.LeftArrowSize / 2, PathGeneratorResult.SourceMarkerPosition.Y); break; } case ConnectorOrientation.Right: { StartPoint = new PointBase(PathGeneratorResult.SourceMarkerPosition.X - ColorViewModel.LeftArrowSize, PathGeneratorResult.SourceMarkerPosition.Y - ColorViewModel.LeftArrowSize / 2); break; } case ConnectorOrientation.Bottom: { StartPoint = new PointBase(PathGeneratorResult.SourceMarkerPosition.X - ColorViewModel.LeftArrowSize / 2, PathGeneratorResult.SourceMarkerPosition.Y - ColorViewModel.LeftArrowSize); break; } default: { StartPoint = PathGeneratorResult.SourceMarkerPosition; break; } } //修正旋转 switch (SinkConnectorInfo.Orientation) { case ConnectorOrientation.Left: { EndPoint = new PointBase(PathGeneratorResult.TargetMarkerPosition.X, PathGeneratorResult.TargetMarkerPosition.Y - ColorViewModel.RightArrowSize / 2); break; } case ConnectorOrientation.Top: { EndPoint = new PointBase(PathGeneratorResult.TargetMarkerPosition.X - ColorViewModel.RightArrowSize / 2, PathGeneratorResult.TargetMarkerPosition.Y); break; } case ConnectorOrientation.Right: { EndPoint = new PointBase(PathGeneratorResult.TargetMarkerPosition.X - ColorViewModel.RightArrowSize, PathGeneratorResult.TargetMarkerPosition.Y - ColorViewModel.RightArrowSize / 2); break; } case ConnectorOrientation.Bottom: { EndPoint = new PointBase(PathGeneratorResult.TargetMarkerPosition.X - ColorViewModel.RightArrowSize / 2, PathGeneratorResult.TargetMarkerPosition.Y - ColorViewModel.RightArrowSize); break; } default: { EndPoint = PathGeneratorResult.TargetMarkerPosition; break; } } StartAngle = PathGeneratorResult.SourceMarkerAngle; EndAngle = PathGeneratorResult.TargetMarkerAngle; } private void DeleteConnection(object args) { if (this.Parent is IDiagramViewModel) { var diagramVM = this.Parent as IDiagramViewModel; diagramVM.RemoveItemCommand.Execute(this); } } private void AddVertex(object parameter) { MouseButtonEventArgs mosueArg = ((EventToCommandArgs)parameter).EventArgs as MouseButtonEventArgs; var position = mosueArg.GetPosition(((EventToCommandArgs)parameter).Sender as IInputElement); var vertice = new LinkVertexModel(this, new PointBase(position.X, position.Y)); vertice.PropertyChanged += new WeakINPCEventHandler(ConnectorViewModel_PropertyChanged).Handler; Vertices.Add(vertice); UpdatePathGeneratorResult(); if (!((Keyboard.Modifiers & ModifierKeys.Control) == ModifierKeys.Control)) { ShouldInsertAnchor = false; } } private (PointBase? source, PointBase? target) FindConnectionPoints(PointBase[] route) { if (IsPortless) // Portless { if (SourceConnectorInfo.DataItem == null || (IsFullConnection && SinkConnectorInfoFully.DataItem == null)) return (null, null); var sourceCenter = SourceConnectorInfo.DataItem.GetBounds().Center; var targetCenter = SinkConnectorInfoFully?.DataItem?.GetBounds().Center ?? OnGoingPosition; var firstPt = route.Length > 0 ? route[0] : targetCenter; var secondPt = route.Length > 0 ? route[0] : sourceCenter; var sourceLine = new LineBase(firstPt, sourceCenter); var targetLine = new LineBase(secondPt, targetCenter); var sourceIntersections = SourceConnectorInfo.DataItem.GetShape().GetIntersectionsWithLine(sourceLine); var targetIntersections = SinkConnectorInfoFully.DataItem.GetShape()?.GetIntersectionsWithLine(targetLine) ?? new PointBase[] { OnGoingPosition }; var sourceIntersection = GetClosestPointTo(sourceIntersections, firstPt); var targetIntersection = GetClosestPointTo(targetIntersections, secondPt); return (sourceIntersection ?? sourceCenter, targetIntersection ?? targetCenter); } else { var source = SourceConnectorInfo.MiddlePosition;//GetPortPositionBasedOnAlignment(SourceConnectorInfo, ColorViewModel.LeftArrowSizeStyle); var target = SinkConnectorInfo.MiddlePosition;// GetPortPositionBasedOnAlignment(SinkConnectorInfoFully, ColorViewModel.RightArrowSizeStyle); return (source, target); } } private PointBase? GetPortPositionBasedOnAlignment(ConnectorInfoBase port, ArrowSizeStyle marker) { if (port == null) return null; if (marker == null) return port.MiddlePosition; var pt = port.Position; switch (port.Orientation) { case ConnectorOrientation.Top: return new PointBase(pt.X + port.ConnectorWidth / 2, pt.Y); case ConnectorOrientation.TopRight: return new PointBase(pt.X + port.ConnectorWidth, pt.Y); case ConnectorOrientation.Right: return new PointBase(pt.X + port.ConnectorWidth, pt.Y + port.ConnectorHeight / 2); case ConnectorOrientation.BottomRight: return new PointBase(pt.X + port.ConnectorWidth, pt.Y + port.ConnectorHeight); case ConnectorOrientation.Bottom: return new PointBase(pt.X + port.ConnectorWidth / 2, pt.Y + port.ConnectorHeight); case ConnectorOrientation.BottomLeft: return new PointBase(pt.X, pt.Y + port.ConnectorHeight); case ConnectorOrientation.Left: return new PointBase(pt.X, pt.Y + port.ConnectorHeight / 2); default: return pt; } } private PointBase? GetClosestPointTo(IEnumerable points, PointBase point) { var minDist = double.MaxValue; PointBase? minPoint = null; foreach (var pt in points) { var dist = pt.DistanceTo(point); if (dist < minDist) { minDist = dist; minPoint = pt; } } return minPoint; } #region 双击添加 protected override void ExecuteEditCommand(object param) { if (this.OutTextItem != null) return; AddText(""); } public void AddText(string text) { if (this.Parent is IDiagramViewModel) { var diagramVM = this.Parent as IDiagramViewModel; TextDesignerItemViewModel textitem = new TextDesignerItemViewModel(); textitem.ItemWidth = Double.NaN; textitem.ItemHeight = double.NaN; //if (this.PathMode == DrawMode.ConnectingLineBoundary.ToString()) //{ // var mid = (int)(ConnectionPoints.Count / 2); // var p = PathGenerators.SegmentMiddlePoint(ConnectionPoints[mid - 1], ConnectionPoints[mid]); // textitem.Left = this.Area.Left + p.X + 2; // textitem.Top = this.Area.Top + p.Y - 15; //} //else { textitem.Left = this.Area.Left + this.Area.Width / 2 - 16; textitem.Top = this.Area.Top + this.Area.Height / 2 - 5; } textitem.Watermark = null; textitem.ZIndex = diagramVM.Items.Count; textitem.ParentId = this.Id; textitem.ParentItem = this; textitem.ColorViewModel.FillColor = new ColorObject() { Color = Colors.White }; textitem.Text = text; diagramVM.DirectAddItemCommand.Execute(textitem); this.OutTextItem = textitem; } } public void OutTextItemLocation(RectangleBase oldArea, RectangleBase newArea) { if (this.OutTextItem is TextDesignerItemViewModel text) { var oldpoint = new PointBase(oldArea.Left + oldArea.Width / 2, oldArea.Top + oldArea.Height / 2); var newpoint = new PointBase(newArea.Left + newArea.Width / 2, newArea.Top + newArea.Height / 2); text.Left = text.Left + newpoint.X - oldpoint.X; text.Top = text.Top + newpoint.Y - oldpoint.Y; } } #endregion } }