templated setup jobs

libraries/render-utils/src/AmbientOcclusionStage.cpp

@@ -10,13 +10,4 @@
 
 #include "AmbientOcclusionStage.h"
 
-#include <gpu/Context.h>
-
 std::string AmbientOcclusionStage::_name { "AMBIENT_OCCLUSION_STAGE" };
-
-void AmbientOcclusionStageSetup::run(const render::RenderContextPointer& renderContext) {
-    auto stage = renderContext->_scene->getStage(AmbientOcclusionStage::getName());
-    if (!stage) {
-        renderContext->_scene->resetStage(AmbientOcclusionStage::getName(), std::make_shared<AmbientOcclusionStage>());
-    }
-}

libraries/render-utils/src/AmbientOcclusionStage.h

@@ -12,20 +12,16 @@
 #define hifi_render_utils_AmbientOcclusionStage_h
 
 #include <graphics/AmbientOcclusion.h>
-#include <render/Forward.h>
-#include <render/DrawTask.h>
 #include <render/Stage.h>
+#include <render/StageSetup.h>
 
 // AmbientOcclusion stage to set up ambientOcclusion-related rendering tasks
 class AmbientOcclusionStage : public render::PointerStage<graphics::AmbientOcclusion, graphics::AmbientOcclusionPointer> {};
 using AmbientOcclusionStagePointer = std::shared_ptr<AmbientOcclusionStage>;
 
-class AmbientOcclusionStageSetup {
+class AmbientOcclusionStageSetup : public render::StageSetup<AmbientOcclusionStage> {
 public:
     using JobModel = render::Job::Model<AmbientOcclusionStageSetup>;
-
-    AmbientOcclusionStageSetup() {}
-    void run(const render::RenderContextPointer& renderContext);
 };
 
 #endif

libraries/render-utils/src/BackgroundStage.cpp

@@ -13,8 +13,6 @@
 
 #include "DeferredLightingEffect.h"
 
-#include <gpu/Context.h>
-
 #include <graphics/ShaderConstants.h>
 
 std::string BackgroundStage::_name { "BACKGROUND_STAGE" };
@@ -74,10 +72,3 @@ void DrawBackgroundStage::run(const render::RenderContextPointer& renderContext,
         args->_batch = nullptr;
     }
 }
-
-void BackgroundStageSetup::run(const render::RenderContextPointer& renderContext) {
-    auto stage = renderContext->_scene->getStage(BackgroundStage::getName());
-    if (!stage) {
-        renderContext->_scene->resetStage(BackgroundStage::getName(), std::make_shared<BackgroundStage>());
-    }
-}

libraries/render-utils/src/BackgroundStage.h

@@ -1,6 +1,6 @@
 //
 //  BackgroundStage.h
-
+//
 //  Created by Sam Gateau on 5/9/2017.
 //  Copyright 2015 High Fidelity, Inc.
 //  Copyright 2024 Overte e.V.
@@ -14,6 +14,7 @@
 
 #include <graphics/Stage.h>
 #include <render/Stage.h>
+#include <render/StageSetup.h>
 
 #include "HazeStage.h"
 #include "LightingModel.h"
@@ -22,12 +23,9 @@
 class BackgroundStage : public render::PointerStage<graphics::SunSkyStage, graphics::SunSkyStagePointer> {};
 using BackgroundStagePointer = std::shared_ptr<BackgroundStage>;
 
