3D-Demo/jsm/renderers/webgpu/nodes/WebGPUNodes.js

83 lines
1.4 KiB
JavaScript
Raw Normal View History

2022-10-14 16:50:42 +08:00
import WebGPUNodeBuilder from './WebGPUNodeBuilder.js';
import NodeFrame from 'three-nodes/core/NodeFrame.js';
class WebGPUNodes {
constructor( renderer, properties ) {
this.renderer = renderer;
this.properties = properties;
this.nodeFrame = new NodeFrame();
}
get( object ) {
const objectProperties = this.properties.get( object );
let nodeBuilder = objectProperties.nodeBuilder;
if ( nodeBuilder === undefined ) {
const fogNode = objectProperties.fogNode;
const lightNode = objectProperties.lightNode;
nodeBuilder = new WebGPUNodeBuilder( object, this.renderer );
nodeBuilder.lightNode = lightNode;
nodeBuilder.fogNode = fogNode;
nodeBuilder.build();
objectProperties.nodeBuilder = nodeBuilder;
}
return nodeBuilder;
}
remove( object ) {
const objectProperties = this.properties.get( object );
delete objectProperties.nodeBuilder;
}
updateFrame() {
this.nodeFrame.update();
}
update( object, camera ) {
const renderer = this.renderer;
const material = object.material;
const nodeBuilder = this.get( object );
const nodeFrame = this.nodeFrame;
nodeFrame.object = object;
nodeFrame.camera = camera;
nodeFrame.renderer = renderer;
nodeFrame.material = material;
for ( const node of nodeBuilder.updateNodes ) {
nodeFrame.updateNode( node );
}
}
dispose() {
this.nodeFrame = new NodeFrame();
}
}
export default WebGPUNodes;