@loaders.gl/gltf 4.0.0-alpha.22 → 4.0.0-alpha.23

package/src/lib/extensions/data-processing.ts

@@ -0,0 +1,264 @@
+import type {GLTFTextureInfoMetadata, GLTFMeshPrimitive} from '../types/gltf-json-schema';
+import type {TypedArray} from '@loaders.gl/schema';
+import type {ImageType} from '@loaders.gl/images';
+
+import {GLTFScenegraph} from '../api/gltf-scenegraph';
+import {getComponentTypeFromArray} from '../gltf-utils/gltf-utils';
+import {getImageData} from '@loaders.gl/images';
+import {emod} from '@loaders.gl/math';
+
+const ATTRIBUTE_TYPE_TO_COMPONENTS = {
+  SCALAR: 1,
+  VEC2: 2,
+  VEC3: 3,
+  VEC4: 4,
+  MAT2: 4,
+  MAT3: 9,
+  MAT4: 16,
+  BOOLEAN: 1,
+  STRING: 1,
+  ENUM: 1
+};
+
+const ATTRIBUTE_COMPONENT_TYPE_TO_ARRAY = {
+  INT8: Int8Array,
+  UINT8: Uint8Array,
+  INT16: Int16Array,
+  UINT16: Uint16Array,
+  INT32: Int32Array,
+  UINT32: Uint32Array,
+  INT64: BigInt64Array,
+  UINT64: BigUint64Array,
+  FLOAT32: Float32Array,
+  FLOAT64: Float64Array
+};
+
+const ATTRIBUTE_COMPONENT_TYPE_TO_BYTE_SIZE = {
+  INT8: 1,
+  UINT8: 1,
+  INT16: 2,
+  UINT16: 2,
+  INT32: 4,
+  UINT32: 4,
+  INT64: 8,
+  UINT64: 8,
+  FLOAT32: 4,
+  FLOAT64: 8
+};
+
+export function getArrayElementByteSize(attributeType, componentType): number {
+  return (
+    ATTRIBUTE_COMPONENT_TYPE_TO_BYTE_SIZE[componentType] *
+    ATTRIBUTE_TYPE_TO_COMPONENTS[attributeType]
+  );
+}
+
+/**
+ * Converts raw bytes that are in the buffer to an array of the type defined by the schema.
+ * @param {Uint8Array} typedArray - raw bytes in the buffer
+ * @param {string} attributeType - SCALAR, VECN, MATN
+ * @param {string} componentType - type of the component in elements, e.g. 'UINT8' or 'FLOAT32'
+ * @param {number} elementCount - number of elements in the array. Default value is 1.
+ * @returns {TypedArray} Data array
+ */
+export function convertRawBufferToMetadataArray(
+  typedArray: Uint8Array,
+  attributeType: string,
+  componentType: string,
+  elementCount: number = 1
+): TypedArray {
+  const numberOfComponents = ATTRIBUTE_TYPE_TO_COMPONENTS[attributeType];
+  const ArrayType = ATTRIBUTE_COMPONENT_TYPE_TO_ARRAY[componentType];
+  const length = elementCount * numberOfComponents;
+  const size = ATTRIBUTE_COMPONENT_TYPE_TO_BYTE_SIZE[componentType];
+  // the buffer view `byteOffset` must be aligned to a multiple of the `componentType` size.
+  const offset =
+    typedArray.byteOffset % size
+      ? Math.ceil(typedArray.byteOffset / size) * size
+      : typedArray.byteOffset;
+  return new ArrayType(typedArray.buffer, offset, length);
+}
+
+/**
+ * Processes data encoded in the texture associated with the primitive.
+ * If Ext_mesh_featues is combined with the Ext_structural_metadata, propertyTable will also be processed.
+ * @param {GLTFScenegraph} scenegraph - Instance of the class for structured access to GLTF data.
+ * @param {GLTFTextureInfoMetadata} textureInfo - reference to the texture where extension data are stored.
+ * @param {GLTFMeshPrimitive} primitive - primitive object in the mesh
+ * @returns {number[] | null} Array of data taken. Null if data can't be taken from the texture.
+ */
+export function getPrimitiveTextureData(
+  scenegraph: GLTFScenegraph,
+  textureInfo: GLTFTextureInfoMetadata,
+  primitive: GLTFMeshPrimitive
+): number[] | null {
+  /*
+    texture.index is an index for the "textures" array.
+    The texture object referenced by this index looks like this:
+    {
+    "sampler": 0,
+    "source": 0
+    }
+    "sampler" is an index for the "samplers" array
+    "source" is an index for the "images" array that contains data. These data are stored in rgba channels of the image.
+
+    texture.texCoord is a number-suffix (like 1) for an attribute like "TEXCOORD_1" in meshes.primitives
+    The value of "TEXCOORD_1" is an accessor that is used to get coordinates. These coordinates are being used to get data from the image.
+  */
+  const json = scenegraph.gltf.json;
+
+  const texCoordAccessorKey = `TEXCOORD_${textureInfo.texCoord || 0}`;
+  const texCoordAccessorIndex = primitive.attributes[texCoordAccessorKey];
+  const texCoordBufferView = scenegraph.getBufferView(texCoordAccessorIndex);
+  const texCoordArray: Uint8Array = scenegraph.getTypedArrayForBufferView(texCoordBufferView);
+
+  // textureCoordinates array contains UV coordinates of the actual data stored in the texture
+  const textureCoordinates: Float32Array = new Float32Array(
+    texCoordArray.buffer,
+    texCoordArray.byteOffset,
+    texCoordArray.length / 4
+  );
+
+  const textureIndex: number = textureInfo.index;
+  const imageIndex = json.textures?.[textureIndex]?.source;
+  if (typeof imageIndex !== 'undefined') {
+    const mimeType = json.images?.[imageIndex]?.mimeType;
+    const parsedImage = scenegraph.gltf.images?.[imageIndex];
+    // Checking for width is to prevent handling Un-processed images (e.g. [analyze] stage, where loadImages option is set to false)
+    if (parsedImage && typeof parsedImage.width !== 'undefined') {
+      const textureData: number[] = [];
+      for (let index = 0; index < textureCoordinates.length; index += 2) {
+        const value = getImageValueByCoordinates(
+          parsedImage,
+          mimeType,
+          textureCoordinates,
+          index,
+          textureInfo.channels
+        );
+        textureData.push(value);
+      }
+      return textureData;
+    }
+  }
+  return null;
+}
+
+/**
+ * Puts property data to attributes.
+ * It creates corresponding buffer, bufferView and accessor
+ * so the data can be accessed like regular data stored in buffers.
+ * @param {GLTFScenegraph} scenegraph - scenegraph object
+ * @param {string} attributeName - name of the attribute
+ * @param {number[]} propertyData - property data to store
+ * @param {number[]} featureTable - an array where unique data from the property data are being stored
+ * @param {GLTFMeshPrimitive} primitive - primitive object
+ */
+export function primitivePropertyDataToAttributes(
+  scenegraph: GLTFScenegraph,
+  attributeName: string,
+  propertyData: number[],
+  featureTable: number[],
+  primitive: GLTFMeshPrimitive
+): void {
+  if (propertyData === null) return;
+  /*
+    featureTable will contain unique values, e.g.
+    propertyData = [24, 35, 28, 24]
+    featureTable = [24, 35, 28]
+    featureIndices will contain indices that refer featureTextureTable, e.g.
+    featureIndices = [0, 1, 2, 0]
+  */
+  const featureIndices: number[] = [];
+  for (const texelData of propertyData) {
+    let index = featureTable.findIndex((item) => item === texelData);
+    if (index === -1) {
+      index = featureTable.push(texelData) - 1;
+    }
+    featureIndices.push(index);
+  }
+  const typedArray = new Uint32Array(featureIndices);
+  const bufferIndex =
+    scenegraph.gltf.buffers.push({
+      arrayBuffer: typedArray.buffer,
+      byteOffset: typedArray.byteOffset,
+      byteLength: typedArray.byteLength
+    }) - 1;
+  const bufferViewIndex = scenegraph.addBufferView(typedArray, bufferIndex, 0);
+  const accessorIndex = scenegraph.addAccessor(bufferViewIndex, {
+    size: 1,
+    componentType: getComponentTypeFromArray(typedArray),
+    count: typedArray.length
+  });
+  primitive.attributes[attributeName] = accessorIndex;
+}
+
+/**
+ * Gets the value from the texture by coordinates provided.
+ * @param {ImageType} parsedImage - image where the data are stored.
+ * @param {string | undefined} mimeType - MIME type
+ * @param {Float32Array} textureCoordinates - uv coordinates to access data in the image.
+ * @param {number} index - index of uv coordinates in the array textureCoordinates
+ * @param {channels} channels - image channels where data are stored. Channels of an RGBA texture are numbered 0..3 respectively.
+ * @returns {number} Value taken from the image.
+ */
+function getImageValueByCoordinates(
+  parsedImage: ImageType,
+  mimeType: string | undefined,
+  textureCoordinates: Float32Array,
+  index: number,
+  channels: number[] = [0]
+) {
+  const CHANNELS_MAP = [
+    {offset: 0, shift: 0},
+    {offset: 1, shift: 8},
+    {offset: 2, shift: 16},
+    {offset: 3, shift: 24}
+  ];
+
+  const u = textureCoordinates[index];
+  const v = textureCoordinates[index + 1];
+
+  let components = 1;
+  if (mimeType && (mimeType.indexOf('image/jpeg') !== -1 || mimeType.indexOf('image/png') !== -1))
+    components = 4;
+  const offset = coordinatesToOffset(u, v, parsedImage, components);
+  let value: number = 0;
+  for (const c of channels) {
+    const map = CHANNELS_MAP[c];
+    const imageOffset = offset + map.offset;
+    const imageData = getImageData(parsedImage);
+    if (imageData.data.length <= imageOffset) {
+      throw new Error(`${imageData.data.length} <= ${imageOffset}`);
+    }
+    const imageValue = imageData.data[imageOffset];
+    value |= imageValue << map.shift;
+  }
+  return value;
+}
+
+/**
+ * Retrieves the offset in the image where the data are stored
+ * @param u - u-coordinate
+ * @param v - v-coordinate
+ * @param parsedImage - image where the data are stored
+ * @param componentsCount - number of components the data consists of.
+ * @returns offset in the image where the data are stored
+ */
+function coordinatesToOffset(
+  u: number,
+  v: number,
+  parsedImage: any,
+  componentsCount: number = 1
+): number {
+  const w = parsedImage.width;
+  const iX = emod(u) * (w - 1);
+  const indX = Math.round(iX);
+
+  const h = parsedImage.height;
+  const iY = emod(v) * (h - 1);
+  const indY = Math.round(iY);
+  const components = parsedImage.components ? parsedImage.components : componentsCount;
+  // components is a number of channels in the image
+  const offset = (indY * w + indX) * components;
+  return offset;
+}

