three
Version:
JavaScript 3D library
105 lines (56 loc) • 1.96 kB
JavaScript
import { addNodeClass } from '../core/Node.js';
import TextureNode from './TextureNode.js';
import { nodeProxy } from '../shadernode/ShaderNode.js';
import { GPUStorageTextureAccess } from '../../renderers/webgpu/utils/WebGPUConstants.js';
class StorageTextureNode extends TextureNode {
constructor( value, uvNode, storeNode = null ) {
super( value, uvNode );
this.storeNode = storeNode;
this.isStorageTextureNode = true;
this.access = GPUStorageTextureAccess.WriteOnly;
}
getInputType( /*builder*/ ) {
return 'storageTexture';
}
setup( builder ) {
super.setup( builder );
const properties = builder.getNodeProperties( this );
properties.storeNode = this.storeNode;
}
setAccess( value ) {
this.access = value;
return this;
}
generate( builder, output ) {
let snippet;
if ( this.storeNode !== null ) {
snippet = this.generateStore( builder );
} else {
snippet = super.generate( builder, output );
}
return snippet;
}
toReadOnly() {
return this.setAccess( GPUStorageTextureAccess.ReadOnly );
}
toWriteOnly() {
return this.setAccess( GPUStorageTextureAccess.WriteOnly );
}
generateStore( builder ) {
const properties = builder.getNodeProperties( this );
const { uvNode, storeNode } = properties;
const textureProperty = super.generate( builder, 'property' );
const uvSnippet = uvNode.build( builder, 'uvec2' );
const storeSnippet = storeNode.build( builder, 'vec4' );
const snippet = builder.generateTextureStore( builder, textureProperty, uvSnippet, storeSnippet );
builder.addLineFlowCode( snippet );
}
}
export default StorageTextureNode;
export const storageTexture = nodeProxy( StorageTextureNode );
export const textureStore = ( value, uvNode, storeNode ) => {
const node = storageTexture( value, uvNode, storeNode );
if ( storeNode !== null ) node.append();
return node;
};
addNodeClass( 'StorageTextureNode', StorageTextureNode );