@uwdata/mosaic-plot/src/marks/util/permute.js

A Mosaic-powered plotting framework based on Observable Plot.

export function indices(length) {
  return Array.from({ length }, (_, i) => i);
}

export function permute(data, order) {
  const ord = order.reduce((acc, val, i) => (acc[val] = i, acc), {});
  const idx = indices(data.length);
  idx.sort((a, b) => ord[data[a]] - ord[data[b]]);
  return idx;
}