flatbuffers/js/flexbuffers/reference.js

Memory Efficient Serialization Library

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.Reference = exports.toReference = void 0;
var bit_width_util_js_1 = require("./bit-width-util.js");
var value_type_js_1 = require("./value-type.js");
var value_type_util_js_1 = require("./value-type-util.js");
var reference_util_js_1 = require("./reference-util.js");
var flexbuffers_util_js_1 = require("./flexbuffers-util.js");
var bit_width_js_1 = require("./bit-width.js");
function toReference(buffer) {
    var len = buffer.byteLength;
    if (len < 3) {
        throw "Buffer needs to be bigger than 3";
    }
    var dataView = new DataView(buffer);
    var byteWidth = dataView.getUint8(len - 1);
    var packedType = dataView.getUint8(len - 2);
    var parentWidth = (0, bit_width_util_js_1.fromByteWidth)(byteWidth);
    var offset = len - byteWidth - 2;
    return new Reference(dataView, offset, parentWidth, packedType, "/");
}
exports.toReference = toReference;
function valueForIndexWithKey(index, key, dataView, offset, parentWidth, byteWidth, length, path) {
    var _indirect = (0, reference_util_js_1.indirect)(dataView, offset, parentWidth);
    var elementOffset = _indirect + index * byteWidth;
    var packedType = dataView.getUint8(_indirect + length * byteWidth + index);
    return new Reference(dataView, elementOffset, (0, bit_width_util_js_1.fromByteWidth)(byteWidth), packedType, "".concat(path, "/").concat(key));
}
var Reference = /** @class */ (function () {
    function Reference(dataView, offset, parentWidth, packedType, path) {
        this.dataView = dataView;
        this.offset = offset;
        this.parentWidth = parentWidth;
        this.packedType = packedType;
        this.path = path;
        this._length = -1;
        this.byteWidth = 1 << (packedType & 3);
        this.valueType = packedType >> 2;
    }
    Reference.prototype.isNull = function () { return this.valueType === value_type_js_1.ValueType.NULL; };
    Reference.prototype.isNumber = function () { return (0, value_type_util_js_1.isNumber)(this.valueType) || (0, value_type_util_js_1.isIndirectNumber)(this.valueType); };
    Reference.prototype.isFloat = function () { return value_type_js_1.ValueType.FLOAT === this.valueType || value_type_js_1.ValueType.INDIRECT_FLOAT === this.valueType; };
    Reference.prototype.isInt = function () { return this.isNumber() && !this.isFloat(); };
    Reference.prototype.isString = function () { return value_type_js_1.ValueType.STRING === this.valueType || value_type_js_1.ValueType.KEY === this.valueType; };
    Reference.prototype.isBool = function () { return value_type_js_1.ValueType.BOOL === this.valueType; };
    Reference.prototype.isBlob = function () { return value_type_js_1.ValueType.BLOB === this.valueType; };
    Reference.prototype.isVector = function () { return (0, value_type_util_js_1.isAVector)(this.valueType); };
    Reference.prototype.isMap = function () { return value_type_js_1.ValueType.MAP === this.valueType; };
    Reference.prototype.boolValue = function () {
        if (this.isBool()) {
            return (0, reference_util_js_1.readInt)(this.dataView, this.offset, this.parentWidth) > 0;
        }
        return null;
    };
    Reference.prototype.intValue = function () {
        if (this.valueType === value_type_js_1.ValueType.INT) {
            return (0, reference_util_js_1.readInt)(this.dataView, this.offset, this.parentWidth);
        }
        if (this.valueType === value_type_js_1.ValueType.UINT) {
            return (0, reference_util_js_1.readUInt)(this.dataView, this.offset, this.parentWidth);
        }
        if (this.valueType === value_type_js_1.ValueType.INDIRECT_INT) {
            return (0, reference_util_js_1.readInt)(this.dataView, (0, reference_util_js_1.indirect)(this.dataView, this.offset, this.parentWidth), (0, bit_width_util_js_1.fromByteWidth)(this.byteWidth));
        }
        if (this.valueType === value_type_js_1.ValueType.INDIRECT_UINT) {
            return (0, reference_util_js_1.readUInt)(this.dataView, (0, reference_util_js_1.indirect)(this.dataView, this.offset, this.parentWidth), (0, bit_width_util_js_1.fromByteWidth)(this.byteWidth));
        }
        return null;
    };
    Reference.prototype.floatValue = function () {
        if (this.valueType === value_type_js_1.ValueType.FLOAT) {
            return (0, reference_util_js_1.readFloat)(this.dataView, this.offset, this.parentWidth);
        }
        if (this.valueType === value_type_js_1.ValueType.INDIRECT_FLOAT) {
            return (0, reference_util_js_1.readFloat)(this.dataView, (0, reference_util_js_1.indirect)(this.dataView, this.offset, this.parentWidth), (0, bit_width_util_js_1.fromByteWidth)(this.byteWidth));
        }
        return null;
    };
    Reference.prototype.numericValue = function () { return this.floatValue() || this.intValue(); };
    Reference.prototype.stringValue = function () {
        if (this.valueType === value_type_js_1.ValueType.STRING || this.valueType === value_type_js_1.ValueType.KEY) {
            var begin = (0, reference_util_js_1.indirect)(this.dataView, this.offset, this.parentWidth);
            return (0, flexbuffers_util_js_1.fromUTF8Array)(new Uint8Array(this.dataView.buffer, begin, this.length()));
        }
        return null;
    };
    Reference.prototype.blobValue = function () {
        if (this.isBlob()) {
            var begin = (0, reference_util_js_1.indirect)(this.dataView, this.offset, this.parentWidth);
            return new Uint8Array(this.dataView.buffer, begin, this.length());
        }
        return null;
    };
    Reference.prototype.get = function (key) {
        var length = this.length();
        if (Number.isInteger(key) && (0, value_type_util_js_1.isAVector)(this.valueType)) {
            if (key >= length || key < 0) {
                throw "Key: [".concat(key, "] is not applicable on ").concat(this.path, " of ").concat(this.valueType, " length: ").concat(length);
            }
            var _indirect = (0, reference_util_js_1.indirect)(this.dataView, this.offset, this.parentWidth);
            var elementOffset = _indirect + key * this.byteWidth;
            var _packedType = this.dataView.getUint8(_indirect + length * this.byteWidth + key);
            if ((0, value_type_util_js_1.isTypedVector)(this.valueType)) {
                var _valueType = (0, value_type_util_js_1.typedVectorElementType)(this.valueType);
                _packedType = (0, value_type_util_js_1.packedType)(_valueType, bit_width_js_1.BitWidth.WIDTH8);
            }
            else if ((0, value_type_util_js_1.isFixedTypedVector)(this.valueType)) {
                var _valueType = (0, value_type_util_js_1.fixedTypedVectorElementType)(this.valueType);
                _packedType = (0, value_type_util_js_1.packedType)(_valueType, bit_width_js_1.BitWidth.WIDTH8);
            }
            return new Reference(this.dataView, elementOffset, (0, bit_width_util_js_1.fromByteWidth)(this.byteWidth), _packedType, "".concat(this.path, "[").concat(key, "]"));
        }
        if (typeof key === 'string') {
            var index = (0, reference_util_js_1.keyIndex)(key, this.dataView, this.offset, this.parentWidth, this.byteWidth, length);
            if (index !== null) {
                return valueForIndexWithKey(index, key, this.dataView, this.offset, this.parentWidth, this.byteWidth, length, this.path);
            }
        }
        throw "Key [".concat(key, "] is not applicable on ").concat(this.path, " of ").concat(this.valueType);
    };
    Reference.prototype.length = function () {
        var size;
        if (this._length > -1) {
            return this._length;
        }
        if ((0, value_type_util_js_1.isFixedTypedVector)(this.valueType)) {
            this._length = (0, value_type_util_js_1.fixedTypedVectorElementSize)(this.valueType);
        }
        else if (this.valueType === value_type_js_1.ValueType.BLOB
            || this.valueType === value_type_js_1.ValueType.MAP
            || (0, value_type_util_js_1.isAVector)(this.valueType)) {
            this._length = (0, reference_util_js_1.readUInt)(this.dataView, (0, reference_util_js_1.indirect)(this.dataView, this.offset, this.parentWidth) - this.byteWidth, (0, bit_width_util_js_1.fromByteWidth)(this.byteWidth));
        }
        else if (this.valueType === value_type_js_1.ValueType.NULL) {
            this._length = 0;
        }
        else if (this.valueType === value_type_js_1.ValueType.STRING) {
            var _indirect = (0, reference_util_js_1.indirect)(this.dataView, this.offset, this.parentWidth);
            var sizeByteWidth = this.byteWidth;
            size = (0, reference_util_js_1.readUInt)(this.dataView, _indirect - sizeByteWidth, (0, bit_width_util_js_1.fromByteWidth)(this.byteWidth));
            while (this.dataView.getInt8(_indirect + size) !== 0) {
                sizeByteWidth <<= 1;
                size = (0, reference_util_js_1.readUInt)(this.dataView, _indirect - sizeByteWidth, (0, bit_width_util_js_1.fromByteWidth)(this.byteWidth));
            }
            this._length = size;
        }
        else if (this.valueType === value_type_js_1.ValueType.KEY) {
            var _indirect = (0, reference_util_js_1.indirect)(this.dataView, this.offset, this.parentWidth);
            size = 1;
            while (this.dataView.getInt8(_indirect + size) !== 0) {
                size++;
            }
            this._length = size;
        }
        else {
            this._length = 1;
        }
        return Number(this._length);
    };
    Reference.prototype.toObject = function () {
        var length = this.length();
        if (this.isVector()) {
            var result = [];
            for (var i = 0; i < length; i++) {
                result.push(this.get(i).toObject());
            }
            return result;
        }
        if (this.isMap()) {
            var result = {};
            for (var i = 0; i < length; i++) {
                var key = (0, reference_util_js_1.keyForIndex)(i, this.dataView, this.offset, this.parentWidth, this.byteWidth);
                result[key] = valueForIndexWithKey(i, key, this.dataView, this.offset, this.parentWidth, this.byteWidth, length, this.path).toObject();
            }
            return result;
        }
        if (this.isNull()) {
            return null;
        }
        if (this.isBool()) {
            return this.boolValue();
        }
        if (this.isNumber()) {
            return this.numericValue();
        }
        return this.blobValue() || this.stringValue();
    };
    return Reference;
}());
exports.Reference = Reference;