Fix quaternion_get_wxyz

xioTechnologies · Sep 26, 2024 · fb13780 · fb13780
1 parent a719bcd
commit fb13780
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Python/Python-C-API/Quaternion.h b/Python/Python-C-API/Quaternion.h
@@ -40,7 +40,10 @@ static void quaternion_free(Quaternion *self) {
 
 static PyObject *quaternion_get_wxyz(Quaternion *self) {
     const npy_intp dims[] = {4};
-    return PyArray_SimpleNewFromData(1, dims, NPY_FLOAT, self->quaternion.array);
+    PyObject* array = PyArray_SimpleNewFromData(1, dims, NPY_FLOAT, self->quaternion.array);
+    Py_INCREF(self);
+    PyArray_SetBaseObject((PyArrayObject *) array, self);
+    return array;
 }
 
 static PyObject *quaternion_get_w(Quaternion *self) {