【基于spark IM 的二次开发笔记】第二天 树形结构

• MessageTreePlugin.java

1. final MessageTreeTab messageTreeTab = new MessageTreeTab();
2. /**
3. * Adds a tab to Spark
4. */
5. private void addTabToSpark(){
6. // Get Workspace UI from SparkManager
7. Workspace workspace = SparkManager.getWorkspace();
8. // Retrieve the Tabbed Pane from the WorkspaceUI.
9. SparkTabbedPane tabbedPane = workspace.getWorkspacePane();
10. // Add own Tab.
11. tabbedPane.addTab("消息", null,<SPAN style="COLOR: #ff0000">messageTreeTab</SPAN>);
12. //messageTreeTab.invokeDialog();
13. }

• MessageTreeTab.java

1. public class MessageTreeTab extends JPanel {
2. private static final long serialVersionUID = -381458717030745122L;
3. private JiveTreeNode msgType;
4. private List<JiveTreeNode> groupNodes = new ArrayList<JiveTreeNode>();
5. private MessageTree msgTree;
6. private JiveTable messageType;
7. public MessageTreeTab(){
8. setLayout(new GridBagLayout());
9. msgType = new JiveTreeNode("分类");
10. msgTree = <SPAN style="COLOR: #ff0000">new MessageTree</SPAN>(msgType);
11. msgTree.setPreferredSize(new Dimension(150, 400));
12. String groupName = Res.getString("status.online");
13. String[] headers = new String[]{"类型","时间","内容"};
14. messageType = new JiveTable(headers, null);
15. messageType.setPreferredSize(new Dimension(350, 400));
16. final JScrollPane pane = new JScrollPane(messageType);
17. pane.setBorder(BorderFactory.createTitledBorder(Res.getString("label.message").replace("&", "")));
18. final JScrollPane treePane = new JScrollPane(msgTree);
19. treePane.setBorder(BorderFactory.createTitledBorder(Res.getString("message.send.to.these.people")));
20. //          treePane.getVerticalScrollBar().setBlockIncrement(200);
21. //          treePane.getVerticalScrollBar().setUnitIncrement(20);
22. // Add to UI
23. add(treePane, new GridBagConstraints(0, 0, 1, 1, 0.5, 1.0, GridBagConstraints.WEST, GridBagConstraints.BOTH, new Insets(5, 5, 5, 0), 0, 0));
24. add(pane, new GridBagConstraints(1, 0, 1, 3, 0.5, 1.0, GridBagConstraints.WEST, GridBagConstraints.BOTH, new Insets(2, 0, 2, 5), 0, 0));
25. msgTree.expandTree();
26. }
27. /**
28. * Displays the message tree dialog.
29. */
30. public void invokeDialog() {
31. JiveTreeNode groupNode = new JiveTreeNode("通知类型1");
32. groupNode.add( new JiveTreeNode("通知类型11"));
33. groupNode.add(new JiveTreeNode("通知类型12"));
34. groupNode.add(new JiveTreeNode("通知类型13"));
35. groupNode.add(new JiveTreeNode("通知类型14"));
36. groupNode.add(new JiveTreeNode("通知类型15"));
37. JiveTreeNode groupNode2 = new JiveTreeNode("通知类型2");
38. groupNode2.add( new JiveTreeNode("通知类型21"));
39. groupNode2.add(new JiveTreeNode("通知类型22"));
40. groupNode2.add(new JiveTreeNode("通知类型23"));
41. groupNode2.add(new JiveTreeNode("通知类型24"));
42. groupNode2.add(new JiveTreeNode("通知类型25"));
43. msgType.add(groupNode);
44. msgType.add(groupNode2);
45. }
46. }

• MessageTree.java

1. public class MessageTree extends JPanel {
2. private static final long serialVersionUID = 5992552368286729600L;
3. private Tree tree;
4. public MessageTree(JiveTreeNode rootNode) {
5. tree = new Tree(rootNode);
6. tree.setCellRenderer(new DefaultTreeCellRenderer());
7. tree.setRowHeight(18);
8. tree.getSelectionModel().setSelectionMode(TreeSelectionModel.SINGLE_TREE_SELECTION);
9. tree.setToggleClickCount(1000);
10. tree.putClientProperty("JTree.lineStyle", "Angled");
11. tree.addMouseListener(new NodeSelectionListener(tree));
12. setLayout(new BorderLayout());
13. add(tree);
14. }
15. class NodeSelectionListener extends MouseAdapter {
16. JTree tree;
17. NodeSelectionListener(JTree tree) {
18. this.tree = tree;
19. }
20. public void mouseClicked(MouseEvent e) {
21. int x = e.getX();
22. int y = e.getY();
23. int row = tree.getRowForLocation(x, y);
24. TreePath path = tree.getPathForRow(row);
25. if (path != null) {
26. JiveTreeNode node = (JiveTreeNode)path.getLastPathComponent();
27. if (!node.isLeaf()) {
28. tree.expandPath(path);
29. }
30. else {
31. tree.collapsePath(path);
32. }
33. ((DefaultTreeModel)tree.getModel()).nodeChanged(node);
34. // I need revalidate if node is root.  but why?
35. tree.revalidate();
36. tree.repaint();
37. }
38. }
39. }
40. public JTree getTree() {
41. return tree;
42. }
43. /**
44. * Call to expand the entire tree.
45. */
46. public void expandTree() {
47. for (int i = 0; i <= tree.getRowCount(); i++) {
48. tree.expandPath(tree.getPathForRow(i));
49. }
50. }
51. }

效果图：