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

package/src/las/lib/laz-perf/laslaz-decoder.ts

@@ -0,0 +1,443 @@
+// loaders.gl
+// SPDX-License-Identifier: MIT
+// Copyright (c) vis.gl contributors
+
+/*
+  Modified from Uday Verma and Howard Butler's plasio
+  https://github.com/verma/plasio/
+  MIT License
+*/
+// laslaz.js - treat as compiled code
+import type {LASHeader} from '../las-types';
+import getModule from '../../libs/laz-perf/laz-perf';
+
+let Module: any = null;
+
+type LASReader = (dv: DataView) => {
+  [LASAttribute: string]: number | number[];
+};
+
+type LASReaders = {
+  [key: number]: LASReader;
+};
+
+type LASData = {
+  buffer: ArrayBuffer;
+  count: number;
+  hasMoreData: boolean;
+};
+
+const POINT_FORMAT_READERS: LASReaders = {
+  0: dv => {
+    return {
+      position: [dv.getInt32(0, true), dv.getInt32(4, true), dv.getInt32(8, true)],
+      intensity: dv.getUint16(12, true),
+      classification: dv.getUint8(15)
+    };
+  },
+  1: dv => {
+    return {
+      position: [dv.getInt32(0, true), dv.getInt32(4, true), dv.getInt32(8, true)],
+      intensity: dv.getUint16(12, true),
+      classification: dv.getUint8(15)
+    };
+  },
+  2: dv => {
+    return {
+      position: [dv.getInt32(0, true), dv.getInt32(4, true), dv.getInt32(8, true)],
+      intensity: dv.getUint16(12, true),
+      classification: dv.getUint8(15),
+      color: [dv.getUint16(20, true), dv.getUint16(22, true), dv.getUint16(24, true)]
+    };
+  },
+  3: dv => {
+    return {
+      position: [dv.getInt32(0, true), dv.getInt32(4, true), dv.getInt32(8, true)],
+      intensity: dv.getUint16(12, true),
+      classification: dv.getUint8(15),
+      color: [dv.getUint16(28, true), dv.getUint16(30, true), dv.getUint16(32, true)]
+    };
+  }
+};
+
+/**
+ * Reads incoming binary data depends on the Type parameter
+ * @param buf
+ * @param Type
+ * @param offset
+ * @param count
+ * @returns number | number[] from incoming binary data
+ */
+function readAs(buf: ArrayBuffer, Type: any = {}, offset: number, count?: number) {
+  count = count === undefined || count === 0 ? 1 : count;
+  const sub = buf.slice(offset, offset + Type.BYTES_PER_ELEMENT * count);
+
+  const r = new Type(sub);
+  if (count === 1) {
+    return r[0];
+  }
+
+  const ret: number[] = [];
+  for (let i = 0; i < count; i++) {
+    ret.push(r[i]);
+  }
+
+  return ret;
+}
+
+/**
+ * Parsing of header's attributes
+ * @param arraybuffer
+ * @returns header as LASHeader
+ */
+function parseLASHeader(arraybuffer: ArrayBuffer): LASHeader {
+  let start = 32 * 3 + 35;
+
+  const o: Partial<LASHeader> = {
+    pointsOffset: readAs(arraybuffer, Uint32Array, 32 * 3),
+    pointsFormatId: readAs(arraybuffer, Uint8Array, 32 * 3 + 8),
+    pointsStructSize: readAs(arraybuffer, Uint16Array, 32 * 3 + 8 + 1),
+    pointsCount: readAs(arraybuffer, Uint32Array, 32 * 3 + 11),
+    scale: readAs(arraybuffer, Float64Array, start, 3)
+  };
+  start += 24; // 8*3
+  o.offset = readAs(arraybuffer, Float64Array, start, 3);
+  start += 24;
+
+  const bounds = readAs(arraybuffer, Float64Array, start, 6);
+  start += 48; // 8*6;
+  o.maxs = [bounds[0], bounds[2], bounds[4]];
+  o.mins = [bounds[1], bounds[3], bounds[5]];
+
+  const colorPointFormats = new Set([2, 3]);
+  o.hasColor = colorPointFormats.has(o.pointsFormatId! & 0x3f);
+
+  return o as LASHeader;
+}
+
+// LAS Loader
+// Loads uncompressed files
+//
+class LASLoader {
+  arraybuffer: ArrayBuffer;
+  readOffset: number = 0;
+  header: LASHeader = {
+    pointsOffset: 0,
+    pointsFormatId: 0,
+    pointsStructSize: 0,
+    pointsCount: 0,
+    scale: [0, 0, 0],
+    offset: [0, 0, 0],
+    maxs: [0],
+    mins: [0],
+    totalToRead: 0,
+    totalRead: 0,
+    versionAsString: '',
+    isCompressed: true,
+    hasColor: false
+  };
+
+  constructor(arraybuffer: ArrayBuffer) {
+    this.arraybuffer = arraybuffer;
+  }
+
+  /**
+   * @returns boolean
+   */
+  open() {
+    // Nothing needs to be done to open this
+    return true;
+  }
+  /**
+   * Parsing of incoming binary
+   * @returns LASHeader
+   */
+  getHeader() {
+    this.header = parseLASHeader(this.arraybuffer);
+    return this.header;
+  }
+
+  /**
+   * Reading data
+   * @param count
+   * @returns new ArrayBuffer, count, hasMoreData
+   */
+  readData(count: number) {
+    const {header, arraybuffer} = this;
+    if (!header) {
+      throw new Error('Cannot start reading data till a header request is issued');
+    }
+
+    let {readOffset} = this;
+    count = Math.min(count, header.pointsCount - readOffset);
+    const start = header.pointsOffset + readOffset * header.pointsStructSize;
+    const end = start + count * header.pointsStructSize;
+    readOffset += count;
+    this.readOffset = readOffset;
+    return {
+      buffer: arraybuffer.slice(start, end),
+      count,
+      hasMoreData: readOffset < header.pointsCount
+    };
+  }
+  /**
+   * Method which brings data to null to close the file
+   * @returns
+   */
+  close() {
+    // @ts-ignore Possibly null
+    this.arraybuffer = null;
+    return true;
+  }
+}
+
+/**
+ * LAZ Loader
+ * Uses NaCL module to load LAZ files
+ */
+class LAZLoader {
+  arraybuffer: ArrayBuffer;
+  instance: any = null; // LASZip instance
+  header: LASHeader | null = null;
+
+  constructor(arraybuffer: ArrayBuffer) {
+    this.arraybuffer = arraybuffer;
+
+    if (!Module) {
+      // Avoid executing laz-perf on import
+      Module = getModule();
+    }
+  }
+
+  /**
+   * Opens the file
+   * @returns boolean
+   */
+  open(): boolean {
+    try {
+      const {arraybuffer} = this;
+      this.instance = new Module.LASZip();
+      const abInt = new Uint8Array(arraybuffer);
+      const buf = Module._malloc(arraybuffer.byteLength);
+
+      this.instance.arraybuffer = arraybuffer;
+      this.instance.buf = buf;
+      Module.HEAPU8.set(abInt, buf);
+      this.instance.open(buf, arraybuffer.byteLength);
+
+      this.instance.readOffset = 0;
+
+      return true;
+    } catch (error) {
+      throw new Error(`Failed to open file: ${(error as Error).message}`);
+    }
+  }
+
+  getHeader(): LASHeader {
+    if (!this.instance) {
+      throw new Error('You need to open the file before trying to read header');
+    }
+
+    try {
+      const header = parseLASHeader(this.instance.arraybuffer);
+      header.pointsFormatId &= 0x3f;
+      this.header = header;
+      return header;
+    } catch (error) {
+      throw new Error(`Failed to get header: ${(error as Error).message}`);
+    }
+  }
+  /**
+   * @param count
+   * @returns Data
+   */
+  readData(count: number): LASData {
+    if (!this.instance) {
+      throw new Error('You need to open the file before trying to read stuff');
+    }
+
+    const {header, instance} = this;
+
+    if (!header) {
+      throw new Error(
+        'You need to query header before reading, I maintain state that way, sorry :('
+      );
+    }
+
+    try {
+      const pointsToRead = Math.min(count, header.pointsCount - instance.readOffset);
+      const thisBuf = new Uint8Array(pointsToRead * header.pointsStructSize);
+      const bufRead = Module._malloc(header.pointsStructSize);
+      for (let i = 0; i < pointsToRead; i++) {
+        instance.getPoint(bufRead);
+        const a = new Uint8Array(Module.HEAPU8.buffer, bufRead, header.pointsStructSize);
+        thisBuf.set(a, i * header.pointsStructSize);
+        instance.readOffset++;
+      }
+      Module._free(bufRead);
+      return {
+        buffer: thisBuf.buffer,
+        count: pointsToRead,
+        hasMoreData: instance.readOffset < header.pointsCount
+      };
+    } catch (error) {
+      throw new Error(`Failed to read data: ${(error as Error).message}`);
+    }
+  }
+
+  /**
+   * Deletes the instance
+   * @returns boolean
+   */
+  close(): boolean {
+    try {
+      if (this.instance !== null) {
+        Module._free(this.instance.buf);
+        this.instance.delete();
+        this.instance = null;
+      }
+      return true;
+    } catch (error) {
+      throw new Error(`Failed to close file: ${(error as Error).message}`);
+    }
+  }
+}
+
+/**
+ * Helper class: Decodes LAS records into points
+ */
+class LASDecoder {
+  arrayb: ArrayBuffer;
+  decoder: (dv: DataView) => {};
+  pointsCount: number;
+  pointSize: number;
+  scale: [number, number, number];
+  offset?: [number, number, number];
+  mins?: number[];
+  maxs?: number[];
+
+  constructor(buffer: ArrayBuffer, len: number, header: LASHeader) {
+    this.arrayb = buffer;
+    this.decoder = POINT_FORMAT_READERS[header.pointsFormatId];
+    this.pointsCount = len;
+    this.pointSize = header.pointsStructSize;
+    this.scale = header.scale;
+    this.offset = header.offset;
+    this.mins = header.mins;
+    this.maxs = header.maxs;
+  }
+
+  /**
+   * Decodes data depends on this point size
+   * @param index
+   * @returns New object
+   */
+  getPoint(index: number): {} {
+    if (index < 0 || index >= this.pointsCount) {
+      throw new Error('Point index out of range');
+    }
+
+    const dv = new DataView(this.arrayb, index * this.pointSize, this.pointSize);
+    return this.decoder(dv);
+  }
+}
+
+/**
+ * A single consistent interface for loading LAS/LAZ files
+ */
+export class LASFile {
+  arraybuffer: ArrayBuffer;
+  formatId: number = 0;
+  loader: LASLoader | LAZLoader;
+  isCompressed: boolean = true;
+  isOpen: boolean = false;
+  version: number = 0;
+  versionAsString: string = '';
+
+  constructor(arraybuffer: ArrayBuffer) {
+    this.arraybuffer = arraybuffer;
+
+    if (this.determineVersion() > 13) {
+      throw new Error('Only file versions <= 1.3 are supported at this time');
+    }
+
+    this.determineFormat();
+    if (POINT_FORMAT_READERS[this.formatId] === undefined) {
+      throw new Error('The point format ID is not supported');
+    }
+
+    this.loader = this.isCompressed
+      ? new LAZLoader(this.arraybuffer)
+      : new LASLoader(this.arraybuffer);
+  }
+
+  /**
+   * Determines format in parameters of LASHeaer
+   */
+  determineFormat(): void {
+    const formatId = readAs(this.arraybuffer, Uint8Array, 32 * 3 + 8);
+    const bit7 = (formatId & 0x80) >> 7;
+    const bit6 = (formatId & 0x40) >> 6;
+
+    if (bit7 === 1 && bit6 === 1) {
+      throw new Error('Old style compression not supported');
+    }
+
+    this.formatId = formatId & 0x3f;
+    this.isCompressed = bit7 === 1 || bit6 === 1;
+  }
+
+  /**
+   * Determines version
+   * @returns version
+   */
+  determineVersion(): number {
+    const ver = new Int8Array(this.arraybuffer, 24, 2);
+    this.version = ver[0] * 10 + ver[1];
+    this.versionAsString = `${ver[0]}.${ver[1]}`;
+    return this.version;
+  }
+
+  /**
+   * Reads if the file is open
+   * @returns boolean
+   */
+  open(): void {
+    if (this.loader.open()) {
+      this.isOpen = true;
+    }
+  }
+  /**
+   * Gets the header
+   * @returns Header
+   */
+  getHeader(): LASHeader {
+    return this.loader.getHeader();
+  }
+
+  /**
+   * @param count
+   * @returns Data
+   */
+  readData(count: number): LASData {
+    return this.loader.readData(count);
+  }
+
+  /**
+   * Closes the file
+   */
+  close(): void {
+    if (this.loader.close()) {
+      this.isOpen = false;
+    }
+  }
+  /**
+   */
+  getUnpacker(): typeof LASDecoder {
+    return LASDecoder;
+  }
+}
+
+export const LASModuleWasLoaded = false;
+
+/* eslint no-use-before-define: 2 */