import type { NumberArray } from 'cheminfo-types';
import { isAnyArray } from 'is-any-array';

import { getOutputArray } from './getOutputArray.ts';

export interface XDivideOptions<ArrayType extends NumberArray = Float64Array> {
  /** output into which the result should be placed if needed. In can be the same as array1 in order to have in-place modification */
  output?: ArrayType;
}

/**
 * This function divide the first array by the second array or a constant value to each element of the first array
 * @param array1 - first array
 * @param array2 - second array or number
 * @param options - options
 */
export function xDivide<ArrayType extends NumberArray = Float64Array>(
  array1: NumberArray,
  array2: NumberArray | number,
  options: XDivideOptions<ArrayType> = {},
): ArrayType {
  let isConstant = false;
  let constant = 0;
  if (isAnyArray(array2)) {
    if (array1.length !== array2.length) {
      throw new RangeError('size of array1 and array2 must be identical');
    }
  } else {
    isConstant = true;
    constant = array2;
  }

  const array3 = getOutputArray(options.output, array1.length);
  if (isConstant) {
    for (let i = 0; i < array1.length; i++) {
      array3[i] = array1[i] / constant;
    }
  } else {
    for (let i = 0; i < array1.length; i++) {
      array3[i] = array1[i] / (array2 as number[])[i];
    }
  }

  return array3;
}