Получить выбранное значение из JXTreeTable

Я создаю treetable , используя JXTreeTabble , и я хочу отключить / возможность меню пунктов в зависимости от выбранного значения. Итак, я попытался поместить этот код в свою модель таблицы :

public Object getValueAt(int index) {
        if (index >= 0 && index < root.getSize()){
            return root.get(index);
        }

        return null;

    }

Проблема

Вышеупомянутое работает, только если содержимое таблицы не расширяется. Поскольку индекс выбранной строки может быть больше, чем размер t модели (модель может иметь два элемента и строку может быть 10, когда все развернуто). Кроме того, тип объекта родителя отличается от дочерних объектов (представьте книгу с главами как дочерние).

Что бы вы посоветовали как способ сделать это правильно?