-class BackgroundStageSetup {
+class BackgroundStageSetup : public render::StageSetup<BackgroundStage> {
 public:
     using JobModel = render::Job::Model<BackgroundStageSetup>;
-
-    BackgroundStageSetup() {}
-    void run(const render::RenderContextPointer& renderContext);
 };
 
 class DrawBackgroundStage {

libraries/render-utils/src/BloomStage.cpp

@@ -8,17 +8,7 @@
 //  Distributed under the Apache License, Version 2.0.
 //  See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html
 //
+
 #include "BloomStage.h"
 
-#include "DeferredLightingEffect.h"
-
-#include <gpu/Context.h>
-
 std::string BloomStage::_name { "BLOOM_STAGE" };
-
-void BloomStageSetup::run(const render::RenderContextPointer& renderContext) {
-    auto stage = renderContext->_scene->getStage(BloomStage::getName());
-    if (!stage) {
-        renderContext->_scene->resetStage(BloomStage::getName(), std::make_shared<BloomStage>());
-    }
-}

libraries/render-utils/src/BloomStage.h

@@ -13,22 +13,16 @@
 #define hifi_render_utils_BloomStage_h
 
 #include <graphics/Bloom.h>
-#include <render/Forward.h>
-#include <render/DrawTask.h>
 #include <render/Stage.h>
+#include <render/StageSetup.h>
 
 // Bloom stage to set up bloom-related rendering tasks
 class BloomStage : public render::PointerStage<graphics::Bloom, graphics::BloomPointer> {};
 using BloomStagePointer = std::shared_ptr<BloomStage>;
 
-class BloomStageSetup {
+class BloomStageSetup : public render::StageSetup<BloomStage> {
 public:
     using JobModel = render::Job::Model<BloomStageSetup>;
-
-    BloomStageSetup() {}
-    void run(const render::RenderContextPointer& renderContext);
-
-protected:
 };
 
 #endif

libraries/render-utils/src/HazeStage.cpp

@@ -8,17 +8,7 @@
 //  Distributed under the Apache License, Version 2.0.
 //  See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html
 //
+
 #include "HazeStage.h"
 
-#include "DeferredLightingEffect.h"
-
-#include <gpu/Context.h>
-
 std::string HazeStage::_name { "HAZE_STAGE" };
-
-void HazeStageSetup::run(const render::RenderContextPointer& renderContext) {
-    auto stage = renderContext->_scene->getStage(HazeStage::getName());
-    if (!stage) {
-        renderContext->_scene->resetStage(HazeStage::getName(), std::make_shared<HazeStage>());
-    }
-}

libraries/render-utils/src/HazeStage.h

@@ -13,22 +13,16 @@
 #define hifi_render_utils_HazeStage_h
 
 #include <graphics/Haze.h>
-#include <render/Forward.h>
-#include <render/DrawTask.h>
 #include <render/Stage.h>
+#include <render/StageSetup.h>
 
 // Haze stage to set up haze-related rendering tasks
 class HazeStage : public render::PointerStage<graphics::Haze, graphics::HazePointer> {};
 using HazeStagePointer = std::shared_ptr<HazeStage>;
 
-class HazeStageSetup {
+class HazeStageSetup : public render::StageSetup<HazeStage> {
 public:
     using JobModel = render::Job::Model<HazeStageSetup>;
-
-    HazeStageSetup() {}
-    void run(const render::RenderContextPointer& renderContext);
-
-protected:
 };
 
 class FetchHazeConfig : public render::Job::Config {

libraries/render-utils/src/LightStage.cpp

@@ -403,15 +403,3 @@ void LightStage::updateLightArrayBuffer(Index lightId) {
         // this should not happen ?
     }
 }
-
-LightStageSetup::LightStageSetup() {
-}
-
-void LightStageSetup::run(const render::RenderContextPointer& renderContext) {
-    if (renderContext->_scene) {
-        auto stage = renderContext->_scene->getStage(LightStage::getName());
-        if (!stage) {
-            renderContext->_scene->resetStage(LightStage::getName(), std::make_shared<LightStage>());
-        }
-    }
-}

libraries/render-utils/src/LightStage.h

@@ -15,8 +15,8 @@
 
 #include <gpu/Framebuffer.h>
 #include <graphics/Light.h>
-#include <render/Engine.h>
 #include <render/Stage.h>
+#include <render/StageSetup.h>
 
 class ViewFrustum;
 
@@ -185,14 +185,9 @@ protected:
 };
 using LightStagePointer = std::shared_ptr<LightStage>;
 
-
-class LightStageSetup {
+class LightStageSetup : public render::StageSetup<LightStage> {
 public:
     using JobModel = render::Job::Model<LightStageSetup>;
-
-    LightStageSetup();
-    void run(const render::RenderContextPointer& renderContext);
 };
 
-
 #endif

libraries/render-utils/src/TonemappingStage.cpp

@@ -7,15 +7,7 @@
 //  Distributed under the Apache License, Version 2.0.
 //  See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html
 //
+
 #include "TonemappingStage.h"
 
-#include <gpu/Context.h>
-
 std::string TonemappingStage::_name { "TONEMAPPING_STAGE" };
-
-void TonemappingStageSetup::run(const render::RenderContextPointer& renderContext) {
-    auto stage = renderContext->_scene->getStage(TonemappingStage::getName());
-    if (!stage) {
-        renderContext->_scene->resetStage(TonemappingStage::getName(), std::make_shared<TonemappingStage>());
-    }
-}

libraries/render-utils/src/TonemappingStage.h

@@ -12,22 +12,16 @@
 #define hifi_render_utils_TonemappingStage_h
 
 #include <graphics/Tonemapping.h>
-#include <render/Forward.h>
-#include <render/DrawTask.h>
 #include <render/Stage.h>
+#include <render/StageSetup.h>
 
 // Tonemapping stage to set up tonemapping-related rendering tasks
 class TonemappingStage : public render::PointerStage<graphics::Tonemapping, graphics::TonemappingPointer> {};
 using TonemappingStagePointer = std::shared_ptr<TonemappingStage>;
 
-class TonemappingStageSetup {
+class TonemappingStageSetup : public render::StageSetup<TonemappingStage> {
 public:
     using JobModel = render::Job::Model<TonemappingStageSetup>;
-
-    TonemappingStageSetup() {}
-    void run(const render::RenderContextPointer& renderContext);
-
-protected:
 };
 
 #endif

libraries/render/src/render/HighlightStage.cpp

@@ -1,6 +1,6 @@
 //
 //  HighlightStage.cpp
-
+//
 //  Created by Olivier Prat on 07/07/2017.
 //  Copyright 2017 High Fidelity, Inc.
 //  Copyright 2024 Overte e.V.
@@ -11,6 +11,8 @@
 
 #include "HighlightStage.h"
 
+#include "Engine.h"
+
 using namespace render;
 
 std::string HighlightStage::_name { "HIGHLIGHT_STAGE" };
@@ -20,7 +22,7 @@ HighlightStage::Index HighlightStage::addHighlight(const std::string& selectionN
     return addElement(outline);
 }
 
-Index HighlightStage::getHighlightIdBySelection(const std::string& selectionName) const {
+HighlightStage::Index HighlightStage::getHighlightIdBySelection(const std::string& selectionName) const {
     for (auto outlineId : _activeElementIDs) {
         const auto& outline = _elements.get(outlineId);
         if (outline._selectionName == selectionName) {

libraries/render/src/render/HighlightStage.h

@@ -12,9 +12,9 @@
 #ifndef hifi_render_utils_HighlightStage_h
 #define hifi_render_utils_HighlightStage_h
 
-#include "Stage.h"
 #include "Engine.h"
 #include "HighlightStyle.h"
+#include "Stage.h"
 
 namespace render {
 

libraries/render/src/render/HighlightStyle.h

@@ -17,6 +17,8 @@
 
 #include <string>
 
+#include <ViewFrustum.h>
+
 namespace render {
 
     // This holds the configuration for a particular outline style

libraries/render/src/render/Stage.h

@@ -108,7 +108,7 @@ namespace render {
         P getElement(Index id) const { return _elements.get(id); }
 
         Index findElement(const P& element) const {
-            auto found = _elementMap.find(haze);
+            auto found = _elementMap.find(element);
             if (found != _elementMap.end()) {
                 return INVALID_INDEX;
             } else {

libraries/render/src/render/StageSetup.h

@@ -0,0 +1,35 @@
+//
+//  StageSetup.h
+//  render/src/render
+//
+//  Created by HifiExperiments on 10/16/24
+//  Copyright 2024 Overte e.V.
+//
+//  Distributed under the Apache License, Version 2.0.
+//  See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html
+//
+
+#ifndef hifi_render_StageSetup_h
+#define hifi_render_StageSetup_h
+
+#include "Engine.h"
+
+namespace render {
+
+    template <typename T>
+    class StageSetup {
+    public:
+        StageSetup() {}
+
+        void run(const RenderContextPointer& renderContext) {
+            if (renderContext->_scene) {
+                auto stage = renderContext->_scene->getStage(T::getName());
+                if (!stage) {
+                    renderContext->_scene->resetStage(T::getName(), std::make_shared<T>());
+                }
+            }
+        }
+    };
+}
+
+#endif // hifi_render_StageSetup_h

libraries/render/src/render/TransitionStage.cpp

@@ -11,8 +11,6 @@
 
 #include "TransitionStage.h"
 
-#include <algorithm>
-
 using namespace render;
 
 std::string TransitionStage::_name { "TRANSITION_STAGE" };
@@ -24,10 +22,3 @@ TransitionStage::Index TransitionStage::addTransition(ItemID itemId, Transition:
     transition.boundItemId = boundId;
     return addElement(transition);
 }
-
-void TransitionStageSetup::run(const RenderContextPointer& renderContext) {
-    auto stage = renderContext->_scene->getStage(TransitionStage::getName());
-    if (!stage) {
-        renderContext->_scene->resetStage(TransitionStage::getName(), std::make_shared<TransitionStage>());
-    }
-}

libraries/render/src/render/TransitionStage.h

@@ -1,6 +1,6 @@
 //
 //  TransitionStage.h
-
+//
 //  Created by Olivier Prat on 07/07/2017.
 //  Copyright 2017 High Fidelity, Inc.
 //  Copyright 2024 Overte e.V.
@@ -13,7 +13,7 @@
 #define hifi_render_TransitionStage_h
 
 #include "Stage.h"
-#include "Engine.h"
+#include "StageSetup.h"
 #include "Transition.h"
 
 namespace render {
@@ -26,13 +26,11 @@ namespace render {
     };
     using TransitionStagePointer = std::shared_ptr<TransitionStage>;
 
-    class TransitionStageSetup {
+    class TransitionStageSetup : public StageSetup<TransitionStage> {
     public:
-        using JobModel = render::Job::Model<TransitionStageSetup>;
-
-        TransitionStageSetup() {}
-        void run(const RenderContextPointer& renderContext);
+        using JobModel = Job::Model<TransitionStageSetup>;
     };
+
 }
 
 #endif // hifi_render_TransitionStage_h