| // Licensed to the Apache Software Foundation (ASF) under one |
| // or more contributor license agreements. See the NOTICE file |
| // distributed with this work for additional information |
| // regarding copyright ownership. The ASF licenses this file |
| // to you under the Apache License, Version 2.0 (the |
| // "License"); you may not use this file except in compliance |
| // with the License. You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // 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. |
| |
| import * as type from './type.js'; |
| import type { Type } from './enum.js'; |
| import type { DataType } from './type.js'; |
| import type { Builder } from './builder.js'; |
| import type { BuilderOptions } from './builder.js'; |
| import type { BoolBuilder } from './builder/bool.js'; |
| import type { NullBuilder } from './builder/null.js'; |
| import type { DateBuilder, DateDayBuilder, DateMillisecondBuilder } from './builder/date.js'; |
| import type { DecimalBuilder } from './builder/decimal.js'; |
| import type { DictionaryBuilder } from './builder/dictionary.js'; |
| import type { FixedSizeBinaryBuilder } from './builder/fixedsizebinary.js'; |
| import type { FloatBuilder, Float16Builder, Float32Builder, Float64Builder } from './builder/float.js'; |
| import type { IntBuilder, Int8Builder, Int16Builder, Int32Builder, Int64Builder, Uint8Builder, Uint16Builder, Uint32Builder, Uint64Builder } from './builder/int.js'; |
| import type { TimeBuilder, TimeSecondBuilder, TimeMillisecondBuilder, TimeMicrosecondBuilder, TimeNanosecondBuilder } from './builder/time.js'; |
| import type { TimestampBuilder, TimestampSecondBuilder, TimestampMillisecondBuilder, TimestampMicrosecondBuilder, TimestampNanosecondBuilder } from './builder/timestamp.js'; |
| import type { IntervalBuilder, IntervalDayTimeBuilder, IntervalYearMonthBuilder } from './builder/interval.js'; |
| import type { Utf8Builder } from './builder/utf8.js'; |
| import type { BinaryBuilder } from './builder/binary.js'; |
| import type { ListBuilder } from './builder/list.js'; |
| import type { FixedSizeListBuilder } from './builder/fixedsizelist.js'; |
| import type { MapBuilder } from './builder/map.js'; |
| import type { StructBuilder } from './builder/struct.js'; |
| import type { UnionBuilder, SparseUnionBuilder, DenseUnionBuilder } from './builder/union.js'; |
| |
| /** @ignore */ type FloatArray = Float32Array | Float64Array; |
| /** @ignore */ type IntArray = Int8Array | Int16Array | Int32Array; |
| /** @ignore */ type UintArray = Uint8Array | Uint16Array | Uint32Array | Uint8ClampedArray; |
| /** @ignore */ |
| export type TypedArray = FloatArray | IntArray | UintArray; |
| /** @ignore */ |
| export type BigIntArray = BigInt64Array | BigUint64Array; |
| |
| /** @ignore */ |
| export interface TypedArrayConstructor<T extends TypedArray> { |
| readonly prototype: T; |
| new(length?: number): T; |
| new(array: Iterable<number>): T; |
| new(buffer: ArrayBufferLike, byteOffset?: number, length?: number): T; |
| /** |
| * The size in bytes of each element in the array. |
| */ |
| readonly BYTES_PER_ELEMENT: number; |
| /** |
| * Returns a new array from a set of elements. |
| * @param items A set of elements to include in the new array object. |
| */ |
| of(...items: number[]): T; |
| /** |
| * Creates an array from an array-like or iterable object. |
| * @param arrayLike An array-like or iterable object to convert to an array. |
| * @param mapfn A mapping function to call on every element of the array. |
| * @param thisArg Value of 'this' used to invoke the mapfn. |
| */ |
| from(arrayLike: ArrayLike<number>, mapfn?: (v: number, k: number) => number, thisArg?: any): T; |
| from<U>(arrayLike: ArrayLike<U>, mapfn: (v: U, k: number) => number, thisArg?: any): T; |
| } |
| |
| /** @ignore */ |
| export interface BigIntArrayConstructor<T extends BigIntArray> { |
| readonly prototype: T; |
| new(length?: number): T; |
| new(array: Iterable<bigint>): T; |
| new(buffer: ArrayBufferLike, byteOffset?: number, length?: number): T; |
| /** |
| * The size in bytes of each element in the array. |
| */ |
| readonly BYTES_PER_ELEMENT: number; |
| /** |
| * Returns a new array from a set of elements. |
| * @param items A set of elements to include in the new array object. |
| */ |
| of(...items: bigint[]): T; |
| /** |
| * Creates an array from an array-like or iterable object. |
| * @param arrayLike An array-like or iterable object to convert to an array. |
| * @param mapfn A mapping function to call on every element of the array. |
| * @param thisArg Value of 'this' used to invoke the mapfn. |
| */ |
| from(arrayLike: ArrayLike<bigint>, mapfn?: (v: bigint, k: number) => bigint, thisArg?: any): T; |
| from<U>(arrayLike: ArrayLike<U>, mapfn: (v: U, k: number) => bigint, thisArg?: any): T; |
| } |
| |
| /** @ignore */ |
| export type BuilderCtorArgs< |
| T extends BuilderType<R, any>, |
| R extends DataType = any, |
| TArgs extends any[] = any[], |
| TCtor extends new (type: R, ...args: TArgs) => T = |
| new (type: R, ...args: TArgs) => T |
| > = TCtor extends new (type: R, ...args: infer TArgs) => T ? TArgs : never; |
| |
| /** |
| * Obtain the constructor function of an instance type |
| * @ignore |
| */ |
| export type ConstructorType< |
| T, |
| TCtor extends new (...args: any[]) => T = |
| new (...args: any[]) => T |
| > = TCtor extends new (...args: any[]) => T ? TCtor : never; |
| |
| /** @ignore */ |
| export type BuilderCtorType< |
| T extends BuilderType<R, any>, |
| R extends DataType = any, |
| TCtor extends new (options: BuilderOptions<R, any>) => T = |
| new (options: BuilderOptions<R, any>) => T |
| > = TCtor extends new (options: BuilderOptions<R, any>) => T ? TCtor : never; |
| |
| /** @ignore */ |
| export type BuilderType<T extends Type | DataType = any, TNull = any> = |
| T extends Type ? TypeToBuilder<T, TNull> : |
| T extends DataType ? DataTypeToBuilder<T, TNull> : |
| Builder<any, TNull> |
| ; |
| |
| /** @ignore */ |
| export type BuilderCtor<T extends Type | DataType = any> = |
| T extends Type ? BuilderCtorType<BuilderType<T>> : |
| T extends DataType ? BuilderCtorType<BuilderType<T>> : |
| BuilderCtorType<Builder> |
| ; |
| |
| /** @ignore */ |
| export type DataTypeCtor<T extends Type | DataType = any> = |
| T extends DataType ? ConstructorType<T> : |
| T extends Type ? ConstructorType<TypeToDataType<T>> : |
| never |
| ; |
| |
| /** @ignore */ |
| export type TypedArrayDataType<T extends TypedArray | BigIntArray> = |
| T extends Int8Array ? type.Int8 : |
| T extends Int16Array ? type.Int16 : |
| T extends Int32Array ? type.Int32 : |
| T extends BigInt64Array ? type.Int64 : |
| T extends Uint8Array ? type.Uint8 : |
| T extends Uint8ClampedArray ? type.Uint8 : |
| T extends Uint16Array ? type.Uint16 : |
| T extends Uint32Array ? type.Uint32 : |
| T extends BigUint64Array ? type.Uint64 : |
| T extends Float32Array ? type.Float32 : |
| T extends Float64Array ? type.Float64 : |
| never; |
| |
| /** @ignore */ |
| export type JavaScriptArrayDataType<T extends readonly unknown[]> = |
| T extends readonly (null | undefined)[] ? type.Null : |
| T extends readonly (null | undefined | boolean)[] ? type.Bool : |
| T extends readonly (null | undefined | string)[] ? type.Dictionary<type.Utf8, type.Int32> : |
| T extends readonly (null | undefined | Date)[] ? type.Date_ : |
| T extends readonly (null | undefined | bigint)[] ? type.Int64 : |
| T extends readonly (null | undefined | number)[] ? type.Float64 : |
| never; |
| |
| /** @ignore */ |
| export type ArrayDataType<T> = |
| T extends TypedArray | BigIntArray ? TypedArrayDataType<T> : |
| T extends readonly unknown[] ? JavaScriptArrayDataType<T> : |
| never; |
| |
| /** @ignore */ |
| export type TypeToDataType<T extends Type> = { |
| [key: number]: type.DataType; |
| [Type.Null]: type.Null; |
| [Type.Bool]: type.Bool; |
| [Type.Int]: type.Int; |
| [Type.Int16]: type.Int16; |
| [Type.Int32]: type.Int32; |
| [Type.Int64]: type.Int64; |
| [Type.Uint8]: type.Uint8; |
| [Type.Uint16]: type.Uint16; |
| [Type.Uint32]: type.Uint32; |
| [Type.Uint64]: type.Uint64; |
| [Type.Int8]: type.Int8; |
| [Type.Float16]: type.Float16; |
| [Type.Float32]: type.Float32; |
| [Type.Float64]: type.Float64; |
| [Type.Float]: type.Float; |
| [Type.Utf8]: type.Utf8; |
| [Type.Binary]: type.Binary; |
| [Type.FixedSizeBinary]: type.FixedSizeBinary; |
| [Type.Date]: type.Date_; |
| [Type.DateDay]: type.DateDay; |
| [Type.DateMillisecond]: type.DateMillisecond; |
| [Type.Timestamp]: type.Timestamp; |
| [Type.TimestampSecond]: type.TimestampSecond; |
| [Type.TimestampMillisecond]: type.TimestampMillisecond; |
| [Type.TimestampMicrosecond]: type.TimestampMicrosecond; |
| [Type.TimestampNanosecond]: type.TimestampNanosecond; |
| [Type.Time]: type.Time; |
| [Type.TimeSecond]: type.TimeSecond; |
| [Type.TimeMillisecond]: type.TimeMillisecond; |
| [Type.TimeMicrosecond]: type.TimeMicrosecond; |
| [Type.TimeNanosecond]: type.TimeNanosecond; |
| [Type.Decimal]: type.Decimal; |
| [Type.Union]: type.Union; |
| [Type.DenseUnion]: type.DenseUnion; |
| [Type.SparseUnion]: type.SparseUnion; |
| [Type.Interval]: type.Interval; |
| [Type.IntervalDayTime]: type.IntervalDayTime; |
| [Type.IntervalYearMonth]: type.IntervalYearMonth; |
| [Type.Map]: type.Map_; |
| [Type.List]: type.List; |
| [Type.Struct]: type.Struct; |
| [Type.Dictionary]: type.Dictionary; |
| [Type.FixedSizeList]: type.FixedSizeList; |
| }[T]; |
| |
| /** @ignore */ |
| type TypeToBuilder<T extends Type = any, TNull = any> = { |
| [key: number]: Builder; |
| [Type.Null]: NullBuilder<TNull>; |
| [Type.Bool]: BoolBuilder<TNull>; |
| [Type.Int8]: Int8Builder<TNull>; |
| [Type.Int16]: Int16Builder<TNull>; |
| [Type.Int32]: Int32Builder<TNull>; |
| [Type.Int64]: Int64Builder<TNull>; |
| [Type.Uint8]: Uint8Builder<TNull>; |
| [Type.Uint16]: Uint16Builder<TNull>; |
| [Type.Uint32]: Uint32Builder<TNull>; |
| [Type.Uint64]: Uint64Builder<TNull>; |
| [Type.Int]: IntBuilder<any, TNull>; |
| [Type.Float16]: Float16Builder<TNull>; |
| [Type.Float32]: Float32Builder<TNull>; |
| [Type.Float64]: Float64Builder<TNull>; |
| [Type.Float]: FloatBuilder<any, TNull>; |
| [Type.Utf8]: Utf8Builder<TNull>; |
| [Type.Binary]: BinaryBuilder<TNull>; |
| [Type.FixedSizeBinary]: FixedSizeBinaryBuilder<TNull>; |
| [Type.Date]: DateBuilder<any, TNull>; |
| [Type.DateDay]: DateDayBuilder<TNull>; |
| [Type.DateMillisecond]: DateMillisecondBuilder<TNull>; |
| [Type.Timestamp]: TimestampBuilder<any, TNull>; |
| [Type.TimestampSecond]: TimestampSecondBuilder<TNull>; |
| [Type.TimestampMillisecond]: TimestampMillisecondBuilder<TNull>; |
| [Type.TimestampMicrosecond]: TimestampMicrosecondBuilder<TNull>; |
| [Type.TimestampNanosecond]: TimestampNanosecondBuilder<TNull>; |
| [Type.Time]: TimeBuilder<any, TNull>; |
| [Type.TimeSecond]: TimeSecondBuilder<TNull>; |
| [Type.TimeMillisecond]: TimeMillisecondBuilder<TNull>; |
| [Type.TimeMicrosecond]: TimeMicrosecondBuilder<TNull>; |
| [Type.TimeNanosecond]: TimeNanosecondBuilder<TNull>; |
| [Type.Decimal]: DecimalBuilder<TNull>; |
| [Type.Union]: UnionBuilder<any, TNull>; |
| [Type.DenseUnion]: DenseUnionBuilder<any, TNull>; |
| [Type.SparseUnion]: SparseUnionBuilder<any, TNull>; |
| [Type.Interval]: IntervalBuilder<any, TNull>; |
| [Type.IntervalDayTime]: IntervalDayTimeBuilder<TNull>; |
| [Type.IntervalYearMonth]: IntervalYearMonthBuilder<TNull>; |
| [Type.Map]: MapBuilder<any, any, TNull>; |
| [Type.List]: ListBuilder<any, TNull>; |
| [Type.Struct]: StructBuilder<any, TNull>; |
| [Type.Dictionary]: DictionaryBuilder<any, TNull>; |
| [Type.FixedSizeList]: FixedSizeListBuilder<any, TNull>; |
| }[T]; |
| |
| /** @ignore */ |
| type DataTypeToBuilder<T extends DataType = any, TNull = any> = { |
| [key: number]: Builder<any, TNull>; |
| [Type.Null]: T extends type.Null ? NullBuilder<TNull> : never; |
| [Type.Bool]: T extends type.Bool ? BoolBuilder<TNull> : never; |
| [Type.Int8]: T extends type.Int8 ? Int8Builder<TNull> : never; |
| [Type.Int16]: T extends type.Int16 ? Int16Builder<TNull> : never; |
| [Type.Int32]: T extends type.Int32 ? Int32Builder<TNull> : never; |
| [Type.Int64]: T extends type.Int64 ? Int64Builder<TNull> : never; |
| [Type.Uint8]: T extends type.Uint8 ? Uint8Builder<TNull> : never; |
| [Type.Uint16]: T extends type.Uint16 ? Uint16Builder<TNull> : never; |
| [Type.Uint32]: T extends type.Uint32 ? Uint32Builder<TNull> : never; |
| [Type.Uint64]: T extends type.Uint64 ? Uint64Builder<TNull> : never; |
| [Type.Int]: T extends type.Int ? IntBuilder<T, TNull> : never; |
| [Type.Float16]: T extends type.Float16 ? Float16Builder<TNull> : never; |
| [Type.Float32]: T extends type.Float32 ? Float32Builder<TNull> : never; |
| [Type.Float64]: T extends type.Float64 ? Float64Builder<TNull> : never; |
| [Type.Float]: T extends type.Float ? FloatBuilder<T, TNull> : never; |
| [Type.Utf8]: T extends type.Utf8 ? Utf8Builder<TNull> : never; |
| [Type.Binary]: T extends type.Binary ? BinaryBuilder<TNull> : never; |
| [Type.FixedSizeBinary]: T extends type.FixedSizeBinary ? FixedSizeBinaryBuilder<TNull> : never; |
| [Type.Date]: T extends type.Date_ ? DateBuilder<T, TNull> : never; |
| [Type.DateDay]: T extends type.DateDay ? DateDayBuilder<TNull> : never; |
| [Type.DateMillisecond]: T extends type.DateMillisecond ? DateMillisecondBuilder<TNull> : never; |
| [Type.Timestamp]: T extends type.Timestamp ? TimestampBuilder<T, TNull> : never; |
| [Type.TimestampSecond]: T extends type.TimestampSecond ? TimestampSecondBuilder<TNull> : never; |
| [Type.TimestampMillisecond]: T extends type.TimestampMillisecond ? TimestampMillisecondBuilder<TNull> : never; |
| [Type.TimestampMicrosecond]: T extends type.TimestampMicrosecond ? TimestampMicrosecondBuilder<TNull> : never; |
| [Type.TimestampNanosecond]: T extends type.TimestampNanosecond ? TimestampNanosecondBuilder<TNull> : never; |
| [Type.Time]: T extends type.Time ? TimeBuilder<T, TNull> : never; |
| [Type.TimeSecond]: T extends type.TimeSecond ? TimeSecondBuilder<TNull> : never; |
| [Type.TimeMillisecond]: T extends type.TimeMillisecond ? TimeMillisecondBuilder<TNull> : never; |
| [Type.TimeMicrosecond]: T extends type.TimeMicrosecond ? TimeMicrosecondBuilder<TNull> : never; |
| [Type.TimeNanosecond]: T extends type.TimeNanosecond ? TimeNanosecondBuilder<TNull> : never; |
| [Type.Decimal]: T extends type.Decimal ? DecimalBuilder<TNull> : never; |
| [Type.Union]: T extends type.Union ? UnionBuilder<T, TNull> : never; |
| [Type.DenseUnion]: T extends type.DenseUnion ? DenseUnionBuilder<T, TNull> : never; |
| [Type.SparseUnion]: T extends type.SparseUnion ? SparseUnionBuilder<T, TNull> : never; |
| [Type.Interval]: T extends type.Interval ? IntervalBuilder<T, TNull> : never; |
| [Type.IntervalDayTime]: T extends type.IntervalDayTime ? IntervalDayTimeBuilder<TNull> : never; |
| [Type.IntervalYearMonth]: T extends type.IntervalYearMonth ? IntervalYearMonthBuilder<TNull> : never; |
| [Type.Map]: T extends type.Map_ ? MapBuilder<T['keyType'], T['valueType'], TNull> : never; |
| [Type.List]: T extends type.List ? ListBuilder<T['valueType'], TNull> : never; |
| [Type.Struct]: T extends type.Struct ? StructBuilder<T['dataTypes'], TNull> : never; |
| [Type.Dictionary]: T extends type.Dictionary ? DictionaryBuilder<T, TNull> : never; |
| [Type.FixedSizeList]: T extends type.FixedSizeList ? FixedSizeListBuilder<T['valueType'], TNull> : never; |
| }[T['TType']]; |