vivian_tree_layout_common.js

function _expandAllNode() {
  clearAutocomplete();
  expandAllNode(chart.nodes());
  chart.update(chart.nodes());
}

function _collapseAllNode() {
  clearAutocomplete();
  collapseAllNode(chart.nodes());
  chart.update(chart.nodes());
}

function _resetAllNode() {
  clearAutocomplete();
  resetAllNode(chart.nodes());
  chart.update(chart.nodes());
}

function expandAllNode(root) {
  expand(root)
  root.children.forEach(expandAll);
}

function collapseAllNode(root) {
  if (root.children) {
    root.children.forEach(collapseAll);
  }
  collapse(root);
}

function resetAllNode(root) {
  expand(root);
  if (root.children !== undefined && root.children) {
    root.children.forEach(collapseAll);
    // Initialize the display to show a few nodes.
    //expandAll(root.children[0]);
  }
}

// Helper functions
function expandAll(d) {
  expand(d);
  if (d.children) {
    d.children.forEach(expandAll);
  }
}

function expand(d) {
  if (d._children) {
    d.children = d._children;
    d._children = null;
  }
}

function collapseAll(d) {
  if (d.children) {
    d.children.forEach(collapseAll);
    collapse(d);
  }
}

function collapse(d) {
  if (d.children) {
    d._children = d.children;
    d.children = null;
  }
}

// Highlight path
var target_node = [];
var target_path = [];

function openSpecificNode(target, root) {
  target_node = [];
  target_path = [];
  expandAllNode(chart.nodes());
  searchForNode(target, root);
}

function searchForNode(target, d) {
  var totalChildren=[]
  if(d.children) {totalChildren = d.children; }
  if(d._children) {totalChildren = $.merge(totalChildren,d._children);}
  for(var i=0; i<totalChildren.length;i++) {
    searchForNode(target, totalChildren[i])
  }
  if (d.name.toUpperCase() == target.toUpperCase()) {
    target_node.push(d);
  }
}

function highlightPath(chart,json) {
  var tree = d3.layout.tree()
  var nodes = tree.nodes(chart.nodes());
  var links = tree.links(nodes);
  collapseAllNode(chart.nodes());
  target_node.forEach(function(targetNode) {
    var target = targetNode
    while (target != null && target.name != nodes[0].name) {
      chart.svg().selectAll("path.link").data(links, function(d) {
        if(d.target == target) {
          target = d.source;
          expand(d.source);
          expand(d.target);
          target_path.push(d)
          }
      });
      if(target == targetNode) {
        $("#option_autocomplete")[0].style.border="solid 4px blue";
        $("#search_result").html("<h5>Target option found in menu, but couldn't be matched.</h5>");
        resetAllNode(chart.nodes())
        target_path = [];
        break;
        }
    }
  });

  if (target_path.length) {
    $("#option_autocomplete")[0].style.border="";
    $("#search_result").html("");
  }
  d3.select("#treeview_placeholder").datum(chart.nodes()).call(chart);
  chart.svg().selectAll("path.link").data(target_path).forEach(highlight);
  d3.select("#treeview_placeholder").datum(chart.nodes()).call(chart);
}

function highlight(d) {
  d.forEach(function(highlightLink) {
    if(highlightLink.classList) {highlightLink.classList.add("target");};
  });
}

function clearAutocomplete() {
  if (document.getElementById("option_autocomplete")) {
    document.getElementById("option_autocomplete").value= '';
    $("#option_autocomplete")[0].style.border="";
    $("#search_result").html("");
    clearHighlightedPath();
  }
}

function clearHighlightedPath() {
  chart.svg().selectAll("path.link").data(target_path).forEach(function(d) {
    for(var i =0; i< d.length; i++) {
      d[i].classList.remove("target");
    }
  });

  target_node = null;
  target_path = [];
}