Add replay renderer flag to enable ambient occlusion. (#2223)

facebookresearch · Oct 2, 2023 · c5f38b3 · c5f38b3
1 parent 30f4cc7
commit c5f38b3
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/src/esp/bindings/SimBindings.cpp b/src/esp/bindings/SimBindings.cpp
@@ -381,6 +381,9 @@ void initSimBindings(py::module& m) {
       .def_readwrite("enable_frustum_culling",
                      &ReplayRendererConfiguration::enableFrustumCulling,
                      R"(Controls whether frustum culling is enabled.)")
+      .def_readwrite(
+          "enable_hbao", &ReplayRendererConfiguration::enableHBAO,
+          R"(Controls whether horizon-based ambient occlusion is enabled.)")
       .def_readwrite(
           "force_separate_semantic_scene_graph",
           &ReplayRendererConfiguration::forceSeparateSemanticSceneGraph,

diff --git a/src/esp/sim/AbstractReplayRenderer.h b/src/esp/sim/AbstractReplayRenderer.h
@@ -57,6 +57,8 @@ class ReplayRendererConfiguration {
 
   bool enableFrustumCulling = true;
 
+  bool enableHBAO = false;
+
   std::vector<std::shared_ptr<sensor::SensorSpec>> sensorSpecifications;
 
   ESP_SMART_POINTERS(ReplayRendererConfiguration)

diff --git a/src/esp/sim/ClassicReplayRenderer.cpp b/src/esp/sim/ClassicReplayRenderer.cpp
@@ -100,6 +100,10 @@ ClassicReplayRenderer::ClassicReplayRenderer(
     }
 
     gfx::Renderer::Flags flags;
+
+    if (config_.enableHBAO)
+      flags |= gfx::Renderer::Flag::HorizonBasedAmbientOcclusion;
+
 #ifdef ESP_BUILD_WITH_BACKGROUND_RENDERER
     if (context_)
       flags |= gfx::Renderer::Flag::BackgroundRenderer;