@babylonjs/core 7.9.0 → 7.10.0

package/Collisions/gpuPicker.js

@@ -0,0 +1,256 @@
+
+import { RenderTargetTexture } from "../Materials/Textures/renderTargetTexture.js";
+import { ShaderMaterial } from "../Materials/shaderMaterial.js";
+import { Color3, Color4 } from "../Maths/math.color.js";
+import { VertexBuffer } from "../Meshes/buffer.js";
+/**
+ * Class used to perform a picking operation using GPU
+ * Please note that GPUPIcker cannot pick instances, only meshes
+ */
+export class GPUPicker {
+    constructor() {
+        this._pickingTexure = null;
+        this._idMap = {};
+        this._idColors = [];
+        this._cachedMaterials = [];
+        this._meshRenderingCount = 0;
+        this._userDefinedList = false;
+    }
+    _createRenderTarget(scene, width, height) {
+        this._pickingTexure = new RenderTargetTexture("pickingTexure", { width: width, height: height }, scene, false, undefined, 0, false, 1);
+    }
+    _createColorMaterial(scene) {
+        if (this._renderMaterial) {
+            this._renderMaterial.dispose();
+        }
+        const defines = [];
+        const options = {
+            attributes: [VertexBuffer.PositionKind],
+            uniforms: ["world", "viewProjection", "color"],
+            needAlphaBlending: false,
+            defines: defines,
+            useClipPlane: null,
+        };
+        this._renderMaterial = new ShaderMaterial("colorShader", scene, "color", options, false);
+        const callback = (mesh) => {
+            if (!mesh) {
+                return;
+            }
+            const effect = this._renderMaterial.getEffect();
+            if (!mesh.hasInstances && !mesh.isAnInstance) {
+                effect.setColor4("color", this._idColors[mesh.uniqueId], 1);
+            }
+            this._meshRenderingCount++;
+        };
+        this._renderMaterial.customBindingObservable.add(callback);
+    }
+    /**
+     * Set the list of meshes to pick from
+     * @param list defines the list of meshes to pick from
+     */
+    setPickingList(list) {
+        if (!list) {
+            if (this._userDefinedList) {
+                for (let index = 0; index < this._pickableMeshes.length; index++) {
+                    const mesh = this._pickableMeshes[index];
+                    if (mesh.hasInstances) {
+                        mesh.removeVerticesData(VertexBuffer.ColorKind);
+                    }
+                }
+            }
+            this._userDefinedList = false;
+            this._pickableMeshes = [];
+            this._idMap = {};
+            this._idColors = [];
+            return;
+        }
+        this._userDefinedList = true;
+        this._pickableMeshes = list;
+        // We will affect colors and create vertex color buffers
+        let id = 1;
+        for (let index = 0; index < this._pickableMeshes.length; index++) {
+            const mesh = this._pickableMeshes[index];
+            mesh.useVertexColors = false;
+            if (mesh.isAnInstance) {
+                continue; // This will be handled by the source mesh
+            }
+            const r = (id & 0xff0000) >> 16;
+            const g = (id & 0x00ff00) >> 8;
+            const b = (id & 0x0000ff) >> 0;
+            this._idMap[`${r}_${g}_${b}`] = index;
+            id++;
+            if (mesh.hasInstances) {
+                const instances = mesh.instances;
+                const colorData = new Float32Array(4 * (instances.length + 1));
+                const engine = mesh.getEngine();
+                colorData[0] = r / 255.0;
+                colorData[1] = g / 255.0;
+                colorData[2] = b / 255.0;
+                colorData[3] = 1.0;
+                for (let i = 0; i < instances.length; i++) {
+                    const instance = instances[i];
+                    const r = (id & 0xff0000) >> 16;
+                    const g = (id & 0x00ff00) >> 8;
+                    const b = (id & 0x0000ff) >> 0;
+                    this._idMap[`${r}_${g}_${b}`] = this._pickableMeshes.indexOf(instance);
+                    colorData[(i + 1) * 4] = r / 255.0;
+                    colorData[(i + 1) * 4 + 1] = g / 255.0;
+                    colorData[(i + 1) * 4 + 2] = b / 255.0;
+                    colorData[(i + 1) * 4 + 3] = 1.0;
+                    id++;
+                }
+                const buffer = new VertexBuffer(engine, colorData, VertexBuffer.ColorKind, false, false, 4, true);
+                mesh.setVerticesBuffer(buffer, true);
+            }
+            else {
+                this._idColors[mesh.uniqueId] = Color3.FromInts(r, g, b);
+            }
+        }
+    }
+    /**
+     * Execute a picking operation
+     * @param x defines the X coordinates where to run the pick
+     * @param y defines the Y coordinates where to run the pick
+     * @param scene defines the scene to pick from
+     * @param disposeWhenDone defines a boolean indicating we do not want to keep resources alive
+     * @returns A promise with the picking results
+     */
+    pickAsync(x, y, scene, disposeWhenDone = true) {
+        const engine = scene.getEngine();
+        const rttSizeW = engine.getRenderWidth();
+        const rttSizeH = engine.getRenderHeight();
+        if (!this._readbuffer) {
+            this._readbuffer = new Uint8Array(engine.isWebGPU ? 256 : 4); // Because of block alignment in WebGPU
+        }
+        this._meshRenderingCount = 0;
+        // Ensure ints
+        x = x >> 0;
+        y = y >> 0;
+        if (x < 0 || y < 0 || x >= rttSizeW || y >= rttSizeH) {
+            return Promise.resolve(null);
+        }
+        // Invert Y
+        y = rttSizeH - y;
+        if (!this._pickingTexure) {
+            this._createRenderTarget(scene, rttSizeW, rttSizeH);
+        }
+        else {
+            const size = this._pickingTexure.getSize();
+            if (size.width !== rttSizeW || size.height !== rttSizeH || this._cachedScene !== scene) {
+                this._pickingTexure.dispose();
+                this._createRenderTarget(scene, rttSizeW, rttSizeH);
+            }
+        }
+        if (!this._cachedScene || this._cachedScene !== scene) {
+            this._createColorMaterial(scene);
+        }
+        this._cachedScene = scene;
+        scene.customRenderTargets.push(this._pickingTexure);
+        this._pickingTexure.renderList = [];
+        this._pickingTexure.clearColor = new Color4(0, 0, 0, 0);
+        // We need to give every mesh an unique color (when there is no picking list)
+        this._pickingTexure.onBeforeRender = () => {
+            if (!this._userDefinedList) {
+                this._pickableMeshes = scene.meshes.filter((m) => m.geometry && m.isPickable && m.onBeforeRenderObservable);
+            }
+            for (let index = 0; index < this._pickableMeshes.length; index++) {
+                const mesh = this._pickableMeshes[index];
+                if (!mesh.isAnInstance) {
+                    this._cachedMaterials[index] = mesh.material;
+                    if (!this._userDefinedList) {
+                        // We need to define the color for each mesh as it was not previously defined
+                        const id = index + 1;
+                        const r = (id & 0xff0000) >> 16;
+                        const g = (id & 0x00ff00) >> 8;
+                        const b = (id & 0x0000ff) >> 0;
+                        this._idMap[`${r}_${g}_${b}`] = index;
+                        this._idColors[mesh.uniqueId] = Color3.FromInts(r, g, b);
+                    }
+                    else {
+                        mesh.useVertexColors = true; // In that case we will be using vertex colors and not an uniform to support instances
+                    }
+                    mesh.material = this._renderMaterial;
+                }
+                this._pickingTexure?.renderList?.push(mesh);
+            }
+            // Enable scissor
+            if (engine.enableScissor) {
+                engine.enableScissor(x, y, 1, 1);
+            }
+        };
+        return new Promise((resolve, reject) => {
+            this._pickingTexure.onAfterRender = async () => {
+                // Disable scissor
+                if (engine.disableScissor) {
+                    engine.disableScissor();
+                }
+                if (!this._pickingTexure) {
+                    reject();
+                }
+                let pickedMesh = null;
+                const wasSuccessfull = this._meshRenderingCount > 0;
+                // Restore materials
+                for (let index = 0; index < this._pickableMeshes.length; index++) {
+                    const mesh = this._pickableMeshes[index];
+                    if (!mesh.isAnInstance) {
+                        mesh.material = this._cachedMaterials[index];
+                        if (this._userDefinedList) {
+                            mesh.useVertexColors = false;
+                        }
+                    }
+                }
+                if (wasSuccessfull) {
+                    // Remove from the active RTTs
+                    const index = scene.customRenderTargets.indexOf(this._pickingTexure);
+                    if (index > -1) {
+                        scene.customRenderTargets.splice(index, 1);
+                    }
+                    // Do the actual picking
+                    if (await this._readTexturePixelsAsync(x, y)) {
+                        const colorId = `${this._readbuffer[0]}_${this._readbuffer[1]}_${this._readbuffer[2]}`;
+                        pickedMesh = this._pickableMeshes[this._idMap[colorId]];
+                    }
+                }
+                // Clean-up
+                if (!this._userDefinedList) {
+                    this._idMap = {};
+                    this._idColors = [];
+                    this._pickableMeshes = [];
+                }
+                this._pickingTexure.renderList = [];
+                if (!wasSuccessfull) {
+                    this._meshRenderingCount = 0;
+                    return; // We need to wait for the shaders to be ready
+                }
+                else {
+                    if (disposeWhenDone) {
+                        this.dispose();
+                    }
+                    resolve(pickedMesh);
+                }
+            };
+        });
+    }
+    async _readTexturePixelsAsync(x, y) {
+        if (!this._cachedScene || !this._pickingTexure?._texture) {
+            return false;
+        }
+        const engine = this._cachedScene.getEngine();
+        await engine._readTexturePixels(this._pickingTexure._texture, 1, 1, -1, 0, this._readbuffer, true, true, x, y);
+        return true;
+    }
+    /** Release the resources */
+    dispose() {
+        if (this._userDefinedList) {
+            this.setPickingList(null);
+        }
+        this._pickableMeshes = [];
+        this._cachedScene = null;
+        // Cleaning up
+        this._pickingTexure?.dispose();
+        this._pickingTexure = null;
+        this._renderMaterial?.dispose();
+        this._renderMaterial = null;
+    }
+}
+//# sourceMappingURL=gpuPicker.js.map

