using System;
using System.Drawing;
using System.Drawing.Drawing2D;
namespace DevComponents.Tree
{
///
/// Represents the connector display for map type.
///
public class MapConnectorDisplay:NodeConnectorDisplay
{
private int m_RootNodeConnectorWidth=4;
public MapConnectorDisplay()
{
}
///
/// Draws connector line between two nodes.
///
/// Connector context information.
public override void DrawConnector(NodeConnectorDisplayInfo info)
{
if(info.LineColor.IsEmpty && info.FillColor.IsEmpty)
return;
DrawCurveConnector(info);
}
private void DrawCurveConnector(NodeConnectorDisplayInfo info)
{
Point pStart=this.GetStartPoint(info);
Point pEnd=this.GetEndPoint(info);
pStart.Offset(info.Offset.X,info.Offset.Y);
pEnd.Offset(info.Offset.X,info.Offset.Y);
// Determine whether curve can be drawn
if(Math.Abs(pStart.X-pEnd.X)<=6 || Math.Abs(pStart.Y-pEnd.Y)<=10)
{
using(Pen pen=this.GetCurvePen(info))
{
info.Graphics.DrawLine(pen,pStart,pEnd);
}
}
else
{
// Create two points in between the start and end point
Point[] p=new Point[4];
int xMulti=1, yMulti=1;
// used for direction control
if(pStart.X>pEnd.X)
xMulti=-1;
if(pStart.Y>pEnd.Y)
yMulti=-1;
p[1].X=pStart.X+(int)(Math.Abs(pStart.X-pEnd.X)*.15f*xMulti);
p[1].Y=pStart.Y+(int)(Math.Abs(pStart.Y-pEnd.Y)*.15f*yMulti);
p[2].X=pStart.X+(int)(Math.Abs(pStart.X-pEnd.X)*.5f*xMulti);
p[2].Y=pEnd.Y-(int)(yMulti*(Math.Abs(pStart.Y-pEnd.Y)*.15f));
p[0]=pStart;
p[3]=pEnd;
if(info.IsRootNode && m_RootNodeConnectorWidth>1)
{
GraphicsPath path=new GraphicsPath();
path.AddCurve(p,.5f);
path.AddLine(p[0].X,p[0].Y,p[0].X+m_RootNodeConnectorWidth*xMulti,p[0].Y);
p[0].X+=(m_RootNodeConnectorWidth*xMulti);
p[1].X+=(m_RootNodeConnectorWidth*xMulti);
p[2].X+=(m_RootNodeConnectorWidth*xMulti);
p[3].Y-=yMulti;
path.AddCurve(p,.5f);
path.AddLine(p[3].X,p[3].Y,p[3].X,p[3].Y+yMulti);
path.CloseAllFigures();
using(Brush brush=this.GetCurveBrush(info))
{
info.Graphics.FillPath(brush,path);
}
}
else
{
using(Pen pen=this.GetCurvePen(info))
{
info.Graphics.DrawCurve(pen,p,.5f);
}
}
}
}
private Pen GetCurvePen(NodeConnectorDisplayInfo info)
{
return new Pen(info.LineColor,1);
}
private Brush GetCurveBrush(NodeConnectorDisplayInfo info)
{
return new SolidBrush(info.LineColor);
}
private Point GetStartPoint(NodeConnectorDisplayInfo info)
{
Point p=Point.Empty;
if(info.IsRootNode)
{
int toMidPoint=info.ToNode.Bounds.Top+info.ToNode.Bounds.Height/2;
if(info.FromNode.Bounds.Top>toMidPoint)
p=new Point(info.FromNode.Bounds.Left+info.FromNode.Bounds.Width/2,info.FromNode.Bounds.Top);
else if(info.FromNode.Bounds.Bottomtarget.Bounds.Left)
return true;
return false;
}
}
}