using System;

 using System.Collections.Generic;

 using System.Text;

 namespace Aga.Controls.Tree

 {

 	internal class InputWithShift: NormalInputState

 	{

 		public InputWithShift(TreeViewAdv tree): base(tree)

 		{

 		}

 		protected override void FocusRow(TreeNodeAdv node)

 		{

 			Tree.SuspendSelectionEvent = true;

 			try

 			{

 				if (Tree.SelectionMode == TreeSelectionMode.Single || Tree.SelectionStart == null)

 					base.FocusRow(node);

 				else if (CanSelect(node))

 				{

 					SelectAllFromStart(node);

 					Tree.CurrentNode = node;

 					Tree.ScrollTo(node);

 				}

 			}

 			finally

 			{

 				Tree.SuspendSelectionEvent = false;

 			}

 		}

 		protected override void DoMouseOperation(TreeNodeAdvMouseEventArgs args)

 		{

 			if (Tree.SelectionMode == TreeSelectionMode.Single || Tree.SelectionStart == null)

 			{

 				base.DoMouseOperation(args);

 			}

 			else if (CanSelect(args.Node))

 			{

 				Tree.SuspendSelectionEvent = true;

 				try

 				{

 					SelectAllFromStart(args.Node);

 				}

 				finally

 				{

 					Tree.SuspendSelectionEvent = false;

 				}

 			}

 		}

 		protected override void MouseDownAtEmptySpace(TreeNodeAdvMouseEventArgs args)

 		{

 		}

 		private void SelectAllFromStart(TreeNodeAdv node)

 		{

 			Tree.ClearSelectionInternal();

 			int a = node.Row;

 			int b = Tree.SelectionStart.Row;

 			for (int i = Math.Min(a, b); i <= Math.Max(a, b); i++)

 			{

 				if (Tree.SelectionMode == TreeSelectionMode.Multi || Tree.RowMap[i].Parent == node.Parent)

 					Tree.RowMap[i].IsSelected = true;

 			}

 		}

 	}

 }