package/Collisions/gpuPicker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gpuPicker.js","sourceRoot":"","sources":["../../../../dev/core/src/Collisions/gpuPicker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,gCAA+B;AAGnD,OAAO,EAAE,mBAAmB,EAAE,qDAAoD;AAElF,OAAO,EAAE,cAAc,EAAE,uCAAsC;AAC/D,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,+BAA8B;AAEvD,OAAO,EAAE,YAAY,EAAE,4BAA2B;AAKlD;;;GAGG;AACH,MAAM,OAAO,SAAS;IAAtB;QACY,mBAAc,GAAkC,IAAI,CAAC;QACrD,WAAM,GAA8B,EAAE,CAAC;QACvC,cAAS,GAAkB,EAAE,CAAC;QAC9B,qBAAgB,GAA8B,EAAE,CAAC;QAKjD,wBAAmB,GAAW,CAAC,CAAC;QAChC,qBAAgB,GAAG,KAAK,CAAC;IA8RrC,CAAC;IA5RW,mBAAmB,CAAC,KAAY,EAAE,KAAa,EAAE,MAAc;QACnE,IAAI,CAAC,cAAc,GAAG,IAAI,mBAAmB,CACzC,eAAe,EACf,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,EAChC,KAAK,EACL,KAAK,EACL,SAAS,EACT,SAAS,CAAC,wBAAwB,EAClC,KAAK,EACL,SAAS,CAAC,uBAAuB,CACpC,CAAC;IACN,CAAC;IAEO,oBAAoB,CAAC,KAAY;QACrC,IAAI,IAAI,CAAC,eAAe,EAAE;YACtB,IAAI,CAAC,eAAe,CAAC,OAAO,EAAE,CAAC;SAClC;QAED,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAG;YACZ,UAAU,EAAE,CAAC,YAAY,CAAC,YAAY,CAAC;YACvC,QAAQ,EAAE,CAAC,OAAO,EAAE,gBAAgB,EAAE,OAAO,CAAC;YAC9C,iBAAiB,EAAE,KAAK;YACxB,OAAO,EAAE,OAAO;YAChB,YAAY,EAAE,IAAI;SACrB,CAAC;QAEF,IAAI,CAAC,eAAe,GAAG,IAAI,cAAc,CAAC,aAAa,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;QAEzF,MAAM,QAAQ,GAAG,CAAC,IAA8B,EAAE,EAAE;YAChD,IAAI,CAAC,IAAI,EAAE;gBACP,OAAO;aACV;YAED,MAAM,MAAM,GAAG,IAAI,CAAC,eAAgB,CAAC,SAAS,EAAE,CAAC;YAEjD,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;gBAC1C,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;aAC/D;YAED,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC/B,CAAC,CAAC;QAEF,IAAI,CAAC,eAAe,CAAC,uBAAuB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC/D,CAAC;IAED;;;OAGG;IACI,cAAc,CAAC,IAAmC;QACrD,IAAI,CAAC,IAAI,EAAE;YACP,IAAI,IAAI,CAAC,gBAAgB,EAAE;gBACvB,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;oBAC9D,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;oBACzC,IAAI,IAAI,CAAC,YAAY,EAAE;wBAClB,IAAa,CAAC,kBAAkB,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;qBAC7D;iBACJ;aACJ;YACD,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC;YAC9B,IAAI,CAAC,eAAe,GAAG,EAAE,CAAC;YAC1B,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;YACjB,IAAI,CAAC,SAAS,GAAG,EAAE,CAAC;YACpB,OAAO;SACV;QACD,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC7B,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;QAE5B,wDAAwD;QACxD,IAAI,EAAE,GAAG,CAAC,CAAC;QACX,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;YAC9D,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YACzC,IAAI,CAAC,eAAe,GAAG,KAAK,CAAC;YAE7B,IAAI,IAAI,CAAC,YAAY,EAAE;gBACnB,SAAS,CAAC,0CAA0C;aACvD;YAED,MAAM,CAAC,GAAG,CAAC,EAAE,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;YAChC,MAAM,CAAC,GAAG,CAAC,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC/B,MAAM,CAAC,GAAG,CAAC,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC/B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC;YACtC,EAAE,EAAE,CAAC;YAEL,IAAI,IAAI,CAAC,YAAY,EAAE;gBACnB,MAAM,SAAS,GAAI,IAAa,CAAC,SAAS,CAAC;gBAC3C,MAAM,SAAS,GAAG,IAAI,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;gBAC/D,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;gBAEhC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;gBACzB,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;gBACzB,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;gBACzB,SAAS,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC;gBACnB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;oBACvC,MAAM,QAAQ,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;oBAC9B,MAAM,CAAC,GAAG,CAAC,EAAE,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;oBAChC,MAAM,CAAC,GAAG,CAAC,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;oBAC/B,MAAM,CAAC,GAAG,CAAC,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;oBAC/B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;oBAEvE,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;oBACnC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;oBACvC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;oBACvC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC;oBACjC,EAAE,EAAE,CAAC;iBACR;gBAED,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,MAAM,EAAE,SAAS,EAAE,YAAY,CAAC,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;gBACjG,IAAa,CAAC,iBAAiB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;aAClD;iBAAM;gBACH,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;aAC5D;SACJ;IACL,CAAC;IAED;;;;;;;OAOG;IACI,SAAS,CAAC,CAAS,EAAE,CAAS,EAAE,KAAY,EAAE,eAAe,GAAG,IAAI;QACvE,MAAM,MAAM,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC;QACjC,MAAM,QAAQ,GAAG,MAAM,CAAC,cAAc,EAAE,CAAC;QACzC,MAAM,QAAQ,GAAG,MAAM,CAAC,eAAe,EAAE,CAAC;QAE1C,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;YACnB,IAAI,CAAC,WAAW,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,uCAAuC;SACxG;QAED,IAAI,CAAC,mBAAmB,GAAG,CAAC,CAAC;QAC7B,cAAc;QACd,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACX,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAEX,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,QAAQ,IAAI,CAAC,IAAI,QAAQ,EAAE;YAClD,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;SAChC;QAED,WAAW;QACX,CAAC,GAAG,QAAQ,GAAG,CAAC,CAAC;QAEjB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE;YACtB,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;SACvD;aAAM;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,CAAC;YAE3C,IAAI,IAAI,CAAC,KAAK,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,KAAK,QAAQ,IAAI,IAAI,CAAC,YAAY,KAAK,KAAK,EAAE;gBACpF,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,CAAC;gBAC9B,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;aACvD;SACJ;QAED,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,YAAY,KAAK,KAAK,EAAE;YACnD,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;SACpC;QAED,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAC1B,KAAK,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,cAAe,CAAC,CAAC;QACrD,IAAI,CAAC,cAAe,CAAC,UAAU,GAAG,EAAE,CAAC;QACrC,IAAI,CAAC,cAAe,CAAC,UAAU,GAAG,IAAI,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAEzD,6EAA6E;QAC7E,IAAI,CAAC,cAAe,CAAC,cAAc,GAAG,GAAG,EAAE;YACvC,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE;gBACxB,IAAI,CAAC,eAAe,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAU,CAAC,QAAQ,IAAI,CAAC,CAAC,UAAU,IAAK,CAAU,CAAC,wBAAwB,CAAW,CAAC;aAC7I;YAED,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;gBAC9D,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;gBACzC,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;oBACpB,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC;oBAC7C,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE;wBACxB,6EAA6E;wBAC7E,MAAM,EAAE,GAAG,KAAK,GAAG,CAAC,CAAC;wBACrB,MAAM,CAAC,GAAG,CAAC,EAAE,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;wBAChC,MAAM,CAAC,GAAG,CAAC,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;wBAC/B,MAAM,CAAC,GAAG,CAAC,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;wBAC/B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC;wBAEtC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;qBAC5D;yBAAM;wBACH,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,CAAC,sFAAsF;qBACtH;oBACD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC;iBACxC;gBACD,IAAI,CAAC,cAAc,EAAE,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;aAC/C;YAED,iBAAiB;YACjB,IAAK,MAAgC,CAAC,aAAa,EAAE;gBAChD,MAAgC,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;aAC/D;QACL,CAAC,CAAC;QAEF,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACnC,IAAI,CAAC,cAAe,CAAC,aAAa,GAAG,KAAK,IAAI,EAAE;gBAC5C,kBAAkB;gBAClB,IAAK,MAAgC,CAAC,cAAc,EAAE;oBACjD,MAAgC,CAAC,cAAc,EAAE,CAAC;iBACtD;gBAED,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE;oBACtB,MAAM,EAAE,CAAC;iBACZ;gBAED,IAAI,UAAU,GAA2B,IAAI,CAAC;gBAC9C,MAAM,cAAc,GAAG,IAAI,CAAC,mBAAmB,GAAG,CAAC,CAAC;gBAEpD,oBAAoB;gBACpB,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;oBAC9D,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;oBACzC,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;wBACpB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;wBAE7C,IAAI,IAAI,CAAC,gBAAgB,EAAE;4BACvB,IAAI,CAAC,eAAe,GAAG,KAAK,CAAC;yBAChC;qBACJ;iBACJ;gBAED,IAAI,cAAc,EAAE;oBAChB,8BAA8B;oBAC9B,MAAM,KAAK,GAAG,KAAK,CAAC,mBAAmB,CAAC,OAAO,CAAC,IAAI,CAAC,cAAe,CAAC,CAAC;oBACtE,IAAI,KAAK,GAAG,CAAC,CAAC,EAAE;wBACZ,KAAK,CAAC,mBAAmB,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;qBAC9C;oBAED,wBAAwB;oBACxB,IAAI,MAAM,IAAI,CAAC,uBAAuB,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;wBAC1C,MAAM,OAAO,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;wBACvF,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;qBAC3D;iBACJ;gBAED,WAAW;gBACX,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE;oBACxB,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;oBACjB,IAAI,CAAC,SAAS,GAAG,EAAE,CAAC;oBAEpB,IAAI,CAAC,eAAe,GAAG,EAAE,CAAC;iBAC7B;gBACD,IAAI,CAAC,cAAe,CAAC,UAAU,GAAG,EAAE,CAAC;gBAErC,IAAI,CAAC,cAAc,EAAE;oBACjB,IAAI,CAAC,mBAAmB,GAAG,CAAC,CAAC;oBAC7B,OAAO,CAAC,8CAA8C;iBACzD;qBAAM;oBACH,IAAI,eAAe,EAAE;wBACjB,IAAI,CAAC,OAAO,EAAE,CAAC;qBAClB;oBACD,OAAO,CAAC,UAAU,CAAC,CAAC;iBACvB;YACL,CAAC,CAAC;QACN,CAAC,CAAC,CAAC;IACP,CAAC;IAEO,KAAK,CAAC,uBAAuB,CAAC,CAAS,EAAE,CAAS;QACtD,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,QAAQ,EAAE;YACtD,OAAO,KAAK,CAAC;SAChB;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,CAAC;QAC7C,MAAM,MAAM,CAAC,kBAAkB,CAAC,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAE/G,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,4BAA4B;IACrB,OAAO;QACV,IAAI,IAAI,CAAC,gBAAgB,EAAE;YACvB,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;SAC7B;QACD,IAAI,CAAC,eAAe,GAAG,EAAE,CAAC;QAC1B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAEzB,cAAc;QACd,IAAI,CAAC,cAAc,EAAE,OAAO,EAAE,CAAC;QAC/B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,IAAI,CAAC,eAAe,EAAE,OAAO,EAAE,CAAC;QAChC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;IAChC,CAAC;CACJ","sourcesContent":["import { Constants } from \"core/Engines/constants\";\r\nimport type { Engine } from \"core/Engines/engine\";\r\nimport type { WebGPUEngine } from \"core/Engines/webgpuEngine\";\r\nimport { RenderTargetTexture } from \"core/Materials/Textures/renderTargetTexture\";\r\nimport type { Material } from \"core/Materials/material\";\r\nimport { ShaderMaterial } from \"core/Materials/shaderMaterial\";\r\nimport { Color3, Color4 } from \"core/Maths/math.color\";\r\nimport type { AbstractMesh } from \"core/Meshes/abstractMesh\";\r\nimport { VertexBuffer } from \"core/Meshes/buffer\";\r\nimport type { Mesh } from \"core/Meshes/mesh\";\r\nimport type { Scene } from \"core/scene\";\r\nimport type { Nullable } from \"core/types\";\r\n\r\n/**\r\n * Class used to perform a picking operation using GPU\r\n * Please note that GPUPIcker cannot pick instances, only meshes\r\n */\r\nexport class GPUPicker {\r\n    private _pickingTexure: Nullable<RenderTargetTexture> = null;\r\n    private _idMap: { [key: string]: number } = {};\r\n    private _idColors: Array<Color3> = [];\r\n    private _cachedMaterials: Array<Nullable<Material>> = [];\r\n    private _cachedScene: Nullable<Scene>;\r\n    private _renderMaterial: Nullable<ShaderMaterial>;\r\n    private _pickableMeshes: Array<AbstractMesh>;\r\n    private _readbuffer: Uint8Array;\r\n    private _meshRenderingCount: number = 0;\r\n    private _userDefinedList = false;\r\n\r\n    private _createRenderTarget(scene: Scene, width: number, height: number) {\r\n        this._pickingTexure = new RenderTargetTexture(\r\n            \"pickingTexure\",\r\n            { width: width, height: height },\r\n            scene,\r\n            false,\r\n            undefined,\r\n            Constants.TEXTURETYPE_UNSIGNED_INT,\r\n            false,\r\n            Constants.TEXTURE_NEAREST_NEAREST\r\n        );\r\n    }\r\n\r\n    private _createColorMaterial(scene: Scene) {\r\n        if (this._renderMaterial) {\r\n            this._renderMaterial.dispose();\r\n        }\r\n\r\n        const defines: string[] = [];\r\n        const options = {\r\n            attributes: [VertexBuffer.PositionKind],\r\n            uniforms: [\"world\", \"viewProjection\", \"color\"],\r\n            needAlphaBlending: false,\r\n            defines: defines,\r\n            useClipPlane: null,\r\n        };\r\n\r\n        this._renderMaterial = new ShaderMaterial(\"colorShader\", scene, \"color\", options, false);\r\n\r\n        const callback = (mesh: AbstractMesh | undefined) => {\r\n            if (!mesh) {\r\n                return;\r\n            }\r\n\r\n            const effect = this._renderMaterial!.getEffect();\r\n\r\n            if (!mesh.hasInstances && !mesh.isAnInstance) {\r\n                effect.setColor4(\"color\", this._idColors[mesh.uniqueId], 1);\r\n            }\r\n\r\n            this._meshRenderingCount++;\r\n        };\r\n\r\n        this._renderMaterial.customBindingObservable.add(callback);\r\n    }\r\n\r\n    /**\r\n     * Set the list of meshes to pick from\r\n     * @param list defines the list of meshes to pick from\r\n     */\r\n    public setPickingList(list: Nullable<Array<AbstractMesh>>) {\r\n        if (!list) {\r\n            if (this._userDefinedList) {\r\n                for (let index = 0; index < this._pickableMeshes.length; index++) {\r\n                    const mesh = this._pickableMeshes[index];\r\n                    if (mesh.hasInstances) {\r\n                        (mesh as Mesh).removeVerticesData(VertexBuffer.ColorKind);\r\n                    }\r\n                }\r\n            }\r\n            this._userDefinedList = false;\r\n            this._pickableMeshes = [];\r\n            this._idMap = {};\r\n            this._idColors = [];\r\n            return;\r\n        }\r\n        this._userDefinedList = true;\r\n        this._pickableMeshes = list;\r\n\r\n        // We will affect colors and create vertex color buffers\r\n        let id = 1;\r\n        for (let index = 0; index < this._pickableMeshes.length; index++) {\r\n            const mesh = this._pickableMeshes[index];\r\n            mesh.useVertexColors = false;\r\n\r\n            if (mesh.isAnInstance) {\r\n                continue; // This will be handled by the source mesh\r\n            }\r\n\r\n            const r = (id & 0xff0000) >> 16;\r\n            const g = (id & 0x00ff00) >> 8;\r\n            const b = (id & 0x0000ff) >> 0;\r\n            this._idMap[`${r}_${g}_${b}`] = index;\r\n            id++;\r\n\r\n            if (mesh.hasInstances) {\r\n                const instances = (mesh as Mesh).instances;\r\n                const colorData = new Float32Array(4 * (instances.length + 1));\r\n                const engine = mesh.getEngine();\r\n\r\n                colorData[0] = r / 255.0;\r\n                colorData[1] = g / 255.0;\r\n                colorData[2] = b / 255.0;\r\n                colorData[3] = 1.0;\r\n                for (let i = 0; i < instances.length; i++) {\r\n                    const instance = instances[i];\r\n                    const r = (id & 0xff0000) >> 16;\r\n                    const g = (id & 0x00ff00) >> 8;\r\n                    const b = (id & 0x0000ff) >> 0;\r\n                    this._idMap[`${r}_${g}_${b}`] = this._pickableMeshes.indexOf(instance);\r\n\r\n                    colorData[(i + 1) * 4] = r / 255.0;\r\n                    colorData[(i + 1) * 4 + 1] = g / 255.0;\r\n                    colorData[(i + 1) * 4 + 2] = b / 255.0;\r\n                    colorData[(i + 1) * 4 + 3] = 1.0;\r\n                    id++;\r\n                }\r\n\r\n                const buffer = new VertexBuffer(engine, colorData, VertexBuffer.ColorKind, false, false, 4, true);\r\n                (mesh as Mesh).setVerticesBuffer(buffer, true);\r\n            } else {\r\n                this._idColors[mesh.uniqueId] = Color3.FromInts(r, g, b);\r\n            }\r\n        }\r\n    }\r\n\r\n    /**\r\n     * Execute a picking operation\r\n     * @param x defines the X coordinates where to run the pick\r\n     * @param y defines the Y coordinates where to run the pick\r\n     * @param scene defines the scene to pick from\r\n     * @param disposeWhenDone defines a boolean indicating we do not want to keep resources alive\r\n     * @returns A promise with the picking results\r\n     */\r\n    public pickAsync(x: number, y: number, scene: Scene, disposeWhenDone = true): Promise<Nullable<AbstractMesh>> {\r\n        const engine = scene.getEngine();\r\n        const rttSizeW = engine.getRenderWidth();\r\n        const rttSizeH = engine.getRenderHeight();\r\n\r\n        if (!this._readbuffer) {\r\n            this._readbuffer = new Uint8Array(engine.isWebGPU ? 256 : 4); // Because of block alignment in WebGPU\r\n        }\r\n\r\n        this._meshRenderingCount = 0;\r\n        // Ensure ints\r\n        x = x >> 0;\r\n        y = y >> 0;\r\n\r\n        if (x < 0 || y < 0 || x >= rttSizeW || y >= rttSizeH) {\r\n            return Promise.resolve(null);\r\n        }\r\n\r\n        // Invert Y\r\n        y = rttSizeH - y;\r\n\r\n        if (!this._pickingTexure) {\r\n            this._createRenderTarget(scene, rttSizeW, rttSizeH);\r\n        } else {\r\n            const size = this._pickingTexure.getSize();\r\n\r\n            if (size.width !== rttSizeW || size.height !== rttSizeH || this._cachedScene !== scene) {\r\n                this._pickingTexure.dispose();\r\n                this._createRenderTarget(scene, rttSizeW, rttSizeH);\r\n            }\r\n        }\r\n\r\n        if (!this._cachedScene || this._cachedScene !== scene) {\r\n            this._createColorMaterial(scene);\r\n        }\r\n\r\n        this._cachedScene = scene;\r\n        scene.customRenderTargets.push(this._pickingTexure!);\r\n        this._pickingTexure!.renderList = [];\r\n        this._pickingTexure!.clearColor = new Color4(0, 0, 0, 0);\r\n\r\n        // We need to give every mesh an unique color (when there is no picking list)\r\n        this._pickingTexure!.onBeforeRender = () => {\r\n            if (!this._userDefinedList) {\r\n                this._pickableMeshes = scene.meshes.filter((m) => (m as Mesh).geometry && m.isPickable && (m as Mesh).onBeforeRenderObservable) as Mesh[];\r\n            }\r\n\r\n            for (let index = 0; index < this._pickableMeshes.length; index++) {\r\n                const mesh = this._pickableMeshes[index];\r\n                if (!mesh.isAnInstance) {\r\n                    this._cachedMaterials[index] = mesh.material;\r\n                    if (!this._userDefinedList) {\r\n                        // We need to define the color for each mesh as it was not previously defined\r\n                        const id = index + 1;\r\n                        const r = (id & 0xff0000) >> 16;\r\n                        const g = (id & 0x00ff00) >> 8;\r\n                        const b = (id & 0x0000ff) >> 0;\r\n                        this._idMap[`${r}_${g}_${b}`] = index;\r\n\r\n                        this._idColors[mesh.uniqueId] = Color3.FromInts(r, g, b);\r\n                    } else {\r\n                        mesh.useVertexColors = true; // In that case we will be using vertex colors and not an uniform to support instances\r\n                    }\r\n                    mesh.material = this._renderMaterial;\r\n                }\r\n                this._pickingTexure?.renderList?.push(mesh);\r\n            }\r\n\r\n            // Enable scissor\r\n            if ((engine as WebGPUEngine | Engine).enableScissor) {\r\n                (engine as WebGPUEngine | Engine).enableScissor(x, y, 1, 1);\r\n            }\r\n        };\r\n\r\n        return new Promise((resolve, reject) => {\r\n            this._pickingTexure!.onAfterRender = async () => {\r\n                // Disable scissor\r\n                if ((engine as WebGPUEngine | Engine).disableScissor) {\r\n                    (engine as WebGPUEngine | Engine).disableScissor();\r\n                }\r\n\r\n                if (!this._pickingTexure) {\r\n                    reject();\r\n                }\r\n\r\n                let pickedMesh: Nullable<AbstractMesh> = null;\r\n                const wasSuccessfull = this._meshRenderingCount > 0;\r\n\r\n                // Restore materials\r\n                for (let index = 0; index < this._pickableMeshes.length; index++) {\r\n                    const mesh = this._pickableMeshes[index];\r\n                    if (!mesh.isAnInstance) {\r\n                        mesh.material = this._cachedMaterials[index];\r\n\r\n                        if (this._userDefinedList) {\r\n                            mesh.useVertexColors = false;\r\n                        }\r\n                    }\r\n                }\r\n\r\n                if (wasSuccessfull) {\r\n                    // Remove from the active RTTs\r\n                    const index = scene.customRenderTargets.indexOf(this._pickingTexure!);\r\n                    if (index > -1) {\r\n                        scene.customRenderTargets.splice(index, 1);\r\n                    }\r\n\r\n                    // Do the actual picking\r\n                    if (await this._readTexturePixelsAsync(x, y)) {\r\n                        const colorId = `${this._readbuffer[0]}_${this._readbuffer[1]}_${this._readbuffer[2]}`;\r\n                        pickedMesh = this._pickableMeshes[this._idMap[colorId]];\r\n                    }\r\n                }\r\n\r\n                // Clean-up\r\n                if (!this._userDefinedList) {\r\n                    this._idMap = {};\r\n                    this._idColors = [];\r\n\r\n                    this._pickableMeshes = [];\r\n                }\r\n                this._pickingTexure!.renderList = [];\r\n\r\n                if (!wasSuccessfull) {\r\n                    this._meshRenderingCount = 0;\r\n                    return; // We need to wait for the shaders to be ready\r\n                } else {\r\n                    if (disposeWhenDone) {\r\n                        this.dispose();\r\n                    }\r\n                    resolve(pickedMesh);\r\n                }\r\n            };\r\n        });\r\n    }\r\n\r\n    private async _readTexturePixelsAsync(x: number, y: number) {\r\n        if (!this._cachedScene || !this._pickingTexure?._texture) {\r\n            return false;\r\n        }\r\n        const engine = this._cachedScene.getEngine();\r\n        await engine._readTexturePixels(this._pickingTexure._texture, 1, 1, -1, 0, this._readbuffer, true, true, x, y);\r\n\r\n        return true;\r\n    }\r\n\r\n    /** Release the resources */\r\n    public dispose() {\r\n        if (this._userDefinedList) {\r\n            this.setPickingList(null);\r\n        }\r\n        this._pickableMeshes = [];\r\n        this._cachedScene = null;\r\n\r\n        // Cleaning up\r\n        this._pickingTexure?.dispose();\r\n        this._pickingTexure = null;\r\n        this._renderMaterial?.dispose();\r\n        this._renderMaterial = null;\r\n    }\r\n}\r\n"]}

