using System;

 using System.Collections.Generic;

 using System.Text;

 using System.Collections;

 namespace Aga.Controls.Tree

 {

 	public class SortedTreeModel: TreeModelBase

 	{

 		private ITreeModel _innerModel;

 		public ITreeModel InnerModel

 		{

 			get { return _innerModel; }

 		}

 		private IComparer _comparer;

 		public IComparer Comparer

 		{

 			get { return _comparer; }

 			set 

 			{ 

 				_comparer = value;

 				OnStructureChanged(new TreePathEventArgs(TreePath.Empty));

 			}

 		}

 		public SortedTreeModel(ITreeModel innerModel)

 		{

 			_innerModel = innerModel;

 			_innerModel.NodesChanged += new EventHandler<TreeModelEventArgs>(_innerModel_NodesChanged);

 			_innerModel.NodesInserted += new EventHandler<TreeModelEventArgs>(_innerModel_NodesInserted);

 			_innerModel.NodesRemoved += new EventHandler<TreeModelEventArgs>(_innerModel_NodesRemoved);

 			_innerModel.StructureChanged += new EventHandler<TreePathEventArgs>(_innerModel_StructureChanged);

 		}

 		void _innerModel_StructureChanged(object sender, TreePathEventArgs e)

 		{

 			OnStructureChanged(e);

 		}

 		void _innerModel_NodesRemoved(object sender, TreeModelEventArgs e)

 		{

 			OnStructureChanged(new TreePathEventArgs(e.Path));

 		}

 		void _innerModel_NodesInserted(object sender, TreeModelEventArgs e)

 		{

 			OnStructureChanged(new TreePathEventArgs(e.Path));

 		}

 		void _innerModel_NodesChanged(object sender, TreeModelEventArgs e)

 		{

 			OnStructureChanged(new TreePathEventArgs(e.Path));

 		}

 		public override IEnumerable GetChildren(TreePath treePath)

 		{

 			if (Comparer != null)

 			{

 				ArrayList list = new ArrayList();

 				IEnumerable res = InnerModel.GetChildren(treePath);

 				if (res != null)

 				{

 					foreach (object obj in res)

 						list.Add(obj);

 					list.Sort(Comparer);

 					return list;

 				}

 				else

 					return null;

 			}

 			else

 				return InnerModel.GetChildren(treePath);

 		}

 		public override bool IsLeaf(TreePath treePath)

 		{

 			return InnerModel.IsLeaf(treePath);

 		}

 	}

 }