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

namespace Tree
{
    public class TreeLayout
    {
        private readonly TreeViewModel treeViewModel;
        private const double XGap = 20;
        private const double YGap = 10;
        private double rootOrigX;
        private double rootOrigY;
        private double rootOffsetX;
        private double rootOffsetY;

        public TreeLayout(TreeViewModel treeViewModel)
        {
            this.treeViewModel = treeViewModel;
        }

        private TreeNodeViewModel LeftMostOffspring(
                TreeNodeViewModel treeNode, int currentLevel, int searchLevel)
        {
            if (currentLevel == searchLevel)
            {
                return treeNode;
            }
            for (int i = 0; i < treeNode.Children.Count; ++i)
            {
                var child = this.treeViewModel.Get(treeNode.Children[i]);
                child.AncestorModify = treeNode.Modify + treeNode.AncestorModify;
                var offspring = LeftMostOffspring(child, currentLevel + 1, searchLevel);
                if (offspring == null)
                {
                    continue;
                }
                return offspring;
            }
            return null;
        }

        private TreeNodeViewModel RightMostOffspring(
                TreeNodeViewModel treeNode, int currentLevel, int searchLevel)
        {
            if (currentLevel == searchLevel)
            {
                return treeNode;
            }
            for (int i = treeNode.Children.Count - 1; i >= 0; --i)
            {
                var child = this.treeViewModel.Get(treeNode.Children[i]);
                child.AncestorModify = treeNode.Modify + treeNode.AncestorModify;
                var offspring = RightMostOffspring(child, currentLevel + 1, searchLevel);
                if (offspring == null)
                {
                    continue;
                }
                return offspring;
            }
            return null;
        }

        private void AjustSubTreeGap(TreeNodeViewModel left, TreeNodeViewModel right)
        {
            double offset = 0;
            TreeNodeViewModel tLeft = left;
            TreeNodeViewModel tRight = right;
            left.AncestorModify = 0;
            right.AncestorModify = 0;
            for (int i = 0; tLeft != null && tRight != null; ++i)
            {
                double tGap = (tRight.Prelim + tRight.AncestorModify) -
                              (tLeft.Prelim + tLeft.AncestorModify);
                if (XGap + TreeNodeViewModel.Width - tGap > offset)
                {
                    offset = XGap + TreeNodeViewModel.Width - tGap;
                }
                tLeft = RightMostOffspring(left, 0, i + 1);
                tRight = LeftMostOffspring(right, 0, i + 1);
            }
            right.Modify += offset;
            right.Prelim += offset;
        }

        private void AjustTreeGap(TreeNodeViewModel treeNode)
        {
            for (int i = 0; i < treeNode.Children.Count - 1; ++i)
            {
                for (int j = i + 1; j < treeNode.Children.Count; ++j)
                {
                    var left = this.treeViewModel.Get(treeNode.Children[i]);
                    var right = this.treeViewModel.Get(treeNode.Children[j]);
                    AjustSubTreeGap(left, right);
                }
            }
        }

        private void CalculatePrelimAndModify(TreeNodeViewModel treeNode)
        {
            foreach (int childId in treeNode.Children)
            {
                TreeNodeViewModel child = this.treeViewModel.Get(childId);
                CalculatePrelimAndModify(child);
            }

            double prelim = 0;
            double modify = 0;

            if (treeNode.IsLeaf)
            {
                if (treeNode.LeftSibling == null)
                {
                    // 如果没有左邻居，不需要设置modify
                    prelim = 0;
                }
                else
                {
                    prelim = treeNode.LeftSibling.Prelim + TreeNodeViewModel.Width + XGap;
                }
            }
            else
            {
                // 调整子树间的间距
                AjustTreeGap(treeNode);
                double childrenCenter = (treeNode.FirstChild.Prelim + treeNode.LastChild.Prelim) / 2;
                if (treeNode.LeftSibling == null)
                {
                    // 如果没有左邻居，不需要设置modify
                    prelim = childrenCenter;
                }
                else
                {
                    prelim = treeNode.LeftSibling.Prelim + TreeNodeViewModel.Width + XGap;
                    modify = prelim - childrenCenter;
                }
            }
            treeNode.Prelim = prelim;
            treeNode.Modify = modify;

            // Log.Debug("Id: " + treeNode.Id + " Prelim: " + treeNode.Prelim + " Modify: " +
            // 	treeNode.Modify);
        }

        private void CalculateRelativeXAndY(
                TreeNodeViewModel treeNode, int level, double totalModify)
        {
            foreach (int childId in treeNode.Children)
            {
                TreeNodeViewModel child = this.treeViewModel.Get(childId);
                CalculateRelativeXAndY(child, level + 1, treeNode.Modify + totalModify);
            }
            if (treeNode.IsLeaf)
            {
                treeNode.X = treeNode.Prelim + totalModify;
            }
            else
            {
                treeNode.X = (treeNode.FirstChild.X + treeNode.LastChild.X) / 2;
            }
            treeNode.Y = level * (TreeNodeViewModel.Height + YGap);
        }

        private void FixXAndY(TreeNodeViewModel treeNode)
        {
            treeNode.X += rootOffsetX;
            treeNode.Y += rootOffsetY;
            foreach (var childId in treeNode.Children)
            {
                TreeNodeViewModel child = this.treeViewModel.Get(childId);
                FixXAndY(child);
            }
        }

        public void ExcuteLayout(TreeNodeViewModel root)
        {
            if (root == null)
            {
                return;
            }
            rootOrigX = root.X;
            rootOrigY = root.Y;
            CalculatePrelimAndModify(root);
            CalculateRelativeXAndY(root, 0, 0);

            rootOffsetX = rootOrigX - root.X;
            rootOffsetY = rootOrigY - root.Y;
            FixXAndY(root);
        }
    }
}