pickle support:

Author: SwayamInSync

quaddtype/numpy_quaddtype/src/dtype.c

@@ -1,4 +1,5 @@
 #include <Python.h>
+#include <structmember.h>
 #include <sleef.h>
 #include <sleefquad.h>
 #include <ctype.h>
@@ -439,13 +440,45 @@ QuadPrecDType_str(QuadPrecDTypeObject *self)
     return PyUnicode_FromFormat("QuadPrecDType(backend='%s')", backend_str);
 }
 
+
+static PyObject *
+quaddtype__reduce__(QuadPrecDTypeObject *self, PyObject *NPY_UNUSED(args))
+{
+    const char *backend_str = (self->backend == BACKEND_SLEEF) ? "sleef" : "longdouble";
+    
+    /* Return (type(self), (backend_str,)) 
+     * This will call type(self).__new__(type(self), backend_str) followed by __init__
+     */
+    PyObject *result = Py_BuildValue("O(s)", Py_TYPE(self), backend_str);
+    
+    return result;
+}
+
+static PyMethodDef QuadPrecDType_methods[] = {
+        {
+                "__reduce__",
+                (PyCFunction)quaddtype__reduce__,
+                METH_NOARGS,
+                "Reduction method for a QuadPrecDType object",
+        },
+        {NULL, NULL, 0, NULL},
+};
+
+static PyMemberDef QuadPrecDType_members[] = {
+        {"backend", T_INT, offsetof(QuadPrecDTypeObject, backend), READONLY,
+         "The backend used for quad precision (0=sleef, 1=longdouble)"},
+        {NULL, 0, 0, 0, NULL},
+};
+
 PyArray_DTypeMeta QuadPrecDType = {
         {{
                 PyVarObject_HEAD_INIT(NULL, 0).tp_name = "numpy_quaddtype.QuadPrecDType",
                 .tp_basicsize = sizeof(QuadPrecDTypeObject),
                 .tp_new = QuadPrecDType_new,
                 .tp_repr = (reprfunc)QuadPrecDType_repr,
                 .tp_str = (reprfunc)QuadPrecDType_str,
+                .tp_methods = QuadPrecDType_methods,
+                .tp_members = QuadPrecDType_members,
         }},
 };