显示隐藏JTree节点

2019独角兽企业重金招聘Python工程师标准>>>

import java.awt.BorderLayout;
import java.awt.Dimension;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.Enumeration;

import javax.swing.JCheckBox;
import javax.swing.JOptionPane;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTree;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeModel;

class FilteredTree {
	private class PlainBelliedSneech {
		public String toString() {
			return "Plain Bellied Sneech";
		}
	}

	private class StarBelliedSneech {
		public String toString() {
			return "Star Bellied Sneech";
		}
	}

	private class FilteredTreeModel extends DefaultTreeModel {
		private boolean mShowStarBelliedSneeches = true;
		private DefaultMutableTreeNode mRoot;

		FilteredTreeModel(DefaultMutableTreeNode root) {
			super(root);
			mRoot = root;
		}

		public Object getChild(Object parent, int index) {
			DefaultMutableTreeNode node = (DefaultMutableTreeNode) parent;

			if (mShowStarBelliedSneeches)
				return node.getChildAt(index);

			int pos = 0;
			for (int i = 0, cnt = 0; i < node.getChildCount(); i++) {
				if (((DefaultMutableTreeNode) node.getChildAt(i)).getUserObject() instanceof PlainBelliedSneech) {
					if (cnt++ == index) {
						pos = i;
						break;
					}
				}
			}

			return node.getChildAt(pos);
		}

		public int getChildCount(Object parent) {
			DefaultMutableTreeNode node = (DefaultMutableTreeNode) parent;

			if (mShowStarBelliedSneeches)
				return node.getChildCount();

			int childCount = 0;
			Enumeration children = node.children();
			while (children.hasMoreElements()) {
				if (((DefaultMutableTreeNode) children.nextElement()).getUserObject() instanceof PlainBelliedSneech)
					childCount++;
			}

			return childCount;
		}

		public boolean getShowStarBelliedSneeches() {
			return mShowStarBelliedSneeches;
		}

		public void setShowStarBelliedSneeches(boolean showStarBelliedSneeches) {
			if (showStarBelliedSneeches != mShowStarBelliedSneeches) {
				mShowStarBelliedSneeches = showStarBelliedSneeches;
				Object[] path = { mRoot };
				int[] childIndices = new int[root.getChildCount()];
				Object[] children = new Object[root.getChildCount()];
				for (int i = 0; i < root.getChildCount(); i++) {
					childIndices[i] = i;
					children[i] = root.getChildAt(i);
				}
				fireTreeStructureChanged(this, path, childIndices, children);

			}
		}
	}

	private FilteredTree() {
		final DefaultMutableTreeNode root = new DefaultMutableTreeNode("Root");

		DefaultMutableTreeNode parent;
		DefaultMutableTreeNode child;

		for (int i = 0; i < 2; i++) {
			parent = new DefaultMutableTreeNode(new PlainBelliedSneech());
			root.add(parent);
			for (int j = 0; j < 2; j++) {
				child = new DefaultMutableTreeNode(new StarBelliedSneech());
				parent.add(child);
				for (int k = 0; k < 2; k++)
					child.add(new DefaultMutableTreeNode(new PlainBelliedSneech()));
			}
			for (int j = 0; j < 2; j++)
				parent.add(new DefaultMutableTreeNode(new PlainBelliedSneech()));

			parent = new DefaultMutableTreeNode(new StarBelliedSneech());
			root.add(parent);
			for (int j = 0; j < 2; j++) {
				child = new DefaultMutableTreeNode(new PlainBelliedSneech());
				parent.add(child);
				for (int k = 0; k < 2; k++)
					child.add(new DefaultMutableTreeNode(new StarBelliedSneech()));
			}
			for (int j = 0; j < 2; j++)
				parent.add(new DefaultMutableTreeNode(new StarBelliedSneech()));

		}

		final FilteredTreeModel model = new FilteredTreeModel(root);
		JTree tree = new JTree(model);
		tree.setShowsRootHandles(true);
		tree.putClientProperty("JTree.lineStyle", "Angled");
		tree.setRootVisible(false);

		JScrollPane sp = new JScrollPane(tree);
		sp.setPreferredSize(new Dimension(200, 400));

		final JCheckBox check = new JCheckBox("Show Star Bellied Sneeches");
		check.setSelected(model.getShowStarBelliedSneeches());

		check.addActionListener(new ActionListener() {
			public void actionPerformed(ActionEvent e) {
				model.setShowStarBelliedSneeches(check.isSelected());
			}
		});

		JPanel panel = new JPanel(new BorderLayout());
		panel.add(check, BorderLayout.NORTH);
		panel.add(sp, BorderLayout.CENTER);

		JOptionPane.showOptionDialog(null, panel, "Sneeches on Beeches", JOptionPane.DEFAULT_OPTION,
				JOptionPane.PLAIN_MESSAGE, null, new String[0], null);

		System.exit(0);
	}

	public static void main(String[] argv) {
		new FilteredTree();
	}
}

转载于:https://my.oschina.net/fbliujia/blog/73515