Overhaul & regenerate imgui bindings

.github/workflows/test_linux.yml

@@ -44,7 +44,7 @@ jobs:
         cmake -DCMAKE_BUILD_TYPE=Debug -DPYTHON_EXECUTABLE=$(python3 -c "import sys; print(sys.executable)") -DPOLYSCOPE_BACKEND_OPENGL3_EGL=ON ..
 
     - name: build
-      run: cd build && make
+      run: cd build && cmake --build . --parallel
 
     - name: run test backend mock
       run: python3 test/polyscope_test.py -v

.github/workflows/test_macos.yml

@@ -37,7 +37,7 @@ jobs:
       run: mkdir build && cd build && cmake -DCMAKE_BUILD_TYPE=Debug -DPOLYSCOPE_BACKEND_OPENGL3_GLFW=ON -DPOLYSCOPE_BACKEND_OPENGL_MOCK=ON ..
 
     - name: build
-      run: cd build && make
+      run: cd build && cmake --build . --parallel
 
     - name: run test
       run: python test/polyscope_test.py -v

.github/workflows/test_windows.yml

@@ -39,7 +39,7 @@ jobs:
       run: mkdir build && cd build && cmake -DCMAKE_BUILD_TYPE=Debug ..
 
     - name: build
-      run: cd build && cmake --build "."
+      run: cd build && cmake --build . --parallel
 
     - name: run test
       run: python test/polyscope_test.py -v

.gitignore

@@ -9,6 +9,9 @@ __pycache__/
 .cache
 *.egg-info
 
