Add more metrics

shg8 · Feb 29, 2024 · 3371de7 · 3371de7
1 parent cc3c39f
commit 3371de7
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/3dgs/GUIManager.cpp b/3dgs/GUIManager.cpp
@@ -3,9 +3,11 @@
 #include <iostream>
 
 static std::shared_ptr<std::unordered_map<std::string, ScrollingBuffer>> metricsMap;
+static std::shared_ptr<std::unordered_map<std::string, float>> textMetricsMap;
 
 GUIManager::GUIManager() {
     metricsMap = std::make_shared<std::unordered_map<std::string, ScrollingBuffer>>();
+    textMetricsMap = std::make_shared<std::unordered_map<std::string, float>>();
 }
 
 void GUIManager::init() {
@@ -45,6 +47,13 @@ void GUIManager::buildGui() {
 
     bool popen = true;
     ImGui::SetNextWindowPos(ImVec2(10, 270), ImGuiCond_FirstUseEver);
+    ImGui::Begin("Metrics", &popen, ImGuiWindowFlags_AlwaysAutoResize);
+    for (auto& [name, value]: *textMetricsMap) {
+        ImGui::Text("%s: %.2f", name.c_str(), value);
+    }
+    ImGui::End();
+
+    ImGui::SetNextWindowPos(ImVec2(10, 310), ImGuiCond_FirstUseEver);
     ImGui::Begin("Controls", &popen, ImGuiWindowFlags_AlwaysAutoResize);
     ImGui::Text("WASD: move");
     ImGui::Text("Space: up");
@@ -59,6 +68,14 @@ void GUIManager::buildGui() {
     }
 }
 
+void GUIManager::pushTextMetric(const std::string& name, float value) {
+    if (!textMetricsMap->contains(name)) {
+        textMetricsMap->insert({name, value});
+    } else {
+        textMetricsMap->at(name) = value;
+    }
+}
+
 void GUIManager::pushMetric(const std::string& name, float value) {
     int maxSize = 600;
     if (!metricsMap->contains(name)) {

diff --git a/3dgs/GUIManager.h b/3dgs/GUIManager.h
@@ -41,6 +41,8 @@ class GUIManager {
 
     void buildGui();
 
+    static void pushTextMetric(const std::string& name, float value);
+
     static void pushMetric(const std::string& name, float value);
 
     static void pushMetric(const std::unordered_map<std::string, float>& name);

diff --git a/3dgs/Renderer.cpp b/3dgs/Renderer.cpp
@@ -526,6 +526,7 @@ bool Renderer::recordRenderCommandBuffer(uint32_t currentFrame) {
     }
 
     uint32_t numInstances = totalSumBufferHost->readOne<uint32_t>();
+    guiManager.pushTextMetric("instances", numInstances);
     if (numInstances > scene->getNumVertices() * sortBufferSizeMultiplier) {
         auto old = sortBufferSizeMultiplier;
         while (numInstances > scene->getNumVertices() * sortBufferSizeMultiplier) {