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.
78 lines
1.3 KiB
78 lines
1.3 KiB
import WebGPUProgrammableStage from './WebGPUProgrammableStage.js';
|
|
|
|
class WebGPUComputePipelines {
|
|
|
|
constructor( device, nodes ) {
|
|
|
|
this.device = device;
|
|
this.nodes = nodes;
|
|
|
|
this.pipelines = new WeakMap();
|
|
this.stages = {
|
|
compute: new WeakMap()
|
|
};
|
|
|
|
}
|
|
|
|
has( computeNode ) {
|
|
|
|
return this.pipelines.get( computeNode ) !== undefined;
|
|
|
|
}
|
|
|
|
get( computeNode ) {
|
|
|
|
let pipeline = this.pipelines.get( computeNode );
|
|
|
|
// @TODO: Reuse compute pipeline if possible, introduce WebGPUComputePipeline
|
|
|
|
if ( pipeline === undefined ) {
|
|
|
|
const device = this.device;
|
|
|
|
// get shader
|
|
|
|
const nodeBuilder = this.nodes.get( computeNode );
|
|
const computeShader = nodeBuilder.computeShader;
|
|
|
|
const shader = {
|
|
computeShader
|
|
};
|
|
|
|
// programmable stage
|
|
|
|
let stageCompute = this.stages.compute.get( shader );
|
|
|
|
if ( stageCompute === undefined ) {
|
|
|
|
stageCompute = new WebGPUProgrammableStage( device, computeShader, 'compute' );
|
|
|
|
this.stages.compute.set( shader, stageCompute );
|
|
|
|
}
|
|
|
|
pipeline = device.createComputePipeline( {
|
|
compute: stageCompute.stage,
|
|
layout: 'auto'
|
|
} );
|
|
|
|
this.pipelines.set( computeNode, pipeline );
|
|
|
|
}
|
|
|
|
return pipeline;
|
|
|
|
}
|
|
|
|
dispose() {
|
|
|
|
this.pipelines = new WeakMap();
|
|
this.stages = {
|
|
compute: new WeakMap()
|
|
};
|
|
|
|
}
|
|
|
|
}
|
|
|
|
export default WebGPUComputePipelines;
|
|
|