@loaders.gl/wkt 3.1.1 → 3.1.5

package/src/lib/encode-wkt.ts

@@ -1,55 +1,53 @@
 // Fork of https://github.com/mapbox/wellknown under ISC license (MIT/BSD-2-clause equivalent)
 // eslint-disable-next-line import/no-unresolved
-import type {Feature, Geometry} from 'geojson';
+import type {Feature, Geometry} from '@loaders.gl/schema';
+
 /**
  * Stringifies a GeoJSON object into WKT
  * @param geojson
  * @returns string
  */
-
-export default function encodeWKT(gj: Geometry | Feature): string {
-  if (gj.type === 'Feature') {
-    gj = gj.geometry;
-  }
-
-  function pairWKT(c: number[]) {
-    return c.join(' ');
-  }
-
-  function ringWKT(r: number[][]) {
-    return r.map(pairWKT).join(', ');
-  }
-
-  function ringsWKT(r: number[][][]) {
-    return r.map(ringWKT).map(wrapParens).join(', ');
-  }
-
-  function multiRingsWKT(r: number[][][][]) {
-    return r.map(ringsWKT).map(wrapParens).join(', ');
-  }
-
-  function wrapParens(s: string) {
-    return `(${s})`;
+export default function encodeWKT(geometry: Geometry | Feature): string {
+  if (geometry.type === 'Feature') {
+    geometry = geometry.geometry;
   }
 
-  switch (gj.type) {
+  switch (geometry.type) {
     case 'Point':
-      return `POINT ${wrapParens(pairWKT(gj.coordinates))}`;
+      return `POINT ${wrapParens(pairWKT(geometry.coordinates))}`;
     case 'LineString':
-      return `LINESTRING ${wrapParens(ringWKT(gj.coordinates))}`;
+      return `LINESTRING ${wrapParens(ringWKT(geometry.coordinates))}`;
     case 'Polygon':
-      return `POLYGON ${wrapParens(ringsWKT(gj.coordinates))}`;
+      return `POLYGON ${wrapParens(ringsWKT(geometry.coordinates))}`;
     case 'MultiPoint':
-      return `MULTIPOINT ${wrapParens(ringWKT(gj.coordinates))}`;
+      return `MULTIPOINT ${wrapParens(ringWKT(geometry.coordinates))}`;
     case 'MultiPolygon':
-      return `MULTIPOLYGON ${wrapParens(multiRingsWKT(gj.coordinates))}`;
+      return `MULTIPOLYGON ${wrapParens(multiRingsWKT(geometry.coordinates))}`;
     case 'MultiLineString':
-      return `MULTILINESTRING ${wrapParens(ringsWKT(gj.coordinates))}`;
+      return `MULTILINESTRING ${wrapParens(ringsWKT(geometry.coordinates))}`;
     case 'GeometryCollection':
-      return `GEOMETRYCOLLECTION ${wrapParens(gj.geometries.map(encodeWKT).join(', '))}`;
+      return `GEOMETRYCOLLECTION ${wrapParens(geometry.geometries.map(encodeWKT).join(', '))}`;
     default:
-      return ((_x: never): never => {
-        throw new Error('stringify requires a valid GeoJSON Feature or geometry object as input');
-      })(gj);
+      throw new Error('stringify requires a valid GeoJSON Feature or geometry object as input');
   }
 }
+
+function pairWKT(c: number[]): string {
+  return c.join(' ');
+}
+
+function ringWKT(r: number[][]): string {
+  return r.map(pairWKT).join(', ');
+}
+
+function ringsWKT(r: number[][][]): string {
+  return r.map(ringWKT).map(wrapParens).join(', ');
+}
+
+function multiRingsWKT(r: number[][][][]): string {
+  return r.map(ringsWKT).map(wrapParens).join(', ');
+}
+
+function wrapParens(s: string): string {
+  return `(${s})`;
+}

package/src/lib/utils/binary-writer.ts

@@ -0,0 +1,125 @@
+// loaders.gl, MIT license
+// Forked from https://github.com/cschwarz/wkx under MIT license, Copyright (c) 2013 Christian Schwarz
+
+const LE = true;
+const BE = false;
+
+export default class BinaryWriter {
+  arrayBuffer: ArrayBuffer;
+  dataView: DataView;
+  byteOffset: number = 0;
+  allowResize: boolean = false;
+
+  constructor(size: number, allowResize?: boolean) {
+    this.arrayBuffer = new ArrayBuffer(size);
+    this.dataView = new DataView(this.arrayBuffer);
+    this.byteOffset = 0;
+    this.allowResize = allowResize || false;
+  }
+
+  writeUInt8(value: number): void {
+    this._ensureSize(1);
+    this.dataView.setUint8(this.byteOffset, value);
+    this.byteOffset += 1;
+  }
+  writeUInt16LE(value: number): void {
+    this._ensureSize(2);
+    this.dataView.setUint16(this.byteOffset, value, LE);
+    this.byteOffset += 2;
+  }
+  writeUInt16BE(value: number): void {
+    this._ensureSize(2);
+    this.dataView.setUint16(this.byteOffset, value, BE);
+    this.byteOffset += 2;
+  }
+  writeUInt32LE(value: number): void {
+    this._ensureSize(4);
+    this.dataView.setUint32(this.byteOffset, value, LE);
+    this.byteOffset += 4;
+  }
+  writeUInt32BE(value: number): void {
+    this._ensureSize(4);
+    this.dataView.setUint32(this.byteOffset, value, BE);
+    this.byteOffset += 4;
+  }
+  writeInt8(value: number): void {
+    this._ensureSize(1);
+    this.dataView.setInt8(this.byteOffset, value);
+    this.byteOffset += 1;
+  }
+  writeInt16LE(value: number): void {
+    this._ensureSize(2);
+    this.dataView.setInt16(this.byteOffset, value, LE);
+    this.byteOffset += 2;
+  }
+  writeInt16BE(value: number): void {
+    this._ensureSize(2);
+    this.dataView.setInt16(this.byteOffset, value, BE);
+    this.byteOffset += 2;
+  }
+  writeInt32LE(value: number): void {
+    this._ensureSize(4);
+    this.dataView.setInt32(this.byteOffset, value, LE);
+    this.byteOffset += 4;
+  }
+  writeInt32BE(value: number): void {
+    this._ensureSize(4);
+    this.dataView.setInt32(this.byteOffset, value, BE);
+    this.byteOffset += 4;
+  }
+  writeFloatLE(value: number): void {
+    this._ensureSize(4);
+    this.dataView.setFloat32(this.byteOffset, value, LE);
+    this.byteOffset += 4;
+  }
+  writeFloatBE(value: number): void {
+    this._ensureSize(4);
+    this.dataView.setFloat32(this.byteOffset, value, BE);
+    this.byteOffset += 4;
+  }
+  writeDoubleLE(value: number): void {
+    this._ensureSize(8);
+    this.dataView.setFloat64(this.byteOffset, value, LE);
+    this.byteOffset += 8;
+  }
+  writeDoubleBE(value: number): void {
+    this._ensureSize(8);
+    this.dataView.setFloat64(this.byteOffset, value, BE);
+    this.byteOffset += 8;
+  }
+
+  /** A varint uses a variable number of bytes */
+  writeVarInt(value: number): number {
+    // TODO - ensure size?
+    let length = 1;
+    while ((value & 0xffffff80) !== 0) {
+      this.writeUInt8((value & 0x7f) | 0x80);
+      value >>>= 7;
+      length++;
+    }
+    this.writeUInt8(value & 0x7f);
+    return length;
+  }
+
+  /** Append another ArrayBuffer to this ArrayBuffer */
+  writeBuffer(arrayBuffer: ArrayBuffer): void {
+    this._ensureSize(arrayBuffer.byteLength);
+    const tempArray = new Uint8Array(this.arrayBuffer);
+    tempArray.set(new Uint8Array(arrayBuffer), this.byteOffset);
+    this.byteOffset += arrayBuffer.byteLength;
+  }
+
+  /** Resizes this.arrayBuffer if not enough space */
+  _ensureSize(size: number) {
+    if (this.arrayBuffer.byteLength < this.byteOffset + size) {
+      if (this.allowResize) {
+        const newArrayBuffer = new ArrayBuffer(this.byteOffset + size);
+        const tempArray = new Uint8Array(newArrayBuffer);
+        tempArray.set(new Uint8Array(this.arrayBuffer));
+        this.arrayBuffer = newArrayBuffer;
+      } else {
+        throw new Error('BinaryWriter overflow');
+      }
+    }
+  }
+}

package/src/wkb-writer.ts

@@ -0,0 +1,19 @@
+import type {Writer} from '@loaders.gl/loader-utils';
+import {VERSION} from './lib/utils/version';
+import encodeWKB from './lib/encode-wkb';
+
+/**
+ * WKT exporter
+ */
+export const WKBWriter: Writer = {
+  name: 'WKB (Well Known Binary)',
+  id: 'wkb',
+  module: 'wkt',
+  version: VERSION,
+  extensions: ['wkb'],
+  // @ts-ignore
+  encode: encodeWKB,
+  options: {
+    wkt: {}
+  }
+};