62 lines
1003 B
JavaScript
62 lines
1003 B
JavaScript
import { Color, Node } from 'three/webgpu';
|
|
import { NodeUpdateType, renderGroup, uniform } from 'three/tsl';
|
|
|
|
/**
|
|
* Batched data node for ambient lights in dynamic lighting mode.
|
|
*
|
|
* @augments Node
|
|
*/
|
|
class AmbientLightDataNode extends Node {
|
|
|
|
static get type() {
|
|
|
|
return 'AmbientLightDataNode';
|
|
|
|
}
|
|
|
|
constructor() {
|
|
|
|
super();
|
|
|
|
this._color = new Color();
|
|
this._lights = [];
|
|
|
|
this.colorNode = uniform( this._color ).setGroup( renderGroup );
|
|
this.updateType = NodeUpdateType.RENDER;
|
|
|
|
}
|
|
|
|
setLights( lights ) {
|
|
|
|
this._lights = lights;
|
|
|
|
return this;
|
|
|
|
}
|
|
|
|
update() {
|
|
|
|
this._color.setScalar( 0 );
|
|
|
|
for ( let i = 0; i < this._lights.length; i ++ ) {
|
|
|
|
const light = this._lights[ i ];
|
|
|
|
this._color.r += light.color.r * light.intensity;
|
|
this._color.g += light.color.g * light.intensity;
|
|
this._color.b += light.color.b * light.intensity;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
setup( builder ) {
|
|
|
|
builder.context.irradiance.addAssign( this.colorNode );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
export default AmbientLightDataNode;
|