ET9_WebGL_Base/CSharp/Modules/BehaviorTree/BehaviorTreeViewModel.cs

using System.ComponentModel.Composition;
using Microsoft.Practices.Prism.ViewModel;
using NLog;
using System.Windows.Input;
using Microsoft.Practices.Prism.Commands;
using System.Collections.ObjectModel;
using System.Windows;
using Google.ProtocolBuffers;

namespace BehaviorTree
{
	[Export(typeof(BehaviorTreeViewModel))]
	[PartCreationPolicy(CreationPolicy.NonShared)]
	class BehaviorTreeViewModel
	{
		private ObservableCollection<TreeNodeViewModel> treeNodes = new ObservableCollection<TreeNodeViewModel>();

		public ObservableCollection<TreeNodeViewModel> TreeNodes
		{
			get
			{
				return treeNodes;
			}
			set
			{
				treeNodes = value;
			}
		}

		public void Add(TreeNode treeNode, TreeNodeViewModel parent)
		{
			var treeNodeViewModel = new TreeNodeViewModel(treeNode, parent);
			treeNodes.Add(treeNodeViewModel);
			if (parent != null)
			{
				parent.Children.Add(treeNodeViewModel);
			}
		}

		public void Remove(TreeNodeViewModel treeNodeViewModel)
		{
			for (int i = 0; i < treeNodeViewModel.Children.Count; ++i)
			{
				Remove(treeNodeViewModel.Children[0]);
			}
			treeNodeViewModel.Parent.Children.Remove(treeNodeViewModel);
			treeNodes.Remove(treeNodeViewModel);
		}
	}
}