@holoscript/engine 6.0.3 → 6.0.4

package/dist/chunk-HNEE36PY.js

@@ -0,0 +1,93 @@
+// src/gpu/GaussianSplatExtractor.ts
+var GaussianSplatExtractor = class {
+  context;
+  device;
+  options;
+  readbackBuffersA = null;
+  readbackBuffersB = null;
+  isUsingA = true;
+  ongoingReadback = null;
+  frameCounter = 0;
+  constructor(context, options) {
+    this.context = context;
+    this.device = context.getDevice();
+    this.options = {
+      maxSplats: options.maxSplats
+    };
+    this.initializeBuffers();
+  }
+  initializeBuffers() {
+    const compressedSize = this.options.maxSplats * 32;
+    const indicesSize = this.options.maxSplats * 4;
+    this.readbackBuffersA = {
+      compressed: this.createReadbackBuffer(compressedSize),
+      indices: this.createReadbackBuffer(indicesSize)
+    };
+    this.readbackBuffersB = {
+      compressed: this.createReadbackBuffer(compressedSize),
+      indices: this.createReadbackBuffer(indicesSize)
+    };
+  }
+  createReadbackBuffer(size) {
+    return this.device.createBuffer({
+      size,
+      usage: GPUBufferUsage.MAP_READ | GPUBufferUsage.COPY_DST
+    });
+  }
+  /**
+   * Captures the sorted state of the GaussianSplatSorter.
+   * Internally pipelines the GPU readback to avoid stalling the main render loop.
+   */
+  extractFrame(sorter, camera, compressedSource, indicesSource) {
+    const stats = sorter.getStats();
+    if (stats.splatCount === 0) return Promise.resolve(null);
+    if (this.ongoingReadback) {
+      return Promise.resolve(null);
+    }
+    const currentBuffers = this.isUsingA ? this.readbackBuffersA : this.readbackBuffersB;
+    this.isUsingA = !this.isUsingA;
+    const compressedSize = stats.splatCount * 32;
+    const indicesSize = stats.splatCount * 4;
+    const currentFrame = ++this.frameCounter;
+    const encoder = this.device.createCommandEncoder({
+      label: "splat-extractor-copy-encoder"
+    });
+    encoder.copyBufferToBuffer(compressedSource, 0, currentBuffers.compressed, 0, compressedSize);
+    encoder.copyBufferToBuffer(indicesSource, 0, currentBuffers.indices, 0, indicesSize);
+    this.device.queue.submit([encoder.finish()]);
+    this.ongoingReadback = Promise.all([
+      currentBuffers.compressed.mapAsync(GPUMapMode.READ, 0, compressedSize),
+      currentBuffers.indices.mapAsync(GPUMapMode.READ, 0, indicesSize)
+    ]).then(() => {
+      const compData = currentBuffers.compressed.getMappedRange(0, compressedSize).slice(0);
+      const indData = currentBuffers.indices.getMappedRange(0, indicesSize).slice(0);
+      currentBuffers.compressed.unmap();
+      currentBuffers.indices.unmap();
+      this.ongoingReadback = null;
+      const packet = {
+        frameId: currentFrame,
+        cameraState: {
+          viewProjectionMatrix: Array.from(camera.viewProjectionMatrix),
+          cameraPosition: [
+            camera.cameraPosition[0],
+            camera.cameraPosition[1],
+            camera.cameraPosition[2]
+          ]
+        },
+        splatCount: stats.splatCount,
+        compressedSplatsBuffer: compData,
+        sortedIndicesBuffer: indData
+      };
+      return packet;
+    }).catch((e) => {
+      console.warn("GaussianSplatExtractor readback failed", e);
+      this.ongoingReadback = null;
+      return null;
+    });
+    return this.ongoingReadback;
+  }
+};
+
+export {
+  GaussianSplatExtractor
+};