锦水大屏
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.
 
 

73 lines
1.1 KiB

import WebGPUNodeBuilder from './WebGPUNodeBuilder.js';
import NodeFrame from '../../nodes/core/NodeFrame.js';
class WebGPUNodes {
constructor( renderer ) {
this.renderer = renderer;
this.nodeFrame = new NodeFrame();
this.builders = new WeakMap();
}
get( object, lightNode ) {
let nodeBuilder = this.builders.get( object );
if ( nodeBuilder === undefined ) {
nodeBuilder = new WebGPUNodeBuilder( object, this.renderer, lightNode ).build();
this.builders.set( object, nodeBuilder );
}
return nodeBuilder;
}
remove( object ) {
this.builders.delete( object );
}
updateFrame() {
this.nodeFrame.update();
}
update( object, camera, lightNode ) {
const renderer = this.renderer;
const material = object.material;
const nodeBuilder = this.get( object, lightNode );
const nodeFrame = this.nodeFrame;
nodeFrame.material = material;
nodeFrame.camera = camera;
nodeFrame.object = object;
nodeFrame.renderer = renderer;
for ( const node of nodeBuilder.updateNodes ) {
nodeFrame.updateNode( node );
}
}
dispose() {
this.builders = new WeakMap();
}
}
export default WebGPUNodes;