Function computeBinning

src/compute-engine/library/statistics.ts:67–109

* Shared binning for `Histogram`/`BinCounts`. Returns the bin edges and the * count in each bin, or `undefined` if the input is not a usable finite * numeric collection. * * The final bin is *closed* on both ends (`[edge, lastEdge]`) so the dataset * maximum is counted — every interior bin is h

(
  xs: Expression,
  binsArg: Expression
)

} from '../boxed-expression/utils.js';
import { numberLiteralOf, toInteger } from '../boxed-expression/numerics.js';
import { randomCount } from './random-utils.js';
import { checkDeadline } from '../../common/interruptible.js';
import { findFit } from '../nonlinear-fit.js';
import {
  distributionMean,
  distributionStandardDeviation,
  distributionVariance,
  isDistributionExpression,
} from './distributions.js';

// Geometric mean:
// Harmonic mean:

/**
 * Shared binning for `Histogram`/`BinCounts`. Returns the bin edges and the
 * count in each bin, or `undefined` if the input is not a usable finite
 * numeric collection.
 *
 * The final bin is *closed* on both ends (`[edge, lastEdge]`) so the dataset
 * maximum is counted — every interior bin is half-open `[edge, next)`.
 * (Previously every bin was half-open, so the max value, which equals the
 * last edge, was never counted.)
 */
function computeBinning(
  xs: Expression,
  binsArg: Expression
): { binEdges: number[]; counts: number[] } | undefined {
  if (!xs.isFiniteCollection) return undefined;

  const data = (Array.from(xs.each()) as Expression[])
    .map((x) => x.re)
    .filter(Number.isFinite);
  if (data.length === 0) return undefined;

  const min = Math.min(...data);
  const max = Math.max(...data);

  let binEdges: number[];
  if (binsArg.isCollection) {
    binEdges = [...binsArg.each()].map((op) => op.re);
  } else {
    const binCount = toInteger(binsArg);
    if (binCount === null || binCount <= 0) return undefined;
    const binWidth = (max - min) / binCount;
    binEdges = Array.from(
      { length: binCount + 1 },