// loaders.gl
// SPDX-License-Identifier: MIT
// Copyright (c) vis.gl contributors

import {TypedArray} from '@math.gl/types';
import * as arrow from 'apache-arrow';
import {getDataTypeFromArray, deserializeArrowType} from '@loaders.gl/schema-utils';

export function isListFixedSizeList(
  vector: arrow.Vector
): vector is arrow.Vector<arrow.FixedSizeList> {
  return vector.type instanceof arrow.FixedSizeList;
}

export function getListFixedSizeListSize(vector: arrow.Vector): number {
  return vector.type instanceof arrow.FixedSizeList ? vector.type.listSize : 1;
}

/** Get Arrow FixedSizeList vector from a typed array */
export function getListFixedSizeListVector(
  indexes: Uint32Array,
  typedArray: TypedArray,
  stride: number
): arrow.Vector<arrow.FixedSizeList> {
  const data = getFixedSizeListData(typedArray, stride);
  return new arrow.Vector<arrow.FixedSizeList>([data]);
}

/** Get Arrow FixedSizeList vector from a typed array */
export function getFixedSizeListData(
  typedArray: TypedArray,
  stride: number
): arrow.Data<arrow.FixedSizeList> {
  const listType = getFixedSizeListType(typedArray, stride);
  const data = new arrow.Data<arrow.FixedSizeList>(listType, 0, typedArray.length / stride, 0, [
    typedArray
  ]);
  return data;
}

/** Get Arrow FixedSizeList vector from a typed array */
export function getFixedSizeListType(typedArray: TypedArray, stride: number): arrow.FixedSizeList {
  const {type} = getDataTypeFromArray(typedArray);
  const arrowType = deserializeArrowType(type);
  const listType = new arrow.FixedSizeList(stride, new arrow.Field('value', arrowType));
  return listType;
}