package/Collisions/index.d.ts

@@ -3,3 +3,4 @@ export * from "./collisionCoordinator";
 export * from "./pickingInfo";
 export * from "./intersectionInfo";
 export * from "./meshCollisionData";
+export * from "./gpuPicker";

package/Collisions/index.js

@@ -3,4 +3,5 @@ export * from "./collisionCoordinator.js";
 export * from "./pickingInfo.js";
 export * from "./intersectionInfo.js";
 export * from "./meshCollisionData.js";
+export * from "./gpuPicker.js";
 //# sourceMappingURL=index.js.map

package/Collisions/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../dev/core/src/Collisions/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAC3B,cAAc,wBAAwB,CAAC;AACvC,cAAc,eAAe,CAAC;AAC9B,cAAc,oBAAoB,CAAC;AACnC,cAAc,qBAAqB,CAAC","sourcesContent":["export * from \"./collider\";\r\nexport * from \"./collisionCoordinator\";\r\nexport * from \"./pickingInfo\";\r\nexport * from \"./intersectionInfo\";\r\nexport * from \"./meshCollisionData\";\r\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../dev/core/src/Collisions/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAC3B,cAAc,wBAAwB,CAAC;AACvC,cAAc,eAAe,CAAC;AAC9B,cAAc,oBAAoB,CAAC;AACnC,cAAc,qBAAqB,CAAC;AACpC,cAAc,aAAa,CAAC","sourcesContent":["export * from \"./collider\";\r\nexport * from \"./collisionCoordinator\";\r\nexport * from \"./pickingInfo\";\r\nexport * from \"./intersectionInfo\";\r\nexport * from \"./meshCollisionData\";\r\nexport * from \"./gpuPicker\";\r\n"]}

