import AnalyticLightNode from './AnalyticLightNode.js';
import { getDistanceAttenuation } from './LightUtils.js';
import { uniform } from '../core/UniformNode.js';
import { lightViewPosition } from '../accessors/Lights.js';
import { positionView } from '../accessors/Position.js';
import { Fn } from '../tsl/TSLBase.js';
import { renderGroup } from '../core/UniformGroupNode.js';
import { pointShadow } from './PointShadowNode.js';
export const directPointLight = Fn( ( { color, lightViewPosition, cutoffDistance, decayExponent }, builder ) => {
const lightingModel = builder.context.lightingModel;
const lVector = lightViewPosition.sub( positionView ); // @TODO: Add it into LightNode
const lightDirection = lVector.normalize();
const lightDistance = lVector.length();
const lightAttenuation = getDistanceAttenuation( {
lightDistance,
cutoffDistance,
decayExponent
} );
const lightColor = color.mul( lightAttenuation );
const reflectedLight = builder.context.reflectedLight;
lightingModel.direct( {
lightDirection,
lightColor,
reflectedLight
}, builder.stack, builder );
} );
/**
* Module for representing point lights as nodes.
*
* @augments AnalyticLightNode
*/
class PointLightNode extends AnalyticLightNode {
static get type() {
return 'PointLightNode';
}
/**
* Constructs a new point light node.
*
* @param {PointLight?} [light=null] - The point light source.
*/
constructor( light = null ) {
super( light );
/**
* Uniform node representing the cutoff distance.
*
* @type {UniformNode<float>}
*/
this.cutoffDistanceNode = uniform( 0 ).setGroup( renderGroup );
/**
* Uniform node representing the decay exponent.
*
* @type {UniformNode<float>}
*/
this.decayExponentNode = uniform( 2 ).setGroup( renderGroup );
}
/**
* Overwritten to updated point light specific uniforms.
*
* @param {NodeFrame} frame - A reference to the current node frame.
*/
update( frame ) {
const { light } = this;
super.update( frame );
this.cutoffDistanceNode.value = light.distance;
this.decayExponentNode.value = light.decay;
}
/**
* Overwritten to setup point light specific shadow.
*
* @return {PointShadowNode}
*/
setupShadowNode() {
return pointShadow( this.light );
}
setup( builder ) {
super.setup( builder );
directPointLight( {
color: this.colorNode,
lightViewPosition: lightViewPosition( this.light ),
cutoffDistance: this.cutoffDistanceNode,
decayExponent: this.decayExponentNode
} ).append();
}
}
export default PointLightNode;