SHJX_GM/SHJX.Service.WorkFlowEdit/Controls/Links/SegmentLink.cs

using System.Linq;
using System.Windows;
using System.Windows.Media;

namespace SHJX.Service.WorkFlowEdit
{
    public class SegmentLink : LinkBase
    {
        static SegmentLink()
        {
            DefaultStyleKeyProperty.OverrideMetadata(
                typeof(SegmentLink), new FrameworkPropertyMetadata(typeof(LinkBase)));
        }

        public override void UpdatePath()
        {
            var linePoints = CalculateSegments();
            if (CheckPoints(linePoints))
            {
                CalculatePositions(linePoints);
                PathGeometry geometry = new PathGeometry();
                PathFigure figure = new PathFigure
                {
                    StartPoint = linePoints[0]
                };
                figure.Segments.Add(new PolyLineSegment(linePoints.Skip(1), true));
                geometry.Figures.Add(figure);
                this.PathGeometry = geometry;
            }
            else
                this.PathGeometry = null;
        }

        protected virtual Point[] CalculateSegments()
        {
            var res = GetEndPoinds();
            if (res != null)
                UpdateEdges(res);
            return res;
        }

        protected Point[] GetEndPoinds()
        {
            Point tc, sc;
            if (Target != null)
                tc = Target.Center;
            else if (TargetPoint != null)
                tc = TargetPoint.Value;
            else
                return null;

            if (Source != null)
                sc = Source.Center;
            else if (SourcePoint != null)
                sc = SourcePoint.Value;
            else
                return null;

            var linePoints = new Point[2];
            linePoints[0] = sc;
            linePoints[1] = tc;
            return linePoints;
        }

        protected void UpdateEdges(Point[] linePoints)
        {
            if (linePoints.Length >= 2)
            {
                if (Source != null)
                    linePoints[0] = Source.GetEdgePoint(linePoints[1]);
                if (Target != null)
                    linePoints[linePoints.Length - 1] = Target.GetEdgePoint(linePoints[linePoints.Length - 2]);
            }
        }

        protected virtual void CalculatePositions(Point[] linePoints)
        {
            StartPoint = linePoints[0];
            EndPoint = linePoints[linePoints.Length - 1];
            StartCapAngle = GeometryHelper.NormalAngle(linePoints[0], linePoints[1]);
            EndCapAngle = GeometryHelper.NormalAngle(linePoints[linePoints.Length - 2], linePoints[linePoints.Length - 1]);
            var mid = (int)(linePoints.Length / 2);
            var p = GeometryHelper.SegmentMiddlePoint(linePoints[mid - 1], linePoints[mid]);
            LabelPosition = new Point(p.X, p.Y - 15);
        }

        private bool CheckPoints(Point[] linePoints)
        {
            if (linePoints != null && linePoints.Length >= 2)
            {
                foreach (var p in linePoints)
                    if (double.IsNaN(p.X) || double.IsNaN(p.Y))
                        return false;
                return true;
            }
            return false;
        }
    }
}