Fix gsplat sorting bug (#6845)

playcanvas · Jul 24, 2024 · dac6e4a · dac6e4a
1 parent 25d05be
commit dac6e4a
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/src/scene/gsplat/gsplat-sorter.js b/src/scene/gsplat/gsplat-sorter.js
@@ -97,11 +97,11 @@ function SortWorker() {
             }
         }
 
-        if (!countBuffer)
+        if (!countBuffer) {
             countBuffer = new Uint32Array(bucketCount);
-
-        for (let i = 0; i < bucketCount; i++)
-            countBuffer[i] = 0;
+        } else {
+            countBuffer.fill(0);
+        }
 
         // generate per vertex distance to camera
         const range = maxDist - minDist;
@@ -121,8 +121,9 @@ function SortWorker() {
         }
 
         // Change countBuffer[i] so that it contains actual position of this digit in outputArray
-        for (let i = 1; i < bucketCount; i++)
+        for (let i = 1; i < bucketCount; i++) {
             countBuffer[i] += countBuffer[i - 1];
+        }
 
         // Build the output array
         for (let i = 0; i < numVertices; i++) {
@@ -137,6 +138,7 @@ function SortWorker() {
             const result = binarySearch(0, numVertices - 1, i => -dist(i));
             return Math.min(numVertices, Math.abs(result));
         };
+        const count = dist(numVertices - 1) >= 0 ? findZero() : numVertices;
 
         // apply mapping
         if (mapping) {
@@ -148,7 +150,7 @@ function SortWorker() {
         // send results
         self.postMessage({
             order: order.buffer,
-            count: dist(numVertices - 1) >= 0 ? findZero() : numVertices
+            count
         }, [order.buffer]);
 
         order = null;