Как получить индекс выбранного узла в selcted уровне в JTree

Question

Я использую JTree, в котором новые узлы необходимо динамически вставлять в корень (подумайте о добавлении дочерних элементов в root). Когда пользователь выбирает узел и нажимает кнопку, новый узел должен быть добавлен после выбранного узла. Если ни один из узлов не выбран, он добавляет новый узел в конец дерева. Ниже мой код

 public void addNodeToRoot(TestCase testCase) { 
     DefaultMutableTreeNode childNode = new DefaultMutableTreeNode(testCase.toString()); 
     int currentNoOfChildren = getTcBuilderTree().getModel().getChildCount(getTcBuilderTree().getModel().getRoot()); 
     TreePath currentSelection = getTcBuilderTree().getSelectionPath(); 
     int currentIndex=0; 
     //if the user has not selected a node add the test case at the end of the tree 
     if (currentSelection == null) { 
      currentIndex = currentNoOfChildren; 
     } 
     //if user has selected a node then insert the new node after the selected node 
     else { 
       int[] currentSelectedIndex = getTcBuilderTree().getSelectionRows(); 
       currentIndex = currentSelectedIndex[0]; 

     } 
     treeModel.insertNodeInto(childNode, getRoot(), currentIndex); 
    } 

работает все нормально, но код дает исключение, когда есть дочерние узлы на уровне 3, а также. Причина в том, что дерево имеет больше уровней, а когда его расширяет, то currentIndex дает неожиданный номер (он подсчитывает все индексы на всех уровнях от корня до выбранного узла), а приложение дает ArrayIndexOutOfBoundsException, так как currentIndex становится больше currentNoOfChildren

Если дерево не разворачивается, все происходит правильно. Пожалуйста, дайте мне знать, как это решить. Есть ли другой способ получить детей из определенного уровня в дереве?

Answer 1

Может быть ниже код может решить вашу проблему.

int currentNoOfChildren = getTcBuilderTree().getVisibleRowCount();