using System;
using System.Collections.Generic;
using System.Text;
using System.Windows.Forms;
using System.Drawing;
using System.Windows.Forms.VisualStyles;
using System.Diagnostics;
using System.Drawing.Drawing2D;
using Aga.Controls.Tree.NodeControls;
namespace Aga.Controls.Tree
{
public partial class TreeViewAdv
{
private void CreatePens()
{
CreateLinePen();
CreateMarkPen();
}
private void CreateMarkPen()
{
GraphicsPath path = new GraphicsPath();
path.AddLines(new Point[] { new Point(0, 0), new Point(1, 1), new Point(-1, 1), new Point(0, 0) });
CustomLineCap cap = new CustomLineCap(null, path);
cap.WidthScale = 1.0f;
_markPen = new Pen(_dragDropMarkColor, _dragDropMarkWidth);
_markPen.CustomStartCap = cap;
_markPen.CustomEndCap = cap;
}
private void CreateLinePen()
{
_linePen = new Pen(_lineColor);
_linePen.DashStyle = DashStyle.Dot;
}
protected override void OnPaint(PaintEventArgs e)
{
BeginPerformanceCount();
DrawContext context = new DrawContext();
context.Graphics = e.Graphics;
context.Font = this.Font;
context.Enabled = Enabled;
int y = 0;
int gridHeight = 0;
if (UseColumns)
{
DrawColumnHeaders(e.Graphics);
y += ColumnHeaderHeight;
if (Columns.Count == 0 || e.ClipRectangle.Height <= y)
return;
}
int firstRowY = _rowLayout.GetRowBounds(FirstVisibleRow).Y;
y -= firstRowY;
e.Graphics.ResetTransform();
e.Graphics.TranslateTransform(-OffsetX, y);
Rectangle displayRect = DisplayRectangle;
for (int row = FirstVisibleRow; row < RowCount; row++)
{
Rectangle rowRect = _rowLayout.GetRowBounds(row);
gridHeight += rowRect.Height;
if (rowRect.Y + y > displayRect.Bottom)
break;
else
DrawRow(e, ref context, row, rowRect);
}
if ((GridLineStyle & GridLineStyle.Vertical) == GridLineStyle.Vertical)
DrawVerticalGridLines(e.Graphics, firstRowY);
if (_dropPosition.Node != null && DragMode && HighlightDropPosition)
DrawDropMark(e.Graphics);
e.Graphics.ResetTransform();
DrawScrollBarsBox(e.Graphics);
if (DragMode && _dragBitmap != null)
e.Graphics.DrawImage(_dragBitmap, PointToClient(MousePosition));
EndPerformanceCount(e);
}
private void DrawRow(PaintEventArgs e, ref DrawContext context, int row, Rectangle rowRect)
{
TreeNodeAdv node = RowMap[row];
context.DrawSelection = DrawSelectionMode.None;
context.CurrentEditorOwner = _currentEditorOwner;
if (DragMode)
{
if ((_dropPosition.Node == node) && _dropPosition.Position == NodePosition.Inside && HighlightDropPosition)
context.DrawSelection = DrawSelectionMode.Active;
}
else
{
if (node.IsSelected && Focused)
context.DrawSelection = DrawSelectionMode.Active;
else if (node.IsSelected && !Focused && !HideSelection)
context.DrawSelection = DrawSelectionMode.Inactive;
}
context.DrawFocus = Focused && CurrentNode == node;
if (FullRowSelect)
{
context.DrawFocus = false;
if (context.DrawSelection == DrawSelectionMode.Active || context.DrawSelection == DrawSelectionMode.Inactive)
{
Rectangle focusRect = new Rectangle(OffsetX, rowRect.Y, ClientRectangle.Width, rowRect.Height);
if (context.DrawSelection == DrawSelectionMode.Active)
{
e.Graphics.FillRectangle(SystemBrushes.Highlight, focusRect);
context.DrawSelection = DrawSelectionMode.FullRowSelect;
}
else
{
e.Graphics.FillRectangle(SystemBrushes.InactiveBorder, focusRect);
context.DrawSelection = DrawSelectionMode.None;
}
}
}
if ((GridLineStyle & GridLineStyle.Horizontal) == GridLineStyle.Horizontal)
e.Graphics.DrawLine(SystemPens.InactiveBorder, 0, rowRect.Bottom, e.Graphics.ClipBounds.Right, rowRect.Bottom);
if (ShowLines)
DrawLines(e.Graphics, node, rowRect);
DrawNode(node, context);
}
private void DrawVerticalGridLines(Graphics gr, int y)
{
int x = 0;
foreach (TreeColumn c in Columns)
{
x += c.Width;
gr.DrawLine(SystemPens.InactiveBorder, x - 1, y, x - 1, gr.ClipBounds.Bottom);
}
}
private void DrawColumnHeaders(Graphics gr)
{
ReorderColumnState reorder = Input as ReorderColumnState;
int x = 0;
TreeColumn.DrawBackground(gr, new Rectangle(0, 0, ClientRectangle.Width + 2, ColumnHeaderHeight - 1), false, false);
gr.TranslateTransform(-OffsetX, 0);
foreach (TreeColumn c in Columns)
{
if (c.IsVisible)
{
Rectangle rect = new Rectangle(x, 0, c.Width, ColumnHeaderHeight - 1);
gr.SetClip(rect);
bool pressed = ((Input is ClickColumnState || reorder != null) && ((Input as ColumnState).Column == c));
c.Draw(gr, rect, Font, pressed, _hotColumn == c);
gr.ResetClip();
if (reorder != null && reorder.DropColumn == c)
TreeColumn.DrawDropMark(gr, rect);
x += c.Width;
}
}
if (reorder != null)
{
if (reorder.DropColumn == null)
TreeColumn.DrawDropMark(gr, new Rectangle(x, 0, 0, ColumnHeaderHeight));
gr.DrawImage(reorder.GhostImage, new Point(reorder.Location.X + + reorder.DragOffset, reorder.Location.Y));
}
}
public void DrawNode(TreeNodeAdv node, DrawContext context)
{
foreach (NodeControlInfo item in GetNodeControls(node))
{
context.Bounds = item.Bounds;
context.Graphics.SetClip(context.Bounds);
item.Control.Draw(node, context);
context.Graphics.ResetClip();
}
}
private void DrawScrollBarsBox(Graphics gr)
{
Rectangle r1 = DisplayRectangle;
Rectangle r2 = ClientRectangle;
gr.FillRectangle(SystemBrushes.Control,
new Rectangle(r1.Right, r1.Bottom, r2.Width - r1.Width, r2.Height - r1.Height));
}
private void DrawDropMark(Graphics gr)
{
if (_dropPosition.Position == NodePosition.Inside)
return;
Rectangle rect = GetNodeBounds(_dropPosition.Node);
int right = DisplayRectangle.Right - LeftMargin + OffsetX;
int y = rect.Y;
if (_dropPosition.Position == NodePosition.After)
y = rect.Bottom;
gr.DrawLine(_markPen, rect.X, y, right, y);
}
private void DrawLines(Graphics gr, TreeNodeAdv node, Rectangle rowRect)
{
if (UseColumns && Columns.Count > 0)
gr.SetClip(new Rectangle(0, rowRect.Y, Columns[0].Width, rowRect.Bottom));
TreeNodeAdv curNode = node;
while (curNode != _root && curNode != null)
{
int level = curNode.Level;
int x = (level - 1) * _indent + NodePlusMinus.ImageSize / 2 + LeftMargin;
int width = NodePlusMinus.Width - NodePlusMinus.ImageSize / 2;
int y = rowRect.Y;
int y2 = y + rowRect.Height;
if (curNode == node)
{
int midy = y + rowRect.Height / 2;
gr.DrawLine(_linePen, x, midy, x + width, midy);
if (curNode.NextNode == null)
y2 = y + rowRect.Height / 2;
}
if (node.Row == 0)
y = rowRect.Height / 2;
if (curNode.NextNode != null || curNode == node)
gr.DrawLine(_linePen, x, y, x, y2);
curNode = curNode.Parent;
}
gr.ResetClip();
}
#region Performance
private float _totalTime;
private int _paintCount;
[Conditional("PERF_TEST")]
private void BeginPerformanceCount()
{
_paintCount++;
TimeCounter.Start();
}
[Conditional("PERF_TEST")]
private void EndPerformanceCount(PaintEventArgs e)
{
float time = TimeCounter.Finish();
_totalTime += time;
string debugText = string.Format("FPS {0:0.0}; Avg. FPS {1:0.0}",
1 / time, 1 / (_totalTime / _paintCount));
e.Graphics.DrawString(debugText, Control.DefaultFont, Brushes.Gray,
new PointF(DisplayRectangle.Width - 150, DisplayRectangle.Height - 20));
}
#endregion
}
}