WinForm TreeView 三種狀態(tài)

昵稱10504424 2013-11-12

展開全文

private void treeView1_NodeMouseClick(object sender, TreeNodeMouseClickEventArgs e)
        {
            var node = e.Node;
            if (node == null) return;
            //根據(jù)父節(jié)點狀態(tài)，全選或全消子節(jié)點
            node.StateImageIndex = node.StateImageIndex == 1 ? 0 : 1;
            UpdateSubNodeCheckStatus(node);

            //確定子節(jié)點的父節(jié)點是否被選中（所有子節(jié)點選中時,，父節(jié)點才選中）
            UpdateParentNodeCheckStatus(node);

            //根節(jié)點級別選擇互斥
            if (node.StateImageIndex != 1) return;
            var rootNode = GetRootByNode(node);
            UncheckNextNode(rootNode);
            UncheckPrevNode(rootNode);
        }
        private static TreeNode GetRootByNode(TreeNode node)
        {
            while (true)
            {
                if (node.Level == 0)
                {
                    return node;
                }
                var parentNode = node.Parent;
                node = parentNode;
            }
        }

        private void UncheckPrevNode(TreeNode node)
        {
            if (node.PrevNode == null) return;
            node.PrevNode.StateImageIndex = 0;
            UpdateSubNodeCheckStatus(node.PrevNode);
            UncheckPrevNode(node.PrevNode);
        }

        private void UncheckNextNode(TreeNode node)
        {
            while (true)
            {
                if (node.NextNode == null) return;
                //node.NextNode.Checked = false;
                node.NextNode.StateImageIndex = 0;
                UpdateSubNodeCheckStatus(node.NextNode);
                node = node.NextNode;

            }
        }

        private static void UpdateParentNodeCheckStatus(TreeNode node)
        {
            var parentNode = node.Parent;
            if (parentNode == null) return;

            if (IsSubNodeAllCheck(parentNode))
            {
                parentNode.StateImageIndex = 1;
            }
            else if (IsSubNodeAllUncheck(parentNode))
            {
                parentNode.StateImageIndex = 0;
            }
            else
            {
                parentNode.StateImageIndex = 2;
            }

            UpdateParentNodeCheckStatus(parentNode);
        }

        private void UpdateSubNodeCheckStatus(TreeNode parentNode)
        {
            var node = parentNode;
            foreach (TreeNode subNode in node.Nodes)
            { 
                subNode.StateImageIndex = node.StateImageIndex;
                UpdateSubNodeCheckStatus(subNode);
            }
        }
        
        //判斷節(jié)點的子節(jié)點是否全選
        private static bool IsSubNodeAllCheck(TreeNode parentNode)
        {
            return parentNode.Nodes.Cast<TreeNode>().All(subNode => subNode.StateImageIndex==1);
        }

        //判斷同級的節(jié)點是否全不選
        private static bool IsSubNodeAllUncheck(TreeNode parentNode) 
        {
            return parentNode.Nodes.Cast<TreeNode>().All(subNode => subNode.StateImageIndex==0);
        }