Spaces:
Sleeping
Sleeping
File size: 4,577 Bytes
be5030f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 |
"use strict";
// 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
|