@luxonis/visualizer-protobuf 3.1.13 → 3.1.14

package/dist/{pointcloudFromDepth.worker-60RA4Ief.js → pointcloudFromDepth.worker-wzSOsJVb.js}

@@ -60,6 +60,8 @@ function depthToPointCloudBuffer(uint16Array, width, height, fx, fy, maxSteroDep
 }
 const WORKGROUP_SIZE_X = 128;
 const POINT_STRIDE_BYTES = 16;
+let cachedGpuContext;
+let gpuContextPromise;
 const shader = `
 struct Intrinsics {
   fx: f32,
@@ -195,6 +197,85 @@ fn main(@builtin(global_invocation_id) global_id: vec3<u32>) {
   }
 }
 `;
+async function getGPUContext() {
+  if (cachedGpuContext) {
+    return cachedGpuContext;
+  }
+  if (gpuContextPromise) {
+    return await gpuContextPromise;
+  }
+  gpuContextPromise = (async () => {
+    if (typeof navigator.gpu === "undefined") {
+      return undefined;
+    }
+    const adapter = await navigator.gpu.requestAdapter();
+    if (!adapter) {
+      return undefined;
+    }
+    const device = await adapter.requestDevice();
+    const bindGroupLayout = device.createBindGroupLayout({
+      label: "Compute Bind Group Layout",
+      entries: [{
+        binding: 0,
+        visibility: GPUShaderStage.COMPUTE,
+        buffer: {
+          type: "read-only-storage"
+        }
+      }, {
+        binding: 1,
+        visibility: GPUShaderStage.COMPUTE,
+        buffer: {
+          type: "read-only-storage"
+        }
+      }, {
+        binding: 2,
+        visibility: GPUShaderStage.COMPUTE,
+        buffer: {
+          type: "storage"
+        }
+      }, {
+        binding: 3,
+        visibility: GPUShaderStage.COMPUTE,
+        buffer: {
+          type: "read-only-storage"
+        }
+      }]
+    });
+    const shaderModule = device.createShaderModule({
+      label: "Compute Shader Module",
+      code: shader
+    });
+    const pipeline = device.createComputePipeline({
+      label: "Compute Pipeline",
+      layout: device.createPipelineLayout({
+        bindGroupLayouts: [bindGroupLayout]
+      }),
+      compute: {
+        module: shaderModule,
+        entryPoint: "main"
+      }
+    });
+    const context = {
+      device,
+      bindGroupLayout,
+      pipeline
+    };
+    void device.lost.then(() => {
+      var _cachedGpuContext;
+      if (((_cachedGpuContext = cachedGpuContext) === null || _cachedGpuContext === void 0 ? void 0 : _cachedGpuContext.device) === device) {
+        cachedGpuContext = undefined;
+      }
+      gpuContextPromise = undefined;
+    });
+    cachedGpuContext = context;
+    return context;
+  })();
+  try {
+    return await gpuContextPromise;
+  } finally {
+    gpuContextPromise = undefined;
+  }
+}
 async function depthToPointcloudGPU(depthFrames, depthWidth, depthHeight, fx, fy, cx, cy, i420Frames, i420Width, i420Height, maxSteroDepth, {
   hasGPU
 }) {
@@ -243,11 +324,11 @@ async function depthToPointcloudGPU(depthFrames, depthWidth, depthHeight, fx, fy
   let readbackBuffer = null;
   try {
     var _i420Frames$0$byteLen, _i420Frames$;
-    const adapter = await navigator.gpu.requestAdapter();
-    if (!adapter) {
+    const gpuContext = await getGPUContext();
+    if (!gpuContext) {
       throw new Error("Failed to get GPU adapter.");
     }
-    device = await adapter.requestDevice();
+    device = gpuContext.device;
     const queue = device.queue;
     const numPixelsPerFrame = depthWidth * depthHeight;
     const outputFrameSize = numPixelsPerFrame * POINT_STRIDE_BYTES;
@@ -340,37 +421,9 @@ async function depthToPointcloudGPU(depthFrames, depthWidth, depthHeight, fx, fy
       size: totalOutputBufferSize,
       usage: GPUBufferUsage.COPY_DST | GPUBufferUsage.MAP_READ
     });
-    const bindGroupLayout = device.createBindGroupLayout({
-      label: "Compute Bind Group Layout",
-      entries: [{
-        binding: 0,
-        visibility: GPUShaderStage.COMPUTE,
-        buffer: {
-          type: "read-only-storage"
-        }
-      }, {
-        binding: 1,
-        visibility: GPUShaderStage.COMPUTE,
-        buffer: {
-          type: "read-only-storage"
-        }
-      }, {
-        binding: 2,
-        visibility: GPUShaderStage.COMPUTE,
-        buffer: {
-          type: "storage"
-        }
-      }, {
-        binding: 3,
-        visibility: GPUShaderStage.COMPUTE,
-        buffer: {
-          type: "read-only-storage"
-        }
-      }]
-    });
     const bindGroup = device.createBindGroup({
       label: "Compute Bind Group",
-      layout: bindGroupLayout,
+      layout: gpuContext.bindGroupLayout,
       entries: [{
         binding: 0,
         resource: {
@@ -393,27 +446,13 @@ async function depthToPointcloudGPU(depthFrames, depthWidth, depthHeight, fx, fy
         }
       }]
     });
-    const shaderModule = device.createShaderModule({
-      label: "Compute Shader Module",
-      code: shader
-    });
-    const pipeline = device.createComputePipeline({
-      label: "Compute Pipeline",
-      layout: device.createPipelineLayout({
-        bindGroupLayouts: [bindGroupLayout]
-      }),
-      compute: {
-        module: shaderModule,
-        entryPoint: "main"
-      }
-    });
     const commandEncoder = device.createCommandEncoder({
       label: "Compute Command Encoder"
     });
     const passEncoder = commandEncoder.beginComputePass({
       label: "Compute Pass"
     });
-    passEncoder.setPipeline(pipeline);
+    passEncoder.setPipeline(gpuContext.pipeline);
     passEncoder.setBindGroup(0, bindGroup);
     const numWorkgroupsX = Math.ceil(numPixelsPerFrame / WORKGROUP_SIZE_X);
     const numWorkgroupsY = numFrames;