treeview.java

JTree的视图展示：记录树的展开状况

package fileTree;

import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;

import javax.swing.tree.TreeNode;

/**
 * 树的视图类
 * @author hyj
 *
 */
public class TreeView {
	
	// 当前树的根节点
	private TreeNode rootNode = null;
	
	// 状态为展开的节点集
	private Set hasExpandedNode = new HashSet();
	
	// 强制展开的节点
	private TreeNode forceExpandNode = null;

	public TreeView() {
	}
	
	public TreeView(TreeNode root) {
		this.rootNode = root;
	}
	
	public TreeView(TreeNode root, TreeNode forceExpandNode) {
		this.rootNode = root;
		this.forceExpandNode = forceExpandNode;
	}

	public TreeNode getRootNode() {
		return rootNode;
	}

	public void setRootNode(TreeNode rootNode) {
		this.rootNode = rootNode;
	}


	public TreeNode getForceExpandNode() {
		return forceExpandNode;
	}

	public void setForceExpandNode(TreeNode forceExpandNode) {
		this.forceExpandNode = forceExpandNode;
	}

	public Set getHasExpandedNode() {
		return hasExpandedNode;
	}

	public void setHasExpandedNode(Set hasExpandedNode) {
		this.hasExpandedNode = hasExpandedNode;
	}

	/**
	 * 
	 * @param obj
	 */
	public void addExpandNodeToSet(Object obj){
		hasExpandedNode.add(obj);
	}
	
	public boolean removeExpandFromSet(Object obj){
		return hasExpandedNode.remove(obj);
	}

	public int getExpandNodeNum(){
		return hasExpandedNode.size();
	}
	
	public void clearExpandNodeSet(){
		hasExpandedNode.clear();
	}
	
//	public String toString() {
//		StringBuffer buffer = new StringBuffer("状态为展开的节点集:");
//		for (Iterator iter = hasExpandedNode.iterator(); iter.hasNext();) {
//			buffer.append(iter.next() + " , ");
//		}
//		
//		return "根节点:"+rootNode.toString() + "\n" + buffer
//				+ "\n" + "强制展开的节点:" + forceExpandNode;
//	}
	
}