[gui] Fix IMGUI when GGUI is running in headless mode (taichi-dev#3357)

sjwsl · Nov 3, 2021 · be73302 · be73302
1 parent 1bd43bc
commit be73302
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/taichi/ui/backends/vulkan/gui.cpp b/taichi/ui/backends/vulkan/gui.cpp
@@ -99,6 +99,13 @@ void Gui::prepare_for_next_frame() {
   ImGui_ImplVulkan_NewFrame();
   if (app_context_->config.show_window) {
     ImGui_ImplGlfw_NewFrame();
+  } else {
+    // io.DisplaySize is set during ImGui_ImplGlfw_NewFrame()
+    // but since we're headless, we do it explicitly here
+    auto w = app_context_->config.width;
+    auto h = app_context_->config.height;
+    ImGuiIO &io = ImGui::GetIO();
+    io.DisplaySize = ImVec2((float)w, (float)h);
   }
   ImGui::NewFrame();
   is_empty_ = true;