| // 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 { Type, Precision, DateUnit, TimeUnit, IntervalUnit, UnionMode } from './enum.js'; |
| import { DataType, Float, Int, Date_, Interval, Time, Timestamp, Union, } from './type.js'; |
| |
| export abstract class Visitor { |
| public visitMany(nodes: any[], ...args: any[][]) { |
| return nodes.map((node, i) => this.visit(node, ...args.map((x) => x[i]))); |
| } |
| public visit(...args: any[]) { |
| return this.getVisitFn(args[0], false).apply(this, args); |
| } |
| public getVisitFn(node: any, throwIfNotFound = true) { |
| return getVisitFn(this, node, throwIfNotFound); |
| } |
| public getVisitFnByTypeId(typeId: Type, throwIfNotFound = true) { |
| return getVisitFnByTypeId(this, typeId, throwIfNotFound); |
| } |
| public visitNull(_node: any, ..._args: any[]): any { return null; } |
| public visitBool(_node: any, ..._args: any[]): any { return null; } |
| public visitInt(_node: any, ..._args: any[]): any { return null; } |
| public visitFloat(_node: any, ..._args: any[]): any { return null; } |
| public visitUtf8(_node: any, ..._args: any[]): any { return null; } |
| public visitBinary(_node: any, ..._args: any[]): any { return null; } |
| public visitFixedSizeBinary(_node: any, ..._args: any[]): any { return null; } |
| public visitDate(_node: any, ..._args: any[]): any { return null; } |
| public visitTimestamp(_node: any, ..._args: any[]): any { return null; } |
| public visitTime(_node: any, ..._args: any[]): any { return null; } |
| public visitDecimal(_node: any, ..._args: any[]): any { return null; } |
| public visitList(_node: any, ..._args: any[]): any { return null; } |
| public visitStruct(_node: any, ..._args: any[]): any { return null; } |
| public visitUnion(_node: any, ..._args: any[]): any { return null; } |
| public visitDictionary(_node: any, ..._args: any[]): any { return null; } |
| public visitInterval(_node: any, ..._args: any[]): any { return null; } |
| public visitFixedSizeList(_node: any, ..._args: any[]): any { return null; } |
| public visitMap(_node: any, ..._args: any[]): any { return null; } |
| } |
| |
| /** @ignore */ |
| function getVisitFn<T extends DataType>(visitor: Visitor, node: any, throwIfNotFound = true) { |
| if (typeof node === 'number') { |
| return getVisitFnByTypeId(visitor, node, throwIfNotFound); |
| } |
| if (typeof node === 'string' && (node in Type)) { |
| return getVisitFnByTypeId(visitor, Type[node as keyof typeof Type], throwIfNotFound); |
| } |
| if (node && (node instanceof DataType)) { |
| return getVisitFnByTypeId(visitor, inferDType(node as T), throwIfNotFound); |
| } |
| if (node?.type && (node.type instanceof DataType)) { |
| return getVisitFnByTypeId(visitor, inferDType(node.type as T), throwIfNotFound); |
| } |
| return getVisitFnByTypeId(visitor, Type.NONE, throwIfNotFound); |
| } |
| |
| /** @ignore */ |
| function getVisitFnByTypeId(visitor: Visitor, dtype: Type, throwIfNotFound = true) { |
| let fn: any = null; |
| switch (dtype) { |
| case Type.Null: fn = visitor.visitNull; break; |
| case Type.Bool: fn = visitor.visitBool; break; |
| case Type.Int: fn = visitor.visitInt; break; |
| case Type.Int8: fn = visitor.visitInt8 || visitor.visitInt; break; |
| case Type.Int16: fn = visitor.visitInt16 || visitor.visitInt; break; |
| case Type.Int32: fn = visitor.visitInt32 || visitor.visitInt; break; |
| case Type.Int64: fn = visitor.visitInt64 || visitor.visitInt; break; |
| case Type.Uint8: fn = visitor.visitUint8 || visitor.visitInt; break; |
| case Type.Uint16: fn = visitor.visitUint16 || visitor.visitInt; break; |
| case Type.Uint32: fn = visitor.visitUint32 || visitor.visitInt; break; |
| case Type.Uint64: fn = visitor.visitUint64 || visitor.visitInt; break; |
| case Type.Float: fn = visitor.visitFloat; break; |
| case Type.Float16: fn = visitor.visitFloat16 || visitor.visitFloat; break; |
| case Type.Float32: fn = visitor.visitFloat32 || visitor.visitFloat; break; |
| case Type.Float64: fn = visitor.visitFloat64 || visitor.visitFloat; break; |
| case Type.Utf8: fn = visitor.visitUtf8; break; |
| case Type.Binary: fn = visitor.visitBinary; break; |
| case Type.FixedSizeBinary: fn = visitor.visitFixedSizeBinary; break; |
| case Type.Date: fn = visitor.visitDate; break; |
| case Type.DateDay: fn = visitor.visitDateDay || visitor.visitDate; break; |
| case Type.DateMillisecond: fn = visitor.visitDateMillisecond || visitor.visitDate; break; |
| case Type.Timestamp: fn = visitor.visitTimestamp; break; |
| case Type.TimestampSecond: fn = visitor.visitTimestampSecond || visitor.visitTimestamp; break; |
| case Type.TimestampMillisecond: fn = visitor.visitTimestampMillisecond || visitor.visitTimestamp; break; |
| case Type.TimestampMicrosecond: fn = visitor.visitTimestampMicrosecond || visitor.visitTimestamp; break; |
| case Type.TimestampNanosecond: fn = visitor.visitTimestampNanosecond || visitor.visitTimestamp; break; |
| case Type.Time: fn = visitor.visitTime; break; |
| case Type.TimeSecond: fn = visitor.visitTimeSecond || visitor.visitTime; break; |
| case Type.TimeMillisecond: fn = visitor.visitTimeMillisecond || visitor.visitTime; break; |
| case Type.TimeMicrosecond: fn = visitor.visitTimeMicrosecond || visitor.visitTime; break; |
| case Type.TimeNanosecond: fn = visitor.visitTimeNanosecond || visitor.visitTime; break; |
| case Type.Decimal: fn = visitor.visitDecimal; break; |
| case Type.List: fn = visitor.visitList; break; |
| case Type.Struct: fn = visitor.visitStruct; break; |
| case Type.Union: fn = visitor.visitUnion; break; |
| case Type.DenseUnion: fn = visitor.visitDenseUnion || visitor.visitUnion; break; |
| case Type.SparseUnion: fn = visitor.visitSparseUnion || visitor.visitUnion; break; |
| case Type.Dictionary: fn = visitor.visitDictionary; break; |
| case Type.Interval: fn = visitor.visitInterval; break; |
| case Type.IntervalDayTime: fn = visitor.visitIntervalDayTime || visitor.visitInterval; break; |
| case Type.IntervalYearMonth: fn = visitor.visitIntervalYearMonth || visitor.visitInterval; break; |
| case Type.FixedSizeList: fn = visitor.visitFixedSizeList; break; |
| case Type.Map: fn = visitor.visitMap; break; |
| } |
| if (typeof fn === 'function') return fn; |
| if (!throwIfNotFound) return () => null; |
| throw new Error(`Unrecognized type '${Type[dtype]}'`); |
| } |
| |
| /** @ignore */ |
| function inferDType<T extends DataType>(type: T): Type { |
| switch (type.typeId) { |
| case Type.Null: return Type.Null; |
| case Type.Int: { |
| const { bitWidth, isSigned } = (type as any as Int); |
| switch (bitWidth) { |
| case 8: return isSigned ? Type.Int8 : Type.Uint8; |
| case 16: return isSigned ? Type.Int16 : Type.Uint16; |
| case 32: return isSigned ? Type.Int32 : Type.Uint32; |
| case 64: return isSigned ? Type.Int64 : Type.Uint64; |
| } |
| // @ts-ignore |
| return Type.Int; |
| } |
| case Type.Float: |
| switch ((type as any as Float).precision) { |
| case Precision.HALF: return Type.Float16; |
| case Precision.SINGLE: return Type.Float32; |
| case Precision.DOUBLE: return Type.Float64; |
| } |
| // @ts-ignore |
| return Type.Float; |
| case Type.Binary: return Type.Binary; |
| case Type.Utf8: return Type.Utf8; |
| case Type.Bool: return Type.Bool; |
| case Type.Decimal: return Type.Decimal; |
| case Type.Time: |
| switch ((type as any as Time).unit) { |
| case TimeUnit.SECOND: return Type.TimeSecond; |
| case TimeUnit.MILLISECOND: return Type.TimeMillisecond; |
| case TimeUnit.MICROSECOND: return Type.TimeMicrosecond; |
| case TimeUnit.NANOSECOND: return Type.TimeNanosecond; |
| } |
| // @ts-ignore |
| return Type.Time; |
| case Type.Timestamp: |
| switch ((type as any as Timestamp).unit) { |
| case TimeUnit.SECOND: return Type.TimestampSecond; |
| case TimeUnit.MILLISECOND: return Type.TimestampMillisecond; |
| case TimeUnit.MICROSECOND: return Type.TimestampMicrosecond; |
| case TimeUnit.NANOSECOND: return Type.TimestampNanosecond; |
| } |
| // @ts-ignore |
| return Type.Timestamp; |
| case Type.Date: |
| switch ((type as any as Date_).unit) { |
| case DateUnit.DAY: return Type.DateDay; |
| case DateUnit.MILLISECOND: return Type.DateMillisecond; |
| } |
| // @ts-ignore |
| return Type.Date; |
| case Type.Interval: |
| switch ((type as any as Interval).unit) { |
| case IntervalUnit.DAY_TIME: return Type.IntervalDayTime; |
| case IntervalUnit.YEAR_MONTH: return Type.IntervalYearMonth; |
| } |
| // @ts-ignore |
| return Type.Interval; |
| case Type.Map: return Type.Map; |
| case Type.List: return Type.List; |
| case Type.Struct: return Type.Struct; |
| case Type.Union: |
| switch ((type as any as Union).mode) { |
| case UnionMode.Dense: return Type.DenseUnion; |
| case UnionMode.Sparse: return Type.SparseUnion; |
| } |
| // @ts-ignore |
| return Type.Union; |
| case Type.FixedSizeBinary: return Type.FixedSizeBinary; |
| case Type.FixedSizeList: return Type.FixedSizeList; |
| case Type.Dictionary: return Type.Dictionary; |
| } |
| throw new Error(`Unrecognized type '${Type[type.typeId]}'`); |
| } |
| |
| export interface Visitor { |
| visitNull(node: any, ...args: any[]): any; |
| visitBool(node: any, ...args: any[]): any; |
| visitInt(node: any, ...args: any[]): any; |
| visitInt8?(node: any, ...args: any[]): any; |
| visitInt16?(node: any, ...args: any[]): any; |
| visitInt32?(node: any, ...args: any[]): any; |
| visitInt64?(node: any, ...args: any[]): any; |
| visitUint8?(node: any, ...args: any[]): any; |
| visitUint16?(node: any, ...args: any[]): any; |
| visitUint32?(node: any, ...args: any[]): any; |
| visitUint64?(node: any, ...args: any[]): any; |
| visitFloat(node: any, ...args: any[]): any; |
| visitFloat16?(node: any, ...args: any[]): any; |
| visitFloat32?(node: any, ...args: any[]): any; |
| visitFloat64?(node: any, ...args: any[]): any; |
| visitUtf8(node: any, ...args: any[]): any; |
| visitBinary(node: any, ...args: any[]): any; |
| visitFixedSizeBinary(node: any, ...args: any[]): any; |
| visitDate(node: any, ...args: any[]): any; |
| visitDateDay?(node: any, ...args: any[]): any; |
| visitDateMillisecond?(node: any, ...args: any[]): any; |
| visitTimestamp(node: any, ...args: any[]): any; |
| visitTimestampSecond?(node: any, ...args: any[]): any; |
| visitTimestampMillisecond?(node: any, ...args: any[]): any; |
| visitTimestampMicrosecond?(node: any, ...args: any[]): any; |
| visitTimestampNanosecond?(node: any, ...args: any[]): any; |
| visitTime(node: any, ...args: any[]): any; |
| visitTimeSecond?(node: any, ...args: any[]): any; |
| visitTimeMillisecond?(node: any, ...args: any[]): any; |
| visitTimeMicrosecond?(node: any, ...args: any[]): any; |
| visitTimeNanosecond?(node: any, ...args: any[]): any; |
| visitDecimal(node: any, ...args: any[]): any; |
| visitList(node: any, ...args: any[]): any; |
| visitStruct(node: any, ...args: any[]): any; |
| visitUnion(node: any, ...args: any[]): any; |
| visitDenseUnion?(node: any, ...args: any[]): any; |
| visitSparseUnion?(node: any, ...args: any[]): any; |
| visitDictionary(node: any, ...args: any[]): any; |
| visitInterval(node: any, ...args: any[]): any; |
| visitIntervalDayTime?(node: any, ...args: any[]): any; |
| visitIntervalYearMonth?(node: any, ...args: any[]): any; |
| visitFixedSizeList(node: any, ...args: any[]): any; |
| visitMap(node: any, ...args: any[]): any; |
| } |
| |
| // Add these here so they're picked up by the externs creator |
| // in the build, and closure-compiler doesn't minify them away |
| (Visitor.prototype as any).visitInt8 = null; |
| (Visitor.prototype as any).visitInt16 = null; |
| (Visitor.prototype as any).visitInt32 = null; |
| (Visitor.prototype as any).visitInt64 = null; |
| (Visitor.prototype as any).visitUint8 = null; |
| (Visitor.prototype as any).visitUint16 = null; |
| (Visitor.prototype as any).visitUint32 = null; |
| (Visitor.prototype as any).visitUint64 = null; |
| (Visitor.prototype as any).visitFloat16 = null; |
| (Visitor.prototype as any).visitFloat32 = null; |
| (Visitor.prototype as any).visitFloat64 = null; |
| (Visitor.prototype as any).visitDateDay = null; |
| (Visitor.prototype as any).visitDateMillisecond = null; |
| (Visitor.prototype as any).visitTimestampSecond = null; |
| (Visitor.prototype as any).visitTimestampMillisecond = null; |
| (Visitor.prototype as any).visitTimestampMicrosecond = null; |
| (Visitor.prototype as any).visitTimestampNanosecond = null; |
| (Visitor.prototype as any).visitTimeSecond = null; |
| (Visitor.prototype as any).visitTimeMillisecond = null; |
| (Visitor.prototype as any).visitTimeMicrosecond = null; |
| (Visitor.prototype as any).visitTimeNanosecond = null; |
| (Visitor.prototype as any).visitDenseUnion = null; |
| (Visitor.prototype as any).visitSparseUnion = null; |
| (Visitor.prototype as any).visitIntervalDayTime = null; |
| (Visitor.prototype as any).visitIntervalYearMonth = null; |
