Add set_current_context

Author: nurpax

docs/index.html

@@ -302,6 +302,9 @@ <h4 id="functions">Functions</h4>
 <div class="api">
   <code><span class='keyword'>def</span> <span class='func_name'>get_current_context</span>() -> <span class='symbol'>Context</span></code>
 </div>
+<div class="api">
+  <code><span class='keyword'>def</span> <span class='func_name'>set_current_context</span>(<span class='arg_name'>arg</span><span class='special'>:</span> <span class='symbol'>Context</span><span class='special'>,</span> /) -> <span class='keyword'>None</span></code>
+</div>
 
 <h3 id="main">Main</h3>
 <h4 id="functions-1">Functions</h4>

gen/checklist.py

@@ -24,6 +24,8 @@
     "label_text_v",             # varargs not relevant in Python
     "bullet_text_v",            # varargs not relevant in Python
     "is_any_mouse_down",        # will obsolete
+    "get_style",    # wrapped in python, using internal funcs
+    "get_io",       # wrapped in python, using internal funcs
 }
 
 

src/slimgui/__init__.py

@@ -46,6 +46,11 @@ def create_context(shared_font_atlas: slimgui_ext.FontAtlas | None = None) -> Wr
 def get_current_context() -> WrappedContext | None:
     return _current_context
 
+def set_current_context(ctx: WrappedContext) -> None:
+    global _current_context
+    _current_context = ctx
+    slimgui_ext.set_current_context(ctx.context)
+
 def destroy_context(ctx: WrappedContext | None):
     global _current_context
     prev_ctx = get_current_context()

src/slimgui/slimgui_ext.pyi

@@ -3652,6 +3652,10 @@ def set_column_width(column_index: int, width: float) -> None:
     ...
 
 
+def set_current_context(arg: Context, /) -> None:
+    ...
+
+
 def set_cursor_pos(local_pos: tuple[float, float]) -> None:
     """[window-local] \""""
     ...

src/slimgui_ext.cpp

@@ -358,8 +358,9 @@ NB_MODULE(slimgui_ext, m) {
 
     m.def("create_context", &ImGui::CreateContext, "shared_font_atlas"_a = nullptr, nb::rv_policy::reference);
     m.def("get_current_context", &ImGui::GetCurrentContext, nb::rv_policy::reference);
+    m.def("set_current_context", &ImGui::SetCurrentContext, nb::rv_policy::reference);
     m.def("destroy_context", &ImGui::DestroyContext);
-    m.def("get_style", &ImGui::GetStyle, nb::rv_policy::reference);
+    m.def("get_style", &ImGui::GetStyle, nb::rv_policy::reference); // TODO docs should refer to slimgui/__init__.py declarations
     m.def("render", &ImGui::Render);
     m.def("new_frame", &ImGui::NewFrame);
     m.def("get_draw_data", &ImGui::GetDrawData, nb::rv_policy::reference);

tests/test_simple.py

@@ -24,7 +24,7 @@ def frame_scope(imgui_context, null_renderer):
 def test_current_context(imgui_context):
     assert imgui_context is not None
     assert imgui.get_current_context() is not None
-    assert imgui_context == imgui.get_current_context()
+    assert imgui_context is imgui.get_current_context()
 
 def test_style_access(imgui_context):
     styles = imgui.get_style()
@@ -37,6 +37,29 @@ def test_style_access(imgui_context):
     styles.window_padding = (2, 4)
     assert styles.window_padding == (2, 4)
 
+def test_set_current_context(imgui_context):
+    assert imgui_context is not None
+    assert imgui.get_current_context() is not None
+    assert imgui_context is imgui.get_current_context()
+
+    ctx2 = imgui.create_context()
+    assert ctx2 is imgui.get_current_context()
+    assert imgui_context is not imgui.get_current_context()
+    c = imgui.get_current_context()
+    assert c is not None
+    assert imgui_context.context is not c.context
+    imgui.set_current_context(imgui_context)
+    assert imgui_context is imgui.get_current_context()
+    c = imgui.get_current_context()
+    assert c is not None
+    assert imgui_context.context is c.context
+    imgui.set_current_context(ctx2)
+    imgui.destroy_context(None)
+    assert imgui.get_current_context() is None
+    imgui.set_current_context(imgui_context)
+    assert imgui_context is imgui.get_current_context()
+
+
 def test_begin_base_window(frame_scope):
     visible, _ = imgui.begin("Window")
     imgui.end()