Skip to content

Commit

Permalink
Implement Renaud suggestions
Browse files Browse the repository at this point in the history
  • Loading branch information
cmhhelgeson committed Aug 25, 2024
1 parent 904991d commit d73dc3d
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 12 deletions.
2 changes: 1 addition & 1 deletion src/nodes/Nodes.js
Original file line number Diff line number Diff line change
Expand Up @@ -172,7 +172,7 @@ export { default as RangeNode, range } from './geometry/RangeNode.js';

// gpgpu
export { default as ComputeNode, compute } from './gpgpu/ComputeNode.js';
export { default as ScopedArrayNode, workgroupArray, privateArray } from './gpgpu/ScopedArrayNode.js';
export { default as WorkgroupInfoNode, workgroupArray } from './gpgpu/WorkgroupInfoNode.js';
export { workgroupBarrier, textureBarrier, storageBarrier } from './gpgpu/BarrierNode.js';

// lighting
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,11 @@ import ArrayElementNode from '../utils/ArrayElementNode.js';
import { nodeObject } from '../shadernode/ShaderNode.js';
import Node, { addNodeClass } from '../core/Node.js';

class ScopedArrayElementNode extends ArrayElementNode {
class WorkgroupInfoElementNode extends ArrayElementNode {

constructor( scopedArrayNode, indexNode ) {
constructor( workgroupInfoNode, indexNode ) {

super( scopedArrayNode, indexNode );
super( workgroupInfoNode, indexNode );

this.isScopedArrayElementNode = true;

Expand Down Expand Up @@ -36,7 +36,7 @@ class ScopedArrayElementNode extends ArrayElementNode {
}


class ScopedArrayNode extends Node {
class WorkgroupInfoNode extends Node {

constructor( scope, bufferType, bufferCount = 0 ) {

Expand All @@ -45,7 +45,7 @@ class ScopedArrayNode extends Node {
this.bufferType = bufferType;
this.bufferCount = bufferCount;

this.isScopedArrayNode = true;
this.isWorkgroupInfoNode = true;

this.scope = scope;

Expand All @@ -59,7 +59,7 @@ class ScopedArrayNode extends Node {

}

getHash( builder ) {
getHash() {

return this.uuid;

Expand All @@ -81,7 +81,7 @@ class ScopedArrayNode extends Node {

element( indexNode ) {

return nodeObject( new ScopedArrayElementNode( this, indexNode ) );
return nodeObject( new WorkgroupInfoElementNode( this, indexNode ) );

}

Expand All @@ -93,10 +93,9 @@ class ScopedArrayNode extends Node {

}

export default ScopedArrayNode;
export default WorkgroupInfoNode;

export const workgroupArray = ( type, count ) => nodeObject( new ScopedArrayNode( 'Workgroup', type, count ) );
export const privateArray = ( type, count ) => nodeObject( new ScopedArrayNode( 'Private', type, count ) );
export const workgroupArray = ( type, count ) => nodeObject( new WorkgroupInfoNode( 'Workgroup', type, count ) );

addNodeClass( 'ScopedArrayNode', ScopedArrayNode );
addNodeClass( 'WorkgroupInfoNode', WorkgroupInfoNode );

0 comments on commit d73dc3d

Please sign in to comment.