118 lines
3.9 KiB
C++
118 lines
3.9 KiB
C++
#include "CesiumGltfSpecUtility.h"
|
|
|
|
int32_t AddBufferToModel(
|
|
CesiumGltf::Model& model,
|
|
const std::string& type,
|
|
const int32_t componentType,
|
|
const std::vector<std::byte>&& values) {
|
|
CesiumGltf::Buffer& buffer = model.buffers.emplace_back();
|
|
buffer.byteLength = values.size();
|
|
buffer.cesium.data = std::move(values);
|
|
|
|
CesiumGltf::BufferView& bufferView = model.bufferViews.emplace_back();
|
|
bufferView.buffer = static_cast<int32_t>(model.buffers.size() - 1);
|
|
bufferView.byteLength = buffer.byteLength;
|
|
bufferView.byteOffset = 0;
|
|
|
|
CesiumGltf::Accessor& accessor = model.accessors.emplace_back();
|
|
accessor.bufferView = static_cast<int32_t>(model.bufferViews.size() - 1);
|
|
accessor.type = type;
|
|
accessor.componentType = componentType;
|
|
|
|
const int64_t elementByteSize =
|
|
CesiumGltf::Accessor::computeByteSizeOfComponent(componentType) *
|
|
CesiumGltf::Accessor::computeNumberOfComponents(type);
|
|
accessor.count = buffer.byteLength / elementByteSize;
|
|
|
|
return static_cast<int32_t>(model.accessors.size() - 1);
|
|
}
|
|
|
|
CesiumGltf::FeatureId& AddFeatureIDsAsAttributeToModel(
|
|
CesiumGltf::Model& model,
|
|
CesiumGltf::MeshPrimitive& primitive,
|
|
const std::vector<uint8_t>& featureIDs,
|
|
const int64_t featureCount,
|
|
const int64_t setIndex) {
|
|
std::vector<std::byte> values = GetValuesAsBytes(featureIDs);
|
|
|
|
CreateAttributeForPrimitive(
|
|
model,
|
|
primitive,
|
|
"_FEATURE_ID_" + std::to_string(setIndex),
|
|
CesiumGltf::AccessorSpec::Type::SCALAR,
|
|
CesiumGltf::AccessorSpec::ComponentType::UNSIGNED_BYTE,
|
|
std::move(values));
|
|
|
|
CesiumGltf::ExtensionExtMeshFeatures* pExtension =
|
|
primitive.getExtension<CesiumGltf::ExtensionExtMeshFeatures>();
|
|
if (pExtension == nullptr) {
|
|
pExtension =
|
|
&primitive.addExtension<CesiumGltf::ExtensionExtMeshFeatures>();
|
|
}
|
|
|
|
CesiumGltf::FeatureId& featureID = pExtension->featureIds.emplace_back();
|
|
featureID.featureCount = featureCount;
|
|
featureID.attribute = setIndex;
|
|
|
|
return featureID;
|
|
}
|
|
|
|
CesiumGltf::FeatureId& AddFeatureIDsAsTextureToModel(
|
|
CesiumGltf::Model& model,
|
|
CesiumGltf::MeshPrimitive& primitive,
|
|
const std::vector<uint8_t>& featureIDs,
|
|
const int64_t featureCount,
|
|
const int32_t imageWidth,
|
|
const int32_t imageHeight,
|
|
const std::vector<glm::vec2>& texCoords,
|
|
const int64_t texcoordSetIndex,
|
|
const int32_t samplerWrapS,
|
|
const int32_t samplerWrapT) {
|
|
CesiumGltf::Image& image = model.images.emplace_back();
|
|
image.pAsset.emplace();
|
|
image.pAsset->bytesPerChannel = 1;
|
|
image.pAsset->channels = 1;
|
|
image.pAsset->width = imageWidth;
|
|
image.pAsset->height = imageHeight;
|
|
|
|
std::vector<std::byte>& data = image.pAsset->pixelData;
|
|
data.resize(imageWidth * imageHeight);
|
|
std::memcpy(data.data(), featureIDs.data(), data.size());
|
|
|
|
CesiumGltf::Sampler& sampler = model.samplers.emplace_back();
|
|
sampler.wrapS = samplerWrapS;
|
|
sampler.wrapT = samplerWrapT;
|
|
|
|
CesiumGltf::Texture& texture = model.textures.emplace_back();
|
|
texture.sampler = 0;
|
|
texture.source = 0;
|
|
|
|
std::vector<std::byte> values = GetValuesAsBytes(texCoords);
|
|
CreateAttributeForPrimitive(
|
|
model,
|
|
primitive,
|
|
"TEXCOORD_" + std::to_string(texcoordSetIndex),
|
|
CesiumGltf::AccessorSpec::Type::VEC2,
|
|
CesiumGltf::AccessorSpec::ComponentType::FLOAT,
|
|
std::move(values));
|
|
|
|
CesiumGltf::ExtensionExtMeshFeatures* pExtension =
|
|
primitive.getExtension<CesiumGltf::ExtensionExtMeshFeatures>();
|
|
if (pExtension == nullptr) {
|
|
pExtension =
|
|
&primitive.addExtension<CesiumGltf::ExtensionExtMeshFeatures>();
|
|
}
|
|
|
|
CesiumGltf::FeatureId& featureID = pExtension->featureIds.emplace_back();
|
|
featureID.featureCount = featureCount;
|
|
|
|
CesiumGltf::FeatureIdTexture featureIDTexture;
|
|
featureIDTexture.channels = {0};
|
|
featureIDTexture.index = 0;
|
|
featureIDTexture.texCoord = texcoordSetIndex;
|
|
|
|
featureID.texture = featureIDTexture;
|
|
|
|
return featureID;
|
|
}
|