v1.4.0 - cython module

fobos123deimos · Sep 26, 2024 · 9a1cad6 · 9a1cad6
1 parent 525c09c
commit 9a1cad6
Show file tree

Hide file tree

Showing 12 changed files with 745 additions and 437 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -3,4 +3,5 @@ mpmath==1.3.0
 numba==0.59.1
 numpy==1.26.4
 scipy==1.13.0
-sympy==1.12
+sympy==1.12
+cython==3.0.10
diff --git a/setup.cfg b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = fast_wave
-version = 1.3.11
+version = 1.4.0
 description = Package for the calculation of the time-independent wavefunction.
 author = Matheus Gomes Cordeiro
 author_email = matheusgomescord@gmail.com
@@ -17,6 +17,7 @@ numba==0.59.1
 numpy==1.26.4
 scipy==1.13.0
 sympy==1.12
+cython==3.0.10
 
 [test_requires]
 pytest = ^7.1.2
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
     long_description = fh.read()
 
 name = "fast_wave"
-version = "1.3.11"
+version = "1.4.0"
 description = "Package for the calculation of the time-independent wavefunction." 
 author_email = "matheusgomescord@gmail.com"
 url = "https://github.com/pikachu123deimos/fast-wave" 
@@ -17,6 +17,7 @@
     "numpy==1.26.4",
     "scipy==1.13.0",
     "sympy==1.12",
+    "cython==3.0.10",
 ]
 
 test_requires = [

diff --git a/src/fast_wave/setup_cython.py b/src/fast_wave/setup_cython.py
@@ -0,0 +1,8 @@
+from setuptools import setup
+from Cython.Build import cythonize
+import numpy as np
+
+setup(
+    ext_modules=cythonize("wavefunction_cython.pyx"),
+    include_dirs=[np.get_include()],
+)