d3-sankey-diagram/src/sortNodes/initial-ordering.js

Sankey diagram d3 plugin

import { map } from 'd3-collection'
import pkg from '@dagrejs/graphlib'; const { alg } = pkg

export default function initialOrdering (G, ranks) {
  const order = []
  if (ranks.length === 0) return order

  // Start with sources & nodes in rank 0
  const start = G.sources()
  const nodeRanks = map()
  ranks.forEach((nodes, i) => {
    order.push([])
    nodes.forEach(u => {
      if (i === 0 && start.indexOf(u) < 0) start.push(u)
      nodeRanks.set(u, i)
    })
  })

  alg.preorder(G, start).forEach(u => {
    order[nodeRanks.get(u)].push(u)
  })

  return order
}