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.
65 lines
1.0 KiB
65 lines
1.0 KiB
import Node from '../core/Node.js';
|
|
import { NodeUpdateType } from '../core/constants.js';
|
|
|
|
class ComputeNode extends Node {
|
|
|
|
constructor( computeNode, count, workgroupSize = [ 64 ] ) {
|
|
|
|
super( 'void' );
|
|
|
|
this.isComputeNode = true;
|
|
|
|
this.computeNode = computeNode;
|
|
|
|
this.count = count;
|
|
this.workgroupSize = workgroupSize;
|
|
this.dispatchCount = 0;
|
|
|
|
this.updateType = NodeUpdateType.OBJECT;
|
|
|
|
this.updateDispatchCount();
|
|
|
|
}
|
|
|
|
updateDispatchCount() {
|
|
|
|
const { count, workgroupSize } = this;
|
|
|
|
let size = workgroupSize[ 0 ];
|
|
|
|
for ( let i = 1; i < workgroupSize.length; i ++ )
|
|
size *= workgroupSize[ i ];
|
|
|
|
this.dispatchCount = Math.ceil( count / size );
|
|
|
|
}
|
|
|
|
onInit() { }
|
|
|
|
update( { renderer } ) {
|
|
|
|
renderer.compute( this );
|
|
|
|
}
|
|
|
|
generate( builder ) {
|
|
|
|
const { shaderStage } = builder;
|
|
|
|
if ( shaderStage === 'compute' ) {
|
|
|
|
const snippet = this.computeNode.build( builder, 'void' );
|
|
|
|
if ( snippet !== '' ) {
|
|
|
|
builder.addFlowCode( snippet );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
export default ComputeNode;
|
|
|