mytree.java

一个C语言子集的编译器

package com.king4solomon.homework.compiler.gui;

import javax.swing.*;
import javax.swing.tree.*;
import javax.swing.filechooser.*;
import javax.swing.event.*;
import java.io.*;
import java.awt.*;
import java.util.*;
import java.awt.event.*;

@SuppressWarnings("serial")
public class MyTree extends JTree implements TreeExpansionListener,
		TreeSelectionListener, MouseListener {
	protected DefaultTreeModel treeModel;

	protected FileSystemView fileSystemView; // 建立文件系统视类对象

	protected FileNode root;

	public MyTree(SourcePane source) {
		this.source = source;
		Font myFont = new Font("宋体", 11, 12);
		fileSystemView = FileSystemView.getFileSystemView();
		root = new FileNode(fileSystemView.getRoots()[0]);
		root.explore();
		treeModel = new DefaultTreeModel(root);
		setModel(treeModel); // 设定树形菜单
		addTreeExpansionListener(this); // 打开/关闭节点事件
		addTreeSelectionListener(this); // 选择的事件
		setCellRenderer(new MyTreeCellRenderer()); // 生成图标
		setFont(myFont);
		setRootVisible(true);
		setRowHeight(18);
		addMouseListener(this);
	}

	// 图标生成类
	protected class MyTreeCellRenderer extends JPanel implements
			TreeCellRenderer {

		BorderLayout borderLayout1 = new BorderLayout();

		JLabel label = new JLabel();

		public MyTreeCellRenderer() {
			setLayout(null);

			add(label);
			label.setBackground(UIManager.getColor("Tree.textBackground"));
			setBackground(UIManager.getColor("Tree.textBackground"));

		}

		public Dimension getPreferredSize() {
			Dimension labelDimension = label.getPreferredSize();
			return new Dimension(labelDimension.width, labelDimension.height);
		}

		public Component getTreeCellRendererComponent(JTree tree, Object value,
				boolean sel, boolean expanded, boolean leaf, int row,
				boolean hasFocus) {
			String stringValue = tree.convertValueToText(value, sel, expanded,
					leaf, row, hasFocus);
			setEnabled(tree.isEnabled());
			label.setFont(tree.getFont());
			// 设置图标为系统的文件类型图标
			FileSystemView fileSystemView = FileSystemView.getFileSystemView();
			label.setIcon(fileSystemView.getSystemIcon(((FileNode) value)
					.getFile()));
			label.setText(stringValue);
			return this;
		}

		public void doLayout() {
			Dimension labelDimension = label.getPreferredSize();
			label.setLocation(0, 0);
			label.setBounds(0, 0, labelDimension.width, labelDimension.height);
		}

	}

	// 节点张开事件
	public void treeExpanded(TreeExpansionEvent event) {
		// 判断是否是叶节点
		// if (this.getLastSelectedPathComponent() == null) {
		// System.out.println("ok");
		// return;
		// }
		setCursor(new Cursor(Cursor.WAIT_CURSOR));
		TreePath path = event.getPath();
		// System.out.println(path.toString());
		FileNode node = (FileNode) path.getLastPathComponent();
		node.explore();
		treeModel.nodeStructureChanged(node);
		setCursor(new Cursor(Cursor.DEFAULT_CURSOR));

	}

	// 节点闭合事件
	public void treeCollapsed(TreeExpansionEvent event) {

	}

	// 文件节点类
	protected class FileNode extends DefaultMutableTreeNode {

		private boolean isSelected = false;

		private boolean explored = false;

		public FileNode(File file) {
			this(file, false);
		}

		public FileNode(File file, boolean bool) {
			super(file);
			this.isSelected = bool;
		}

		//
		public boolean isSelected() {
			return isSelected;
		}

		public void setSelected(boolean isSelected) {
			this.isSelected = isSelected;
			if (children != null) {
				Enumeration enum1 = children.elements();
				while (enum1.hasMoreElements()) {
					FileNode node = (FileNode) enum1.nextElement();
					node.setSelected(isSelected);
				}
			}
		}

		public boolean getAllowsChildren() {
			return isDirectory();
		}

		public boolean isLeaf() {
			return !isDirectory();
		}

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

		public boolean isExplored() {
			return explored;
		}

		public void setExplored(boolean b) {
			explored = b;
		}

		public boolean isDirectory() {
			return getFile().isDirectory();
		}

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

		public void explore() {
			if (!isExplored()) {
				File file = getFile();
				File[] children = file.listFiles();
				if (children == null || children.length == 0)
					return;
				for (int i = 0; i < children.length; ++i) {
					if (children[i].getName().endsWith("txt")||children[i].isDirectory())
						add(new FileNode(children[i], isSelected));
				}
				explored = true;
			}
		}

	}

	public void valueChanged(TreeSelectionEvent e) {
		@SuppressWarnings("unused")
		String sFilePath = "";
		Object myobj = this.getLastSelectedPathComponent();
		if (myobj != null) {
			sFilePath = ((File) (((DefaultMutableTreeNode) (myobj))
					.getUserObject())).getPath();
		}
		// System.out.println(sFilePath);
	}

	public void mouseClicked(MouseEvent e) {
		int count = e.getClickCount();

		if (count >= 2) {
			TreePath path = this.getSelectionPath();
			if (path == null)
				return;
			DefaultMutableTreeNode selectedNode = (DefaultMutableTreeNode) path
					.getLastPathComponent();
			File f = (File) selectedNode.getUserObject();
			if (!f.isDirectory())
				source.open(f);

		} else {
			int x = e.getX();
			int y = e.getY();
			int row = this.getRowForLocation(x, y);
			TreePath path = this.getPathForRow(row);
			if (path != null) {
				FileNode node = (FileNode) path.getLastPathComponent();
				boolean isSelected = !(node.isSelected());
				node.setSelected(isSelected);
				((DefaultTreeModel) this.getModel()).nodeChanged(node);
			}
		}

	}

	public void mousePressed(MouseEvent e) {
	}

	public void mouseReleased(MouseEvent e) {
	}

	public void mouseEntered(MouseEvent e) {
	}

	public void mouseExited(MouseEvent e) {
	}

	private SourcePane source;
}