@rtzh/engine 0.0.9 → 0.0.10

package/Source/Scene/Model/LightingPipelineStage.js

@@ -1,4 +1,4 @@
-import defined from "../../Core/defined.js";
+import defined from "../../Core/defined.js";
 import ShaderDestination from "../../Renderer/ShaderDestination.js";
 import LightingStageFS from "../../Shaders/Model/LightingStageFS.js";
 import LightingModel from "./LightingModel.js";
@@ -24,10 +24,15 @@ const LightingPipelineStage = {
  * </ul>
  * @param {PrimitiveRenderResources} renderResources The render resources for the primitive
  * @param {ModelComponents.Primitive} primitive The primitive to be rendered
+ * @param {FrameState} frameState The frame state.
  *
  * @private
  */
-LightingPipelineStage.process = function (renderResources, primitive) {
+LightingPipelineStage.process = function (
+  renderResources,
+  primitive,
+  frameState,
+) {
   const { model, lightingOptions, shaderBuilder } = renderResources;
 
   if (defined(model.lightColor)) {
@@ -67,6 +72,16 @@ LightingPipelineStage.process = function (renderResources, primitive) {
     );
   }
 
+  // Add POINT_LIGHTS define if point lights are present
+  const pointLights = frameState.pointLights;
+  if (defined(pointLights) && pointLights.length > 0) {
+    shaderBuilder.addDefine(
+      "POINT_LIGHTS",
+      undefined,
+      ShaderDestination.FRAGMENT,
+    );
+  }
+
   shaderBuilder.addFragmentLines(LightingStageFS);
 };