Как свернуть дерево D3JS на основе выбранной пользователем глубины?

Question

У меня есть дерево D3JS, которое нужно свернуть в зависимости от уровня, выбранного пользователем. В раскрывающемся списке уровней (с помощью Select2) пользователь может выполнить выбор.

Я могу это сделать, но проблема в том, что при вызове функции свертывания возникает задержка в несколько секунд (Expand работает нормально). Вот что я делаю:

$("#drop_down_levels").on("select2-selecting", function(e) {                
        var nodes = tree.nodes(root).reverse();
        nodes.forEach(function(d) {
            if (d.depth >=e.object.id){
                 d._children = d.children;
                 d.children = null;
            }else{
                expand(d);
            }
            update(root);
            centerNode(root);
        });
    })

Есть ли способ, которым это можно оптимизировать? Большое спасибо заранее.