+# Stubs that get generated as part of the build process
+src/polyscope_bindings/*.pyi
+
 # Editor and OS things
 imgui.ini
 .polyscope.ini

CMakeLists.txt

@@ -42,16 +42,110 @@ nanobind_add_module(polyscope_bindings
   src/cpp/floating_quantities.cpp
   src/cpp/implicit_helpers.cpp
   src/cpp/managed_buffer.cpp
+
+  src/cpp/utils.h
 
-  src/cpp/imgui.cpp
-  src/cpp/implot.cpp
+  # ImGui related things
+  src/cpp/imgui/imgui.cpp
+
+  src/cpp/imgui/imgui_enums.cpp
+  src/cpp/imgui/imgui_structs.cpp
+  src/cpp/imgui/imgui_structs_fonts.cpp
+  src/cpp/imgui/imgui_macros.cpp
+  src/cpp/imgui/imgui_structs_io.cpp
+  src/cpp/imgui/imgui_structs_style.cpp
+  src/cpp/imgui/imgui_structs_drawlist.cpp
+
+  src/cpp/imgui/imgui_api_main.cpp
+  src/cpp/imgui/imgui_api_context_creation.cpp
+  src/cpp/imgui/imgui_api_demo_debug.cpp
+  src/cpp/imgui/imgui_api_styles.cpp
+  src/cpp/imgui/imgui_api_windows.cpp
+  src/cpp/imgui/imgui_api_child_windows.cpp
+  src/cpp/imgui/imgui_api_window_utilities.cpp
+  src/cpp/imgui/imgui_api_window_manipulation.cpp
+  src/cpp/imgui/imgui_api_scrolling.cpp
+  src/cpp/imgui/imgui_api_parameter_stacks.cpp
+  src/cpp/imgui/imgui_api_style_read.cpp
+  src/cpp/imgui/imgui_api_cursor_layout.cpp
+  src/cpp/imgui/imgui_api_id_stack.cpp
+  src/cpp/imgui/imgui_api_widgets_text.cpp
+  src/cpp/imgui/imgui_api_widgets_main.cpp
+  src/cpp/imgui/imgui_api_widgets_images.cpp
+  src/cpp/imgui/imgui_api_widgets_combo.cpp
+  src/cpp/imgui/imgui_api_widgets_drag.cpp
+  src/cpp/imgui/imgui_api_widgets_sliders.cpp
+  src/cpp/imgui/imgui_api_widgets_input.cpp
+  src/cpp/imgui/imgui_api_widgets_color.cpp
+  src/cpp/imgui/imgui_api_widgets_trees.cpp
+  src/cpp/imgui/imgui_api_widgets_selectables.cpp
+  src/cpp/imgui/imgui_api_widgets_listbox.cpp
+  src/cpp/imgui/imgui_api_data_plotting.cpp
+  src/cpp/imgui/imgui_api_menus.cpp
+  src/cpp/imgui/imgui_api_tooltips.cpp
+  src/cpp/imgui/imgui_api_popups.cpp
+  src/cpp/imgui/imgui_api_tables.cpp
+  src/cpp/imgui/imgui_api_columns_legacy.cpp
+  src/cpp/imgui/imgui_api_tab_bars.cpp
+  src/cpp/imgui/imgui_api_logging.cpp
+  src/cpp/imgui/imgui_api_drag_drop.cpp
+  src/cpp/imgui/imgui_api_disabling.cpp
+  src/cpp/imgui/imgui_api_clipping.cpp
+  src/cpp/imgui/imgui_api_focus_activation.cpp
+  src/cpp/imgui/imgui_api_overlapping_items.cpp
+  src/cpp/imgui/imgui_api_item_query.cpp
+  src/cpp/imgui/imgui_api_viewports.cpp
+  src/cpp/imgui/imgui_api_draw_lists.cpp
+  src/cpp/imgui/imgui_api_misc_utils.cpp
+  src/cpp/imgui/imgui_api_text_utils.cpp
+  src/cpp/imgui/imgui_api_color_utils.cpp
+  src/cpp/imgui/imgui_api_inputs_keyboard.cpp
+  src/cpp/imgui/imgui_api_inputs_mouse.cpp
+  src/cpp/imgui/imgui_api_clipboard.cpp
+  src/cpp/imgui/imgui_api_settings.cpp
+  src/cpp/imgui/imgui_api_debug.cpp
+  src/cpp/imgui/imgui_api_allocators.cpp
+
+  src/cpp/imgui/implot.cpp
+
+  src/cpp/imgui/imgui_utils.h
 
-  src/cpp/utils.h
-  src/cpp/imgui_utils.h
 )
 set_target_properties(polyscope_bindings PROPERTIES CXX_VISIBILITY_PRESET "default")
 
 target_include_directories(polyscope_bindings PUBLIC "${EIGEN3_INCLUDE_DIR}")
 target_link_libraries(polyscope_bindings PRIVATE polyscope)
 
+# Generate type stubs
+nanobind_add_stub(
+  polyscope_stub
+  MODULE polyscope_bindings
+
+  PYTHON_PATH $<TARGET_FILE_DIR:polyscope_bindings>
+  DEPENDS polyscope_bindings
+
+  MARKER_FILE py.typed
+
+  RECURSIVE
+
+  OUTPUT_PATH .
+  OUTPUT
+      __init__.pyi
+      imgui.pyi
+      implot.pyi
+)
+
+# Copy stub files into the appropriate place in the source directory after build
+set(STUB_FILES __init__.pyi imgui.pyi implot.pyi)
+foreach(STUB_FILE ${STUB_FILES})
+  add_custom_command(
+    TARGET polyscope_stub POST_BUILD
+    COMMAND ${CMAKE_COMMAND} -E copy_if_different
+        ${CMAKE_CURRENT_BINARY_DIR}/polyscope_bindings/${STUB_FILE}
+        ${CMAKE_CURRENT_SOURCE_DIR}/src/polyscope_bindings/${STUB_FILE}
+    VERBATIM
+  )
+endforeach()
+
+
 install(TARGETS polyscope_bindings LIBRARY DESTINATION .)

MANIFEST.in

@@ -8,4 +8,5 @@ recursive-include deps/pybind11 *.h CMakeLists.txt *.cmake
 recursive-include src *.cpp *.h
 recursive-include src/polyscope *.py *.pyi
 recursive-include src/polyscope_bindings *.py *.pyi
+global-include py.typed
 exclude dist