package/src/lib/extensions/deprecated/EXT_feature_metadata.ts

@@ -14,11 +14,11 @@ import {getImageData} from '@loaders.gl/images';
 import {GLTFMeshPrimitive} from '../../types/gltf-json-schema';
 import {getComponentTypeFromArray} from '../../gltf-utils/gltf-utils';
 import {GLTFLoaderOptions} from '../../../gltf-loader';
+import {emod} from '@loaders.gl/math';
 
 /** Extension name */
-const EXT_FEATURE_METADATA = 'EXT_feature_metadata';
-
-export const name = EXT_FEATURE_METADATA;
+const EXT_FEATURE_METADATA_NAME = 'EXT_feature_metadata';
+export const name = EXT_FEATURE_METADATA_NAME;
 
 export async function decode(gltfData: {json: GLTF}, options: GLTFLoaderOptions): Promise<void> {
   const scenegraph = new GLTFScenegraph(gltfData);
@@ -31,7 +31,7 @@ export async function decode(gltfData: {json: GLTF}, options: GLTFLoaderOptions)
  */
 function decodeExtFeatureMetadata(scenegraph: GLTFScenegraph, options: GLTFLoaderOptions): void {
   const extension: GLTF_EXT_feature_metadata_GLTF | null =
-    scenegraph.getExtension(EXT_FEATURE_METADATA);
+    scenegraph.getExtension(EXT_FEATURE_METADATA_NAME);
   if (!extension) return;
 
   const schemaClasses = extension.schema?.classes;
@@ -62,7 +62,7 @@ function decodeExtFeatureMetadata(scenegraph: GLTFScenegraph, options: GLTFLoade
 }
 
 /**
- * Navigate throw all properies in feature table and gets properties data.
+ * Navigates through all properies in feature table and gets properties data.
  * @param scenegraph
  * @param featureTable
  * @param schemaClass
@@ -90,7 +90,7 @@ function handleFeatureTableProperties(
 }
 
 /**
- * Navigate throw all properies in feature texture and gets properties data.
+ * Navigates through all properies in feature texture and gets properties data.
  * Data will be stored in featureTexture.properties[propertyName].data
  * @param scenegraph
  * @param featureTexture
@@ -327,17 +327,6 @@ function coordinatesToOffset(
   return offset;
 }
 
-// The following is taken from tile-converter\src\i3s-converter\helpers\batch-ids-extensions.ts
-/**
- * Handle UVs if they are out of range [0,1].
- * @param n
- * @param m
- */
-function emod(n: number): number {
-  const a = ((n % 1) + 1) % 1;
-  return a;
-}
-
 /**
  * Find the feature table by class name.
  * @param featureTables

package/src/lib/types/gltf-ext-mesh-features-schema.ts

@@ -0,0 +1,48 @@
+import {GLTFTextureInfoMetadata} from './gltf-json-schema';
+/* eslint-disable camelcase */
+
+/**
+ * EXT_mesh_features extension types
+ * This is a primitive-level extension
+ * @see https://github.com/CesiumGS/glTF/tree/3d-tiles-next/extensions/2.0/Vendor/EXT_mesh_features
+ * or https://github.com/CesiumGS/glTF/blob/c38f7f37e894004353c15cd0481bc5b7381ce841/extensions/2.0/Vendor/EXT_mesh_features/schema/mesh.primitive.EXT_mesh_features.schema.json
+ * An object describing feature IDs for a mesh primitive.
+ */
+export type GLTF_EXT_mesh_features = {
+  /** An array of feature ID sets. */
+  featureIds: GLTF_EXT_mesh_features_featureId[];
+  extensions?: Record<string, unknown>;
+  extras?: unknown;
+  /** For internal usage */
+  dataAttributeNames?: string[];
+};
+
+/**
+ * @see https://github.com/CesiumGS/glTF/blob/c38f7f37e894004353c15cd0481bc5b7381ce841/extensions/2.0/Vendor/EXT_mesh_features/schema/featureId.schema.json
+ * Feature IDs stored in an attribute or texture.
+ */
+export type GLTF_EXT_mesh_features_featureId = {
+  /** The number of unique features in the attribute or texture. */
+  featureCount: number;
+  /** A value that indicates that no feature is associated with this vertex or texel. */
+  nullFeatureId?: number;
+  /** A label assigned to this feature ID set. Labels must be alphanumeric identifiers matching the regular expression `^[a-zA-Z_][a-zA-Z0-9_]*$`. */
+  label?: string;
+  /**
+   * An attribute containing feature IDs.
+   * When `attribute` and `texture` are omitted the feature IDs are assigned to vertices by their index.
+   * @see https://github.com/CesiumGS/glTF/blob/3d-tiles-next/extensions/2.0/Vendor/EXT_mesh_features/schema/featureIdAttribute.schema.json
+   * An integer value used to construct a string in the format `_FEATURE_ID_<set index>` which is a reference to a key in `mesh.primitives.attributes`
+   * (e.g. a value of `0` corresponds to `_FEATURE_ID_0`).
+   */
+  attribute?: number;
+  /** A texture containing feature IDs. */
+  texture?: GLTFTextureInfoMetadata;
+  /** The index of the property table containing per-feature property values. Only applicable when using the `EXT_structural_metadata` extension. */
+  propertyTable?: number;
+  extensions?: Record<string, unknown>;
+  extras?: unknown;
+
+  /** For internal usage */
+  data?: unknown;
+};