cn.featherfly.common.structure.tree.util.TreeNodeTestUtils Maven / Gradle / Ivy

Go to download

/**
 * @author 钟冀 - yufei
 *             Jul 17, 2009 
 */
package cn.featherfly.common.structure.tree.util;

import java.io.PrintStream;
import java.util.Iterator;

import cn.featherfly.common.structure.tree.Tree;
import cn.featherfly.common.structure.tree.TreeNode;


/**
 * 
 * 
 * 用来测试树的一些工具方法
 * 
 * 
 * @author 钟冀
 */
public final class TreeNodeTestUtils {
    
    private static final PrintStream DEFAULT_PRINT = System.out;
    
    private TreeNodeTestUtils() {
    }
    /**
     * 返回树的结构字符串
     * @param  泛型参数
     * @param tree 树
     * @return  树的结构字符串
     */
    public static  String constract(Tree tree) {
        return constract(tree.getRootNode());
    }
    /**
     * 返回树节点的结构字符串
     * @param  泛型参数
     * @param treeNode 树节点
     * @return  树节点的结构字符串
     */
    public static  String constract(TreeNode treeNode) {
        StringBuilder sb = new StringBuilder();
        constract(treeNode, sb);
        return sb.toString();
    }
    
    private static  void constract(TreeNode treeNode, StringBuilder sb) {
        StringBuffer space = new StringBuffer("");
        for (int i = 0; i < treeNode.getDepth(); i++) {
            if (i == (treeNode.getDepth() - 1)) {
                space = space.append("|_");
            } else {
                space = space.append(" ");
            }
        }
        sb.append(space + treeNode.getId().toString() + "\n");                
        if (!treeNode.isLeaf()) {
            Iterator> iter = treeNode.getChildNodes().iterator();
            TreeNode node = null;
            while (iter.hasNext()) {
                node = iter.next();
                constract(node, sb);
            }
        }
    }
    /**
     * 显示树的结构
     * @param  泛型
     * @param tree 树
     */
    public static  void show(Tree tree) {
        show(tree, DEFAULT_PRINT);
    }
    /**
     * 显示节点的结构
     * @param  泛型
     * @param node 树节点
     */
    public static  void show(TreeNode node) {
        show(node, DEFAULT_PRINT);
    }
    /**
     * 显示树的结构
     * @param  泛型
     * @param tree 树
     * @param print 打印流
     */
    public static  void show(Tree tree, PrintStream print) {
        show(tree.getRootNode(), print);
    }
    /**
     * 显示节点的结构
     * @param  泛型
     * @param node 树节点
     * @param print 打印流
     */
    public static  void show(TreeNode node, PrintStream print) {
        show(0, node, print);
    }

    private static  void show(int level, TreeNode treeNode, PrintStream print) {
        StringBuffer space = new StringBuffer();
        for (int i = 0; i < level; i++) {
            space = space.append(" ");
        }        
        print.println("level:" + level + " " + space + treeNode.getId() + "  ");
        
        if (!treeNode.isLeaf()) {
            Iterator> iter = treeNode.getChildNodes().iterator();
            TreeNode node = null;
            while (iter.hasNext()) {
                node = iter.next();
                show(level + 1, node, print);
            }
        }
    }
}