Add a pybind11 type caster for SketchParams

QuLogic · QuLogic · commit 13aa334d1b4e · 2023-10-21T04:53:01.000-04:00
diff --git a/src/_path_wrapper.cpp b/src/_path_wrapper.cpp
@@ -302,14 +302,8 @@ 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, e_snap_mode snap_mode, double stroke_width,
-                std::optional<bool> simplify, bool return_curves, py::object sketch_obj)
+                std::optional<bool> simplify, bool return_curves, SketchParams sketch)
 {
-    SketchParams sketch;
-
-    if (!convert_sketch_params(sketch_obj.ptr(), &sketch)) {
-        throw py::error_already_set();
-    }
-
     if (!simplify.has_value()) {
         simplify = path.should_simplify();
     }
@@ -367,18 +361,13 @@ postfix : bool
 static py::object
 Py_convert_to_string(mpl::PathIterator path, agg::trans_affine trans,
                      agg::rect_d cliprect, std::optional<bool> simplify,
-                     py::object sketch_obj, int precision,
+                     SketchParams sketch, int precision,
                      std::array<std::string, 5> codes_obj, bool postfix)
 {
-    SketchParams sketch;
     char *codes[5];
     std::string buffer;
     bool status;
 
-    if (!convert_sketch_params(sketch_obj.ptr(), &sketch)) {
-        throw py::error_already_set();
-    }
-
     for(auto i = 0; i < 5; ++i) {
         codes[i] = const_cast<char *>(codes_obj[i].c_str());
     }
diff --git a/src/py_converters_11.h b/src/py_converters_11.h
@@ -18,6 +18,7 @@ namespace py = pybind11;
 #define NO_IMPORT_ARRAY
 #endif
 #include "py_adaptors.h"
+#include "_backend_agg_basic_types.h"
 #ifdef KEEP_NO_IMPORT_ARRAY
 #undef _KEEP_NO_IMPORT_ARRAY
 #else
@@ -136,6 +137,25 @@ namespace PYBIND11_NAMESPACE { namespace detail {
             return true;
         }
     };
+
+    template <> struct type_caster<SketchParams> {
+    public:
+        PYBIND11_TYPE_CASTER(SketchParams, const_name("SketchParams"));
+
+        bool load(handle src, bool) {
+            if (src.is_none()) {
+                value.scale = 0.0;
+                return true;
+            }
+
+            auto params = src.cast<std::tuple<double, double, double>>();
+            value.scale = std::get<0>(params);
+            value.length = std::get<1>(params);
+            value.randomness = std::get<2>(params);
+
+            return true;
+        }
+    };
 }} // namespace PYBIND11_NAMESPACE::detail
 
 #endif /* MPL_PY_CONVERTERS_11_H */
