ndarray-methods/src/polyfill.ts

Convenient methods for JavaScript built-in multi-dimensional arrays.

import * as Library from "./main";
import { buildShape, nestedSplit, nestedJoin } from "./main";

Object.entries(Library).forEach(([name, method]) => {
  if (!(name in Array.prototype))
    Object.defineProperty(Array.prototype, name, {
      value(...args: unknown[]) {
        return (method as Function)(this, ...args);
      },
      // https://tc39.es/ecma262/#sec-ecmascript-standard-built-in-objects
      writable: true,
      enumerable: false,
      configurable: true,
    });
});

Object.entries({ fromShape: buildShape, nestedSplit, nestedJoin }).forEach(([name, value]) => {
  if (!(name in Array))
    Object.defineProperty(Array, name, { value, writable: true, enumerable: false, configurable: true });
});

// prettier-ignore
export { Infinity, NDArray, MDArray, ArrayOfLength, BuildIndices, Indices, Shape, Map, ElementType, Access, Parent, Cast } from ".";