@animech-public/playcanvas/build/playcanvas.dbg/src/platform/graphics/webgpu/webgpu-compute-pipeline.js

PlayCanvas WebGL game engine

import { DebugHelper, Debug } from '../../../core/debug.js';
import { TRACEID_COMPUTEPIPELINE_ALLOC } from '../../../core/constants.js';
import { WebgpuDebug } from './webgpu-debug.js';
import { WebgpuPipeline } from './webgpu-pipeline.js';

let _pipelineId = 0;

/**
 * @ignore
 */
class WebgpuComputePipeline extends WebgpuPipeline {
  /** @private */
  get(shader, bindGroupFormat) {
    // pipeline layout
    const pipelineLayout = this.getPipelineLayout([bindGroupFormat.impl]);

    // TODO: this could be cached

    const pipeline = this.create(shader, pipelineLayout);
    return pipeline;
  }
  create(shader, pipelineLayout) {
    const wgpu = this.device.wgpu;

    /** @type {import('./webgpu-shader.js').WebgpuShader} */
    const webgpuShader = shader.impl;

    /** @type {GPUComputePipelineDescriptor} */
    const descr = {
      compute: {
        module: webgpuShader.getComputeShaderModule(),
        entryPoint: webgpuShader.computeEntryPoint
      },
      // uniform / texture binding layout
      layout: pipelineLayout
    };
    WebgpuDebug.validate(this.device);
    _pipelineId++;
    DebugHelper.setLabel(descr, `ComputePipelineDescr-${_pipelineId}`);
    const pipeline = wgpu.createComputePipeline(descr);
    DebugHelper.setLabel(pipeline, `ComputePipeline-${_pipelineId}`);
    Debug.trace(TRACEID_COMPUTEPIPELINE_ALLOC, `Alloc: Id ${_pipelineId}`, descr);
    WebgpuDebug.end(this.device, {
      computePipeline: this,
      descr,
      shader
    });
    return pipeline;
  }
}

export { WebgpuComputePipeline };