Найти дочерние и дочерние элементы дерева в JavaFx

Этот код дает мне только первый уровень в иерархии дерева. Как я могу проверить узел до точки, в которой узел является листом, и добавить все эти дочерние элементы в качестве значения для родителя в TreeMap

    for (TreeItem node : root.getChildren()) {

        mytreemap.put(node.getValue().toString(), node.getChildren());
    }

Это дает

{a = [TreeItem [ value: 1, [ TreeItem [ value: 2], b= [TreeItem [ value:b1]} 

Мне нужно иметь {a = [1,2] b= [b1,b2,b2-1,b2-2]}

a
  1
  2
b
  b1
 -b2
    b2-1
    b2-2