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;
//	}
	
}