将子节点添加到选择路径[]

Question

我有一棵树显示网络设备。树被过滤，以便设备按位置和设备类型显示。例如，节点“办公室”将具有子节点“计算机”和“打印机”，每个节点都带有设备。

当我选择位置节点时，我想选择它下面的所有节点，并将其添加到我的树的选择路径[]中，以使它们都高亮显示。下面是我的代码，但我无法弄清楚如何让这个工作。

private class DefaultDeviceTreeCellRenderer extends DefaultTreeCellRenderer { 

    private JLabel label; 

    DefaultDeviceTreeCellRenderer() { 
     super(); 
     label = new JLabel(); 
    } 

    @Override 
    public Component getTreeCellRendererComponent(JTree tree, Object value, boolean selected, boolean expanded, 
      boolean leaf, int row, boolean hasFocus) { 
     DefaultMutableTreeNode node = (DefaultMutableTreeNode) value; 
     Object o = node.getUserObject(); 
     if (o instanceof DefaultDevice) { 
      DefaultDevice defaultDevice = (DefaultDevice) o; 
      label.setIcon(defaultDevice.getIcon()); 
      label.setText(defaultDevice.toString()); 
     } else { 
      System.out.println("Cell Class= " + o.getClass()); 
      //Set all children of this Component as selected. 
      label.setIcon(null); 
      label.setText("" + value); 
      TreePath[] selectionPaths = tree.getSelectionPaths(); 
      if (selectionPaths == null) { 
       System.out.println("Nothing Selected. "); 
      } else { 
       //keep current selection path 
       //if there are leaf nodes, add their selection paths. 
       int i = 0; 
       while (i < node.getChildCount()) { 
        //add node.getNextLeaf().getPath() to selection Paths 
        ArrayList<TreePath> arrayList = new ArrayList<>(Arrays.asList(selectionPaths)); 
        arrayList.add(new TreePath(((DefaultMutableTreeNode)node.getChildAt(i)).getPath())); 
        TreePath[] toArray = arrayList.toArray(new TreePath[arrayList.size()]); 
        this.printSelectionPath(selectionPaths); 
        this.printSelectionPath(toArray); 
        // tree.setSelectionPaths(toArray); 
        i++; 
       } 
      } 
      // System.out.println("Selection Paths.size="+selectionPaths.length); 
      // ArrayList arrayList = new ArrayList(Arrays.asList(selectionPaths)); 
      // System.out.println("ChildNode Path=" + node.getPath()[a]); 

     } 
     label.setOpaque(true); 
     if (selected) { 

      label.setBackground(this.backgroundSelectionColor); 
      label.setForeground(this.textSelectionColor); 
     } else { 
      label.setBackground(this.backgroundNonSelectionColor); 
      label.setForeground(this.textNonSelectionColor); 
     } 
     return label; 
    } 

    private void printSelectionPath(TreePath[] selectionPaths) { 
     System.out.println("\nTreePath:"); 
     for (int i = 0; i < selectionPaths.length; i++) { 
      System.out.println("Selection Path= " + selectionPaths[i]); 
      //TreePath treePath = new TreePath(node.getPath()); 
      // System.out.println("TreePath= " + treePath); 
     } 
} 

Answer 1

我通过扩展DefaultTreeSelectionModel并修改了一些方法来实现这一点。这是我的扩展类。

public static class MyTreeSelectionModel extends DefaultTreeSelectionModel { 

    public MyTreeSelectionModel() { 
     super(); 
    } 

    @Override 
    public void removeSelectionPath(TreePath path) { 
     //remove path and its children 
     Object lastPathComponent = path.getLastPathComponent(); 
     ArrayList<TreePath> childPaths = new ArrayList(); 
     if (lastPathComponent instanceof DefaultMutableTreeNode) { 
      DefaultMutableTreeNode dn = (DefaultMutableTreeNode) lastPathComponent; 
      this.getDecendents(dn, childPaths); 
     } else { 
      System.out.println("Not a DefaultMutableTreeNode"); 
     } 
     childPaths.add(path); 
     this.removeSelectionPaths(childPaths.toArray(new TreePath[childPaths.size()])); 
    } 

    @Override 
    public void addSelectionPath(TreePath path) { 
     //add the path and its children 
     Object lastPathComponent = path.getLastPathComponent(); 
     ArrayList<TreePath> childPaths = new ArrayList(); 
     if (lastPathComponent instanceof DefaultMutableTreeNode) { 
      DefaultMutableTreeNode dn = (DefaultMutableTreeNode) lastPathComponent; 
      System.out.println("Last Path Component= " + dn); 
      //get decendents 
      this.getDecendents(dn, childPaths); 
     } else { 
      System.out.println("Not a DefaultMutableTreeNode"); 
     } 
     childPaths.add(path); 
     this.addSelectionPaths(childPaths.toArray(new TreePath[childPaths.size()])); 
     // super.addSelectionPath(path); //To change body of generated methods, choose Tools | Templates. 
    } 

    private void getDecendents(DefaultMutableTreeNode dn, ArrayList<TreePath> childPaths) { 
     Enumeration children = dn.children(); 
     if (children != null) { 
      while (children.hasMoreElements()) { 
       DefaultMutableTreeNode childNode = (DefaultMutableTreeNode) children.nextElement(); 
       System.out.println("Adding " + childNode); 
       TreeNode[] path1 = childNode.getPath(); 
       childPaths.add(new TreePath(path1)); 
       getDecendents(childNode, childPaths); 
      } 
     } 

    } 

}