las-laz-js 0.1.0 → 0.2.0

package/src/write/writeLas.js

@@ -30,10 +30,16 @@ function setU64(v, pos, value) {
 
 function isGeographic(epsg) { return epsg >= 4000 && epsg < 5000; }
 
-function crsRecords(crs) {
+// LAS 1.4 R15: point formats 6-10 carry the CRS as OGC WKT only; GeoTIFF
+// keys belong to formats 0-5. The library does not turn an EPSG code into
+// WKT (that needs a projection database), so an EPSG-only CRS is refused there.
+function crsRecords(crs, fmt) {
 	var out = [];
 	if (!crs) return out;
-	if (crs.epsg) {
+	if (fmt >= 6 && !crs.wkt && crs.epsg) {
+		throw new LasError(ERROR_CODES.INVALID_ARGUMENT, "point format " + fmt + " must carry its coordinate system as WKT (LAS 1.4); pass crs.wkt — an EPSG code alone is GeoTIFF keys, which formats 6-10 do not allow");
+	}
+	if (crs.epsg && fmt < 6) {
 		var geo = isGeographic(crs.epsg);
 		var keys = [[1024, geo ? 2 : 1], [1025, 1], [geo ? 2048 : 3072, crs.epsg]];
 		var b = new Uint8Array(8 + keys.length * 8);
@@ -73,7 +79,7 @@ function freshTotals() {
 function headerBytes(s, so, totals) {
 	var o = s.opts;
 	var headerSize = HEADER_SIZE[s.version];
-	var vlrs = crsRecords(o.crs);
+	var vlrs = crsRecords(o.crs, s.fmt);
 	var vlrSize = vlrs.reduce(function (n, r) { return n + 54 + r.bytes.length; }, 0);
 	var out = new Uint8Array(headerSize + vlrSize);
 	var v = new DataView(out.buffer);
@@ -150,19 +156,38 @@ export async function writeLas(points, opts) {
 	if (o.scale && !o.offset) so = { scale: o.scale, offset: chooseScaleOffset(boundsOfPoints(points, o.localOrigin)).offset };
 	var len = POINT_FORMATS[s.fmt].size;
 	var totals = freshTotals();
-	var records = new Uint8Array(points.count * len);
-	encodeRecords(points, s.fmt, len, so, o.localOrigin, new DataView(records.buffer), 0, totals);
+	// One buffer: records are encoded in place after the header, whose
+	// size does not depend on the totals; the header is filled in last
+	var headLength = headerBytes(s, so, totals).length;
+	var out = new Uint8Array(headLength + points.count * len);
+	encodeRecords(points, s.fmt, len, so, o.localOrigin, new DataView(out.buffer), headLength, totals);
 	totals.count = points.count;
-	var head = headerBytes(s, so, totals);
-	var out = new Uint8Array(head.length + records.length);
-	out.set(head, 0);
-	out.set(records, head.length);
+	out.set(headerBytes(s, so, totals), 0);
 	return o.as === "blob" ? new Blob([out], { type: "application/octet-stream" }) : out;
 }
 
+// One positioned write for any sink: an OPFS scratch writer (writeAt),
+// a Node FileHandle (write(buffer, offset, length, position), which may
+// write short), or a plain { write(bytes, position) }
+function positionedWrite(sink) {
+	if (typeof sink.writeAt === "function") return async function (bytes, pos) { await sink.writeAt(bytes, pos); };
+	if (typeof sink.fd === "number" && typeof sink.write === "function") {
+		return async function (bytes, pos) {
+			for (var done = 0; done < bytes.length;) {
+				var r = await sink.write(bytes, done, bytes.length - done, pos + done);
+				if (!(r && r.bytesWritten > 0)) throw new LasError(ERROR_CODES.TRUNCATED, "the file handle stopped writing", pos + done);
+				done += r.bytesWritten;
+			}
+		};
+	}
+	if (typeof sink.write === "function") return async function (bytes, pos) { await sink.write(bytes, pos); };
+	throw new LasError(ERROR_CODES.INVALID_ARGUMENT, "createLasWriter needs a sink with writeAt(bytes, position) or write(bytes, position)");
+}
+
 /**
  * Stream a LAS file to a sink. Header totals are patched on close().
- * @param {{write: function(Uint8Array, number): (void|Promise)}} sink
+ * @param {object} sink - an OPFS scratch writer (openScratchSyncWriter), a Node
+ *   FileHandle, or any { write(bytes, position) }
  * @param {object} opts - as writeLas; scale and offset are required (the bounds are not known yet)
  * @returns {Promise<{writePoints: function(object): Promise, close: function(): Promise<{count:number}>}>}
  */
@@ -173,17 +198,25 @@ export async function createLasWriter(sink, opts) {
 	var so = { scale: opts.scale, offset: opts.offset };
 	var len = POINT_FORMATS[s.fmt].size;
 	var totals = freshTotals();
+	var put = positionedWrite(sink);
 	var start = headerBytes(s, so, totals).length;
-	await sink.write(headerBytes(s, so, totals), 0);
+	var pending = [];
+	await put(headerBytes(s, so, totals), 0);
 	return {
 		writePoints: async function (points) {
 			var buf = new Uint8Array(points.count * len);
 			encodeRecords(points, s.fmt, len, so, opts.localOrigin, new DataView(buf.buffer), 0, totals);
-			await sink.write(buf, start + totals.count * len);
+			// Reserve this batch's place BEFORE awaiting, so calls made without
+			// waiting for each other never land on the same bytes
+			var pos = start + totals.count * len;
 			totals.count += points.count;
+			var job = put(buf, pos);
+			pending.push(job);
+			await job;
 		},
 		close: async function () {
-			await sink.write(headerBytes(s, so, totals), 0);
+			await Promise.all(pending);
+			await put(headerBytes(s, so, totals), 0);
 			return { count: totals.count };
 		}
 	};