timgentry/organogram

var appendLegendHeading = function (g, text, leftJustified) {
  g.append('text')
    .attr('dy', '.31em')
    .attr('text-anchor', leftJustified ? 'start' : 'end')
    .attr('transform', leftJustified ? 'translate(-3)' :  'translate(3)')
    .attr('font-weight', 'bold')
    .text(text)
}

var bindLegends = function (g, data) {
  return g.selectAll('.legend')
    .data(data)
    .enter()
    .append('g')
    .attr('class', 'legend')
    .attr('transform', function (d, i) {
      return 'translate(0,' + ((i + 1) * 10) + ')'
    })
}

export function appendColourLegend (svg, values, scale, text) {
  var g = svg.append('g').attr('transform', 'translate(40, 60)')

  appendLegendHeading(g, text, true)

  var node = bindLegends(g, values)

  node.append('circle')
    .attr('r', 3)
    .attr('fill', scale)

  node.append('text')
    .attr('dy', '.31em')
    .attr('text-anchor', 'start')
    .attr('transform', 'translate(6)')
    .style('fill', function (d) {
      return typeof d !== 'undefined' ? 'black' : '#999'
    })
    .text(function (d) {
      return typeof d !== 'undefined' ? d : 'Blank'
    })
}

export function appendWteLegend (svg, scale, text) {
  var ticks = scale.ticks(6)
  // Remove the 0 tick, we don't need it
  ticks.shift()

  var g = svg.append('g').attr('transform', 'translate(' + (+svg.attr('width') - 40) + ', 60)')

  appendLegendHeading(g, text, false)

  var node = bindLegends(g, ticks)

  node.append('circle')
    .attr('r', scale)
    .attr('fill', '#999')

  node.append('text')
    .attr('dy', '.31em')
    .attr('text-anchor', 'end')
    .attr('transform', 'translate(-6)')
    .text(function (d, i) {
      return d
    })
}