pandora_cms/static/ontology/index.js

var margin = {
        top: 20,
        right: 120,
        bottom: 20,
        left: 120
    },
    width = 3000 - margin.right - margin.left,
    height = window.innerHeight - margin.top - margin.bottom;

var i = 0,
    duration = 750,
    root;

var tree = d3.layout.tree()
    .size([height, width]);

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

var svg = d3.select("body").append("svg")
    .attr("width", width)
    .attr("height", height + margin.top + margin.bottom)
    .append("g")
    .attr("transform", "translate(" + margin.left + "," + margin.top + ")");

d3.json("sized_ontology.json", function(error, nodes) {
  root = nodes;
  root.x0 = height / 2;
  root.y0 = 0;

  function collapse(d) {
    if (d.children) {
      if (!d.children[0].children) {
          d._children = d.children;
          d._children.forEach(collapse);
          d.children = null;
      } else {
          d.children.forEach(collapse);
      }
    }
  }

  root.children.forEach(collapse);

  update(root);
});

function resize() {
  root.x0 = height / 2;
  root.y0 = 0;
  update(root);
}

function zoom(size) {
    height = size;
    tree.size([height, width]);
    svg.attr("width", width).attr("height", height + margin.top + margin.bottom)
    document.querySelector('svg').attributes.height.value = height + margin.top + margin.bottom
    resize()
}


document.querySelector('.zoom button.plus').addEventListener('click', function(event) {
    zoom(height*1.5)
})
document.querySelector('.zoom button.minus').addEventListener('click', function(event) {
    zoom(height/1.5)
})


function get_color(d) {
  if (d.children || d._children) {
      return '#fff'
  }
  return "lightsteelblue";
}
function update(source) {

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

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

  // Update the nodes…
  var node = svg.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("g")
      .attr("class", "node")
      .attr("transform", function(d) {
          return "translate(" + source.y0 + "," + source.x0 + ")";
      })
      .on("click", click);

  nodeEnter.append("circle")
      .attr("r", 1e-6)
      .style("fill", get_color)

  nodeEnter.append("text")
      .attr("x", function(d) {
          return d.children || d._children ? -10 : 10;
      })
      .attr("dy", ".35em")
      .attr("text-anchor", function(d) {
          return d.children || d._children ? "end" : "start";
      })
      .text(function(d) {
          if (d.children || d._children) {
              //return d.name;
              return d.name + '(' + (d.children || d._children).length +')';
          }
          return d.name + ' (' + d.size + ')';
      })
      .style("fill-opacity", 1e-6);


  // 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", get_color)

  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 = svg.selectAll("path.link")
      .data(links, function(d) {
          return d.target.id;
      });

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

  // 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 on click.
function click(d) {
  if (d.children) {
    d._children = d.children;
    d.children = null;
  } else {
    d.children = d._children;
    d._children = null;
  }
  if (!d.children && ! d._children) {
    var name = d.parent.name + ': ' + d.name
    var url = '/clip/created/keywords=' + name
    if (d.parent.name == 'other') {
      url = '/clip/created/keywords==' + d.name
    }
    window.open(url, '_blank');
  } else {
    update(d);
  }
}