to-grid/index.ts

generate an array of elements split into groups in specified layout

export enum Order {
  N,
  Z
}

function toGrid<T>(
  arr: T[],
  numOfColumns: number,
  layout: Order = Order.Z
): T[][] {
  let result = [] as T[][];
  const numOfRows = Math.ceil(arr.length / numOfColumns);
  arr.forEach((it, index) => {
    let rowIndex, columnIndex;
    if (layout === Order.Z) {
      rowIndex = Math.floor(index / numOfColumns);
      columnIndex = index % numOfColumns;
    } else {
      rowIndex = index % numOfRows;
      columnIndex = Math.floor(index / numOfRows);
    }
    result[rowIndex] = result[rowIndex] || [];
    result[rowIndex][columnIndex] = it;
  });
  return result;
}

export default toGrid;