Problema ao atualizar o gráfico de dispersão da matriz interativa d3 para v4

Estou tentando melhorar minhas habilidades com o D3.js. atualizando vários scripts da v3 que me interessam na v4, mas fiquei paralisado ao tentar "portar" a matriz de dispersão interativa que Mike Bostok postou aqui:https://bl.ocks.org/mbostock/4063663

Embora eu tenha portado sem problemas a versão estática do código (sem pincel), ao tentar implementar o pincel da mesma maneira que na v3, encontrei um problema que parece um problema real do D3 mais do que relacionado a meu D3 noob-ness: o pincel parece grudar na célula errada na matriz do gráfico de dispersão! Em particular, se eu remover a parte do pincel e apenas registrar para consolar as quantidades p.i e p.j (que identificam em qual célula da matriz de dispersão estamos escovando), obtive um índice i = 3 preso.

var width = 960,
  size = 230,
  padding = 20;

var x = d3.scaleLinear()
  .range([padding / 2, size - padding / 2]);

var y = d3.scaleLinear()
  .range([size - padding / 2, padding / 2]);

var xAxis = d3.axisBottom()
  .scale(x)
  .ticks(6);

var yAxis = d3.axisLeft()
  .scale(y)
  .ticks(6);

var color = d3.scaleOrdinal(d3.schemeCategory10);

//d3.csv("flowers.csv", function(error, data) {
//  if (error) throw error;

data = iris;

var domainByTrait = {},
  traits = d3.keys(data[0]).filter(function(d) { return d !== "species"; }),
  n = traits.length;

traits.forEach(function(trait) {
  domainByTrait[trait] = d3.extent(data, function(d) { return d[trait]; });
});

xAxis.tickSize(size * n);
yAxis.tickSize(-size * n);

var brush = d3.brush()
  .on("start", brushstart)
  .on("brush", brushmove)
  .on("end", brushend);

var svg = d3.select("body").append("svg")
  .attr("width", size * n + padding)
  .attr("height", size * n + padding)
  .append("g")
  .attr("transform", "translate(" + padding + "," + padding / 2 + ")");

svg.selectAll(".x.axis")
  .data(traits)
  .enter().append("g")
  .attr("class", "x axis")
  .attr("transform", function(d, i) { return "translate(" + (n - i - 1) * size + ",0)"; })
  .each(function(d) { x.domain(domainByTrait[d]); d3.select(this).call(xAxis); });

svg.selectAll(".y.axis")
  .data(traits)
  .enter().append("g")
  .attr("class", "y axis")
  .attr("transform", function(d, i) { return "translate(0," + i * size + ")"; })
  .each(function(d) { y.domain(domainByTrait[d]); d3.select(this).call(yAxis); });

var cell = svg.selectAll(".cell")
  .data(cross(traits, traits))
  .enter().append("g")
  .attr("class", "cell")
  .attr("transform", function(d) { return "translate(" + (n - d.i - 1) * size + "," + d.j * size + ")"; })
  .each(plot);

// Titles for the diagonal.
cell.filter(function(d) { return d.i === d.j; }).append("text")
  .attr("x", padding)
  .attr("y", padding)
  .attr("dy", ".71em")
  .text(function(d) { return d.x; });

cell.call(brush);

function plot(p) {
  var cell = d3.select(this);

  x.domain(domainByTrait[p.x]);
  y.domain(domainByTrait[p.y]);

  cell.append("rect")
    .attr("class", "frame")
    .attr("x", padding / 2)
    .attr("y", padding / 2)
    .attr("width", size - padding)
    .attr("height", size - padding);

  cell.selectAll("circle")
    .data(data)
    .enter().append("circle")
    .attr("cx", function(d) { return x(d[p.x]); })
    .attr("cy", function(d) { return y(d[p.y]); })
    .attr("r", 4)
    .style("fill", function(d) { return color(d.species); });
}

var brushCell;

// Clear the previously-active brush, if any.
function brushstart(p) {
  if (brushCell !== this) {
    d3.select(brushCell).call(brush.move, null);
    x.domain(domainByTrait[p.x]);
    y.domain(domainByTrait[p.y]);
    brushCell = this;
  }
}

// Highlight the selected circles.
function brushmove(p) {
  // ??
  console.log(p.i +" " + p.j)
}

// If the brush is empty, select all circles.
function brushend(p) {
  if (!d3.event.selection) svg.selectAll(".hidden").classed("hidden", false);
}
//});

function cross(a, b) {
  var c = [], n = a.length, m = b.length, i, j;
  for (i = -1; ++i < n;) for (j = -1; ++j < m;) c.push({x: a[i], i: i, y: b[j], j: j});
  return c;
}

O código está disponível para revisão no JSFiddle, além do objeto iris:https://jsfiddle.net/fabio_p/pmpjawmm/

Observe que a célula da escova variável definida na função brushstart também é a incorreta (sugerindo que é o "this" passado para as funções da escova como incorreto)

Ainda mais estranho (pelo menos para os meus olhos inexperientes), as coisas parecem ir melhor se eu mudar a ordem em que adiciono células, como você pode ver neste outro violino:https://jsfiddle.net/fabio_p/7pf4cqrg/ Aqui, mudei os índices na linha 220 (e a consistência da escala na linha 206), e o índice i não está mais preso em 3 ...

Alguma idéia sobre o que estou fazendo de errado ou sobre onde o D3 está errado?

questionAnswers(1)

yourAnswerToTheQuestion