//
// Copyright 2020-2023 Picovoice Inc.
//
// You may not use this file except in compliance with the license. A copy of the license is located in the "LICENSE"
// file accompanying this source.
//
// Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
// an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
// specific language governing permissions and limitations under the License.
//
'use strict';

function chunkArray(array: Int16Array, size: number): Int16Array[] {
  return Array.from({ length: Math.ceil(array.length / size) }, (v, index) =>
    array.slice(index * size, index * size + size)
  );
}

export function checkWaveFile(
  waveFile: any,
  engineSampleRate: number
): boolean {
  let valid = true;

  if (waveFile.bitDepth !== '16') {
    // eslint-disable-next-line no-console
    console.error('Audio bit depth must be 16-bit');
    valid = false;
  }

  if (waveFile.fmt.numChannels !== 1) {
    // eslint-disable-next-line no-console
    console.error('Audio must be single channel');
    valid = false;
  }

  if (waveFile.fmt.sampleRate !== engineSampleRate) {
    // eslint-disable-next-line no-console
    console.error(
      `Audio sample rate ${waveFile.sampleRate} does not match Rhino sample rate ${engineSampleRate}`
    );
    valid = false;
  }

  return valid;
}

export function getInt16Frames(
  waveFile: any,
  frameLength: number
): Int16Array[] {
  const samples = waveFile.getSamples(false, Int16Array);

  const frames = chunkArray(samples, frameLength);

  // throw out remainder / partial frame
  if (frames[frames.length - 1].length !== frameLength) {
    frames.pop();
  }

  return frames;
}