Spaces:
Sleeping
Sleeping
| ; | |
| // 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. | |
| Object.defineProperty(exports, "__esModule", { value: true }); | |
| const data_1 = require("../data"); | |
| const vector_1 = require("../vector"); | |
| const base_1 = require("./base"); | |
| const index_1 = require("./index"); | |
| const type_1 = require("../type"); | |
| /** @ignore */ | |
| class FloatVector extends base_1.BaseVector { | |
| /** @nocollapse */ | |
| static from(input) { | |
| let ArrowType = vectorTypeToDataType(this); | |
| if ((input instanceof ArrayBuffer) || ArrayBuffer.isView(input)) { | |
| let InputType = arrayTypeToDataType(input.constructor) || ArrowType; | |
| // Special case, infer the Arrow DataType from the input if calling the base | |
| // FloatVector.from with a TypedArray, e.g. `FloatVector.from(new Float32Array())` | |
| if (ArrowType === null) { | |
| ArrowType = InputType; | |
| } | |
| // If the DataType inferred from the Vector constructor matches the | |
| // DataType inferred from the input arguments, return zero-copy view | |
| if (ArrowType && ArrowType === InputType) { | |
| let type = new ArrowType(); | |
| let length = input.byteLength / type.ArrayType.BYTES_PER_ELEMENT; | |
| // If the ArrowType is Float16 but the input type isn't a Uint16Array, | |
| // let the Float16Builder handle casting the input values to Uint16s. | |
| if (!convertTo16Bit(ArrowType, input.constructor)) { | |
| return vector_1.Vector.new(data_1.Data.Float(type, 0, length, 0, null, input)); | |
| } | |
| } | |
| } | |
| if (ArrowType) { | |
| // If the DataType inferred from the Vector constructor is different than | |
| // the DataType inferred from the input TypedArray, or if input isn't a | |
| // TypedArray, use the Builders to construct the result Vector | |
| return index_1.vectorFromValuesWithType(() => new ArrowType(), input); | |
| } | |
| if ((input instanceof DataView) || (input instanceof ArrayBuffer)) { | |
| throw new TypeError(`Cannot infer float type from instance of ${input.constructor.name}`); | |
| } | |
| throw new TypeError('Unrecognized FloatVector input'); | |
| } | |
| } | |
| exports.FloatVector = FloatVector; | |
| /** @ignore */ | |
| class Float16Vector extends FloatVector { | |
| // Since JS doesn't have half floats, `toArray()` returns a zero-copy slice | |
| // of the underlying Uint16Array data. This behavior ensures we don't incur | |
| // extra compute or copies if you're calling `toArray()` in order to create | |
| // a buffer for something like WebGL. Buf if you're using JS and want typed | |
| // arrays of 4-to-8-byte precision, these methods will enumerate the values | |
| // and clamp to the desired byte lengths. | |
| toFloat32Array() { return new Float32Array(this); } | |
| toFloat64Array() { return new Float64Array(this); } | |
| } | |
| exports.Float16Vector = Float16Vector; | |
| /** @ignore */ | |
| class Float32Vector extends FloatVector { | |
| } | |
| exports.Float32Vector = Float32Vector; | |
| /** @ignore */ | |
| class Float64Vector extends FloatVector { | |
| } | |
| exports.Float64Vector = Float64Vector; | |
| const convertTo16Bit = (typeCtor, dataCtor) => { | |
| return (typeCtor === type_1.Float16) && (dataCtor !== Uint16Array); | |
| }; | |
| /** @ignore */ | |
| const arrayTypeToDataType = (ctor) => { | |
| switch (ctor) { | |
| case Uint16Array: return type_1.Float16; | |
| case Float32Array: return type_1.Float32; | |
| case Float64Array: return type_1.Float64; | |
| default: return null; | |
| } | |
| }; | |
| /** @ignore */ | |
| const vectorTypeToDataType = (ctor) => { | |
| switch (ctor) { | |
| case Float16Vector: return type_1.Float16; | |
| case Float32Vector: return type_1.Float32; | |
| case Float64Vector: return type_1.Float64; | |
| default: return null; | |
| } | |
| }; | |
| //# sourceMappingURL=float.js.map | |