@onerjs/core 8.52.8 → 8.53.1

package/Engines/WebGPU/Extensions/engine.computeShader.pure.js

@@ -1,10 +1,158 @@
 /** This file must only contain pure code and pure imports */
 import { Logger } from "../../../Misc/logger.js";
 import { ComputeEffect } from "../../../Compute/computeEffect.js";
+
 import { WebGPUComputeContext } from "../webgpuComputeContext.js";
 import { WebGPUComputePipelineContext } from "../webgpuComputePipelineContext.js";
 import { WebGPUEngine } from "../../webgpuEngine.pure.js";
 let _Registered = false;
+const _GetComputeStorageBufferType = (source, group, binding) => {
+    const bindingPattern = `@binding\\(${binding}\\)\\s*@group\\(${group}\\)`;
+    const groupPattern = `@group\\(${group}\\)\\s*@binding\\(${binding}\\)`;
+    const declarationPattern = `(?:${bindingPattern}|${groupPattern})\\s*var<storage\\s*,\\s*(read|read_write)>`;
+    const access = source.match(new RegExp(declarationPattern))?.[1];
+    return access === "read" ? "read-only-storage" /* WebGPUConstants.BufferBindingType.ReadOnlyStorage */ : "storage" /* WebGPUConstants.BufferBindingType.Storage */;
+};
+const _GetComputeTextureViewDimension = (source, group, binding) => {
+    const bindingPattern = `@binding\\(${binding}\\)\\s*@group\\(${group}\\)`;
+    const groupPattern = `@group\\(${group}\\)\\s*@binding\\(${binding}\\)`;
+    const declaration = source.match(new RegExp(`(?:${bindingPattern}|${groupPattern})\\s*var\\s+\\w+\\s*:\\s*(texture_\\w+)`))?.[1];
+    switch (declaration) {
+        case "texture_2d_array":
+        case "texture_depth_2d_array":
+            return "2d-array" /* WebGPUConstants.TextureViewDimension.E2dArray */;
+        case "texture_cube":
+            return "cube" /* WebGPUConstants.TextureViewDimension.Cube */;
+        case "texture_cube_array":
+            return "cube-array" /* WebGPUConstants.TextureViewDimension.CubeArray */;
+        case "texture_3d":
+            return "3d" /* WebGPUConstants.TextureViewDimension.E3d */;
+        default:
+            return "2d" /* WebGPUConstants.TextureViewDimension.E2d */;
+    }
+};
+const _GetInternalTexture = (texture) => {
+    const internalTexture = texture;
+    if (internalTexture._hardwareTexture !== undefined) {
+        return internalTexture;
+    }
+    return texture._texture;
+};
+const _GetComputeTextureSampleType = (texture, textureFloatLinearFiltering) => {
+    const internalTexture = _GetInternalTexture(texture);
+    if (!internalTexture) {
+        return "float" /* WebGPUConstants.TextureSampleType.Float */;
+    }
+    const textureIsDepth = internalTexture.format >= 13 && internalTexture.format <= 18;
+    if (textureIsDepth) {
+        return "depth" /* WebGPUConstants.TextureSampleType.Depth */;
+    }
+    if (internalTexture.type === 1 && !textureFloatLinearFiltering) {
+        return "unfilterable-float" /* WebGPUConstants.TextureSampleType.UnfilterableFloat */;
+    }
+    return "float" /* WebGPUConstants.TextureSampleType.Float */;
+};
+const _GetComputePipelineLayout = (device, bindings, bindingsMapping, shaderSource, textureFloatLinearFiltering) => {
+    if (!bindingsMapping) {
+        return null;
+    }
+    const bindGroupLayoutEntries = [];
+    const addEntry = (group, entry) => {
+        const entries = (bindGroupLayoutEntries[group] ?? (bindGroupLayoutEntries[group] = []));
+        const existingIndex = entries.findIndex((item) => item.binding === entry.binding);
+        if (existingIndex >= 0) {
+            entries[existingIndex] = entry;
+        }
+        else {
+            entries.push(entry);
+        }
+    };
+    for (const key in bindings) {
+        const binding = bindings[key];
+        const location = bindingsMapping[key];
+        if (!location) {
+            continue;
+        }
+        const group = location.group;
+        const bindingIndex = location.binding;
+        switch (binding.type) {
+            case 0 /* ComputeBindingType.Texture */:
+            case 4 /* ComputeBindingType.TextureWithoutSampler */:
+            case 8 /* ComputeBindingType.InternalTexture */: {
+                const sampleType = _GetComputeTextureSampleType(binding.object, textureFloatLinearFiltering);
+                if (binding.type === 0 /* ComputeBindingType.Texture */) {
+                    addEntry(group, {
+                        binding: bindingIndex - 1,
+                        visibility: 4 /* WebGPUConstants.ShaderStage.Compute */,
+                        sampler: {
+                            type: sampleType === "unfilterable-float" /* WebGPUConstants.TextureSampleType.UnfilterableFloat */
+                                ? "non-filtering" /* WebGPUConstants.SamplerBindingType.NonFiltering */
+                                : "filtering" /* WebGPUConstants.SamplerBindingType.Filtering */,
+                        },
+                    });
+                }
+                addEntry(group, {
+                    binding: bindingIndex,
+                    visibility: 4 /* WebGPUConstants.ShaderStage.Compute */,
+                    texture: {
+                        sampleType,
+                        viewDimension: _GetComputeTextureViewDimension(shaderSource, group, bindingIndex),
+                        multisampled: false,
+                    },
+                });
+                break;
+            }
+            case 1 /* ComputeBindingType.StorageTexture */: {
+                addEntry(group, {
+                    binding: bindingIndex,
+                    visibility: 4 /* WebGPUConstants.ShaderStage.Compute */,
+                    storageTexture: {
+                        access: "write-only" /* WebGPUConstants.StorageTextureAccess.WriteOnly */,
+                        format: binding.object._texture._hardwareTexture.format,
+                        viewDimension: _GetComputeTextureViewDimension(shaderSource, group, bindingIndex),
+                    },
+                });
+                break;
+            }
+            case 2 /* ComputeBindingType.UniformBuffer */:
+                addEntry(group, {
+                    binding: bindingIndex,
+                    visibility: 4 /* WebGPUConstants.ShaderStage.Compute */,
+                    buffer: { type: "uniform" /* WebGPUConstants.BufferBindingType.Uniform */ },
+                });
+                break;
+            case 3 /* ComputeBindingType.StorageBuffer */:
+            case 7 /* ComputeBindingType.DataBuffer */:
+                addEntry(group, {
+                    binding: bindingIndex,
+                    visibility: 4 /* WebGPUConstants.ShaderStage.Compute */,
+                    buffer: { type: _GetComputeStorageBufferType(shaderSource, group, bindingIndex) },
+                });
+                break;
+            case 5 /* ComputeBindingType.Sampler */:
+                addEntry(group, {
+                    binding: bindingIndex,
+                    visibility: 4 /* WebGPUConstants.ShaderStage.Compute */,
+                    sampler: { type: "filtering" /* WebGPUConstants.SamplerBindingType.Filtering */ },
+                });
+                break;
+            case 6 /* ComputeBindingType.ExternalTexture */:
+                addEntry(group, {
+                    binding: bindingIndex,
+                    visibility: 4 /* WebGPUConstants.ShaderStage.Compute */,
+                    externalTexture: {},
+                });
+                break;
+        }
+    }
+    const bindGroupLayouts = [];
+    for (let i = 0; i < bindGroupLayoutEntries.length; i++) {
+        const entries = bindGroupLayoutEntries[i] ?? [];
+        entries.sort((a, b) => a.binding - b.binding);
+        bindGroupLayouts[i] = device.createBindGroupLayout({ entries });
+    }
+    return device.createPipelineLayout({ bindGroupLayouts });
+};
 /**
  * Register side effects for enginesWebGPUExtensionsEngineComputeShader.
  * Safe to call multiple times; only the first call has an effect.
@@ -21,7 +169,7 @@ export function RegisterEnginesWebGPUExtensionsEngineComputeShader() {
     };
     WebGPUEngine.prototype.createComputeEffect = function (baseName, options) {
         const compute = typeof baseName === "string" ? baseName : baseName.computeToken || baseName.computeSource || baseName.computeElement || baseName.compute;
-        const name = compute + "@" + options.defines;
+        const name = compute + "@" + options.defines + (options.useExplicitComputePipelineLayout ? "@explicitComputePipelineLayout" : "");
         if (this._compiledComputeEffects[name]) {
             const compiledEffect = this._compiledComputeEffects[name];
             if (options.onCompiled && compiledEffect.isReady()) {
@@ -56,8 +204,12 @@ export function RegisterEnginesWebGPUExtensionsEngineComputeShader() {
         const contextPipeline = effect._pipelineContext;
         const computeContext = context;
         if (!contextPipeline.computePipeline) {
+            const explicitLayout = effect._useExplicitComputePipelineLayout
+                ? _GetComputePipelineLayout(this._device, bindings, bindingsMapping, contextPipeline.sources?.compute ?? "", this._caps.textureFloatLinearFiltering)
+                : null;
+            computeContext.clear();
             contextPipeline.computePipeline = this._device.createComputePipeline({
-                layout: "auto" /* WebGPUConstants.AutoLayoutMode.Auto */,
+                layout: explicitLayout ?? "auto" /* WebGPUConstants.AutoLayoutMode.Auto */,
                 compute: contextPipeline.stage,
             });
         }

