Use unique symbol for ufunc API

Author: ngoldbaum

asciidtype/asciidtype/src/asciidtype_main.c

@@ -1,10 +1,12 @@
 #include <Python.h>
 
 #define PY_ARRAY_UNIQUE_SYMBOL asciidtype_ARRAY_API
+#define PY_UFUNC_UNIQUE_SYMBOL asciidtype_UFUNC_API
 #define NPY_NO_DEPRECATED_API NPY_2_0_API_VERSION
 #define NPY_TARGET_VERSION NPY_2_0_API_VERSION
 #include "numpy/ndarraytypes.h"
 #include "numpy/arrayobject.h"
+#include "numpy/ufuncobject.h"
 #include "numpy/dtype_api.h"
 
 #include "dtype.h"
@@ -21,6 +23,7 @@ PyMODINIT_FUNC
 PyInit__asciidtype_main(void)
 {
     import_array();
+    import_umath();
 
     PyObject *m = PyModule_Create(&moduledef);
     if (m == NULL) {
@@ -47,14 +50,10 @@ PyInit__asciidtype_main(void)
         goto error;
     }
 
-    PyObject *numpy = init_ufuncs();
-
-    if (numpy == NULL) {
+    if (init_ufuncs() == -1) {
         goto error;
     }
 
-    Py_DECREF(numpy);
-
     return m;
 
 error:

asciidtype/asciidtype/src/casts.c

@@ -1,9 +1,11 @@
 #include <Python.h>
 
 #define PY_ARRAY_UNIQUE_SYMBOL asciidtype_ARRAY_API
+#define PY_UFUNC_UNIQUE_SYMBOL asciidtype_UFUNC_API
 #define NPY_NO_DEPRECATED_API NPY_2_0_API_VERSION
 #define NPY_TARGET_VERSION NPY_2_0_API_VERSION
 #define NO_IMPORT_ARRAY
+#define NO_IMPORT_UFUNC
 #include "numpy/arrayobject.h"
 #include "numpy/dtype_api.h"
 #include "numpy/ndarraytypes.h"

asciidtype/asciidtype/src/dtype.c

@@ -4,9 +4,11 @@
 // clang-format on
 
 #define PY_ARRAY_UNIQUE_SYMBOL asciidtype_ARRAY_API
+#define PY_UFUNC_UNIQUE_SYMBOL asciidtype_UFUNC_API
 #define NPY_NO_DEPRECATED_API NPY_2_0_API_VERSION
 #define NPY_TARGET_VERSION NPY_2_0_API_VERSION
 #define NO_IMPORT_ARRAY
+#define NO_IMPORT_UFUNC
 #include "numpy/ndarraytypes.h"
 #include "numpy/arrayobject.h"
 #include "numpy/ufuncobject.h"

asciidtype/asciidtype/src/umath.c

@@ -1,9 +1,11 @@
 #include <Python.h>
 
 #define PY_ARRAY_UNIQUE_SYMBOL asciidtype_ARRAY_API
+#define PY_UFUNC_UNIQUE_SYMBOL asciidtype_UFUNC_API
 #define NPY_NO_DEPRECATED_API NPY_2_0_API_VERSION
 #define NPY_TARGET_VERSION NPY_2_0_API_VERSION
 #define NO_IMPORT_ARRAY
+#define NO_IMPORT_UFUNC
 #include "numpy/ndarraytypes.h"
 #include "numpy/arrayobject.h"
 #include "numpy/ufuncobject.h"
@@ -219,14 +221,12 @@ init_equal_ufunc(PyObject *numpy)
     return 0;
 }
 
-PyObject *
+int
 init_ufuncs(void)
 {
-    import_umath();
-  
     PyObject *numpy = PyImport_ImportModule("numpy");
     if (numpy == NULL) {
-        return NULL;
+        return -1;
     }
 
     if (init_add_ufunc(numpy) < 0) {
@@ -237,9 +237,9 @@ init_ufuncs(void)
         goto error;
     }
 
-    return numpy;
+    return 0;
 
 error:
     Py_DECREF(numpy);
-    return NULL;
+    return -1;
 }

asciidtype/asciidtype/src/umath.h

@@ -1,7 +1,7 @@
 #ifndef _NPY_UFUNC_H
 #define _NPY_UFUNC_H
 
-PyObject *
+int
 init_ufuncs(void);
 
 #endif /*_NPY_UFUNC_H */

metadatadtype/metadatadtype/src/casts.c

@@ -1,9 +1,11 @@
 #include <Python.h>
 
 #define PY_ARRAY_UNIQUE_SYMBOL metadatadtype_ARRAY_API
+#define PY_UFUNC_UNIQUE_SYMBOL metadatadtype_UFUNC_API
 #define NPY_NO_DEPRECATED_API NPY_2_0_API_VERSION
 #define NPY_TARGET_VERSION NPY_2_0_API_VERSION
 #define NO_IMPORT_ARRAY
+#define NO_IMPORT_UFUNC
 #include "numpy/arrayobject.h"
 #include "numpy/dtype_api.h"
 #include "numpy/ndarraytypes.h"

metadatadtype/metadatadtype/src/dtype.c

@@ -1,7 +1,14 @@
+// clang-format off
+#include <Python.h>
+#include "structmember.h"
+// clang-format on
+
 #define PY_ARRAY_UNIQUE_SYMBOL metadatadtype_ARRAY_API
+#define PY_UFUNC_UNIQUE_SYMBOL metadatadtype_UFUNC_API
 #define NPY_NO_DEPRECATED_API NPY_2_0_API_VERSION
 #define NPY_TARGET_VERSION NPY_2_0_API_VERSION
 #define NO_IMPORT_ARRAY
+#define NO_IMPORT_UFUNC
 #include "numpy/arrayobject.h"
 #include "numpy/dtype_api.h"
 #include "numpy/ndarraytypes.h"

metadatadtype/metadatadtype/src/dtype.h

@@ -1,11 +1,6 @@
 #ifndef _NPY_DTYPE_H
 #define _NPY_DTYPE_H
 
-// clang-format off
-#include <Python.h>
-#include "structmember.h"
-// clang-format on
-
 typedef struct {
     PyArray_Descr base;
     PyObject *metadata;

metadatadtype/metadatadtype/src/metadatadtype_main.c

@@ -1,9 +1,11 @@
 #include <Python.h>
 
 #define PY_ARRAY_UNIQUE_SYMBOL metadatadtype_ARRAY_API
+#define PY_UFUNC_UNIQUE_SYMBOL metadatadtype_UFUNC_API
 #define NPY_NO_DEPRECATED_API NPY_2_0_API_VERSION
 #define NPY_TARGET_VERSION NPY_2_0_API_VERSION
 #include "numpy/arrayobject.h"
+#include "numpy/ufuncobject.h"
 #include "numpy/dtype_api.h"
 
 #include "umath.h"
@@ -20,6 +22,7 @@ PyMODINIT_FUNC
 PyInit__metadatadtype_main(void)
 {
     import_array();
+    import_umath();
 
     PyObject *m = PyModule_Create(&moduledef);
     if (m == NULL) {
@@ -47,7 +50,7 @@ PyInit__metadatadtype_main(void)
         goto error;
     }
 
-    if (init_ufuncs(m) == NULL) {
+    if (init_ufuncs() == -1) {
         goto error;
     }
 

metadatadtype/metadatadtype/src/umath.c

@@ -1,9 +1,11 @@
 #include <Python.h>
 
 #define PY_ARRAY_UNIQUE_SYMBOL metadatadtype_ARRAY_API
+#define PY_UFUNC_UNIQUE_SYMBOL metadatadtype_UFUNC_API
 #define NPY_NO_DEPRECATED_API NPY_2_0_API_VERSION
 #define NPY_TARGET_VERSION NPY_2_0_API_VERSION
 #define NO_IMPORT_ARRAY
+#define NO_IMPORT_UFUNC
 #include "numpy/ndarraytypes.h"
 #include "numpy/arrayobject.h"
 #include "numpy/ufuncobject.h"
@@ -103,11 +105,9 @@ add_wrapping_loop(const char *ufunc_name, PyArray_DTypeMeta **dtypes,
     return res;
 }
 
-PyObject *
-init_ufuncs(PyObject *module)
+int
+init_ufuncs(void)
 {
-    import_umath();
-
     PyArray_DTypeMeta *binary_orig_dtypes[3] = {&MetadataDType, &MetadataDType,
                                                 &MetadataDType};
     PyArray_DTypeMeta *binary_wrapped_dtypes[3] = {
@@ -126,9 +126,8 @@ init_ufuncs(PyObject *module)
         goto error;
     }
 
-    return module;
+    return 0;
 error:
 
-    Py_DECREF(module);
-    return NULL;
+    return -1;
 }