Add AOV visualization to offline render settings

Author: favreau

tsd/apps/tools/tsdRender.cpp

@@ -6,6 +6,7 @@
 #include <tsd/core/scene/Scene.hpp>
 // tsd_rendering
 #include <tsd/rendering/pipeline/RenderPipeline.h>
+#include <tsd/rendering/pipeline/passes/VisualizeAOVPass.h>
 #include <tsd/rendering/index/RenderIndexAllLayers.hpp>
 #include <tsd/rendering/view/ManipulatorToAnari.hpp>
 // tsd_app
@@ -253,6 +254,22 @@ static void setupRenderPipeline()
   arp->setCamera(g_camera);
   arp->setRunAsync(false);
 
+  // Add AOV visualization pass if enabled
+  if (g_core->offline.aov.aovType != tsd::rendering::AOVType::NONE) {
+    auto *aovPass =
+        g_renderPipeline->emplace_back<tsd::rendering::VisualizeAOVPass>();
+    aovPass->setAOVType(g_core->offline.aov.aovType);
+    aovPass->setDepthRange(
+        g_core->offline.aov.depthMin, g_core->offline.aov.depthMax);
+
+    // Enable necessary frame channels
+    if (g_core->offline.aov.aovType == tsd::rendering::AOVType::ALBEDO) {
+      arp->setEnableAlbedo(true);
+    } else if (g_core->offline.aov.aovType == tsd::rendering::AOVType::NORMAL) {
+      arp->setEnableNormals(true);
+    }
+  }
+
   anari::release(g_device, r);
 
   g_timer.end();

tsd/src/tsd/app/Core.cpp

@@ -750,6 +750,11 @@ void OfflineRenderSequenceConfig::saveSettings(tsd::core::DataNode &root)
   auto &outputRoot = root["output"];
   outputRoot["outputDirectory"] = output.outputDirectory;
   outputRoot["filePrefix"] = output.filePrefix;
+
+  auto &aovRoot = root["aov"];
+  aovRoot["aovType"] = static_cast<int>(aov.aovType);
+  aovRoot["depthMin"] = aov.depthMin;
+  aovRoot["depthMax"] = aov.depthMax;
 }
 
 void OfflineRenderSequenceConfig::loadSettings(tsd::core::DataNode &root)
@@ -786,6 +791,13 @@ void OfflineRenderSequenceConfig::loadSettings(tsd::core::DataNode &root)
   auto &outputRoot = root["output"];
   outputRoot["outputDirectory"].getValue(ANARI_STRING, &output.outputDirectory);
   outputRoot["filePrefix"].getValue(ANARI_STRING, &output.filePrefix);
+
+  auto &aovRoot = root["aov"];
+  int aovTypeInt = static_cast<int>(aov.aovType);
+  aovRoot["aovType"].getValue(ANARI_INT32, &aovTypeInt);
+  aov.aovType = static_cast<tsd::rendering::AOVType>(aovTypeInt);
+  aovRoot["depthMin"].getValue(ANARI_FLOAT32, &aov.depthMin);
+  aovRoot["depthMax"].getValue(ANARI_FLOAT32, &aov.depthMax);
 }
 
 } // namespace tsd::app

tsd/src/tsd/app/Core.h

@@ -8,6 +8,7 @@
 #include "tsd/core/scene/Scene.hpp"
 // tsd_rendering
 #include "tsd/rendering/index/RenderIndex.hpp"
+#include "tsd/rendering/pipeline/passes/VisualizeAOVPass.h"
 #include "tsd/rendering/view/Manipulator.hpp"
 // std
 #include <map>
@@ -196,6 +197,13 @@ struct OfflineRenderSequenceConfig
     std::string filePrefix{"frame_"};
   } output;
 
+  struct AOVSettings
+  {
+    tsd::rendering::AOVType aovType{tsd::rendering::AOVType::NONE};
+    float depthMin{0.f};
+    float depthMax{1.f};
+  } aov;
+
   void saveSettings(tsd::core::DataNode &root);
   void loadSettings(tsd::core::DataNode &root);
 };

tsd/src/tsd/app/renderAnimationSequence.cpp

@@ -9,6 +9,7 @@
 // tsd_rendering
 #include "tsd/rendering/index/RenderIndex.hpp"
 #include "tsd/rendering/pipeline/RenderPipeline.h"
+#include "tsd/rendering/pipeline/passes/VisualizeAOVPass.h"
 // std
 #include <filesystem>
 #include <iomanip>
@@ -95,6 +96,21 @@ static OfflineRenderRig setupRig(tsd::app::Core &core)
   rig.anariPass->setRenderer(r);
   rig.anariPass->setCamera(c);
 
+  // Add AOV visualization pass if enabled
+  if (config.aov.aovType != tsd::rendering::AOVType::NONE) {
+    auto *aovPass =
+        rig.pipeline->emplace_back<tsd::rendering::VisualizeAOVPass>();
+    aovPass->setAOVType(config.aov.aovType);
+    aovPass->setDepthRange(config.aov.depthMin, config.aov.depthMax);
+
+    // Enable necessary frame channels
+    if (config.aov.aovType == tsd::rendering::AOVType::ALBEDO) {
+      rig.anariPass->setEnableAlbedo(true);
+    } else if (config.aov.aovType == tsd::rendering::AOVType::NORMAL) {
+      rig.anariPass->setEnableNormals(true);
+    }
+  }
+
   rig.saveToFilePass->setEnabled(true);
   rig.saveToFilePass->setSingleShotMode(false);
 

tsd/src/tsd/ui/imgui/modals/AppSettingsDialog.cpp

@@ -266,6 +266,31 @@ void AppSettingsDialog::buildUI_offlineRenderSettings()
     ImGui::Unindent(tsd::ui::INDENT_AMOUNT);
   }
 
+  // AOV Visualization //
+
+  ImGui::Separator();
+  ImGui::Text("==== AOV Visualization ====");
+
+  const char *aovItems[] = {"none", "depth", "albedo", "normal"};
+  int aovIdx = static_cast<int>(core->offline.aov.aovType);
+  if (ImGui::Combo("AOV type", &aovIdx, aovItems, IM_ARRAYSIZE(aovItems))) {
+    core->offline.aov.aovType = static_cast<tsd::rendering::AOVType>(aovIdx);
+  }
+
+  ImGui::BeginDisabled(
+      core->offline.aov.aovType != tsd::rendering::AOVType::DEPTH);
+  ImGui::DragFloat("depth min",
+      &core->offline.aov.depthMin,
+      0.1f,
+      0.f,
+      core->offline.aov.depthMax);
+  ImGui::DragFloat("depth max",
+      &core->offline.aov.depthMax,
+      0.1f,
+      core->offline.aov.depthMin,
+      1e20f);
+  ImGui::EndDisabled();
+
   ImGui::Unindent(tsd::ui::INDENT_AMOUNT);
 }