add chroma.cpp Python bindings

Author: wjakob

include/nanogui/chroma.h

@@ -21,10 +21,10 @@
 
 NAMESPACE_BEGIN(nanogui)
 
-Matrix3f chroma_to_rec709_matrix(const std::array<Vector2f, 4>& chroma);
+extern NANOGUI_EXPORT Matrix3f chroma_to_rec709_matrix(const std::array<Vector2f, 4>& chroma);
 
-std::array<nanogui::Vector2f, 4> chroma_from_wp_primaries(int wp_primaries);
-std::string_view wp_primaries_to_string(int wp_primaties);
+extern NANOGUI_EXPORT std::array<nanogui::Vector2f, 4> chroma_from_wp_primaries(int wp_primaries);
+extern NANOGUI_EXPORT std::string_view wp_primaries_to_string(int wp_primaties);
 
 // Partial implementation of https://www.itu.int/rec/T-REC-H.273-202407-I/en
 NAMESPACE_BEGIN(ituth273)
@@ -44,10 +44,12 @@ enum class ColorPrimaries : uint8_t {
     Weird = 22, // The spec says "No corresponding industry specification identified"
 };
 
-std::string_view to_string(const ColorPrimaries primaries);
-std::array<nanogui::Vector2f, 4> chroma(const ColorPrimaries primaries);
+extern NANOGUI_EXPORT std::string_view to_string(const ColorPrimaries primaries);
+extern NANOGUI_EXPORT std::array<nanogui::Vector2f, 4> chroma(const ColorPrimaries primaries);
 
-ColorPrimaries from_wp_primaries(int wp_primaries);
+extern NANOGUI_EXPORT ColorPrimaries from_wp_primaries(int wp_primaries);
+
+extern NANOGUI_EXPORT ColorPrimaries from_screen(const Screen *screen);
 
 NAMESPACE_END(ituth273)
 

src/chroma.cpp

@@ -11,6 +11,8 @@
 
 #include <nanogui/chroma.h>
 #include <nanogui/vector.h>
+#include <nanogui/screen.h>
+#include <nanogui/opengl.h>
 
 #include <array>
 #include <stdexcept>
@@ -315,6 +317,10 @@ ColorPrimaries from_wp_primaries(int wp_primaries) {
     throw std::invalid_argument{"Unknown wp color primaries: " + std::to_string(wp_primaries)};
 }
 
+ColorPrimaries from_screen(const Screen *screen) {
+    return from_wp_primaries(glfwGetWindowPrimaries(screen->glfw_window()));
+}
+
 NAMESPACE_END(ituth273)
 
 NAMESPACE_END(nanogui)

src/python/CMakeLists.txt

@@ -32,6 +32,7 @@ nanobind_add_module(nanogui-python MODULE
   render.cpp
   quad.cpp
   vector.cpp
+  chroma.cpp
   python.h
   py_doc.h)
 

src/python/chroma.cpp

@@ -0,0 +1,32 @@
+#ifdef NANOGUI_PYTHON
+
+#include "python.h"
+#include <nanogui/chroma.h>
+#include <nanobind/stl/array.h>
+
+using namespace nanogui::ituth273;
+
+void register_chroma(nb::module_ &m_) {
+    auto m = m_.def_submodule("ituth273");
+
+    nb::enum_<ColorPrimaries>(m, "ColorPrimaries")
+        .value("BT709", ColorPrimaries::BT709)
+        .value("Unspecified", ColorPrimaries::Unspecified)
+        .value("BT470M", ColorPrimaries::BT470M)
+        .value("BT470BG", ColorPrimaries::BT470BG)
+        .value("SMTPE170M", ColorPrimaries::SMPTE170M)
+        .value("SMTP240M", ColorPrimaries::SMPTE240M)
+        .value("Film", ColorPrimaries::Film)
+        .value("BT2020", ColorPrimaries::BT2020)
+        .value("SMTPE428", ColorPrimaries::SMPTE428)
+        .value("SMTPE431", ColorPrimaries::SMPTE431)
+        .value("SMTPE432", ColorPrimaries::SMPTE432)
+        .value("Weird", ColorPrimaries::Weird);
+
+
+    m.def("chroma_to_rec709_matrix", &chroma_to_rec709_matrix);
+    m.def("chroma", &chroma);
+    m.def("from_screen", &from_screen);
+};
+
+#endif

src/python/main.cpp

@@ -55,6 +55,7 @@ extern void register_misc(nb::module_ &m);
 extern void register_nanovg(nb::module_ &m);
 extern void register_render(nb::module_ &m);
 extern void register_quad(nb::module_ &m);
+extern void register_chroma(nb::module_ &m);
 
 #if defined(__APPLE__) || defined(__linux__)
 static void (*sigint_handler_prev)(int) = nullptr;
@@ -160,6 +161,7 @@ NB_MODULE(nanogui_ext, m_) {
     register_nanovg(m);
     register_render(m);
     register_quad(m);
+    register_chroma(m);
 }
 
 #endif

src/python/py_doc.h

@@ -2004,6 +2004,8 @@ static const char *__doc_nanogui_Screen_mouse_button_callback_event = R"doc()doc
 
 static const char *__doc_nanogui_Screen_mouse_pos = R"doc(Return the last observed mouse position value)doc";
 
+static const char *__doc_nanogui_Screen_mouse_motion_event_f = R"doc(Like mouse_motion_event(), but also capture fractional motion)doc";
+
 static const char *__doc_nanogui_Screen_move_window_to_front = R"doc()doc";
 
 static const char *__doc_nanogui_Screen_nvg_context = R"doc(Return a pointer to the underlying NanoVG draw context)doc";

src/python/vector.cpp

@@ -163,6 +163,7 @@ void register_vector(nb::module_ &m) {
     register_vector_type<Vector3f>(m, "Vector3f");
     register_vector_type<Vector4f>(m, "Vector4f");
 
+    register_matrix_type<Matrix3f>(m, "Matrix3f");
     register_matrix_type<Matrix4f>(m, "Matrix4f")
         .def_static("translate", &Matrix4f::translate)
         .def_static(

src/python/widget.cpp

@@ -235,6 +235,8 @@ void register_widget(nb::module_ &m) {
         .def("pixel_format", &Screen::pixel_format, D(Screen, pixel_format))
         .def("component_format", &Screen::component_format, D(Screen, component_format))
         .def("nvg_flush", &Screen::nvg_flush, D(Screen, nvg_flush))
+        .def("mouse_motion_event_f", &Screen::mouse_motion_event_f, "p"_a, "rel"_a,
+             "button"_a, "modifiers"_a, D(Screen, mouse_motion_event_f))
 #if defined(NANOGUI_USE_METAL)
         .def("metal_layer", &Screen::metal_layer)
         .def("metal_texture", &Screen::metal_texture)

src/screen.cpp

@@ -291,7 +291,9 @@ Screen::Screen(const Vector2i &size, const std::string &caption, bool resizable,
     if (env_sdr_white != nullptr)
         m_display_sdr_white_level_override = display_sdr_white_level = std::stof(env_sdr_white);
 
-    m_wants_color_management = display_primaries != 1 || display_transfer_function != 10 || display_sdr_white_level != 80.0f;
+    m_wants_color_management = display_primaries != 1 ||
+                               display_transfer_function != 10 ||
+                               display_sdr_white_level != 80.0f;
 
     glfwGetFramebufferSize(m_glfw_window, &m_fbsize[0], &m_fbsize[1]);