Initial Work on Cython Code

2022-12-06 21:15:09 -03:00
parent be8f3fba42
commit 4b863c0ad8
25 changed files with 758 additions and 327 deletions
--- a/setup_cython.py
+++ b/setup_cython.py
@@ -0,0 +1,29 @@
+import numpy
+from Cython.Build import cythonize
+from setuptools import Extension, setup
+
+ext_modules = [
+    Extension(
+        "yoshi_seals.shared.array",
+        [
+            "yoshi_seals/shared/array.pyx",
+        ],
+        extra_compile_args=["-O3", "-fopenmp"],
+        extra_link_args=['-fopenmp'],
+        define_macros=[("NPY_NO_DEPRECATED_API", "NPY_1_7_API_VERSION")],
+    ),
+    Extension(
+        "yoshi_seals.process.process",
+        [
+            "yoshi_seals/process/process.pyx",
+        ],
+        extra_compile_args=["-O3", "-fopenmp"],
+        extra_link_args=['-fopenmp'],
+        define_macros=[("NPY_NO_DEPRECATED_API", "NPY_1_7_API_VERSION")]
+    ),
+]
+
+setup(
+    ext_modules=cythonize(ext_modules),
+    include_dirs=[numpy.get_include()]
+)