Add a pybind11 type caster for agg::rgba

QuLogic · QuLogic · commit a47e26bd8583 · 2024-09-13T20:06:43.000-04:00
diff --git a/src/py_converters_11.h b/src/py_converters_11.h
@@ -9,6 +9,7 @@
 namespace py = pybind11;
 
 #include "agg_basics.h"
+#include "agg_color_rgba.h"
 #include "agg_trans_affine.h"
 #include "path_converters.h"
 
@@ -58,6 +59,36 @@ namespace PYBIND11_NAMESPACE { namespace detail {
         }
     };
 
+    template <> struct type_caster<agg::rgba> {
+    public:
+        PYBIND11_TYPE_CASTER(agg::rgba, const_name("rgba"));
+
+        bool load(handle src, bool) {
+            if (src.is_none()) {
+                value.r = 0.0;
+                value.g = 0.0;
+                value.b = 0.0;
+                value.a = 0.0;
+            } else {
+                auto rgbatuple = src.cast<py::tuple>();
+                value.r = rgbatuple[0].cast<double>();
+                value.g = rgbatuple[1].cast<double>();
+                value.b = rgbatuple[2].cast<double>();
+                switch (rgbatuple.size()) {
+                case 4:
+                    value.a = rgbatuple[3].cast<double>();
+                    break;
+                case 3:
+                    value.a = 1.0;
+                    break;
+                default:
+                    throw py::value_error("RGBA value must be 3- or 4-tuple");
+                }
+            }
+            return true;
+        }
+    };
+
     template <> struct type_caster<agg::trans_affine> {
     public:
         PYBIND11_TYPE_CASTER(agg::trans_affine, const_name("trans_affine"));
