123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169
import type { Constructor, TypedArrayConstructors } from "./utility-types.ts";import { getLog2 } from "./utilities.ts";
/** * Creates a Grid class extending a given Array-like class. */export function GridMixin<  ItemType = number,  U extends Constructor<Array<ItemType>> | TypedArrayConstructors = Constructor<    Array<ItemType>  >,>(Base: U) {  /**   * Extends built-in indexed collections to handle 2 dimensional data.   */  return class Grid extends Base {    [key: number]: ItemType
    size = 0;
    static get [Symbol.species]() {      return Base;    }
    /**     * Number of columns in the grid.     */    get columns() {      return 1 << this.size;    }
    /**     * Specifies the number of columns of the grid.     */    set columns(columns) {      this.size = getLog2(columns);    }
    /**     * Number of rows in the grid.     */    get rows() {      //@ts-ignore 2339      return this.length >> this.size;    }
    /**     * Creates a grid of specified dimensions.     *     * @param rows the amount of rows     * @param columns the amount of columns     * @return a new grid     */    static create<T extends typeof Grid>(      this: T,      rows: number,      columns = 1,    ): InstanceType<T> {      const offset = getLog2(columns);      const length = rows << offset;      const grid = new this(length);      grid.size = offset;      return grid as InstanceType<T>;    }
    /**     * Creates a grid from an array of arrays.     *     * @param arrays the array of arrays     * @return a new grid     */    static fromArrays<T extends typeof Grid>(      this: T,      arrays: Array<Array<ItemType>>,    ): InstanceType<T> {      const rows = arrays.length;
      // find longest array to get the column size      let columns = arrays[0].length; // if !arrays[0].length      for (let i = 0; i < rows; i++) {        if (arrays[i].length > columns) columns = arrays[i].length;      }      const offset = getLog2(columns);      columns = 1 << offset;
      // create grid of the required length      const grid = this.create(rows, columns);
      // fill the grid with values from arrays      for (let i = 0; i < rows; i++) {        const rowId = i << offset;        for (let j = 0; j < arrays[i].length; j++) {          grid[rowId + j] = arrays[i][j];        }      }
      return grid;    }
    /**     * Returns the length of the underlying Array required to hold the grid of specified dimensions.     *     * @param rows the amount of rows     * @param columns the amount of columns     * @return the required length     */    static getLength(rows: number, columns = 1): number {      return rows << getLog2(columns);    }
    getCoordinates(index: number): [row: number, column: number] {      return [index >> this.size, index & ((1 << this.size) - 1)];    }
    /**     * Returns the index of an element at given coordinates.     *     * @param rows the row index     * @param columns the column index     * @return the element index     */    getIndex(row: number, column = 1): number {      return (row << this.size) + column;    }
    /**     * Returns the element at given coordinates.     *     * @param rows the row index     * @param columns the column index     * @return the element     */    getValue(row: number, column: number): ItemType {      return this[this.getIndex(row, column)];    }
    /**     * Sets the element at given coordinates.     *     * @param rows the row index     * @param columns the column index     * @param value the element     * @return the grid     */    setValue(row: number, column: number, value: ItemType): this {      this[this.getIndex(row, column)] = value;      return this;    }
    /**     * Creates an array of arrays representing rows of the grid.     *     * @return an array of arrays     */    toArrays(): Array<Array<ItemType>> {      const { rows, columns, size } = this;      const result: Array<Array<ItemType>> = [];      for (let i = 0; i < rows; i++) {        const rowOffset = i << size;        result[i] = [];        for (let j = 0; j < columns; j++) {          result[i][j] = this[rowOffset + j];        }      }      return result;    }  };}