You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
427 lines
9.2 KiB
JavaScript
427 lines
9.2 KiB
JavaScript
import DataMap from '../DataMap.js';
|
|
import ChainMap from '../ChainMap.js';
|
|
import NodeBuilderState from './NodeBuilderState.js';
|
|
import { NoToneMapping, EquirectangularReflectionMapping, EquirectangularRefractionMapping } from 'three';
|
|
import { NodeFrame, cubeTexture, texture, rangeFog, densityFog, reference, toneMapping, equirectUV, viewportBottomLeft, normalWorld } from '../../../nodes/Nodes.js';
|
|
|
|
class Nodes extends DataMap {
|
|
|
|
constructor( renderer, backend ) {
|
|
|
|
super();
|
|
|
|
this.renderer = renderer;
|
|
this.backend = backend;
|
|
this.nodeFrame = new NodeFrame();
|
|
this.nodeBuilderCache = new Map();
|
|
this.frameHashCache = new ChainMap();
|
|
|
|
}
|
|
|
|
getForRenderCacheKey( renderObject ) {
|
|
|
|
return renderObject.initialCacheKey;
|
|
|
|
}
|
|
|
|
getForRender( renderObject ) {
|
|
|
|
const renderObjectData = this.get( renderObject );
|
|
|
|
let nodeBuilderState = renderObjectData.nodeBuilderState;
|
|
|
|
if ( nodeBuilderState === undefined ) {
|
|
|
|
const { nodeBuilderCache } = this;
|
|
|
|
const cacheKey = this.getForRenderCacheKey( renderObject );
|
|
|
|
nodeBuilderState = nodeBuilderCache.get( cacheKey );
|
|
|
|
if ( nodeBuilderState === undefined ) {
|
|
|
|
const nodeBuilder = this.backend.createNodeBuilder( renderObject.object, this.renderer, renderObject.scene );
|
|
nodeBuilder.material = renderObject.material;
|
|
nodeBuilder.context.material = renderObject.material;
|
|
nodeBuilder.lightsNode = renderObject.lightsNode;
|
|
nodeBuilder.environmentNode = this.getEnvironmentNode( renderObject.scene );
|
|
nodeBuilder.fogNode = this.getFogNode( renderObject.scene );
|
|
nodeBuilder.toneMappingNode = this.getToneMappingNode();
|
|
nodeBuilder.build();
|
|
|
|
nodeBuilderState = this._createNodeBuilderState( nodeBuilder );
|
|
|
|
nodeBuilderCache.set( cacheKey, nodeBuilderState );
|
|
|
|
}
|
|
|
|
nodeBuilderState.usedTimes ++;
|
|
|
|
renderObjectData.nodeBuilderState = nodeBuilderState;
|
|
|
|
}
|
|
|
|
return nodeBuilderState;
|
|
|
|
}
|
|
|
|
delete( object ) {
|
|
|
|
if ( object.isRenderObject ) {
|
|
|
|
const nodeBuilderState = this.get( object ).nodeBuilderState;
|
|
nodeBuilderState.usedTimes --;
|
|
|
|
if ( nodeBuilderState.usedTimes === 0 ) {
|
|
|
|
this.nodeBuilderCache.delete( this.getForRenderCacheKey( object ) );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return super.delete( object );
|
|
|
|
}
|
|
|
|
getForCompute( computeNode ) {
|
|
|
|
const computeData = this.get( computeNode );
|
|
|
|
let nodeBuilderState = computeData.nodeBuilderState;
|
|
|
|
if ( nodeBuilderState === undefined ) {
|
|
|
|
const nodeBuilder = this.backend.createNodeBuilder( computeNode, this.renderer );
|
|
nodeBuilder.build();
|
|
|
|
nodeBuilderState = this._createNodeBuilderState( nodeBuilder );
|
|
|
|
computeData.nodeBuilderState = nodeBuilder;
|
|
|
|
}
|
|
|
|
return nodeBuilderState;
|
|
|
|
}
|
|
|
|
_createNodeBuilderState( nodeBuilder ) {
|
|
|
|
return new NodeBuilderState(
|
|
nodeBuilder.vertexShader,
|
|
nodeBuilder.fragmentShader,
|
|
nodeBuilder.computeShader,
|
|
nodeBuilder.getAttributesArray(),
|
|
nodeBuilder.getBindings(),
|
|
nodeBuilder.updateNodes,
|
|
nodeBuilder.updateBeforeNodes
|
|
);
|
|
|
|
}
|
|
|
|
getEnvironmentNode( scene ) {
|
|
|
|
return scene.environmentNode || this.get( scene ).environmentNode || null;
|
|
|
|
}
|
|
|
|
getBackgroundNode( scene ) {
|
|
|
|
return scene.backgroundNode || this.get( scene ).backgroundNode || null;
|
|
|
|
}
|
|
|
|
getFogNode( scene ) {
|
|
|
|
return scene.fogNode || this.get( scene ).fogNode || null;
|
|
|
|
}
|
|
|
|
getToneMappingNode() {
|
|
|
|
if ( this.isToneMappingState === false ) return null;
|
|
|
|
return this.renderer.toneMappingNode || this.get( this.renderer ).toneMappingNode || null;
|
|
|
|
}
|
|
|
|
getCacheKey( scene, lightsNode ) {
|
|
|
|
const chain = [ scene, lightsNode ];
|
|
const frameId = this.nodeFrame.frameId;
|
|
|
|
let cacheKeyData = this.frameHashCache.get( chain );
|
|
|
|
if ( cacheKeyData === undefined || cacheKeyData.frameId !== frameId ) {
|
|
|
|
const environmentNode = this.getEnvironmentNode( scene );
|
|
const fogNode = this.getFogNode( scene );
|
|
const toneMappingNode = this.getToneMappingNode();
|
|
|
|
const cacheKey = [];
|
|
|
|
if ( lightsNode ) cacheKey.push( lightsNode.getCacheKey() );
|
|
if ( environmentNode ) cacheKey.push( environmentNode.getCacheKey() );
|
|
if ( fogNode ) cacheKey.push( fogNode.getCacheKey() );
|
|
if ( toneMappingNode ) cacheKey.push( toneMappingNode.getCacheKey() );
|
|
|
|
cacheKeyData = {
|
|
frameId,
|
|
cacheKey: cacheKey.join( ',' )
|
|
};
|
|
|
|
this.frameHashCache.set( chain, cacheKeyData );
|
|
|
|
}
|
|
|
|
return cacheKeyData.cacheKey;
|
|
|
|
}
|
|
|
|
updateScene( scene ) {
|
|
|
|
this.updateEnvironment( scene );
|
|
this.updateFog( scene );
|
|
this.updateBackground( scene );
|
|
this.updateToneMapping();
|
|
|
|
}
|
|
|
|
get isToneMappingState() {
|
|
|
|
const renderer = this.renderer;
|
|
const renderTarget = renderer.getRenderTarget();
|
|
|
|
return renderTarget && renderTarget.isCubeRenderTarget ? false : true;
|
|
|
|
}
|
|
|
|
updateToneMapping() {
|
|
|
|
const renderer = this.renderer;
|
|
const rendererData = this.get( renderer );
|
|
const rendererToneMapping = renderer.toneMapping;
|
|
|
|
if ( this.isToneMappingState && rendererToneMapping !== NoToneMapping ) {
|
|
|
|
if ( rendererData.toneMapping !== rendererToneMapping ) {
|
|
|
|
const rendererToneMappingNode = rendererData.rendererToneMappingNode || toneMapping( rendererToneMapping, reference( 'toneMappingExposure', 'float', renderer ) );
|
|
rendererToneMappingNode.toneMapping = rendererToneMapping;
|
|
|
|
rendererData.rendererToneMappingNode = rendererToneMappingNode;
|
|
rendererData.toneMappingNode = rendererToneMappingNode;
|
|
rendererData.toneMapping = rendererToneMapping;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// Don't delete rendererData.rendererToneMappingNode
|
|
delete rendererData.toneMappingNode;
|
|
delete rendererData.toneMapping;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
updateBackground( scene ) {
|
|
|
|
const sceneData = this.get( scene );
|
|
const background = scene.background;
|
|
|
|
if ( background ) {
|
|
|
|
if ( sceneData.background !== background ) {
|
|
|
|
let backgroundNode = null;
|
|
|
|
if ( background.isCubeTexture === true ) {
|
|
|
|
backgroundNode = cubeTexture( background, normalWorld );
|
|
|
|
} else if ( background.isTexture === true ) {
|
|
|
|
let nodeUV = null;
|
|
|
|
if ( background.mapping === EquirectangularReflectionMapping || background.mapping === EquirectangularRefractionMapping ) {
|
|
|
|
nodeUV = equirectUV();
|
|
|
|
} else {
|
|
|
|
nodeUV = viewportBottomLeft;
|
|
|
|
}
|
|
|
|
backgroundNode = texture( background, nodeUV ).setUpdateMatrix( true );
|
|
|
|
} else if ( background.isColor !== true ) {
|
|
|
|
console.error( 'WebGPUNodes: Unsupported background configuration.', background );
|
|
|
|
}
|
|
|
|
sceneData.backgroundNode = backgroundNode;
|
|
sceneData.background = background;
|
|
|
|
}
|
|
|
|
} else if ( sceneData.backgroundNode ) {
|
|
|
|
delete sceneData.backgroundNode;
|
|
delete sceneData.background;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
updateFog( scene ) {
|
|
|
|
const sceneData = this.get( scene );
|
|
const fog = scene.fog;
|
|
|
|
if ( fog ) {
|
|
|
|
if ( sceneData.fog !== fog ) {
|
|
|
|
let fogNode = null;
|
|
|
|
if ( fog.isFogExp2 ) {
|
|
|
|
fogNode = densityFog( reference( 'color', 'color', fog ), reference( 'density', 'float', fog ) );
|
|
|
|
} else if ( fog.isFog ) {
|
|
|
|
fogNode = rangeFog( reference( 'color', 'color', fog ), reference( 'near', 'float', fog ), reference( 'far', 'float', fog ) );
|
|
|
|
} else {
|
|
|
|
console.error( 'WebGPUNodes: Unsupported fog configuration.', fog );
|
|
|
|
}
|
|
|
|
sceneData.fogNode = fogNode;
|
|
sceneData.fog = fog;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
delete sceneData.fogNode;
|
|
delete sceneData.fog;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
updateEnvironment( scene ) {
|
|
|
|
const sceneData = this.get( scene );
|
|
const environment = scene.environment;
|
|
|
|
if ( environment ) {
|
|
|
|
if ( sceneData.environment !== environment ) {
|
|
|
|
let environmentNode = null;
|
|
|
|
if ( environment.isCubeTexture === true ) {
|
|
|
|
environmentNode = cubeTexture( environment );
|
|
|
|
} else if ( environment.isTexture === true ) {
|
|
|
|
environmentNode = texture( environment );
|
|
|
|
} else {
|
|
|
|
console.error( 'Nodes: Unsupported environment configuration.', environment );
|
|
|
|
}
|
|
|
|
sceneData.environmentNode = environmentNode;
|
|
sceneData.environment = environment;
|
|
|
|
}
|
|
|
|
} else if ( sceneData.environmentNode ) {
|
|
|
|
delete sceneData.environmentNode;
|
|
delete sceneData.environment;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
getNodeFrame( renderer = this.renderer, scene = null, object = null, camera = null, material = null ) {
|
|
|
|
const nodeFrame = this.nodeFrame;
|
|
nodeFrame.renderer = renderer;
|
|
nodeFrame.scene = scene;
|
|
nodeFrame.object = object;
|
|
nodeFrame.camera = camera;
|
|
nodeFrame.material = material;
|
|
|
|
return nodeFrame;
|
|
|
|
}
|
|
|
|
getNodeFrameForRender( renderObject ) {
|
|
|
|
return this.getNodeFrame( renderObject.renderer, renderObject.scene, renderObject.object, renderObject.camera, renderObject.material );
|
|
|
|
}
|
|
|
|
updateBefore( renderObject ) {
|
|
|
|
const nodeFrame = this.getNodeFrameForRender( renderObject );
|
|
const nodeBuilder = renderObject.getNodeBuilderState();
|
|
|
|
for ( const node of nodeBuilder.updateBeforeNodes ) {
|
|
|
|
nodeFrame.updateBeforeNode( node );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
updateForCompute( computeNode ) {
|
|
|
|
const nodeFrame = this.getNodeFrame();
|
|
const nodeBuilder = this.getForCompute( computeNode );
|
|
|
|
for ( const node of nodeBuilder.updateNodes ) {
|
|
|
|
nodeFrame.updateNode( node );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
updateForRender( renderObject ) {
|
|
|
|
const nodeFrame = this.getNodeFrameForRender( renderObject );
|
|
const nodeBuilder = renderObject.getNodeBuilderState();
|
|
|
|
for ( const node of nodeBuilder.updateNodes ) {
|
|
|
|
nodeFrame.updateNode( node );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
dispose() {
|
|
|
|
super.dispose();
|
|
|
|
this.nodeFrame = new NodeFrame();
|
|
this.nodeBuilderCache = new Map();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
export default Nodes;
|