package/Engines/WebGPU/Extensions/engine.computeShader.pure.js.map

@@ -1 +1 @@
-{"version":3,"file":"engine.computeShader.pure.js","sourceRoot":"","sources":["../../../../../../dev/core/src/Engines/WebGPU/Extensions/engine.computeShader.pure.ts"],"names":[],"mappings":"AAAA,6DAA6D;AAG7D,OAAO,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AAC1C,OAAO,EAA+D,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAK5H,OAAO,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,EAAE,4BAA4B,EAAE,MAAM,iCAAiC,CAAC;AAG/E,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAEvD,IAAI,WAAW,GAAG,KAAK,CAAC;AACxB;;;GAGG;AACH,MAAM,UAAU,kDAAkD;IAC9D,IAAI,WAAW,EAAE,CAAC;QACd,OAAO;IACX,CAAC;IACD,WAAW,GAAG,IAAI,CAAC;IAEnB,gEAAgE;IAChE,MAAM,qBAAqB,GAA6B,EAAE,CAAC;IAE3D,YAAY,CAAC,SAAS,CAAC,oBAAoB,GAAG;QAC1C,OAAO,IAAI,oBAAoB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;IACtE,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,mBAAmB,GAAG,UACzC,QAAmE,EACnE,OAAsC;QAEtC,MAAM,OAAO,GAAG,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,YAAY,IAAI,QAAQ,CAAC,aAAa,IAAI,QAAQ,CAAC,cAAc,IAAI,QAAQ,CAAC,OAAO,CAAC;QAEzJ,MAAM,IAAI,GAAG,OAAO,GAAG,GAAG,GAAG,OAAO,CAAC,OAAO,CAAC;QAC7C,IAAI,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,EAAE,CAAC;YACrC,MAAM,cAAc,GAAG,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC;YAC1D,IAAI,OAAO,CAAC,UAAU,IAAI,cAAc,CAAC,OAAO,EAAE,EAAE,CAAC;gBACjD,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;YACvC,CAAC;YAED,OAAO,cAAc,CAAC;QAC1B,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QAChE,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC;QAE5C,OAAO,MAAM,CAAC;IAClB,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,4BAA4B,GAAG;QAClD,OAAO,IAAI,4BAA4B,CAAC,IAAI,CAAC,CAAC;IAClD,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,yBAAyB,GAAG;QAC/C,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,CAAC;YAEjD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC;gBACpB,OAAO,KAAK,CAAC;YACjB,CAAC;QACL,CAAC;QAED,OAAO,IAAI,CAAC;IAChB,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,eAAe,GAAG,UACrC,MAAqB,EACrB,OAAwB,EACxB,QAA4B,EAC5B,CAAS,EACT,CAAC,GAAG,CAAC,EACL,CAAC,GAAG,CAAC,EACL,eAAuC,EACvC,cAAkC;QAElC,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,eAAe,EAAE,cAAc,CAAC,CAAC;IACrH,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,uBAAuB,GAAG,UAC7C,MAAqB,EACrB,OAAwB,EACxB,QAA4B,EAC5B,MAAkB,EAClB,SAAiB,CAAC,EAClB,eAAuC,EACvC,cAAkC;QAElC,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,eAAe,EAAE,cAAc,CAAC,CAAC;IACvI,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,gBAAgB,GAAG,UACtC,MAAqB,EACrB,OAAwB,EACxB,QAA4B,EAC5B,CAAU,EACV,CAAU,EACV,CAAU,EACV,MAAmB,EACnB,MAAe,EACf,eAAuC,EACvC,cAAkC;QAElC,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAE7B,MAAM,eAAe,GAAG,MAAM,CAAC,gBAAgD,CAAC;QAChF,MAAM,cAAc,GAAG,OAA+B,CAAC;QAEvD,IAAI,CAAC,eAAe,CAAC,eAAe,EAAE,CAAC;YACnC,eAAe,CAAC,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC;gBACjE,MAAM,kDAAqC;gBAC3C,OAAO,EAAE,eAAe,CAAC,KAAM;aAClC,CAAC,CAAC;QACP,CAAC;QAED,IAAI,cAAc,EAAE,CAAC;YACjB,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,qBAAqB,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;QAChF,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAAC,qBAAqB,CAAC,CAAC;QAEhF,WAAW,CAAC,WAAW,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;QAEzD,MAAM,UAAU,GAAG,cAAc,CAAC,aAAa,CAAC,QAAQ,EAAE,eAAe,CAAC,eAAe,EAAE,eAAe,CAAC,CAAC;QAC5G,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;YACzC,MAAM,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;YAChC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACb,SAAS;YACb,CAAC;YACD,WAAW,CAAC,YAAY,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;QAC3C,CAAC;QAED,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACvB,WAAW,CAAC,0BAA0B,CAAC,MAAM,CAAC,kBAAkB,EAAU,MAAM,CAAC,CAAC;QACtF,CAAC;aAAM,CAAC;YACJ,IAAY,CAAC,GAAW,CAAC,GAAW,CAAC,GAAG,CAAC,EAAE,CAAC;gBACxC,WAAW,CAAC,kBAAkB,CAAS,CAAC,EAAU,CAAC,EAAU,CAAC,CAAC,CAAC;YACpE,CAAC;QACL,CAAC;QACD,WAAW,CAAC,GAAG,EAAE,CAAC;QAElB,IAAI,cAAc,EAAE,CAAC;YACjB,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,EAAE,cAAc,CAAC,CAAC;YACnE,IAAI,CAAC,eAAe,IAAI,CAAC,CAAC;QAC9B,CAAC;IACL,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,qBAAqB,GAAG;QAC3C,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAC9C,MAAM,4BAA4B,GAAG,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC,kBAAkB,EAAkC,CAAC;YAC7H,IAAI,CAAC,6BAA6B,CAAC,4BAA4B,CAAC,CAAC;QACrE,CAAC;QAED,IAAI,CAAC,uBAAuB,GAAG,EAAE,CAAC;IACtC,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,8BAA8B,GAAG,UACpD,eAAwC,EACxC,iBAAyB,EACzB,oBAA4B,EAC5B,OAAyB,EACzB,UAAkB;QAElB,MAAM,aAAa,GAAG,eAA+C,CAAC;QAEtE,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACzB,MAAM,CAAC,GAAG,CAAC,OAAQ,CAAC,CAAC;YACrB,MAAM,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;QAClC,CAAC;QAED,aAAa,CAAC,OAAO,GAAG;YACpB,OAAO,EAAE,iBAAiB;YAC1B,UAAU,EAAE,oBAAoB;SACnC,CAAC;QAEF,aAAa,CAAC,KAAK,GAAG,IAAI,CAAC,qCAAqC,CAAC,iBAAiB,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;IAC7G,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,qBAAqB,GAAG,UAAU,MAAqB;QAC1E,IAAI,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5C,OAAO,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAEjD,IAAI,CAAC,6BAA6B,CAAC,MAAM,CAAC,kBAAkB,EAAkC,CAAC,CAAC;QACpG,CAAC;IACL,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,sBAAsB,GAAG;QAC5C,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,CAAC;YAEjD,MAAM,CAAC,gBAAgB,GAAG,IAAI,CAAC;YAC/B,MAAM,CAAC,mBAAmB,GAAG,KAAK,CAAC;YACnC,MAAM,CAAC,cAAc,EAAE,CAAC;QAC5B,CAAC;IACL,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,kCAAkC,GAAG,UACxD,eAA6C,EAC7C,MAAgE;QAEhE,mFAAmF;QACnF,eAAe,CAAC,KAAM,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE;YAC7D,MAAM,mBAAmB,GAA+B;gBACpD,SAAS,EAAE,CAAC;gBACZ,QAAQ,EAAE,EAAE;aACf,CAAC;YACF,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClC,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;oBAC3B,mBAAmB,CAAC,SAAS,EAAE,CAAC;gBACpC,CAAC;gBACD,mBAAmB,CAAC,QAAQ,CAAC,IAAI,CAAC;oBAC9B,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,IAAI,EAAE,OAAO,CAAC,OAAO;oBACrB,IAAI,EAAE,OAAO,CAAC,OAAO;oBACrB,MAAM,EAAE,OAAO,CAAC,OAAO;oBACvB,MAAM,EAAE,OAAO,CAAC,MAAM;oBACtB,MAAM,EAAE,OAAO,CAAC,MAAM;iBACzB,CAAC,CAAC;YACP,CAAC;YACD,MAAM,CAAC,mBAAmB,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACP,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,6BAA6B,GAAG,UAAU,eAAwC;QACrG,MAAM,qBAAqB,GAAG,eAA+C,CAAC;QAC9E,IAAI,qBAAqB,EAAE,CAAC;YACxB,eAAe,CAAC,OAAO,EAAE,CAAC;QAC9B,CAAC;IACL,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,qCAAqC,GAAG,UAAU,aAAqB,EAAE,OAAyB,EAAE,UAAkB;QACzI,IAAI,OAAO,EAAE,CAAC;YACV,OAAO,GAAG,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC;QAC7D,CAAC;aAAM,CAAC;YACJ,OAAO,GAAG,EAAE,CAAC;QACjB,CAAC;QACD,OAAO;YACH,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC;gBACpC,IAAI,EAAE,OAAO,GAAG,aAAa;aAChC,CAAC;YACF,UAAU;SACb,CAAC;IACN,CAAC,CAAC;AACN,CAAC","sourcesContent":["/** This file must only contain pure code and pure imports */\r\n\r\nimport * as WebGPUConstants from \"../webgpuConstants\";\r\nimport { Logger } from \"core/Misc/logger\";\r\nimport { type IComputeEffectCreationOptions, type IComputeShaderPath, ComputeEffect } from \"../../../Compute/computeEffect\";\r\nimport { type IComputeContext } from \"../../../Compute/IComputeContext\";\r\nimport { type IComputePipelineContext } from \"../../../Compute/IComputePipelineContext\";\r\nimport { type Nullable } from \"../../../types\";\r\nimport { type ComputeBindingList, type ComputeBindingMapping, type ComputeCompilationMessages } from \"../../Extensions/engine.computeShader.pure\";\r\nimport { WebGPUComputeContext } from \"../webgpuComputeContext\";\r\nimport { WebGPUComputePipelineContext } from \"../webgpuComputePipelineContext\";\r\nimport { type WebGPUPerfCounter } from \"../webgpuPerfCounter\";\r\nimport { type DataBuffer } from \"../../../Buffers/dataBuffer\";\r\nimport { WebGPUEngine } from \"../../webgpuEngine.pure\";\r\n\r\nlet _Registered = false;\r\n/**\r\n * Register side effects for enginesWebGPUExtensionsEngineComputeShader.\r\n * Safe to call multiple times; only the first call has an effect.\r\n */\r\nexport function RegisterEnginesWebGPUExtensionsEngineComputeShader(): void {\r\n    if (_Registered) {\r\n        return;\r\n    }\r\n    _Registered = true;\r\n\r\n    // eslint-disable-next-line @typescript-eslint/naming-convention\r\n    const ComputePassDescriptor: GPUComputePassDescriptor = {};\r\n\r\n    WebGPUEngine.prototype.createComputeContext = function (): IComputeContext | undefined {\r\n        return new WebGPUComputeContext(this._device, this._cacheSampler);\r\n    };\r\n\r\n    WebGPUEngine.prototype.createComputeEffect = function (\r\n        baseName: string | (IComputeShaderPath & { computeToken?: string }),\r\n        options: IComputeEffectCreationOptions\r\n    ): ComputeEffect {\r\n        const compute = typeof baseName === \"string\" ? baseName : baseName.computeToken || baseName.computeSource || baseName.computeElement || baseName.compute;\r\n\r\n        const name = compute + \"@\" + options.defines;\r\n        if (this._compiledComputeEffects[name]) {\r\n            const compiledEffect = this._compiledComputeEffects[name];\r\n            if (options.onCompiled && compiledEffect.isReady()) {\r\n                options.onCompiled(compiledEffect);\r\n            }\r\n\r\n            return compiledEffect;\r\n        }\r\n        const effect = new ComputeEffect(baseName, options, this, name);\r\n        this._compiledComputeEffects[name] = effect;\r\n\r\n        return effect;\r\n    };\r\n\r\n    WebGPUEngine.prototype.createComputePipelineContext = function (): IComputePipelineContext {\r\n        return new WebGPUComputePipelineContext(this);\r\n    };\r\n\r\n    WebGPUEngine.prototype.areAllComputeEffectsReady = function (): boolean {\r\n        for (const key in this._compiledComputeEffects) {\r\n            const effect = this._compiledComputeEffects[key];\r\n\r\n            if (!effect.isReady()) {\r\n                return false;\r\n            }\r\n        }\r\n\r\n        return true;\r\n    };\r\n\r\n    WebGPUEngine.prototype.computeDispatch = function (\r\n        effect: ComputeEffect,\r\n        context: IComputeContext,\r\n        bindings: ComputeBindingList,\r\n        x: number,\r\n        y = 1,\r\n        z = 1,\r\n        bindingsMapping?: ComputeBindingMapping,\r\n        gpuPerfCounter?: WebGPUPerfCounter\r\n    ): void {\r\n        this._computeDispatch(effect, context, bindings, x, y, z, undefined, undefined, bindingsMapping, gpuPerfCounter);\r\n    };\r\n\r\n    WebGPUEngine.prototype.computeDispatchIndirect = function (\r\n        effect: ComputeEffect,\r\n        context: IComputeContext,\r\n        bindings: ComputeBindingList,\r\n        buffer: DataBuffer,\r\n        offset: number = 0,\r\n        bindingsMapping?: ComputeBindingMapping,\r\n        gpuPerfCounter?: WebGPUPerfCounter\r\n    ): void {\r\n        this._computeDispatch(effect, context, bindings, undefined, undefined, undefined, buffer, offset, bindingsMapping, gpuPerfCounter);\r\n    };\r\n\r\n    WebGPUEngine.prototype._computeDispatch = function (\r\n        effect: ComputeEffect,\r\n        context: IComputeContext,\r\n        bindings: ComputeBindingList,\r\n        x?: number,\r\n        y?: number,\r\n        z?: number,\r\n        buffer?: DataBuffer,\r\n        offset?: number,\r\n        bindingsMapping?: ComputeBindingMapping,\r\n        gpuPerfCounter?: WebGPUPerfCounter\r\n    ): void {\r\n        this._endCurrentRenderPass();\r\n\r\n        const contextPipeline = effect._pipelineContext as WebGPUComputePipelineContext;\r\n        const computeContext = context as WebGPUComputeContext;\r\n\r\n        if (!contextPipeline.computePipeline) {\r\n            contextPipeline.computePipeline = this._device.createComputePipeline({\r\n                layout: WebGPUConstants.AutoLayoutMode.Auto,\r\n                compute: contextPipeline.stage!,\r\n            });\r\n        }\r\n\r\n        if (gpuPerfCounter) {\r\n            this._timestampQuery.startPass(ComputePassDescriptor, this._timestampIndex);\r\n        }\r\n\r\n        const computePass = this._renderEncoder.beginComputePass(ComputePassDescriptor);\r\n\r\n        computePass.setPipeline(contextPipeline.computePipeline);\r\n\r\n        const bindGroups = computeContext.getBindGroups(bindings, contextPipeline.computePipeline, bindingsMapping);\r\n        for (let i = 0; i < bindGroups.length; ++i) {\r\n            const bindGroup = bindGroups[i];\r\n            if (!bindGroup) {\r\n                continue;\r\n            }\r\n            computePass.setBindGroup(i, bindGroup);\r\n        }\r\n\r\n        if (buffer !== undefined) {\r\n            computePass.dispatchWorkgroupsIndirect(buffer.underlyingResource, <number>offset);\r\n        } else {\r\n            if (<number>x + <number>y + <number>z > 0) {\r\n                computePass.dispatchWorkgroups(<number>x, <number>y, <number>z);\r\n            }\r\n        }\r\n        computePass.end();\r\n\r\n        if (gpuPerfCounter) {\r\n            this._timestampQuery.endPass(this._timestampIndex, gpuPerfCounter);\r\n            this._timestampIndex += 2;\r\n        }\r\n    };\r\n\r\n    WebGPUEngine.prototype.releaseComputeEffects = function () {\r\n        for (const name in this._compiledComputeEffects) {\r\n            const webGPUPipelineContextCompute = this._compiledComputeEffects[name].getPipelineContext() as WebGPUComputePipelineContext;\r\n            this._deleteComputePipelineContext(webGPUPipelineContextCompute);\r\n        }\r\n\r\n        this._compiledComputeEffects = {};\r\n    };\r\n\r\n    WebGPUEngine.prototype._prepareComputePipelineContext = function (\r\n        pipelineContext: IComputePipelineContext,\r\n        computeSourceCode: string,\r\n        rawComputeSourceCode: string,\r\n        defines: Nullable<string>,\r\n        entryPoint: string\r\n    ): void {\r\n        const webGpuContext = pipelineContext as WebGPUComputePipelineContext;\r\n\r\n        if (this.dbgShowShaderCode) {\r\n            Logger.Log(defines!);\r\n            Logger.Log(computeSourceCode);\r\n        }\r\n\r\n        webGpuContext.sources = {\r\n            compute: computeSourceCode,\r\n            rawCompute: rawComputeSourceCode,\r\n        };\r\n\r\n        webGpuContext.stage = this._createComputePipelineStageDescriptor(computeSourceCode, defines, entryPoint);\r\n    };\r\n\r\n    WebGPUEngine.prototype._releaseComputeEffect = function (effect: ComputeEffect): void {\r\n        if (this._compiledComputeEffects[effect._key]) {\r\n            delete this._compiledComputeEffects[effect._key];\r\n\r\n            this._deleteComputePipelineContext(effect.getPipelineContext() as WebGPUComputePipelineContext);\r\n        }\r\n    };\r\n\r\n    WebGPUEngine.prototype._rebuildComputeEffects = function (): void {\r\n        for (const key in this._compiledComputeEffects) {\r\n            const effect = this._compiledComputeEffects[key];\r\n\r\n            effect._pipelineContext = null;\r\n            effect._wasPreviouslyReady = false;\r\n            effect._prepareEffect();\r\n        }\r\n    };\r\n\r\n    WebGPUEngine.prototype._executeWhenComputeStateIsCompiled = function (\r\n        pipelineContext: WebGPUComputePipelineContext,\r\n        action: (messages: Nullable<ComputeCompilationMessages>) => void\r\n    ): void {\r\n        // eslint-disable-next-line @typescript-eslint/no-floating-promises, github/no-then\r\n        pipelineContext.stage!.module.getCompilationInfo().then((info) => {\r\n            const compilationMessages: ComputeCompilationMessages = {\r\n                numErrors: 0,\r\n                messages: [],\r\n            };\r\n            for (const message of info.messages) {\r\n                if (message.type === \"error\") {\r\n                    compilationMessages.numErrors++;\r\n                }\r\n                compilationMessages.messages.push({\r\n                    type: message.type,\r\n                    text: message.message,\r\n                    line: message.lineNum,\r\n                    column: message.linePos,\r\n                    length: message.length,\r\n                    offset: message.offset,\r\n                });\r\n            }\r\n            action(compilationMessages);\r\n        });\r\n    };\r\n\r\n    WebGPUEngine.prototype._deleteComputePipelineContext = function (pipelineContext: IComputePipelineContext): void {\r\n        const webgpuPipelineContext = pipelineContext as WebGPUComputePipelineContext;\r\n        if (webgpuPipelineContext) {\r\n            pipelineContext.dispose();\r\n        }\r\n    };\r\n\r\n    WebGPUEngine.prototype._createComputePipelineStageDescriptor = function (computeShader: string, defines: Nullable<string>, entryPoint: string): GPUProgrammableStage {\r\n        if (defines) {\r\n            defines = \"//\" + defines.split(\"\\n\").join(\"\\n//\") + \"\\n\";\r\n        } else {\r\n            defines = \"\";\r\n        }\r\n        return {\r\n            module: this._device.createShaderModule({\r\n                code: defines + computeShader,\r\n            }),\r\n            entryPoint,\r\n        };\r\n    };\r\n}\r\n"]}
+{"version":3,"file":"engine.computeShader.pure.js","sourceRoot":"","sources":["../../../../../../dev/core/src/Engines/WebGPU/Extensions/engine.computeShader.pure.ts"],"names":[],"mappings":"AAAA,6DAA6D;AAG7D,OAAO,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AAC1C,OAAO,EAA+D,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAK5H,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,EAAE,4BAA4B,EAAE,MAAM,iCAAiC,CAAC;AAG/E,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAIvD,IAAI,WAAW,GAAG,KAAK,CAAC;AAExB,MAAM,4BAA4B,GAAG,CAAC,MAAc,EAAE,KAAa,EAAE,OAAe,EAAqC,EAAE;IACvH,MAAM,cAAc,GAAG,cAAc,OAAO,mBAAmB,KAAK,KAAK,CAAC;IAC1E,MAAM,YAAY,GAAG,YAAY,KAAK,qBAAqB,OAAO,KAAK,CAAC;IACxE,MAAM,kBAAkB,GAAG,MAAM,cAAc,IAAI,YAAY,6CAA6C,CAAC;IAC7G,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,kBAAkB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAEjE,OAAO,MAAM,KAAK,MAAM,CAAC,CAAC,6EAAmD,CAAC,0DAA0C,CAAC;AAC7H,CAAC,CAAC;AAEF,MAAM,+BAA+B,GAAG,CAAC,MAAc,EAAE,KAAa,EAAE,OAAe,EAAwC,EAAE;IAC7H,MAAM,cAAc,GAAG,cAAc,OAAO,mBAAmB,KAAK,KAAK,CAAC;IAC1E,MAAM,YAAY,GAAG,YAAY,KAAK,qBAAqB,OAAO,KAAK,CAAC;IACxE,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,MAAM,cAAc,IAAI,YAAY,yCAAyC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAEjI,QAAQ,WAAW,EAAE,CAAC;QAClB,KAAK,kBAAkB,CAAC;QACxB,KAAK,wBAAwB;YACzB,sEAAqD;QACzD,KAAK,cAAc;YACf,8DAAiD;QACrD,KAAK,oBAAoB;YACrB,yEAAsD;QAC1D,KAAK,YAAY;YACb,2DAAgD;QACpD;YACI,2DAAgD;IACxD,CAAC;AACL,CAAC,CAAC;AAEF,MAAM,mBAAmB,GAAG,CAAC,OAAsC,EAA6B,EAAE;IAC9F,MAAM,eAAe,GAAG,OAA0B,CAAC;IACnD,IAAI,eAAe,CAAC,gBAAgB,KAAK,SAAS,EAAE,CAAC;QACjD,OAAO,eAAe,CAAC;IAC3B,CAAC;IAED,OAAQ,OAAuB,CAAC,QAAQ,CAAC;AAC7C,CAAC,CAAC;AAEF,MAAM,4BAA4B,GAAG,CAAC,OAAsC,EAAE,2BAAoC,EAAqC,EAAE;IACrJ,MAAM,eAAe,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;IAErD,IAAI,CAAC,eAAe,EAAE,CAAC;QACnB,6DAA+C;IACnD,CAAC;IAED,MAAM,cAAc,GAAG,eAAe,CAAC,MAAM,IAAI,SAAS,CAAC,8BAA8B,IAAI,eAAe,CAAC,MAAM,IAAI,SAAS,CAAC,mCAAmC,CAAC;IACrK,IAAI,cAAc,EAAE,CAAC;QACjB,6DAA+C;IACnD,CAAC;IAED,IAAI,eAAe,CAAC,IAAI,KAAK,SAAS,CAAC,iBAAiB,IAAI,CAAC,2BAA2B,EAAE,CAAC;QACvF,sFAA2D;IAC/D,CAAC;IAED,6DAA+C;AACnD,CAAC,CAAC;AAEF,MAAM,yBAAyB,GAAG,CAC9B,MAAiB,EACjB,QAA4B,EAC5B,eAAkD,EAClD,YAAoB,EACpB,2BAAoC,EACT,EAAE;IAC7B,IAAI,CAAC,eAAe,EAAE,CAAC;QACnB,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,MAAM,sBAAsB,GAAgC,EAAE,CAAC;IAE/D,MAAM,QAAQ,GAAG,CAAC,KAAa,EAAE,KAA8B,EAAE,EAAE;QAC/D,MAAM,OAAO,GAAG,CAAC,sBAAsB,CAAC,KAAK,MAA5B,sBAAsB,CAAC,KAAK,IAAM,EAAE,EAAC,CAAC;QACvD,MAAM,aAAa,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,KAAK,KAAK,CAAC,OAAO,CAAC,CAAC;QAClF,IAAI,aAAa,IAAI,CAAC,EAAE,CAAC;YACrB,OAAO,CAAC,aAAa,CAAC,GAAG,KAAK,CAAC;QACnC,CAAC;aAAM,CAAC;YACJ,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxB,CAAC;IACL,CAAC,CAAC;IAEF,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QACzB,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;QAC9B,MAAM,QAAQ,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,SAAS;QACb,CAAC;QAED,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;QAC7B,MAAM,YAAY,GAAG,QAAQ,CAAC,OAAO,CAAC;QAEtC,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACnB,wCAAgC;YAChC,sDAA8C;YAC9C,+CAAuC,CAAC,CAAC,CAAC;gBACtC,MAAM,UAAU,GAAG,4BAA4B,CAAC,OAAO,CAAC,MAAuC,EAAE,2BAA2B,CAAC,CAAC;gBAE9H,IAAI,OAAO,CAAC,IAAI,uCAA+B,EAAE,CAAC;oBAC9C,QAAQ,CAAC,KAAK,EAAE;wBACZ,OAAO,EAAE,YAAY,GAAG,CAAC;wBACzB,UAAU,6CAAqC;wBAC/C,OAAO,EAAE;4BACL,IAAI,EACA,UAAU,mFAAwD;gCAC9D,CAAC;gCACD,CAAC,+DAA6C;yBACzD;qBACJ,CAAC,CAAC;gBACP,CAAC;gBAED,QAAQ,CAAC,KAAK,EAAE;oBACZ,OAAO,EAAE,YAAY;oBACrB,UAAU,6CAAqC;oBAC/C,OAAO,EAAE;wBACL,UAAU;wBACV,aAAa,EAAE,+BAA+B,CAAC,YAAY,EAAE,KAAK,EAAE,YAAY,CAAC;wBACjF,YAAY,EAAE,KAAK;qBACtB;iBACJ,CAAC,CAAC;gBACH,MAAM;YACV,CAAC;YACD,8CAAsC,CAAC,CAAC,CAAC;gBACrC,QAAQ,CAAC,KAAK,EAAE;oBACZ,OAAO,EAAE,YAAY;oBACrB,UAAU,6CAAqC;oBAC/C,cAAc,EAAE;wBACZ,MAAM,mEAAgD;wBACtD,MAAM,EAAI,OAAO,CAAC,MAAsB,CAAC,QAAS,CAAC,gBAAwB,CAAC,MAAM;wBAClF,aAAa,EAAE,+BAA+B,CAAC,YAAY,EAAE,KAAK,EAAE,YAAY,CAAC;qBACpF;iBACJ,CAAC,CAAC;gBACH,MAAM;YACV,CAAC;YACD;gBACI,QAAQ,CAAC,KAAK,EAAE;oBACZ,OAAO,EAAE,YAAY;oBACrB,UAAU,6CAAqC;oBAC/C,MAAM,EAAE,EAAE,IAAI,2DAA2C,EAAE;iBAC9D,CAAC,CAAC;gBACH,MAAM;YACV,8CAAsC;YACtC;gBACI,QAAQ,CAAC,KAAK,EAAE;oBACZ,OAAO,EAAE,YAAY;oBACrB,UAAU,6CAAqC;oBAC/C,MAAM,EAAE,EAAE,IAAI,EAAE,4BAA4B,CAAC,YAAY,EAAE,KAAK,EAAE,YAAY,CAAC,EAAE;iBACpF,CAAC,CAAC;gBACH,MAAM;YACV;gBACI,QAAQ,CAAC,KAAK,EAAE;oBACZ,OAAO,EAAE,YAAY;oBACrB,UAAU,6CAAqC;oBAC/C,OAAO,EAAE,EAAE,IAAI,gEAA8C,EAAE;iBAClE,CAAC,CAAC;gBACH,MAAM;YACV;gBACI,QAAQ,CAAC,KAAK,EAAE;oBACZ,OAAO,EAAE,YAAY;oBACrB,UAAU,6CAAqC;oBAC/C,eAAe,EAAE,EAAE;iBACtB,CAAC,CAAC;gBACH,MAAM;QACd,CAAC;IACL,CAAC;IAED,MAAM,gBAAgB,GAAyB,EAAE,CAAC;IAClD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,sBAAsB,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACrD,MAAM,OAAO,GAAG,sBAAsB,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAChD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC;QAC9C,gBAAgB,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,qBAAqB,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;IACpE,CAAC;IAED,OAAO,MAAM,CAAC,oBAAoB,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAC;AAC7D,CAAC,CAAC;AAEF;;;GAGG;AACH,MAAM,UAAU,kDAAkD;IAC9D,IAAI,WAAW,EAAE,CAAC;QACd,OAAO;IACX,CAAC;IACD,WAAW,GAAG,IAAI,CAAC;IAEnB,gEAAgE;IAChE,MAAM,qBAAqB,GAA6B,EAAE,CAAC;IAE3D,YAAY,CAAC,SAAS,CAAC,oBAAoB,GAAG;QAC1C,OAAO,IAAI,oBAAoB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;IACtE,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,mBAAmB,GAAG,UACzC,QAAmE,EACnE,OAAsC;QAEtC,MAAM,OAAO,GAAG,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,YAAY,IAAI,QAAQ,CAAC,aAAa,IAAI,QAAQ,CAAC,cAAc,IAAI,QAAQ,CAAC,OAAO,CAAC;QAEzJ,MAAM,IAAI,GAAG,OAAO,GAAG,GAAG,GAAG,OAAO,CAAC,OAAO,GAAG,CAAC,OAAO,CAAC,gCAAgC,CAAC,CAAC,CAAC,gCAAgC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAClI,IAAI,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,EAAE,CAAC;YACrC,MAAM,cAAc,GAAG,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC;YAC1D,IAAI,OAAO,CAAC,UAAU,IAAI,cAAc,CAAC,OAAO,EAAE,EAAE,CAAC;gBACjD,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;YACvC,CAAC;YAED,OAAO,cAAc,CAAC;QAC1B,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QAChE,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC;QAE5C,OAAO,MAAM,CAAC;IAClB,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,4BAA4B,GAAG;QAClD,OAAO,IAAI,4BAA4B,CAAC,IAAI,CAAC,CAAC;IAClD,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,yBAAyB,GAAG;QAC/C,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,CAAC;YAEjD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC;gBACpB,OAAO,KAAK,CAAC;YACjB,CAAC;QACL,CAAC;QAED,OAAO,IAAI,CAAC;IAChB,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,eAAe,GAAG,UACrC,MAAqB,EACrB,OAAwB,EACxB,QAA4B,EAC5B,CAAS,EACT,CAAC,GAAG,CAAC,EACL,CAAC,GAAG,CAAC,EACL,eAAuC,EACvC,cAAkC;QAElC,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,eAAe,EAAE,cAAc,CAAC,CAAC;IACrH,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,uBAAuB,GAAG,UAC7C,MAAqB,EACrB,OAAwB,EACxB,QAA4B,EAC5B,MAAkB,EAClB,SAAiB,CAAC,EAClB,eAAuC,EACvC,cAAkC;QAElC,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,eAAe,EAAE,cAAc,CAAC,CAAC;IACvI,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,gBAAgB,GAAG,UACtC,MAAqB,EACrB,OAAwB,EACxB,QAA4B,EAC5B,CAAU,EACV,CAAU,EACV,CAAU,EACV,MAAmB,EACnB,MAAe,EACf,eAAuC,EACvC,cAAkC;QAElC,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAE7B,MAAM,eAAe,GAAG,MAAM,CAAC,gBAAgD,CAAC;QAChF,MAAM,cAAc,GAAG,OAA+B,CAAC;QAEvD,IAAI,CAAC,eAAe,CAAC,eAAe,EAAE,CAAC;YACnC,MAAM,cAAc,GAAG,MAAM,CAAC,iCAAiC;gBAC3D,CAAC,CAAC,yBAAyB,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,eAAe,EAAE,eAAe,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,2BAA2B,CAAC;gBACpJ,CAAC,CAAC,IAAI,CAAC;YACX,cAAc,CAAC,KAAK,EAAE,CAAC;YACvB,eAAe,CAAC,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC;gBACjE,MAAM,EAAE,cAAc,oDAAuC;gBAC7D,OAAO,EAAE,eAAe,CAAC,KAAM;aAClC,CAAC,CAAC;QACP,CAAC;QAED,IAAI,cAAc,EAAE,CAAC;YACjB,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,qBAAqB,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;QAChF,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAAC,qBAAqB,CAAC,CAAC;QAEhF,WAAW,CAAC,WAAW,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;QAEzD,MAAM,UAAU,GAAG,cAAc,CAAC,aAAa,CAAC,QAAQ,EAAE,eAAe,CAAC,eAAe,EAAE,eAAe,CAAC,CAAC;QAC5G,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;YACzC,MAAM,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;YAChC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACb,SAAS;YACb,CAAC;YACD,WAAW,CAAC,YAAY,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;QAC3C,CAAC;QAED,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACvB,WAAW,CAAC,0BAA0B,CAAC,MAAM,CAAC,kBAAkB,EAAU,MAAM,CAAC,CAAC;QACtF,CAAC;aAAM,CAAC;YACJ,IAAY,CAAC,GAAW,CAAC,GAAW,CAAC,GAAG,CAAC,EAAE,CAAC;gBACxC,WAAW,CAAC,kBAAkB,CAAS,CAAC,EAAU,CAAC,EAAU,CAAC,CAAC,CAAC;YACpE,CAAC;QACL,CAAC;QACD,WAAW,CAAC,GAAG,EAAE,CAAC;QAElB,IAAI,cAAc,EAAE,CAAC;YACjB,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,EAAE,cAAc,CAAC,CAAC;YACnE,IAAI,CAAC,eAAe,IAAI,CAAC,CAAC;QAC9B,CAAC;IACL,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,qBAAqB,GAAG;QAC3C,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAC9C,MAAM,4BAA4B,GAAG,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC,kBAAkB,EAAkC,CAAC;YAC7H,IAAI,CAAC,6BAA6B,CAAC,4BAA4B,CAAC,CAAC;QACrE,CAAC;QAED,IAAI,CAAC,uBAAuB,GAAG,EAAE,CAAC;IACtC,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,8BAA8B,GAAG,UACpD,eAAwC,EACxC,iBAAyB,EACzB,oBAA4B,EAC5B,OAAyB,EACzB,UAAkB;QAElB,MAAM,aAAa,GAAG,eAA+C,CAAC;QAEtE,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACzB,MAAM,CAAC,GAAG,CAAC,OAAQ,CAAC,CAAC;YACrB,MAAM,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;QAClC,CAAC;QAED,aAAa,CAAC,OAAO,GAAG;YACpB,OAAO,EAAE,iBAAiB;YAC1B,UAAU,EAAE,oBAAoB;SACnC,CAAC;QAEF,aAAa,CAAC,KAAK,GAAG,IAAI,CAAC,qCAAqC,CAAC,iBAAiB,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;IAC7G,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,qBAAqB,GAAG,UAAU,MAAqB;QAC1E,IAAI,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5C,OAAO,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAEjD,IAAI,CAAC,6BAA6B,CAAC,MAAM,CAAC,kBAAkB,EAAkC,CAAC,CAAC;QACpG,CAAC;IACL,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,sBAAsB,GAAG;QAC5C,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,CAAC;YAEjD,MAAM,CAAC,gBAAgB,GAAG,IAAI,CAAC;YAC/B,MAAM,CAAC,mBAAmB,GAAG,KAAK,CAAC;YACnC,MAAM,CAAC,cAAc,EAAE,CAAC;QAC5B,CAAC;IACL,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,kCAAkC,GAAG,UACxD,eAA6C,EAC7C,MAAgE;QAEhE,mFAAmF;QACnF,eAAe,CAAC,KAAM,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE;YAC7D,MAAM,mBAAmB,GAA+B;gBACpD,SAAS,EAAE,CAAC;gBACZ,QAAQ,EAAE,EAAE;aACf,CAAC;YACF,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClC,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;oBAC3B,mBAAmB,CAAC,SAAS,EAAE,CAAC;gBACpC,CAAC;gBACD,mBAAmB,CAAC,QAAQ,CAAC,IAAI,CAAC;oBAC9B,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,IAAI,EAAE,OAAO,CAAC,OAAO;oBACrB,IAAI,EAAE,OAAO,CAAC,OAAO;oBACrB,MAAM,EAAE,OAAO,CAAC,OAAO;oBACvB,MAAM,EAAE,OAAO,CAAC,MAAM;oBACtB,MAAM,EAAE,OAAO,CAAC,MAAM;iBACzB,CAAC,CAAC;YACP,CAAC;YACD,MAAM,CAAC,mBAAmB,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACP,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,6BAA6B,GAAG,UAAU,eAAwC;QACrG,MAAM,qBAAqB,GAAG,eAA+C,CAAC;QAC9E,IAAI,qBAAqB,EAAE,CAAC;YACxB,eAAe,CAAC,OAAO,EAAE,CAAC;QAC9B,CAAC;IACL,CAAC,CAAC;IAEF,YAAY,CAAC,SAAS,CAAC,qCAAqC,GAAG,UAAU,aAAqB,EAAE,OAAyB,EAAE,UAAkB;QACzI,IAAI,OAAO,EAAE,CAAC;YACV,OAAO,GAAG,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC;QAC7D,CAAC;aAAM,CAAC;YACJ,OAAO,GAAG,EAAE,CAAC;QACjB,CAAC;QACD,OAAO;YACH,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC;gBACpC,IAAI,EAAE,OAAO,GAAG,aAAa;aAChC,CAAC;YACF,UAAU;SACb,CAAC;IACN,CAAC,CAAC;AACN,CAAC","sourcesContent":["/** This file must only contain pure code and pure imports */\r\n\r\nimport * as WebGPUConstants from \"../webgpuConstants\";\r\nimport { Logger } from \"core/Misc/logger\";\r\nimport { type IComputeEffectCreationOptions, type IComputeShaderPath, ComputeEffect } from \"../../../Compute/computeEffect\";\r\nimport { type IComputeContext } from \"../../../Compute/IComputeContext\";\r\nimport { type IComputePipelineContext } from \"../../../Compute/IComputePipelineContext\";\r\nimport { type Nullable } from \"../../../types\";\r\nimport { ComputeBindingType, type ComputeBindingList, type ComputeBindingMapping, type ComputeCompilationMessages } from \"../../Extensions/engine.computeShader.pure\";\r\nimport { Constants } from \"../../constants\";\r\nimport { WebGPUComputeContext } from \"../webgpuComputeContext\";\r\nimport { WebGPUComputePipelineContext } from \"../webgpuComputePipelineContext\";\r\nimport { type WebGPUPerfCounter } from \"../webgpuPerfCounter\";\r\nimport { type DataBuffer } from \"../../../Buffers/dataBuffer\";\r\nimport { WebGPUEngine } from \"../../webgpuEngine.pure\";\r\nimport { type InternalTexture } from \"../../../Materials/Textures/internalTexture\";\r\nimport { type BaseTexture } from \"../../../Materials/Textures/baseTexture.pure\";\r\n\r\nlet _Registered = false;\r\n\r\nconst _GetComputeStorageBufferType = (source: string, group: number, binding: number): WebGPUConstants.BufferBindingType => {\r\n    const bindingPattern = `@binding\\\\(${binding}\\\\)\\\\s*@group\\\\(${group}\\\\)`;\r\n    const groupPattern = `@group\\\\(${group}\\\\)\\\\s*@binding\\\\(${binding}\\\\)`;\r\n    const declarationPattern = `(?:${bindingPattern}|${groupPattern})\\\\s*var<storage\\\\s*,\\\\s*(read|read_write)>`;\r\n    const access = source.match(new RegExp(declarationPattern))?.[1];\r\n\r\n    return access === \"read\" ? WebGPUConstants.BufferBindingType.ReadOnlyStorage : WebGPUConstants.BufferBindingType.Storage;\r\n};\r\n\r\nconst _GetComputeTextureViewDimension = (source: string, group: number, binding: number): WebGPUConstants.TextureViewDimension => {\r\n    const bindingPattern = `@binding\\\\(${binding}\\\\)\\\\s*@group\\\\(${group}\\\\)`;\r\n    const groupPattern = `@group\\\\(${group}\\\\)\\\\s*@binding\\\\(${binding}\\\\)`;\r\n    const declaration = source.match(new RegExp(`(?:${bindingPattern}|${groupPattern})\\\\s*var\\\\s+\\\\w+\\\\s*:\\\\s*(texture_\\\\w+)`))?.[1];\r\n\r\n    switch (declaration) {\r\n        case \"texture_2d_array\":\r\n        case \"texture_depth_2d_array\":\r\n            return WebGPUConstants.TextureViewDimension.E2dArray;\r\n        case \"texture_cube\":\r\n            return WebGPUConstants.TextureViewDimension.Cube;\r\n        case \"texture_cube_array\":\r\n            return WebGPUConstants.TextureViewDimension.CubeArray;\r\n        case \"texture_3d\":\r\n            return WebGPUConstants.TextureViewDimension.E3d;\r\n        default:\r\n            return WebGPUConstants.TextureViewDimension.E2d;\r\n    }\r\n};\r\n\r\nconst _GetInternalTexture = (texture: BaseTexture | InternalTexture): Nullable<InternalTexture> => {\r\n    const internalTexture = texture as InternalTexture;\r\n    if (internalTexture._hardwareTexture !== undefined) {\r\n        return internalTexture;\r\n    }\r\n\r\n    return (texture as BaseTexture)._texture;\r\n};\r\n\r\nconst _GetComputeTextureSampleType = (texture: BaseTexture | InternalTexture, textureFloatLinearFiltering: boolean): WebGPUConstants.TextureSampleType => {\r\n    const internalTexture = _GetInternalTexture(texture);\r\n\r\n    if (!internalTexture) {\r\n        return WebGPUConstants.TextureSampleType.Float;\r\n    }\r\n\r\n    const textureIsDepth = internalTexture.format >= Constants.TEXTUREFORMAT_DEPTH24_STENCIL8 && internalTexture.format <= Constants.TEXTUREFORMAT_DEPTH32FLOAT_STENCIL8;\r\n    if (textureIsDepth) {\r\n        return WebGPUConstants.TextureSampleType.Depth;\r\n    }\r\n\r\n    if (internalTexture.type === Constants.TEXTURETYPE_FLOAT && !textureFloatLinearFiltering) {\r\n        return WebGPUConstants.TextureSampleType.UnfilterableFloat;\r\n    }\r\n\r\n    return WebGPUConstants.TextureSampleType.Float;\r\n};\r\n\r\nconst _GetComputePipelineLayout = (\r\n    device: GPUDevice,\r\n    bindings: ComputeBindingList,\r\n    bindingsMapping: ComputeBindingMapping | undefined,\r\n    shaderSource: string,\r\n    textureFloatLinearFiltering: boolean\r\n): Nullable<GPUPipelineLayout> => {\r\n    if (!bindingsMapping) {\r\n        return null;\r\n    }\r\n\r\n    const bindGroupLayoutEntries: GPUBindGroupLayoutEntry[][] = [];\r\n\r\n    const addEntry = (group: number, entry: GPUBindGroupLayoutEntry) => {\r\n        const entries = (bindGroupLayoutEntries[group] ??= []);\r\n        const existingIndex = entries.findIndex((item) => item.binding === entry.binding);\r\n        if (existingIndex >= 0) {\r\n            entries[existingIndex] = entry;\r\n        } else {\r\n            entries.push(entry);\r\n        }\r\n    };\r\n\r\n    for (const key in bindings) {\r\n        const binding = bindings[key];\r\n        const location = bindingsMapping[key];\r\n        if (!location) {\r\n            continue;\r\n        }\r\n\r\n        const group = location.group;\r\n        const bindingIndex = location.binding;\r\n\r\n        switch (binding.type) {\r\n            case ComputeBindingType.Texture:\r\n            case ComputeBindingType.TextureWithoutSampler:\r\n            case ComputeBindingType.InternalTexture: {\r\n                const sampleType = _GetComputeTextureSampleType(binding.object as BaseTexture | InternalTexture, textureFloatLinearFiltering);\r\n\r\n                if (binding.type === ComputeBindingType.Texture) {\r\n                    addEntry(group, {\r\n                        binding: bindingIndex - 1,\r\n                        visibility: WebGPUConstants.ShaderStage.Compute,\r\n                        sampler: {\r\n                            type:\r\n                                sampleType === WebGPUConstants.TextureSampleType.UnfilterableFloat\r\n                                    ? WebGPUConstants.SamplerBindingType.NonFiltering\r\n                                    : WebGPUConstants.SamplerBindingType.Filtering,\r\n                        },\r\n                    });\r\n                }\r\n\r\n                addEntry(group, {\r\n                    binding: bindingIndex,\r\n                    visibility: WebGPUConstants.ShaderStage.Compute,\r\n                    texture: {\r\n                        sampleType,\r\n                        viewDimension: _GetComputeTextureViewDimension(shaderSource, group, bindingIndex),\r\n                        multisampled: false,\r\n                    },\r\n                });\r\n                break;\r\n            }\r\n            case ComputeBindingType.StorageTexture: {\r\n                addEntry(group, {\r\n                    binding: bindingIndex,\r\n                    visibility: WebGPUConstants.ShaderStage.Compute,\r\n                    storageTexture: {\r\n                        access: WebGPUConstants.StorageTextureAccess.WriteOnly,\r\n                        format: ((binding.object as BaseTexture)._texture!._hardwareTexture as any).format,\r\n                        viewDimension: _GetComputeTextureViewDimension(shaderSource, group, bindingIndex),\r\n                    },\r\n                });\r\n                break;\r\n            }\r\n            case ComputeBindingType.UniformBuffer:\r\n                addEntry(group, {\r\n                    binding: bindingIndex,\r\n                    visibility: WebGPUConstants.ShaderStage.Compute,\r\n                    buffer: { type: WebGPUConstants.BufferBindingType.Uniform },\r\n                });\r\n                break;\r\n            case ComputeBindingType.StorageBuffer:\r\n            case ComputeBindingType.DataBuffer:\r\n                addEntry(group, {\r\n                    binding: bindingIndex,\r\n                    visibility: WebGPUConstants.ShaderStage.Compute,\r\n                    buffer: { type: _GetComputeStorageBufferType(shaderSource, group, bindingIndex) },\r\n                });\r\n                break;\r\n            case ComputeBindingType.Sampler:\r\n                addEntry(group, {\r\n                    binding: bindingIndex,\r\n                    visibility: WebGPUConstants.ShaderStage.Compute,\r\n                    sampler: { type: WebGPUConstants.SamplerBindingType.Filtering },\r\n                });\r\n                break;\r\n            case ComputeBindingType.ExternalTexture:\r\n                addEntry(group, {\r\n                    binding: bindingIndex,\r\n                    visibility: WebGPUConstants.ShaderStage.Compute,\r\n                    externalTexture: {},\r\n                });\r\n                break;\r\n        }\r\n    }\r\n\r\n    const bindGroupLayouts: GPUBindGroupLayout[] = [];\r\n    for (let i = 0; i < bindGroupLayoutEntries.length; i++) {\r\n        const entries = bindGroupLayoutEntries[i] ?? [];\r\n        entries.sort((a, b) => a.binding - b.binding);\r\n        bindGroupLayouts[i] = device.createBindGroupLayout({ entries });\r\n    }\r\n\r\n    return device.createPipelineLayout({ bindGroupLayouts });\r\n};\r\n\r\n/**\r\n * Register side effects for enginesWebGPUExtensionsEngineComputeShader.\r\n * Safe to call multiple times; only the first call has an effect.\r\n */\r\nexport function RegisterEnginesWebGPUExtensionsEngineComputeShader(): void {\r\n    if (_Registered) {\r\n        return;\r\n    }\r\n    _Registered = true;\r\n\r\n    // eslint-disable-next-line @typescript-eslint/naming-convention\r\n    const ComputePassDescriptor: GPUComputePassDescriptor = {};\r\n\r\n    WebGPUEngine.prototype.createComputeContext = function (): IComputeContext | undefined {\r\n        return new WebGPUComputeContext(this._device, this._cacheSampler);\r\n    };\r\n\r\n    WebGPUEngine.prototype.createComputeEffect = function (\r\n        baseName: string | (IComputeShaderPath & { computeToken?: string }),\r\n        options: IComputeEffectCreationOptions\r\n    ): ComputeEffect {\r\n        const compute = typeof baseName === \"string\" ? baseName : baseName.computeToken || baseName.computeSource || baseName.computeElement || baseName.compute;\r\n\r\n        const name = compute + \"@\" + options.defines + (options.useExplicitComputePipelineLayout ? \"@explicitComputePipelineLayout\" : \"\");\r\n        if (this._compiledComputeEffects[name]) {\r\n            const compiledEffect = this._compiledComputeEffects[name];\r\n            if (options.onCompiled && compiledEffect.isReady()) {\r\n                options.onCompiled(compiledEffect);\r\n            }\r\n\r\n            return compiledEffect;\r\n        }\r\n        const effect = new ComputeEffect(baseName, options, this, name);\r\n        this._compiledComputeEffects[name] = effect;\r\n\r\n        return effect;\r\n    };\r\n\r\n    WebGPUEngine.prototype.createComputePipelineContext = function (): IComputePipelineContext {\r\n        return new WebGPUComputePipelineContext(this);\r\n    };\r\n\r\n    WebGPUEngine.prototype.areAllComputeEffectsReady = function (): boolean {\r\n        for (const key in this._compiledComputeEffects) {\r\n            const effect = this._compiledComputeEffects[key];\r\n\r\n            if (!effect.isReady()) {\r\n                return false;\r\n            }\r\n        }\r\n\r\n        return true;\r\n    };\r\n\r\n    WebGPUEngine.prototype.computeDispatch = function (\r\n        effect: ComputeEffect,\r\n        context: IComputeContext,\r\n        bindings: ComputeBindingList,\r\n        x: number,\r\n        y = 1,\r\n        z = 1,\r\n        bindingsMapping?: ComputeBindingMapping,\r\n        gpuPerfCounter?: WebGPUPerfCounter\r\n    ): void {\r\n        this._computeDispatch(effect, context, bindings, x, y, z, undefined, undefined, bindingsMapping, gpuPerfCounter);\r\n    };\r\n\r\n    WebGPUEngine.prototype.computeDispatchIndirect = function (\r\n        effect: ComputeEffect,\r\n        context: IComputeContext,\r\n        bindings: ComputeBindingList,\r\n        buffer: DataBuffer,\r\n        offset: number = 0,\r\n        bindingsMapping?: ComputeBindingMapping,\r\n        gpuPerfCounter?: WebGPUPerfCounter\r\n    ): void {\r\n        this._computeDispatch(effect, context, bindings, undefined, undefined, undefined, buffer, offset, bindingsMapping, gpuPerfCounter);\r\n    };\r\n\r\n    WebGPUEngine.prototype._computeDispatch = function (\r\n        effect: ComputeEffect,\r\n        context: IComputeContext,\r\n        bindings: ComputeBindingList,\r\n        x?: number,\r\n        y?: number,\r\n        z?: number,\r\n        buffer?: DataBuffer,\r\n        offset?: number,\r\n        bindingsMapping?: ComputeBindingMapping,\r\n        gpuPerfCounter?: WebGPUPerfCounter\r\n    ): void {\r\n        this._endCurrentRenderPass();\r\n\r\n        const contextPipeline = effect._pipelineContext as WebGPUComputePipelineContext;\r\n        const computeContext = context as WebGPUComputeContext;\r\n\r\n        if (!contextPipeline.computePipeline) {\r\n            const explicitLayout = effect._useExplicitComputePipelineLayout\r\n                ? _GetComputePipelineLayout(this._device, bindings, bindingsMapping, contextPipeline.sources?.compute ?? \"\", this._caps.textureFloatLinearFiltering)\r\n                : null;\r\n            computeContext.clear();\r\n            contextPipeline.computePipeline = this._device.createComputePipeline({\r\n                layout: explicitLayout ?? WebGPUConstants.AutoLayoutMode.Auto,\r\n                compute: contextPipeline.stage!,\r\n            });\r\n        }\r\n\r\n        if (gpuPerfCounter) {\r\n            this._timestampQuery.startPass(ComputePassDescriptor, this._timestampIndex);\r\n        }\r\n\r\n        const computePass = this._renderEncoder.beginComputePass(ComputePassDescriptor);\r\n\r\n        computePass.setPipeline(contextPipeline.computePipeline);\r\n\r\n        const bindGroups = computeContext.getBindGroups(bindings, contextPipeline.computePipeline, bindingsMapping);\r\n        for (let i = 0; i < bindGroups.length; ++i) {\r\n            const bindGroup = bindGroups[i];\r\n            if (!bindGroup) {\r\n                continue;\r\n            }\r\n            computePass.setBindGroup(i, bindGroup);\r\n        }\r\n\r\n        if (buffer !== undefined) {\r\n            computePass.dispatchWorkgroupsIndirect(buffer.underlyingResource, <number>offset);\r\n        } else {\r\n            if (<number>x + <number>y + <number>z > 0) {\r\n                computePass.dispatchWorkgroups(<number>x, <number>y, <number>z);\r\n            }\r\n        }\r\n        computePass.end();\r\n\r\n        if (gpuPerfCounter) {\r\n            this._timestampQuery.endPass(this._timestampIndex, gpuPerfCounter);\r\n            this._timestampIndex += 2;\r\n        }\r\n    };\r\n\r\n    WebGPUEngine.prototype.releaseComputeEffects = function () {\r\n        for (const name in this._compiledComputeEffects) {\r\n            const webGPUPipelineContextCompute = this._compiledComputeEffects[name].getPipelineContext() as WebGPUComputePipelineContext;\r\n            this._deleteComputePipelineContext(webGPUPipelineContextCompute);\r\n        }\r\n\r\n        this._compiledComputeEffects = {};\r\n    };\r\n\r\n    WebGPUEngine.prototype._prepareComputePipelineContext = function (\r\n        pipelineContext: IComputePipelineContext,\r\n        computeSourceCode: string,\r\n        rawComputeSourceCode: string,\r\n        defines: Nullable<string>,\r\n        entryPoint: string\r\n    ): void {\r\n        const webGpuContext = pipelineContext as WebGPUComputePipelineContext;\r\n\r\n        if (this.dbgShowShaderCode) {\r\n            Logger.Log(defines!);\r\n            Logger.Log(computeSourceCode);\r\n        }\r\n\r\n        webGpuContext.sources = {\r\n            compute: computeSourceCode,\r\n            rawCompute: rawComputeSourceCode,\r\n        };\r\n\r\n        webGpuContext.stage = this._createComputePipelineStageDescriptor(computeSourceCode, defines, entryPoint);\r\n    };\r\n\r\n    WebGPUEngine.prototype._releaseComputeEffect = function (effect: ComputeEffect): void {\r\n        if (this._compiledComputeEffects[effect._key]) {\r\n            delete this._compiledComputeEffects[effect._key];\r\n\r\n            this._deleteComputePipelineContext(effect.getPipelineContext() as WebGPUComputePipelineContext);\r\n        }\r\n    };\r\n\r\n    WebGPUEngine.prototype._rebuildComputeEffects = function (): void {\r\n        for (const key in this._compiledComputeEffects) {\r\n            const effect = this._compiledComputeEffects[key];\r\n\r\n            effect._pipelineContext = null;\r\n            effect._wasPreviouslyReady = false;\r\n            effect._prepareEffect();\r\n        }\r\n    };\r\n\r\n    WebGPUEngine.prototype._executeWhenComputeStateIsCompiled = function (\r\n        pipelineContext: WebGPUComputePipelineContext,\r\n        action: (messages: Nullable<ComputeCompilationMessages>) => void\r\n    ): void {\r\n        // eslint-disable-next-line @typescript-eslint/no-floating-promises, github/no-then\r\n        pipelineContext.stage!.module.getCompilationInfo().then((info) => {\r\n            const compilationMessages: ComputeCompilationMessages = {\r\n                numErrors: 0,\r\n                messages: [],\r\n            };\r\n            for (const message of info.messages) {\r\n                if (message.type === \"error\") {\r\n                    compilationMessages.numErrors++;\r\n                }\r\n                compilationMessages.messages.push({\r\n                    type: message.type,\r\n                    text: message.message,\r\n                    line: message.lineNum,\r\n                    column: message.linePos,\r\n                    length: message.length,\r\n                    offset: message.offset,\r\n                });\r\n            }\r\n            action(compilationMessages);\r\n        });\r\n    };\r\n\r\n    WebGPUEngine.prototype._deleteComputePipelineContext = function (pipelineContext: IComputePipelineContext): void {\r\n        const webgpuPipelineContext = pipelineContext as WebGPUComputePipelineContext;\r\n        if (webgpuPipelineContext) {\r\n            pipelineContext.dispose();\r\n        }\r\n    };\r\n\r\n    WebGPUEngine.prototype._createComputePipelineStageDescriptor = function (computeShader: string, defines: Nullable<string>, entryPoint: string): GPUProgrammableStage {\r\n        if (defines) {\r\n            defines = \"//\" + defines.split(\"\\n\").join(\"\\n//\") + \"\\n\";\r\n        } else {\r\n            defines = \"\";\r\n        }\r\n        return {\r\n            module: this._device.createShaderModule({\r\n                code: defines + computeShader,\r\n            }),\r\n            entryPoint,\r\n        };\r\n    };\r\n}\r\n"]}

