qd-changjing/public/static/Build/CesiumUnminified/Workers/decodeDraco.js

406 lines
12 KiB
JavaScript
Raw Normal View History

2022-07-05 16:56:29 +08:00
/**
* Cesium - https://github.com/CesiumGS/cesium
*
* Copyright 2011-2020 Cesium Contributors
*
* Licensed 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.
*
* Columbus View (Pat. Pend.)
*
* Portions licensed separately.
* See https://github.com/CesiumGS/cesium/blob/main/LICENSE.md for full licensing details.
*/
define(['./ComponentDatatype-aad54330', './when-4bbc8319', './IndexDatatype-6739e544', './RuntimeError-5b082e8f', './createTaskProcessorWorker', './WebGLConstants-508b9636'], (function (ComponentDatatype, when, IndexDatatype, RuntimeError, createTaskProcessorWorker, WebGLConstants) { 'use strict';
/* global require */
let draco;
function decodeIndexArray(dracoGeometry, dracoDecoder) {
const numPoints = dracoGeometry.num_points();
const numFaces = dracoGeometry.num_faces();
const faceIndices = new draco.DracoInt32Array();
const numIndices = numFaces * 3;
const indexArray = IndexDatatype.IndexDatatype.createTypedArray(numPoints, numIndices);
let offset = 0;
for (let i = 0; i < numFaces; ++i) {
dracoDecoder.GetFaceFromMesh(dracoGeometry, i, faceIndices);
indexArray[offset + 0] = faceIndices.GetValue(0);
indexArray[offset + 1] = faceIndices.GetValue(1);
indexArray[offset + 2] = faceIndices.GetValue(2);
offset += 3;
}
draco.destroy(faceIndices);
return {
typedArray: indexArray,
numberOfIndices: numIndices,
};
}
function decodeQuantizedDracoTypedArray(
dracoGeometry,
dracoDecoder,
dracoAttribute,
quantization,
vertexArrayLength
) {
let vertexArray;
let attributeData;
if (quantization.quantizationBits <= 8) {
attributeData = new draco.DracoUInt8Array();
vertexArray = new Uint8Array(vertexArrayLength);
dracoDecoder.GetAttributeUInt8ForAllPoints(
dracoGeometry,
dracoAttribute,
attributeData
);
} else {
attributeData = new draco.DracoUInt16Array();
vertexArray = new Uint16Array(vertexArrayLength);
dracoDecoder.GetAttributeUInt16ForAllPoints(
dracoGeometry,
dracoAttribute,
attributeData
);
}
for (let i = 0; i < vertexArrayLength; ++i) {
vertexArray[i] = attributeData.GetValue(i);
}
draco.destroy(attributeData);
return vertexArray;
}
function decodeDracoTypedArray(
dracoGeometry,
dracoDecoder,
dracoAttribute,
vertexArrayLength
) {
let vertexArray;
let attributeData;
// Some attribute types are casted down to 32 bit since Draco only returns 32 bit values
switch (dracoAttribute.data_type()) {
case 1:
case 11: // DT_INT8 or DT_BOOL
attributeData = new draco.DracoInt8Array();
vertexArray = new Int8Array(vertexArrayLength);
dracoDecoder.GetAttributeInt8ForAllPoints(
dracoGeometry,
dracoAttribute,
attributeData
);
break;
case 2: // DT_UINT8
attributeData = new draco.DracoUInt8Array();
vertexArray = new Uint8Array(vertexArrayLength);
dracoDecoder.GetAttributeUInt8ForAllPoints(
dracoGeometry,
dracoAttribute,
attributeData
);
break;
case 3: // DT_INT16
attributeData = new draco.DracoInt16Array();
vertexArray = new Int16Array(vertexArrayLength);
dracoDecoder.GetAttributeInt16ForAllPoints(
dracoGeometry,
dracoAttribute,
attributeData
);
break;
case 4: // DT_UINT16
attributeData = new draco.DracoUInt16Array();
vertexArray = new Uint16Array(vertexArrayLength);
dracoDecoder.GetAttributeUInt16ForAllPoints(
dracoGeometry,
dracoAttribute,
attributeData
);
break;
case 5:
case 7: // DT_INT32 or DT_INT64
attributeData = new draco.DracoInt32Array();
vertexArray = new Int32Array(vertexArrayLength);
dracoDecoder.GetAttributeInt32ForAllPoints(
dracoGeometry,
dracoAttribute,
attributeData
);
break;
case 6:
case 8: // DT_UINT32 or DT_UINT64
attributeData = new draco.DracoUInt32Array();
vertexArray = new Uint32Array(vertexArrayLength);
dracoDecoder.GetAttributeUInt32ForAllPoints(
dracoGeometry,
dracoAttribute,
attributeData
);
break;
case 9:
case 10: // DT_FLOAT32 or DT_FLOAT64
attributeData = new draco.DracoFloat32Array();
vertexArray = new Float32Array(vertexArrayLength);
dracoDecoder.GetAttributeFloatForAllPoints(
dracoGeometry,
dracoAttribute,
attributeData
);
break;
}
for (let i = 0; i < vertexArrayLength; ++i) {
vertexArray[i] = attributeData.GetValue(i);
}
draco.destroy(attributeData);
return vertexArray;
}
function decodeAttribute(dracoGeometry, dracoDecoder, dracoAttribute) {
const numPoints = dracoGeometry.num_points();
const numComponents = dracoAttribute.num_components();
let quantization;
let transform = new draco.AttributeQuantizationTransform();
if (transform.InitFromAttribute(dracoAttribute)) {
const minValues = new Array(numComponents);
for (let i = 0; i < numComponents; ++i) {
minValues[i] = transform.min_value(i);
}
quantization = {
quantizationBits: transform.quantization_bits(),
minValues: minValues,
range: transform.range(),
octEncoded: false,
};
}
draco.destroy(transform);
transform = new draco.AttributeOctahedronTransform();
if (transform.InitFromAttribute(dracoAttribute)) {
quantization = {
quantizationBits: transform.quantization_bits(),
octEncoded: true,
};
}
draco.destroy(transform);
const vertexArrayLength = numPoints * numComponents;
let vertexArray;
if (when.defined(quantization)) {
vertexArray = decodeQuantizedDracoTypedArray(
dracoGeometry,
dracoDecoder,
dracoAttribute,
quantization,
vertexArrayLength
);
} else {
vertexArray = decodeDracoTypedArray(
dracoGeometry,
dracoDecoder,
dracoAttribute,
vertexArrayLength
);
}
const componentDatatype = ComponentDatatype.ComponentDatatype.fromTypedArray(vertexArray);
return {
array: vertexArray,
data: {
componentsPerAttribute: numComponents,
componentDatatype: componentDatatype,
byteOffset: dracoAttribute.byte_offset(),
byteStride:
ComponentDatatype.ComponentDatatype.getSizeInBytes(componentDatatype) * numComponents,
normalized: dracoAttribute.normalized(),
quantization: quantization,
},
};
}
function decodePointCloud(parameters) {
const dracoDecoder = new draco.Decoder();
if (parameters.dequantizeInShader) {
dracoDecoder.SkipAttributeTransform(draco.POSITION);
dracoDecoder.SkipAttributeTransform(draco.NORMAL);
}
const buffer = new draco.DecoderBuffer();
buffer.Init(parameters.buffer, parameters.buffer.length);
const geometryType = dracoDecoder.GetEncodedGeometryType(buffer);
if (geometryType !== draco.POINT_CLOUD) {
throw new RuntimeError.RuntimeError("Draco geometry type must be POINT_CLOUD.");
}
const dracoPointCloud = new draco.PointCloud();
const decodingStatus = dracoDecoder.DecodeBufferToPointCloud(
buffer,
dracoPointCloud
);
if (!decodingStatus.ok() || dracoPointCloud.ptr === 0) {
throw new RuntimeError.RuntimeError(
`Error decoding draco point cloud: ${decodingStatus.error_msg()}`
);
}
draco.destroy(buffer);
const result = {};
const properties = parameters.properties;
for (const propertyName in properties) {
if (properties.hasOwnProperty(propertyName)) {
let dracoAttribute;
if (propertyName === "POSITION" || propertyName === "NORMAL") {
const dracoAttributeId = dracoDecoder.GetAttributeId(
dracoPointCloud,
draco[propertyName]
);
dracoAttribute = dracoDecoder.GetAttribute(
dracoPointCloud,
dracoAttributeId
);
} else {
const attributeId = properties[propertyName];
dracoAttribute = dracoDecoder.GetAttributeByUniqueId(
dracoPointCloud,
attributeId
);
}
result[propertyName] = decodeAttribute(
dracoPointCloud,
dracoDecoder,
dracoAttribute
);
}
}
draco.destroy(dracoPointCloud);
draco.destroy(dracoDecoder);
return result;
}
function decodePrimitive(parameters) {
const dracoDecoder = new draco.Decoder();
// Skip all parameter types except generic
const attributesToSkip = ["POSITION", "NORMAL", "COLOR", "TEX_COORD"];
if (parameters.dequantizeInShader) {
for (let i = 0; i < attributesToSkip.length; ++i) {
dracoDecoder.SkipAttributeTransform(draco[attributesToSkip[i]]);
}
}
const bufferView = parameters.bufferView;
const buffer = new draco.DecoderBuffer();
buffer.Init(parameters.array, bufferView.byteLength);
const geometryType = dracoDecoder.GetEncodedGeometryType(buffer);
if (geometryType !== draco.TRIANGULAR_MESH) {
throw new RuntimeError.RuntimeError("Unsupported draco mesh geometry type.");
}
const dracoGeometry = new draco.Mesh();
const decodingStatus = dracoDecoder.DecodeBufferToMesh(buffer, dracoGeometry);
if (!decodingStatus.ok() || dracoGeometry.ptr === 0) {
throw new RuntimeError.RuntimeError(
`Error decoding draco mesh geometry: ${decodingStatus.error_msg()}`
);
}
draco.destroy(buffer);
const attributeData = {};
const compressedAttributes = parameters.compressedAttributes;
for (const attributeName in compressedAttributes) {
if (compressedAttributes.hasOwnProperty(attributeName)) {
const compressedAttribute = compressedAttributes[attributeName];
const dracoAttribute = dracoDecoder.GetAttributeByUniqueId(
dracoGeometry,
compressedAttribute
);
attributeData[attributeName] = decodeAttribute(
dracoGeometry,
dracoDecoder,
dracoAttribute
);
}
}
const result = {
indexArray: decodeIndexArray(dracoGeometry, dracoDecoder),
attributeData: attributeData,
};
draco.destroy(dracoGeometry);
draco.destroy(dracoDecoder);
return result;
}
function decode(parameters) {
if (when.defined(parameters.bufferView)) {
return decodePrimitive(parameters);
}
return decodePointCloud(parameters);
}
function initWorker(dracoModule) {
draco = dracoModule;
self.onmessage = createTaskProcessorWorker(decode);
self.postMessage(true);
}
function decodeDraco(event) {
const data = event.data;
// Expect the first message to be to load a web assembly module
const wasmConfig = data.webAssemblyConfig;
if (when.defined(wasmConfig)) {
// Require and compile WebAssembly module, or use fallback if not supported
return require([wasmConfig.modulePath], function (dracoModule) {
if (when.defined(wasmConfig.wasmBinaryFile)) {
if (!when.defined(dracoModule)) {
dracoModule = self.DracoDecoderModule;
}
dracoModule(wasmConfig).then(function (compiledModule) {
initWorker(compiledModule);
});
} else {
initWorker(dracoModule());
}
});
}
}
return decodeDraco;
}));
//# sourceMappingURL=decodeDraco.js.map