ET9_WebGL_Dev/CSharp/Modules/BehaviorTree/BehaviorTreeView.xaml.cs

using System;
using System.ComponentModel.Composition;
using System.Windows;
using System.Windows.Input;
using Infrastructure;

namespace BehaviorTree
{
	/// <summary>
	/// BehaviorTreeView.xaml 的交互逻辑
	/// </summary>
	[ViewExport(RegionName = "TreeCanvasRegion")]
	[PartCreationPolicy(CreationPolicy.NonShared)]
	public partial class BehaviorTreeView
	{
		private const double DragThreshold = 5;

		private bool isControlDown;
		private bool isDragging;
		private bool isLeftButtonDown;
		private Point origMouseDownPoint;

		public BehaviorTreeView()
		{
			InitializeComponent();
		}

		[Import]
		private BehaviorTreeViewModel ViewModel
		{
			get
			{
				return DataContext as BehaviorTreeViewModel;
			}
			set
			{
				DataContext = value;
			}
		}

		private void MenuNewNode_Executed(object sender, ExecutedRoutedEventArgs e)
		{
			Point point = Mouse.GetPosition(listBox);
			var treeNode = new TreeNode(point.X, point.Y);

			// one root node
			if (ViewModel.TreeNodes.Count == 0)
			{
				ViewModel.Add(treeNode, null);
			}
			else
			{
				if (listBox.SelectedItem != null)
				{
					var treeNodeViewModel = listBox.SelectedItem as TreeNodeViewModel;
					ViewModel.Add(treeNode, treeNodeViewModel);
				}
			}
			listBox.SelectedItem = null;
			e.Handled = true;
		}

		private void MenuDeleteNode_Executed(object sender, ExecutedRoutedEventArgs e)
		{
			if (listBox.SelectedItem == null)
			{
				return;
			}
			var treeNodeViewModel = listBox.SelectedItem as TreeNodeViewModel;
			ViewModel.Remove(treeNodeViewModel);
			listBox.SelectedItem = null;
			e.Handled = true;
		}

		private void ListBoxItem_MouseDown(object sender, MouseButtonEventArgs e)
		{
			if (e.ChangedButton != MouseButton.Left)
			{
				return;
			}
			isLeftButtonDown = true;

			isControlDown = (Keyboard.Modifiers & ModifierKeys.Control) != 0;

			var item = (FrameworkElement) sender;
			var treeNodeViewModel = item.DataContext as TreeNodeViewModel;

			if (!isControlDown && !listBox.SelectedItems.Contains(treeNodeViewModel))
			{
				listBox.SelectedItems.Clear();
				listBox.SelectedItems.Add(treeNodeViewModel);
			}

			origMouseDownPoint = e.GetPosition(this);

			item.CaptureMouse();
			e.Handled = true;
		}

		private void ListBoxItem_MouseUp(object sender, MouseButtonEventArgs e)
		{
			if (!isLeftButtonDown)
			{
				isDragging = false;
				return;
			}

			var item = (FrameworkElement) sender;
			var treeNodeViewModel = item.DataContext as TreeNodeViewModel;

			if (isControlDown)
			{
				if (!listBox.SelectedItems.Contains(treeNodeViewModel))
				{
					listBox.SelectedItems.Add(treeNodeViewModel);
				}
				else
				{
					listBox.SelectedItems.Remove(treeNodeViewModel);
				}
			}
			else if (!isDragging)
			{
				if (listBox.SelectedItems.Count != 1 || listBox.SelectedItem != treeNodeViewModel)
				{
					listBox.SelectedItems.Clear();
					listBox.SelectedItem = treeNodeViewModel;
					listBox.SelectedItems.Add(treeNodeViewModel);
				}
			}

			isLeftButtonDown = false;
			isControlDown = false;
			isDragging = false;

			item.ReleaseMouseCapture();
			e.Handled = true;
		}

		private void ListBoxItem_MouseMove(object sender, MouseEventArgs e)
		{
			Point curMouseDownPoint;
			Vector dragDelta;
			if (isDragging)
			{
				curMouseDownPoint = e.GetPosition(this);
				dragDelta = curMouseDownPoint - origMouseDownPoint;

				origMouseDownPoint = curMouseDownPoint;

				foreach (TreeNodeViewModel selectedItem in listBox.SelectedItems)
				{
					selectedItem.X += dragDelta.X;
					selectedItem.Y += dragDelta.Y;
				}
				return;
			}

			if (!isLeftButtonDown)
			{
				return;
			}

			var item = (FrameworkElement) sender;
			var treeNodeViewModel = item.DataContext as TreeNodeViewModel;

			if (!listBox.SelectedItems.Contains(treeNodeViewModel))
			{
				return;
			}

			curMouseDownPoint = e.GetPosition(this);
			dragDelta = curMouseDownPoint - origMouseDownPoint;
			double dragDistance = Math.Abs(dragDelta.Length);
			if (dragDistance > DragThreshold)
			{
				isDragging = true;
			}
			e.Handled = true;
		}
	}
}