filetreenode.java

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

package fileTree;

import java.io.File;

import javax.swing.tree.DefaultMutableTreeNode;

public class FileTreeNode extends DefaultMutableTreeNode implements ExplorerTreeNode{
	
	// 标识节点是否已经展开过
	private boolean hasExpanded = false;

	public FileTreeNode(File file) {
		setUserObject(file);
	}
	
	public boolean isLeaf() {
		return !isDirectory();
	}

	public File getFile() {
		return (File) getUserObject();
	}

	public boolean hasExpanded() {
		return hasExpanded;
	}

	public boolean isDirectory() {
		File file = getFile();
		return file.isDirectory();
	}

	public String toString() {
		File file = getFile();
		String filename = file.toString();
		int index = filename.lastIndexOf("\\");
		return (index != -1 && index != filename.length() - 1) ? filename.substring(index + 1) : filename;
	}

	public String getString() {
		File file = getFile();
		String filename = file.getAbsolutePath();
		return filename;
	}

	/**
	 * 展开节点
	 */
	public void expand() {
		if (!isDirectory()) {
			return;
		}
		if (!hasExpanded()) {
			File file = getFile();
			File[] children = file.listFiles();
			FileTreeNode treeNode;
			for (int i = 0; i < children.length; ++i) {
				// 如果没有这个判断条件，则不显示所有文件，只显示路径文件
				if (!children[i].isHidden()) { // 不显示隐藏文件
					treeNode = new FileTreeNode(children[i]);
					add(treeNode);
				}
			}

			hasExpanded = true;
		}
	}
}