package/Debug/physicsViewer.d.ts

@@ -133,18 +133,20 @@ export declare class PhysicsViewer {
     /**
      * Hides a body from the physics engine.
      * @param body - The body to hide.
+     * @returns true if body actually removed
      *
      * This function is useful for hiding a body from the physics engine.
      * It removes the body from the utility layer scene and disposes the mesh associated with it.
      * It also unregisters the render function if the number of bodies is 0.
      * This is useful for hiding a body from the physics engine without deleting it.
      */
-    hideBody(body: Nullable<PhysicsBody>): void;
+    hideBody(body: Nullable<PhysicsBody>): boolean;
     /**
      * Hides a body's inertia from the viewer utility layer
      * @param body the body to hide
+     * @returns true if inertia actually removed
      */
-    hideInertia(body: Nullable<PhysicsBody>): void;
+    hideInertia(body: Nullable<PhysicsBody>): boolean;
     /**
      * Hide a physics constraint from the viewer utility layer
      * @param constraint the constraint to hide

package/Debug/physicsViewer.js

@@ -115,21 +115,29 @@ export class PhysicsViewer {
      */
     _updateDebugMeshesV2() {
         const plugin = this._physicsEnginePlugin;
-        for (let i = 0; i < this._numBodies; i++) {
+        for (let i = 0; i < this._numBodies;) {
             const body = this._bodies[i];
+            if (body && body.isDisposed && this.hideBody(body)) {
+                continue;
+            }
             const transform = this._bodyMeshes[i];
             if (body && transform) {
                 plugin.syncTransform(body, transform);
             }
+            i++;
         }
     }
     _updateInertiaMeshes() {
-        for (let i = 0; i < this._numInertiaBodies; i++) {
+        for (let i = 0; i < this._numInertiaBodies;) {
             const body = this._inertiaBodies[i];
+            if (body && body.isDisposed && this.hideInertia(body)) {
+                continue;
+            }
             const mesh = this._inertiaMeshes[i];
             if (body && mesh) {
                 this._updateDebugInertia(body, mesh);
             }
+            i++;
         }
     }
     _updateDebugInertia(body, inertiaMesh) {
@@ -376,6 +384,7 @@ export class PhysicsViewer {
     /**
      * Hides a body from the physics engine.
      * @param body - The body to hide.
+     * @returns true if body actually removed
      *
      * This function is useful for hiding a body from the physics engine.
      * It removes the body from the utility layer scene and disposes the mesh associated with it.
@@ -384,7 +393,7 @@ export class PhysicsViewer {
      */
     hideBody(body) {
         if (!body || !this._scene || !this._utilityLayer) {
-            return;
+            return false;
         }
         let removed = false;
         const utilityLayerScene = this._utilityLayer.utilityLayerScene;
@@ -414,14 +423,16 @@ export class PhysicsViewer {
         if (removed && this._numBodies === 0) {
             this._scene.unregisterBeforeRender(this._renderFunction);
         }
+        return removed;
     }
     /**
      * Hides a body's inertia from the viewer utility layer
      * @param body the body to hide
+     * @returns true if inertia actually removed
      */
     hideInertia(body) {
         if (!body || !this._scene || !this._utilityLayer) {
-            return;
+            return false;
         }
         let removed = false;
         const utilityLayerScene = this._utilityLayer.utilityLayerScene;
@@ -443,6 +454,7 @@ export class PhysicsViewer {
         if (removed && this._numInertiaBodies === 0) {
             this._scene.unregisterBeforeRender(this._inertiaRenderFunction);
         }
+        return removed;
     }
     /**
      * Hide a physics constraint from the viewer utility layer