ET9_WebGL_Base/CSharp/App/Modules/Tree/TreeViewModel.cs

using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.ComponentModel.Composition;
using Microsoft.Practices.Prism.Mvvm;

namespace Modules.Tree
{
    [Export(typeof (TreeViewModel)), PartCreationPolicy(CreationPolicy.NonShared)]
    public class TreeViewModel: BindableBase
    {
        private readonly ObservableCollection<TreeNodeViewModel> treeNodes =
                new ObservableCollection<TreeNodeViewModel>();

        private readonly Dictionary<int, TreeNodeViewModel> treeNodeDict = new Dictionary<int, TreeNodeViewModel>(); 

        public ObservableCollection<TreeNodeViewModel> TreeNodes
        {
            get
            {
                return this.treeNodes;
            }
        }

        public int TreeId { get; set; }

        public TreeViewModel(AllTreeViewModel allTreeViewModel)
        {
            this.AllTreeViewModel = allTreeViewModel;
            this.TreeId = ++allTreeViewModel.MaxTreeId;
            TreeNodeViewModel treeNodeViewModel = new TreeNodeViewModel(this, 300, 100);
            this.treeNodes.Add(treeNodeViewModel);
            this.treeNodeDict[treeNodeViewModel.Id] = treeNodeViewModel;

            var treeLayout = new TreeLayout(this);
            treeLayout.ExcuteLayout(this.Root);
        }

        public TreeViewModel(List<TreeNodeData> treeNodeDatas)
        {
            foreach (TreeNodeData treeNodeData in treeNodeDatas)
            {
                TreeNodeViewModel treeNodeViewModel = new TreeNodeViewModel(this, treeNodeData);
                this.treeNodes.Add(treeNodeViewModel);
                this.treeNodeDict[treeNodeViewModel.Id] = treeNodeViewModel;
            }
            var treeLayout = new TreeLayout(this);
            treeLayout.ExcuteLayout(this.Root);
        }

        public List<TreeNodeData> GetDatas()
        {
            List<TreeNodeData> treeNodeDatas = new List<TreeNodeData>();
            foreach (TreeNodeViewModel treeNodeViewModel in treeNodes)
            {
                treeNodeDatas.Add(treeNodeViewModel.Data);
            }
            return treeNodeDatas;
        }

        public AllTreeViewModel AllTreeViewModel { get; set; }

        public TreeNodeViewModel Root
        {
            get
            {
                return this.treeNodes.Count == 0? null : this.treeNodes[0];
            }
        }

        public TreeNodeViewModel Get(int id)
        {
            TreeNodeViewModel node;
            this.treeNodeDict.TryGetValue(id, out node);
            return node;
        }

        public void Add(TreeNodeViewModel treeNode, TreeNodeViewModel parent)
        {
            // 如果父节点是折叠的,需要先展开父节点
            if (parent != null && parent.IsFold)
            {
                this.UnFold(parent);
            }

            this.treeNodes.Add(treeNode);
            this.treeNodeDict[treeNode.Id] = treeNode;

            if (parent != null)
            {
                parent.Children.Add(treeNode.Id);
            }

            var treeLayout = new TreeLayout(this);
            treeLayout.ExcuteLayout(this.Root);
        }

        private void GetChildrenIdAndSelf(TreeNodeViewModel treeNodeViewModel, List<int> children)
        {
            children.Add(treeNodeViewModel.Id);
            this.GetAllChildrenId(treeNodeViewModel, children);
        }

        private void GetAllChildrenId(TreeNodeViewModel treeNodeViewModel, List<int> children)
        {
            foreach (int childId in treeNodeViewModel.Children)
            {
                TreeNodeViewModel child = this.Get(childId);
                children.Add(child.Id);
                this.GetAllChildrenId(child, children);
            }
        }

        public void Remove(TreeNodeViewModel treeNodeViewModel)
        {
            List<int> allId = new List<int>();
            this.GetChildrenIdAndSelf(treeNodeViewModel, allId);

            foreach (int childId in allId)
            {
                TreeNodeViewModel child = this.Get(childId);
                this.treeNodes.Remove(child);
                this.treeNodes.Remove(treeNodeViewModel);
            }

            TreeNodeViewModel parent = treeNodeViewModel.Parent;
            if (parent != null)
            {
                parent.Children.Remove(treeNodeViewModel.Id);
            }

            var treeLayout = new TreeLayout(this);
            treeLayout.ExcuteLayout(this.Root);
        }

        private void RecursionMove(
                TreeNodeViewModel treeNodeViewModel, double offsetX, double offsetY)
        {
            treeNodeViewModel.X += offsetX;
            treeNodeViewModel.Y += offsetY;
            foreach (var childId in treeNodeViewModel.Children)
            {
                TreeNodeViewModel child = this.Get(childId);
                this.RecursionMove(child, offsetX, offsetY);
            }
        }

        public void MoveToPosition(double offsetX, double offsetY)
        {
            this.RecursionMove(this.Root, offsetX, offsetY);
        }

        public void MoveToNode(TreeNodeViewModel from, TreeNodeViewModel to)
        {
            // from节点不能是to节点的父级节点
            TreeNodeViewModel tmpNode = to;
            while (tmpNode != null)
            {
                if (tmpNode.IsRoot)
                {
                    break;
                }
                if (tmpNode.Id == from.Id)
                {
                    return;
                }
                tmpNode = tmpNode.Parent;
            }

            if (from.IsFold)
            {
                this.UnFold(from);
            }

            if (to.IsFold)
            {
                this.UnFold(to);
            }
            from.Parent.Children.Remove(from.Id);
            to.Children.Add(from.Id);
            from.Parent = to;
            var treeLayout = new TreeLayout(this);
            treeLayout.ExcuteLayout(this.Root);
        }

        /// <summary>
        /// 折叠节点
        /// </summary>
        /// <param name="treeNodeViewModel"></param>
        public void Fold(TreeNodeViewModel treeNodeViewModel)
        {
            List<int> allChildId = new List<int>();
            this.GetAllChildrenId(treeNodeViewModel, allChildId);

            foreach (int childId in allChildId)
            {
                TreeNodeViewModel child = this.Get(childId);
                this.treeNodes.Remove(child);
            }

            treeNodeViewModel.IsFold = true;

            var treeLayout = new TreeLayout(this);
            treeLayout.ExcuteLayout(this.Root);
        }

        /// <summary>
        /// 展开节点,一级一级展开,一次只展开下层子节点,比如下层节点是折叠的,那下下层节点不展开
        /// </summary>
        /// <param name="treeNodeViewModel"></param>
        public void UnFold(TreeNodeViewModel treeNodeViewModel)
        {
            treeNodeViewModel.IsFold = false;

            List<int> allChildId = new List<int>();
            this.GetAllChildrenId(treeNodeViewModel, allChildId);

            foreach (int childId in allChildId)
            {
                TreeNodeViewModel child = this.Get(childId);
                this.treeNodes.Add(child);
            }

            var treeLayout = new TreeLayout(this);
            treeLayout.ExcuteLayout(this.Root);
        }
    }
}