@polygonjs/polygonjs/src/engine/scene/utils/raymarching/PointLight.ts

node-based WebGL 3D engine https://polygonjs.com

import {IUniform, PointLight} from 'three';
import {
	PenumbraUniformElement,
	updateUserDataPenumbra,
	UniformsUpdateFunction,
	ShadowUniformElement,
	updateUserDataShadowBias,
} from './_Base';

export interface PointLightRayMarchingUniformElement extends PenumbraUniformElement, ShadowUniformElement {}
interface PointLightRayMarchingUniforms extends Array<PointLightRayMarchingUniformElement> {
	needsUpdate?: boolean;
}
export interface PointLightRayMarchingUniform extends IUniform {
	value: PointLightRayMarchingUniforms;
}
export function _createPointLightUniform(): PointLightRayMarchingUniformElement {
	return {
		penumbra: 0,
		shadowBiasAngle: 0,
		shadowBiasDistance: 0,
	};
}

let pointLightIndex = 0;

export const _updateUniformsWithPointLight: UniformsUpdateFunction<PointLight> = (
	object: PointLight,
	pointLightsRayMarching: PointLightRayMarchingUniform
) => {
	updateUserDataPenumbra(object as PointLight, pointLightsRayMarching, pointLightIndex, _createPointLightUniform);
	updateUserDataShadowBias(object as PointLight, pointLightsRayMarching, pointLightIndex, _createPointLightUniform);
	pointLightIndex++;
};
export function _resetPointLightIndex() {
	pointLightIndex = 0;
}