JTree,将树的每个节点设置成不同的图标

http://xieruilin.iteye.com/blog/717858

 

MyDefaultTreeCellRenderer.java

Java代码

1. import java.awt.Component; 
3. import javax.swing.ImageIcon; 
4. import javax.swing.JTree; 
5. import javax.swing.tree.DefaultMutableTreeNode; 
6. import javax.swing.tree.DefaultTreeCellRenderer; 
8. /**
9. * 自定义树描述类,将树的每个节点设置成不同的图标
10. * @author RuiLin.Xie - xKF24276
11. *
12. */
13. public class MyDefaultTreeCellRenderer extends DefaultTreeCellRenderer 
14. { 
15. /**
16.      * ID
17.      */
18. private static final long   serialVersionUID    = 1L; 
20. /**
21.      * 重写父类DefaultTreeCellRenderer的方法
22.      */
23. @Override
24. public Component getTreeCellRendererComponent(JTree tree, Object value, 
25. boolean sel, boolean expanded, boolean leaf, int row, 
26. boolean hasFocus) 
27.     { 
29. //执行父类原型操作
30. super.getTreeCellRendererComponent(tree, value, sel, expanded, leaf, 
31.                 row, hasFocus); 
33.         setText(value.toString()); 
35. if (sel) 
36.         { 
37.             setForeground(getTextSelectionColor()); 
38.         } 
39. else
40.         { 
41.             setForeground(getTextNonSelectionColor()); 
42.         } 
44. //得到每个节点的TreeNode
45.         DefaultMutableTreeNode node = (DefaultMutableTreeNode) value; 
47. //得到每个节点的text
48.         String str = node.toString();        
50. //判断是哪个文本的节点设置对应的值（这里如果节点传入的是一个实体,则可以根据实体里面的一个类型属性来显示对应的图标）
51. if (str == "a") 
52.         { 
53. this.setIcon(new ImageIcon("treeimg/a.GIF")); 
54.         } 
55. if (str == "b") 
56.         { 
57. this.setIcon(new ImageIcon("treeimg/b.GIF")); 
58.         } 
59. if (str == "c") 
60.         { 
61. this.setIcon(new ImageIcon("treeimg/c.GIF")); 
62.         } 
64. return this; 
65.     } 
67. } 

TreeIconDemo.java

Java代码

1. import java.awt.BorderLayout; 
2. import javax.swing.JFrame; 
3. import javax.swing.JScrollPane; 
4. import javax.swing.JTree; 
5. import javax.swing.tree.DefaultMutableTreeNode; 
6. import javax.swing.tree.DefaultTreeModel; 
8. public class TreeIconDemo extends JFrame 
9. { 
11. /**
12.      * ID
13.      */
14. private static final long   serialVersionUID    = 1L; 
16. private JTree   tree; 
18. /**
19.      * Launch the application
20.      *
21.      * @param args
22.      */
23. public static void main(String args[]) 
24.     { 
25. try
26.         { 
27.             TreeIconDemo frame = new TreeIconDemo(); 
28.             frame.setVisible(true); 
29.         } 
30. catch (Exception e) 
31.         { 
32.             e.printStackTrace(); 
33.         } 
34.     } 
36. /**
37.      * Create the frame
38.      */
39. public TreeIconDemo() 
40.     { 
41. super(); 
42.         setBounds(100, 100, 500, 375); 
43.         setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE); 
45. final JScrollPane scrollPane = new JScrollPane(); 
46.         getContentPane().add(scrollPane, BorderLayout.CENTER); 
48.         tree = new JTree(); 
49.         scrollPane.setViewportView(tree); 
51. //创建数据
52.         DefaultTreeModel defaultTreeModel = createModel(); 
54. //设置数据
55.         tree.setModel(defaultTreeModel); 
57. //设置自定义描述类
58.         tree.setCellRenderer(new MyDefaultTreeCellRenderer()); 
59.     } 
61. /**
62.      * 创建树节点模型
63.      * @return
64.      */
65. public DefaultTreeModel createModel() 
66.     { 
67.         DefaultMutableTreeNode root = new DefaultMutableTreeNode("a"); 
68.         DefaultMutableTreeNode b = new DefaultMutableTreeNode("b"); 
69.         DefaultMutableTreeNode c = new DefaultMutableTreeNode("c"); 
70.         root.add(b); 
71.         root.add(c); 
72. return new DefaultTreeModel(root); 
73.     } 

}