Add a pybind11 type caster for e_snap_mode

QuLogic · QuLogic · commit 78c8f04487b3 · 2024-02-23T04:43:34.000-05:00
diff --git a/src/_path_wrapper.cpp b/src/_path_wrapper.cpp
@@ -287,15 +287,11 @@ Py_convert_path_to_polygons(mpl::PathIterator path, agg::trans_affine trans,
 
 static py::tuple
 Py_cleanup_path(mpl::PathIterator path, agg::trans_affine trans, bool remove_nans,
-                agg::rect_d clip_rect, py::object snap_mode_obj, double stroke_width,
+                agg::rect_d clip_rect, e_snap_mode snap_mode, double stroke_width,
                 std::optional<bool> simplify, bool return_curves, py::object sketch_obj)
 {
-    e_snap_mode snap_mode;
     SketchParams sketch;
 
-    if (!convert_snap(snap_mode_obj.ptr(), &snap_mode)) {
-        throw py::error_already_set();
-    }
     if (!convert_sketch_params(sketch_obj.ptr(), &sketch)) {
         throw py::error_already_set();
     }
diff --git a/src/py_converters_11.h b/src/py_converters_11.h
@@ -10,6 +10,7 @@ namespace py = pybind11;
 
 #include "agg_basics.h"
 #include "agg_trans_affine.h"
+#include "path_converters.h"
 
 void convert_trans_affine(const py::object& transform, agg::trans_affine& affine);
 
@@ -85,6 +86,22 @@ namespace PYBIND11_NAMESPACE { namespace detail {
         }
     };
 
+    template <> struct type_caster<e_snap_mode> {
+    public:
+        PYBIND11_TYPE_CASTER(e_snap_mode, const_name("e_snap_mode"));
+
+        bool load(handle src, bool) {
+            if (src.is_none()) {
+                value = SNAP_AUTO;
+                return true;
+            }
+
+            value = src.cast<bool>() ? SNAP_TRUE : SNAP_FALSE;
+
+            return true;
+        }
+    };
+
 /* Remove all this macro magic after dropping NumPy usage and just include `py_adaptors.h`. */
 #ifdef MPL_PY_ADAPTORS_H
     template <> struct type_caster<mpl::PathIterator> {
