dnl.utils.text.tree.TextTree Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of j-text-utils Show documentation

Gives text representations of common data views such as tree and table.

The newest version!

package dnl.utils.text.tree;

import java.io.PrintStream;

import javax.swing.tree.TreeModel;

/**
 * Implements a tree that renders into text.
 * 
 * @author Daniel Orr
 */
public class TextTree {

	private TreeModel model;
	private boolean showRoot = true;
	private StringBuilder sb;

	public TextTree(TreeModel model) {
		this.model = model;
	}

	/**
	 * @return true if this tree is set to show the root.
	 */
	public boolean isShowRoot() {
		return showRoot;
	}

	/**
	 * Sets wether to show the root node or not.
	 * 
	 * @param showRoot
	 */
	public void setShowRoot(boolean showRoot) {
		this.showRoot = showRoot;
	}

	/**
	 * Prints this tree to System.out.
	 */
	public void printTree() {
		System.out.println(toString());
	}

	/**
	 * Prints this tree to the given stream.
	 * 
	 * @param printStream
	 */
	public void printTree(PrintStream printStream) {
		printStream.println(toString());
	}

	@Override
	public String toString() {
		sb = new StringBuilder();
		Object root = model.getRoot();
		printNode(root, "", "", showRoot);
		return sb.toString();
	}

	private void printNode(Object node, String pref, String bud, boolean render) {
		if (render) {
			sb.append(pref);
			sb.append(bud);
			sb.append("-- ");
			sb.append(node.toString());
			sb.append("\n");
		}
		String pref1 = pref + "   ";
		String pref2 = pref + "   |";
		String pref3 = pref + "";
		int childCount = model.getChildCount(node);
		String ind;
		String bud2;
		if (childCount > 0) {
			for (int i = 0; i < childCount; i++) {
				boolean lastElement = i == childCount - 1;
				if (lastElement) {
					ind = pref1;
					bud2 = "`";
				} else {
					ind = pref2;
					bud2 = "";
				}
				printNode(model.getChild(node, i), !render ? pref3 : ind, !render ? "" : bud2, true);
			}
		}
	}

}