package/Engines/WebGPU/webgpuCacheRenderPipeline.d.ts

@@ -1,6 +1,6 @@
 import { type Effect } from "../../Materials/effect.js";
 import { type InternalTexture } from "../../Materials/Textures/internalTexture.js";
-import { VertexBuffer } from "../../Buffers/buffer.js";
+import { VertexBuffer } from "../../Buffers/buffer.pure.js";
 import { type DataBuffer } from "../../Buffers/dataBuffer.js";
 import { type Nullable } from "../../types.js";
 /** @internal */

package/Engines/WebGPU/webgpuCacheRenderPipeline.js

@@ -2,7 +2,7 @@
 /* eslint-disable babylonjs/available */
 /* eslint-disable jsdoc/require-jsdoc */
 
-import { VertexBuffer } from "../../Buffers/buffer.js";
+import { VertexBuffer } from "../../Buffers/buffer.pure.js";
 import { WebGPUTextureHelper } from "./webgpuTextureHelper.js";
 import { renderableTextureFormatToIndex } from "./webgpuTextureManager.js";
 import { checkNonFloatVertexBuffers } from "../../Buffers/buffer.nonFloatVertexBuffers.js";
@@ -525,6 +525,16 @@ export class WebGPUCacheRenderPipeline {
                         return normalized ? "unorm16x4" /* WebGPUConstants.VertexFormat.Unorm16x4 */ : "uint16x4" /* WebGPUConstants.VertexFormat.Uint16x4 */;
                 }
                 break;
+            case VertexBuffer.HALF_FLOAT:
+                switch (size) {
+                    case 1:
+                    case 2:
+                        return "float16x2" /* WebGPUConstants.VertexFormat.Float16x2 */;
+                    case 3:
+                    case 4:
+                        return "float16x4" /* WebGPUConstants.VertexFormat.Float16x4 */;
+                }
+                break;
             case VertexBuffer.INT:
                 switch (size) {
                     case 1: