brk/website_next/heatmap/capacity.js

import { packCells } from "./pack.js";

/**
 * @param {number} weight
 * @param {number} capacity
 * @param {number} columns
 */
function weightToSpan(weight, capacity, columns) {
  const cellWeight = capacity / (columns * columns);
  const span = Math.sqrt(weight / cellWeight);

  return Math.max(1, Math.round(span));
}

/**
 * @template {CapacityCell} Cell
 * @param {readonly Cell[]} cells
 * @param {number} capacity
 * @param {number} columns
 */
function resolveCapacityCells(cells, capacity, columns) {
  return cells.map((cell) => ({
    ...cell,
    span: weightToSpan(cell.weight ?? 0, capacity, columns),
  }));
}

/**
 * @template {CapacityCell} Cell
 * @param {readonly Cell[]} cells
 * @param {number} capacity
 * @param {number} columns
 */
function fitCapacityCells(cells, capacity, columns) {
  let resolvedCells = resolveCapacityCells(cells, capacity, columns);
  let layouts = packCells(resolvedCells, columns, columns);

  while (layouts === null) {
    const largest = Math.max(...resolvedCells.map(({ span }) => span));

    if (largest <= 1) break;

    resolvedCells = resolvedCells.map((cell) => ({
      ...cell,
      span: cell.span === largest ? largest - 1 : cell.span,
    }));
    layouts = packCells(resolvedCells, columns, columns);
  }

  return { resolvedCells, layouts };
}

/**
 * @template {CapacityCell} Cell
 * @param {readonly Cell[]} cells
 * @param {number} capacity
 * @param {number} columns
 */
export function createSquareLayout(cells, capacity, columns) {
  const { resolvedCells, layouts } = fitCapacityCells(cells, capacity, columns);

  return { columns, resolvedCells, layouts: /** @type {NonNullable<typeof layouts>} */ (layouts) };
}

/**
 * @typedef {Object} CapacityCell
 * @property {number} span
 * @property {number | undefined} weight
 */