ET9_WebGL_Base/CSharp/App/Modules/BehaviorTreeModule/TreeNodeViewModel.cs

using System;
using System.Collections.Generic;
using Microsoft.Practices.Prism.Mvvm;

namespace Modules.BehaviorTreeModule
{
	public class TreeNodeViewModel: BindableBase
	{
		private static double width = 80;
		private static double height = 50;
		private readonly TreeViewModel treeViewModel;
		private readonly TreeNodeData data;
		private double x;
		private double y;
		private double connectorX2;
		private double connectorY2;
		private double prelim;
		private double modify;
		private double ancestorModify;
		private bool isFold;

		public TreeNodeViewModel(TreeViewModel treeViewModel, double x, double y)
		{
			this.treeViewModel = treeViewModel;
			this.x = x;
			this.y = y;
			this.data = new TreeNodeData();
			this.data.Id = ++treeViewModel.AllTreeViewModel.MaxNodeId;
			this.data.TreeId = treeViewModel.TreeId;
			this.data.Parent = 0;
			this.connectorX2 = 0;
			this.connectorY2 = Height / 2;
		}

		public TreeNodeViewModel(TreeViewModel treeViewModel, TreeNodeViewModel parent)
		{
			this.treeViewModel = treeViewModel;
			this.data = new TreeNodeData();
			this.data.Id = ++treeViewModel.AllTreeViewModel.MaxNodeId;
			this.data.TreeId = treeViewModel.TreeId;
			this.Parent = parent;

			this.connectorX2 = Width + this.Parent.X - this.X;
			this.connectorY2 = Height / 2 + this.Parent.Y - this.Y;
		}

		public TreeNodeViewModel(TreeViewModel treeViewModel, TreeNodeData data)
		{
			this.treeViewModel = treeViewModel;
			this.data = data;
			if (this.IsRoot)
			{
				this.x = 300;
				this.y = 100;
				this.connectorX2 = 0;
				this.connectorY2 = Height / 2;
			}
			else
			{
				this.connectorX2 = Width + this.Parent.X - this.X;
				this.connectorY2 = Height / 2 + this.Parent.Y - this.Y;
			}
		}

		public TreeNodeData Data
		{
			get
			{
				return this.data;
			}
		}

		public int Id
		{
			get
			{
				return this.data.Id;
			}
			set
			{
				if (this.data.Id == value)
				{
					return;
				}
				this.data.Id = value;
				this.OnPropertyChanged("Id");
			}
		}

		public string Comment
		{
			get
			{
				return this.data.Comment;
			}
			set
			{
				if (this.data.Comment == value)
				{
					return;
				}
				this.data.Comment = value;
				this.OnPropertyChanged("Comment");
			}
		}

		public static double Width
		{
			get
			{
				return width;
			}
			set
			{
				width = value;
			}
		}

		public static double Height
		{
			get
			{
				return height;
			}
			set
			{
				height = value;
			}
		}

		public bool IsRoot
		{
			get
			{
				return this.Parent == null;
			}
		}

		public double Prelim
		{
			get
			{
				return this.prelim;
			}
			set
			{
				this.prelim = value;
			}
		}

		public double Modify
		{
			get
			{
				return this.modify;
			}
			set
			{
				this.modify = value;
			}
		}

		public double X
		{
			get
			{
				return this.x;
			}
			set
			{
				if (Math.Abs(this.x - value) < 0.1)
				{
					return;
				}
				this.x = value;
				this.OnPropertyChanged("X");

				if (this.Parent != null)
				{
					this.ConnectorX2 = Width / 2 + this.Parent.X - this.X;
				}

				foreach (var childId in this.Children)
				{
					TreeNodeViewModel child = this.treeViewModel.Get(childId);
					child.ConnectorX2 = Width / 2 + this.X - child.X;
				}
			}
		}

		public double Y
		{
			get
			{
				return this.y;
			}
			set
			{
				if (Math.Abs(this.Y - value) < 0.1)
				{
					return;
				}

				this.y = value;
				this.OnPropertyChanged("Y");

				if (this.Parent != null)
				{
					this.ConnectorY2 = Height + this.Parent.Y - this.Y;
				}

				foreach (var childId in this.Children)
				{
					TreeNodeViewModel child = this.treeViewModel.Get(childId);
					child.ConnectorY2 = Height + this.Y - child.Y;
				}
			}
		}

		public double ConnectorX1
		{
			get
			{
				return Width / 2;
			}
		}

		public double ConnectorY1
		{
			get
			{
				return 0;
			}
		}

		public double ConnectorX2
		{
			get
			{
				return this.IsRoot? Width / 2 : this.connectorX2;
			}
			set
			{
				this.SetProperty(ref this.connectorX2, value);
			}
		}

		public double ConnectorY2
		{
			get
			{
				return this.IsRoot? 0 : this.connectorY2;
			}
			set
			{
				this.SetProperty(ref this.connectorY2, value);
			}
		}

		public int Type
		{
			get
			{
				return this.data.Type;
			}
			set
			{
				if (this.data.Type == value)
				{
					return;
				}
				this.data.Type = value;
				this.OnPropertyChanged("Type");
			}
		}

		public List<string> Args
		{
			get
			{
				return this.data.Args;
			}
			set
			{
				if (this.data.Args == value)
				{
					return;
				}
				this.data.Args = value;
				this.OnPropertyChanged("Args");
			}
		}

		public int TreeId
		{
			get
			{
				return this.data.TreeId;
			}
		}

		public TreeNodeViewModel Parent
		{
			get
			{
				if (this.data.Parent == 0)
				{
					return null;
				}
				TreeNodeViewModel parent = this.treeViewModel.Get(this.data.Parent);
				return parent;
			}
			set
			{
				if (value == null)
				{
					this.data.Parent = 0;
				}
				this.data.Parent = value.Id;
			}
		}

		/// <summary>
		/// 节点是否折叠
		/// </summary>
		public bool IsFold
		{
			get
			{
				return this.isFold;
			}
			set
			{
				if (this.isFold == value)
				{
					return;
				}
				this.isFold = value;
				this.OnPropertyChanged("IsFold");
			}
		}

		public List<int> Children
		{
			get
			{
				if (this.isFold)
				{
					return new List<int>();
				}
				return this.data.Children;
			}
		}

		public TreeNodeViewModel LeftSibling
		{
			get
			{
				if (this.IsRoot)
				{
					return null;
				}

				int index = this.Parent.Children.IndexOf(this.Id);
				return index == 0? null : this.treeViewModel.Get(this.Parent.Children[index - 1]);
			}
		}

		public TreeNodeViewModel LastChild
		{
			get
			{
				if (this.Children.Count == 0)
				{
					return null;
				}

				int maxIndex = this.Children.Count - 1;
				return this.treeViewModel.Get(this.Children[maxIndex]);
			}
		}

		public TreeNodeViewModel FirstChild
		{
			get
			{
				return this.Children.Count == 0? null : this.treeViewModel.Get(this.Children[0]);
			}
		}

		public bool IsLeaf
		{
			get
			{
				return this.Children.Count == 0;
			}
		}

		public double AncestorModify
		{
			get
			{
				return this.ancestorModify;
			}
			set
			{
				this.ancestorModify = value;
			}
		}
	}
}