Add ImGui context sharing for UI rendering in Python MuJoCo Studio

PiperOrigin-RevId: 930507935 Change-Id: I7615931f9a71793b049e8b6b442be6a2d726f0b2
2026-06-11 07:09:44 -07:00
parent 684eccc146
commit 41ac783d3a
3 changed files with 19 additions and 1 deletions
@@ -95,6 +95,13 @@ PYBIND11_MODULE(ux, m) {
      },
      py::arg("theme"), "Set up Dear ImGui visual theme.");

+  m.def(
+      "set_imgui_context",
+      [](intptr_t ptr) {
+        ImGui::SetCurrentContext(reinterpret_cast<ImGuiContext*>(ptr));
+      },
+      py::arg("ptr"), "Set ImGui context pointer.");
+
  m.def(
      "configure_docking_layout",
      []() {