@loaders.gl/pointclouds 0.0.0-stage → 5.0.0-beta.10

package/src/las/lib/laz-rs-wasm/parse-las.ts

@@ -0,0 +1,411 @@
+// loaders.gl
+// SPDX-License-Identifier: MIT
+// Copyright (c) vis.gl contributors
+
+// ported and es6-ified from https://github.com/verma/plasio/
+import {getLoaderLog} from '@loaders.gl/core/loader-utils';
+import type {LASLoaderOptions} from '../../las-loader-types';
+import type {LASMesh, LASHeader} from '../las-types';
+import type {MeshAttributes} from '@loaders.gl/core/schema';
+import {getMeshBoundingBox} from '@loaders.gl/core/schema-utils';
+import {concatenateArrayBuffersAsync} from '@loaders.gl/core/loader-utils';
+import {getLASSchema} from '../get-las-schema';
+import {LASFile} from './laslaz-decoder';
+
+type LASChunk = {
+  count: number;
+  buffer: ArrayBuffer;
+  hasMoreData: boolean;
+};
+
+type LASDecoder = {
+  pointsCount: number;
+  getPoint(index: number): {
+    position: [number, number, number];
+    color?: [number, number, number];
+    intensity: number;
+    classification: number;
+  };
+};
+
+type LASDecodedChunk = {
+  decoder: LASDecoder;
+  header: LASHeader;
+};
+
+const DEFAULT_BATCH_SIZE = 1000 * 100;
+
+/**
+ * Parsing of .las file
+ * @param arrayBuffer
+ * @param options
+ * @returns LASMesh
+ */
+export function parseLAS(arrayBuffer: ArrayBuffer, options?: LASLoaderOptions): LASMesh {
+  return parseLASMesh(arrayBuffer, options);
+  // This code breaks pointcloud example on the website
+  // const mesh = parseLASMesh(arrayBuffer, options);
+  // return convertMeshToTable(mesh, options?.las?.shape || 'mesh') as LASMesh | ArrowTable | ColumnarTable;
+}
+
+/**
+ * Parse LAS/LAZ data in decoded point batches.
+ * @param arrayBufferIterator Iterator yielding LAS/LAZ binary chunks
+ * @param options LAS loader options
+ * @returns Async iterable of mesh batches
+ */
+export async function* parseLASInBatches(
+  arrayBufferIterator:
+    | AsyncIterable<ArrayBufferLike | ArrayBufferView>
+    | Iterable<ArrayBufferLike | ArrayBufferView>,
+  options: LASLoaderOptions = {}
+): AsyncIterable<LASMesh> {
+  const arrayBuffer = await concatenateArrayBuffersAsync(arrayBufferIterator);
+  const batchSize = getBatchSize(options);
+
+  for (const {decoder, header} of parseLASChunkedIterator(arrayBuffer, batchSize)) {
+    yield parseLASMeshBatch(decoder, header, options);
+  }
+}
+
+/**
+ * Parsing of .las file
+ * @param arrayBuffer
+ * @param options
+ * @returns LASHeader
+ */
+function parseLASMesh(arrayBuffer: ArrayBuffer, options: LASLoaderOptions = {}): LASMesh {
+  let pointIndex: number = 0;
+
+  let positions: Float32Array | Float64Array;
+  let colors: Uint8Array | null;
+  let intensities: Uint16Array;
+  let classifications: Uint8Array;
+  let originalHeader: any;
+
+  const lasMesh: LASMesh = {
+    loader: 'las',
+    loaderData: {} as LASHeader,
+    // shape: 'mesh',
+    schema: {fields: [], metadata: {}},
+    header: {
+      vertexCount: 0,
+      boundingBox: [
+        [0, 0, 0],
+        [0, 0, 0]
+      ]
+    },
+    attributes: {},
+    topology: 'point-list',
+    mode: 0 // GL.POINTS
+  };
+
+  /* eslint-disable max-statements */
+  // @ts-ignore Possibly undefined
+  parseLASChunked(arrayBuffer, (decoder: any = {}, lasHeader: LASHeader) => {
+    if (!originalHeader) {
+      originalHeader = lasHeader;
+      const total = lasHeader.totalToRead;
+
+      const PositionsType = options.las?.fp64 ? Float64Array : Float32Array;
+      positions = new PositionsType(total * 3);
+      // laslaz-decoder.js `pointFormatReaders`
+      colors = lasHeader.hasColor ? new Uint8Array(total * 4) : null;
+      intensities = new Uint16Array(total);
+      classifications = new Uint8Array(total);
+
+      lasMesh.loaderData = lasHeader;
+      lasMesh.attributes = {
+        POSITION: {value: positions, size: 3},
+        // non-gltf attributes, use non-capitalized names for now
+        intensity: {value: intensities, size: 1},
+        classification: {value: classifications, size: 1}
+      };
+
+      if (colors) {
+        lasMesh.attributes.COLOR_0 = {value: colors, size: 4};
+      }
+    }
+
+    const batchSize = decoder.pointsCount;
+    const {
+      scale: [scaleX, scaleY, scaleZ],
+      offset: [offsetX, offsetY, offsetZ]
+    } = lasHeader;
+
+    const twoByteColor = detectTwoByteColors(decoder, batchSize, options);
+
+    for (let i = 0; i < batchSize; i++) {
+      const {position, color, intensity, classification} = decoder.getPoint(i);
+
+      positions[pointIndex * 3] = position[0] * scaleX + offsetX;
+      positions[pointIndex * 3 + 1] = position[1] * scaleY + offsetY;
+      positions[pointIndex * 3 + 2] = position[2] * scaleZ + offsetZ;
+
+      if (color && colors) {
+        if (twoByteColor) {
+          colors[pointIndex * 4] = color[0] / 256;
+          colors[pointIndex * 4 + 1] = color[1] / 256;
+          colors[pointIndex * 4 + 2] = color[2] / 256;
+        } else {
+          colors[pointIndex * 4] = color[0];
+          colors[pointIndex * 4 + 1] = color[1];
+          colors[pointIndex * 4 + 2] = color[2];
+        }
+        colors[pointIndex * 4 + 3] = 255;
+      }
+
+      intensities[pointIndex] = intensity;
+      classifications[pointIndex] = classification;
+
+      pointIndex++;
+    }
+
+    const meshBatch = {
+      ...lasMesh,
+      header: {
+        vertexCount: lasHeader.totalRead
+      },
+      progress: lasHeader.totalRead / lasHeader.totalToRead
+    };
+
+    options?.onProgress?.({
+      ...meshBatch,
+      phase: 'decode',
+      completed: lasHeader.totalRead,
+      total: lasHeader.totalToRead,
+      unit: 'points'
+    });
+  });
+  /* eslint-enable max-statements */
+
+  lasMesh.header = {
+    vertexCount: originalHeader.totalToRead,
+    boundingBox: getMeshBoundingBox(lasMesh?.attributes || {}) ?? undefined
+  };
+
+  if (lasMesh) {
+    lasMesh.schema = getLASSchema(lasMesh.loaderData, lasMesh.attributes);
+  }
+  return lasMesh;
+}
+
+/**
+ * Convert one decoded LAS/LAZ chunk into a mesh.
+ * @param decoder Point decoder for the current chunk
+ * @param lasHeader LAS file header with current progress
+ * @param options LAS loader options
+ * @returns Mesh containing one decoded point batch
+ */
+function parseLASMeshBatch(
+  decoder: LASDecoder,
+  lasHeader: LASHeader,
+  options: LASLoaderOptions = {}
+): LASMesh {
+  const batchSize = decoder.pointsCount;
+  const PositionsType = options.las?.fp64 ? Float64Array : Float32Array;
+  const positions = new PositionsType(batchSize * 3);
+  const colors = lasHeader.hasColor ? new Uint8Array(batchSize * 4) : null;
+  const intensities = new Uint16Array(batchSize);
+  const classifications = new Uint8Array(batchSize);
+
+  populateLASAttributes(decoder, lasHeader, options, {
+    positions,
+    colors,
+    intensities,
+    classifications,
+    pointOffset: 0
+  });
+
+  const attributes: MeshAttributes = {
+    POSITION: {value: positions, size: 3},
+    intensity: {value: intensities, size: 1},
+    classification: {value: classifications, size: 1}
+  };
+
+  if (colors) {
+    attributes.COLOR_0 = {value: colors, size: 4};
+  }
+
+  const batchHeader = {...lasHeader};
+  const lasMesh: LASMesh = {
+    loader: 'las',
+    loaderData: batchHeader,
+    schema: {fields: [], metadata: {}},
+    header: {
+      vertexCount: batchSize,
+      boundingBox: getMeshBoundingBox(attributes)
+    },
+    attributes,
+    topology: 'point-list',
+    mode: 0,
+    progress: lasHeader.totalRead / lasHeader.totalToRead
+  } as LASMesh & {progress: number};
+
+  lasMesh.schema = getLASSchema(batchHeader, lasMesh.attributes);
+  return lasMesh;
+}
+
+/**
+ * Fill typed LAS attribute arrays from a decoded chunk.
+ * @param decoder Point decoder for the current chunk
+ * @param lasHeader LAS file header
+ * @param options LAS loader options
+ * @param target Target attribute arrays and write offset
+ */
+function populateLASAttributes(
+  decoder: LASDecoder,
+  lasHeader: LASHeader,
+  options: LASLoaderOptions,
+  target: {
+    positions: Float32Array | Float64Array;
+    colors: Uint8Array | null;
+    intensities: Uint16Array;
+    classifications: Uint8Array;
+    pointOffset: number;
+  }
+): void {
+  const batchSize = decoder.pointsCount;
+  const {
+    scale: [scaleX, scaleY, scaleZ],
+    offset: [offsetX, offsetY, offsetZ]
+  } = lasHeader;
+  const twoByteColor = detectTwoByteColors(decoder, batchSize, options);
+
+  for (let i = 0; i < batchSize; i++) {
+    const {position, color, intensity, classification} = decoder.getPoint(i);
+    const pointIndex = target.pointOffset + i;
+
+    target.positions[pointIndex * 3] = position[0] * scaleX + offsetX;
+    target.positions[pointIndex * 3 + 1] = position[1] * scaleY + offsetY;
+    target.positions[pointIndex * 3 + 2] = position[2] * scaleZ + offsetZ;
+
+    if (color && target.colors) {
+      if (twoByteColor) {
+        target.colors[pointIndex * 4] = color[0] / 256;
+        target.colors[pointIndex * 4 + 1] = color[1] / 256;
+        target.colors[pointIndex * 4 + 2] = color[2] / 256;
+      } else {
+        target.colors[pointIndex * 4] = color[0];
+        target.colors[pointIndex * 4 + 1] = color[1];
+        target.colors[pointIndex * 4 + 2] = color[2];
+      }
+      target.colors[pointIndex * 4 + 3] = 255;
+    }
+
+    target.intensities[pointIndex] = intensity;
+    target.classifications[pointIndex] = classification;
+  }
+}
+
+/**
+ * parse laz data
+ * @param rawData
+ * @param onParseData
+ * @return parsed point cloud
+ */
+/* eslint-enable max-statements */
+export function parseLASChunked(
+  rawData: ArrayBuffer,
+  onParseData: any = {},
+  batchSize: number = DEFAULT_BATCH_SIZE
+): void {
+  for (const {decoder, header} of parseLASChunkedIterator(rawData, batchSize)) {
+    onParseData(decoder, header);
+  }
+}
+
+/**
+ * Decode LAS/LAZ data into point chunks.
+ * @param rawData Complete LAS/LAZ file data
+ * @param batchSize Number of returned points per chunk
+ * @returns Iterable of decoded point chunks
+ */
+function* parseLASChunkedIterator(
+  rawData: ArrayBuffer,
+  batchSize: number = DEFAULT_BATCH_SIZE
+): Iterable<LASDecodedChunk> {
+  const dataHandler = new LASFile(rawData);
+
+  try {
+    // open data
+    dataHandler.open();
+
+    const header = dataHandler.getHeader();
+    // start loading
+    const Unpacker = dataHandler.getUnpacker();
+
+    header.totalToRead = header.pointsCount;
+    let totalRead = 0;
+
+    /* eslint-disable no-constant-condition */
+    while (true) {
+      const chunk: LASChunk = dataHandler.readData(batchSize);
+
+      totalRead += chunk.count;
+
+      header.totalRead = totalRead;
+
+      const unpacker = new Unpacker(chunk.buffer, chunk.count, header);
+
+      // surface unpacker and progress via call back
+      // use unpacker.pointsCount and unpacker.getPoint(i) to handle data in app
+      yield {decoder: unpacker, header};
+
+      if (!chunk.hasMoreData || totalRead >= header.totalToRead) {
+        break;
+      }
+    }
+  } catch (e) {
+    throw e;
+  } finally {
+    dataHandler.close();
+  }
+}
+
+/**
+ * Get the requested LAS point batch size.
+ * @param options LAS loader options
+ * @returns Point count per decoded batch
+ */
+function getBatchSize(options: LASLoaderOptions): number {
+  const batchSize = options.core?.batchSize;
+  return typeof batchSize === 'number' ? batchSize : DEFAULT_BATCH_SIZE;
+}
+
+/**
+ * @param decoder
+ * @param batchSize
+ * @param colorDepth
+ * @returns boolean
+ */
+function detectTwoByteColors(
+  decoder: any = {},
+  batchSize: number,
+  options: LASLoaderOptions
+): boolean {
+  let twoByteColor = false;
+  switch (options.las?.colorDepth) {
+    case 8:
+      twoByteColor = false;
+      break;
+    case 16:
+      twoByteColor = true;
+      break;
+    case 'auto':
+      if (decoder.getPoint(0).color) {
+        for (let i = 0; i < batchSize; i++) {
+          const {color} = decoder.getPoint(i);
+          // eslint-disable-next-line max-depth
+          if (color[0] > 255 || color[1] > 255 || color[2] > 255) {
+            twoByteColor = true;
+          }
+        }
+      }
+      break;
+    default:
+      getLoaderLog(options).warn('las: illegal value for options.las.colorDepth')();
+      break;
+  }
+  return twoByteColor;
+}