You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
1.1 KiB
67 lines
1.1 KiB
import WebGPUProgrammableStage from './WebGPUProgrammableStage.js';
|
|
|
|
class WebGPUComputePipelines {
|
|
|
|
constructor( device, glslang ) {
|
|
|
|
this.device = device;
|
|
this.glslang = glslang;
|
|
|
|
this.pipelines = new WeakMap();
|
|
this.stages = {
|
|
compute: new WeakMap()
|
|
};
|
|
|
|
}
|
|
|
|
get( param ) {
|
|
|
|
let pipeline = this.pipelines.get( param );
|
|
|
|
// @TODO: Reuse compute pipeline if possible, introduce WebGPUComputePipeline
|
|
|
|
if ( pipeline === undefined ) {
|
|
|
|
const device = this.device;
|
|
const glslang = this.glslang;
|
|
|
|
const shader = {
|
|
computeShader: param.shader
|
|
};
|
|
|
|
// programmable stage
|
|
|
|
let stageCompute = this.stages.compute.get( shader );
|
|
|
|
if ( stageCompute === undefined ) {
|
|
|
|
stageCompute = new WebGPUProgrammableStage( device, glslang, shader.computeShader, 'compute' );
|
|
|
|
this.stages.compute.set( shader, stageCompute );
|
|
|
|
}
|
|
|
|
pipeline = device.createComputePipeline( {
|
|
compute: stageCompute.stage
|
|
} );
|
|
|
|
this.pipelines.set( param, pipeline );
|
|
|
|
}
|
|
|
|
return pipeline;
|
|
|
|
}
|
|
|
|
dispose() {
|
|
|
|
this.pipelines = new WeakMap();
|
|
this.stages = {
|
|
compute: new WeakMap()
|
|
};
|
|
|
|
}
|
|
|
|
}
|
|
|
|
export default WebGPUComputePipelines;
|
|
|