zktest/src/main/webapp/js/dtree.js

Summary

Maintainability
C
1 day
Test Coverage

var m = [20, 120, 20, 120],
    i = 0,
    root;

var tree;

var diagonal = d3.svg.diagonal()
    .projection(function(d) { return [d.y, d.x]; });

var vis, w, h;

function getMaxNodeSize(node) {
    var cn = 0;
    if (node.children) {
        cn = node.children.length;
        var levelInfos = {depth: 0, nodeSizes: [cn]};
        for (var i = 0; i < cn; i++) {
            if (node.children[i].children) {
                getNodeSize(node.children[i], levelInfos);
                levelInfos.depth--;
            }
        }
        var maxSize = 0;
        for (var ns = levelInfos.nodeSizes, i = 0, j = ns.length; i < j; i++)
            if (ns[i] > maxSize)
                maxSize = ns[i];
        return maxSize;
    }
    return 0;
}
function getNodeSize(node, levelInfos) {
    if (node.children) {
        var cn = node.children.length;
        levelInfos.depth++;
        if (levelInfos.nodeSizes[levelInfos.depth]) {
            levelInfos.nodeSizes[levelInfos.depth] += cn;
        } else {
            levelInfos.nodeSizes[levelInfos.depth] = cn;
        }
        for (var i = 0; i < cn; i++) {
            if (node.children[i].children) {
                getNodeSize(node.children[i], levelInfos);
                levelInfos.depth--;
            }
        }
    }
}
window.DrawTree = function(json, hostId) {
    var maxNodeSize = getMaxNodeSize(json);
    var zkcontent = jq('$' + hostId);
     w = jq.innerWidth() - 50 - m[1] - m[3];
     h = jq.innerHeight() - 100 - zkcontent.height() - m[0] - m[2];
     h = Math.max(110 * maxNodeSize, h);
  if (!vis) {
     tree = d3.layout.tree().size([h, w]);
      vis = d3.select("#body").append("svg:svg")
        .attr("width", w + m[1] + m[3])
        .attr("height", h + m[0] + m[2])
      .append("svg:g")
        .attr("transform", "translate(" + m[3] + "," + m[0] + ")");
  }
  root = json;
  root.x0 = h / 2;
  root.y0 = 0;

  function toggleAll(d) {
    if (d.children) {
      d.children.forEach(toggleAll);
      toggle(d);
    }
  }

  // Initialize the display to show a few nodes.
  //root.children.forEach(toggleAll);

  update(root);
};

function update(source) {
  var duration = d3.event && d3.event.altKey ? 5000 : 500;

  // Compute the new tree layout.
  var nodes = tree.nodes(root).reverse();

  // Normalize for fixed-depth.
  nodes.forEach(function(d) { d.y = d.depth * 260;});

  // Update the nodes…
  var node = vis.selectAll("g.node")
      .data(nodes, function(d) { return d.id || (d.id = ++i); });

  // Enter any new nodes at the parent's previous position.
  var nodeEnter = node.enter().append("svg:g")
      .attr("class", "node")
      .attr("transform", function(d) { return "translate(" + source.y0 + "," + source.x0 + ")"; })
      .on("click", function(d) { toggle(d); update(d); });

  nodeEnter.append("svg:circle")
      .attr("r", 1e-6)
      .attr('class', function(d) {return d.real ? 'component' : ''})
      .style("fill", function(d) { return d._children ? "lightsteelblue" : "#fff"; });

  nodeEnter.append("svg:text")
      .attr("x", function(d) { return d.depth ? 10 : -10; })
      .attr("dy", ".35em")
      .attr("text-anchor", function(d) { return d.depth ? "start" : "end"; })
      .text(function(d) { return d.name; })
      .style("fill-opacity", 1e-6);

  nodeEnter.append("svg:foreignObject")
  .attr("x", function(d) { return d.depth ? 10 : -100; })
  .attr("y", function(d) {
      var base = -20; 
      return base * (d.allChildren ? (d.allChildren.length / 70 + 1) : 1);
  })
  .attr("dy", ".35em")
  .attr("class", 'allChildren')
  .attr("text-anchor", "start")
  .attr('width', "300")
  .attr('height', "200")
  .append("div")
  .style("fill-opacity", function (d) { return d.real ? 0 : 1})
  .attr('style', 'font-size:12px;color:rgb(148, 88, 145);font-weight:bold')
  .text(function(d) { return d['allChildren']; });

  var indent = 0;
  jq(['prev', 'first', 'last', 'next']).each(function () {
      nodeEnter.append("svg:text")
          .attr("x", 10)
          .attr("y", indent+=15)
          .attr("dy", ".35em")
          .attr("class", this)
          .attr("text-anchor", "start")
          .attr('xml:space', 'preserve')
          .style("display", function (d) { return d.real ? 'none' : 'block'})
          .text((function (v) {return function(d) { return v + ':     ' + (d[v] == 'null' ? '' : d[v]); };})(this));
  });
  
  // Transition nodes to their new position.
  var nodeUpdate = node.transition()
      .duration(duration)
      .attr("transform", function(d) { return "translate(" + d.y + "," + d.x + ")"; });

  nodeUpdate.select("circle")
      .attr("r", 4.5)
      .style("fill", function(d) { return d._children ? "lightsteelblue" : "#fff"; });

  nodeUpdate.select("text")
      .style("fill-opacity", 1);

  // Transition exiting nodes to the parent's new position.
  var nodeExit = node.exit().transition()
      .duration(duration)
      .attr("transform", function(d) { return "translate(" + source.y + "," + source.x + ")"; })
      .remove();

  nodeExit.select("circle")
      .attr("r", 1e-6);

  nodeExit.select("text")
      .style("fill-opacity", 1e-6);

  // Update the links…
  var link = vis.selectAll("path.link")
      .data(tree.links(nodes), function(d) { return d.target.id; });

  // Enter any new links at the parent's previous position.
  link.enter().insert("svg:path", "g")
      .attr("class", "link")
      .attr("d", function(d) {
        var o = {x: source.x0, y: source.y0};
        return diagonal({source: o, target: o});
      })
    .transition()
      .duration(duration)
      .attr("d", diagonal);

  // Transition links to their new position.
  link.transition()
      .duration(duration)
      .attr("d", diagonal);

  // Transition exiting nodes to the parent's new position.
  link.exit().transition()
      .duration(duration)
      .attr("d", function(d) {
        var o = {x: source.x, y: source.y};
        return diagonal({source: o, target: o});
      })
      .remove();

  // Stash the old positions for transition.
  nodes.forEach(function(d) {
    d.x0 = d.x;
    d.y0 = d.y;
  });
}

// Toggle children.
function toggle(d) {
  if (d.children) {
    d._children = d.children;
    d.children = null;
  } else {
    d.children = d._children;
    d._children = null;
  }
}