Skip to content

Commit

Permalink
normalized hermite - 2
Browse files Browse the repository at this point in the history
  • Loading branch information
fobos123deimos committed Sep 10, 2024
1 parent aecfd00 commit 0c0c9d4
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions src/fast_wave/wavefunction.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@
@nb.jit(nopython=True, looplift=True, nogil=True, boundscheck=False, cache=True)
def create_normalized_hermite_coefficients_matrix(n_max: np.uint64) -> np.ndarray:
"""
Create a matrix of normalized coefficients for Hermite polynomials up to order `n_max`.
Create a matrix of coefficients for normalized Hermite polynomials up to order `n_max`.
Parameters
----------
Expand Down Expand Up @@ -93,7 +93,7 @@ def create_normalized_hermite_coefficients_matrix(n_max: np.uint64) -> np.ndarra
def wavefunction_smod(n: np.uint64, x:np.float64, more_fast:bool = True) -> np.float64:

"""
Compute the wavefunction to a real scalar x using a pre-computed matrix of Hermite polynomial normalized coefficients until n=60 and
Compute the wavefunction to a real scalar x using a pre-computed matrix of normalized Hermite polynomial coefficients until n=60 and
then use the adapted recursion relation for multidimensional M-mode wavefunction for higher orders.
Parameters
Expand Down Expand Up @@ -150,7 +150,7 @@ def wavefunction_smod(n: np.uint64, x:np.float64, more_fast:bool = True) -> np.f
def c_wavefunction_smod(n: np.uint64, x: np.complex128, more_fast:bool = True) -> np.complex128:

"""
Compute the wavefunction to a complex scalar x using a pre-computed matrix of Hermite polynomial normalized coefficients until n=60 and
Compute the wavefunction to a complex scalar x using a pre-computed matrix of normalized Hermite polynomial coefficients until n=60 and
then use the adapted recursion relation for multidimensional M-mode wavefunction for higher orders.
Parameters
Expand Down Expand Up @@ -207,7 +207,7 @@ def c_wavefunction_smod(n: np.uint64, x: np.complex128, more_fast:bool = True) -
def wavefunction_smmd(n: np.uint64, x: np.ndarray[np.float64], more_fast: bool = True) -> np.ndarray[np.float64]:

"""
Compute the wavefunction to a real vector x using a pre-computed matrix of Hermite polynomial normalized coefficients until n=60 and
Compute the wavefunction to a real vector x using a pre-computed matrix of normalized Hermite polynomial coefficients until n=60 and
then use the adapted recursion relation for multidimensional M-mode wavefunction for higher orders.
Parameters
Expand Down Expand Up @@ -269,7 +269,7 @@ def wavefunction_smmd(n: np.uint64, x: np.ndarray[np.float64], more_fast: bool =
def c_wavefunction_smmd(n: np.uint64, x: np.ndarray[np.complex128], more_fast: bool = True) -> np.ndarray[np.complex128]:

"""
Compute the wavefunction to a complex vector x using a pre-computed matrix of Hermite polynomial normalized coefficients until n=60 and
Compute the wavefunction to a complex vector x using a pre-computed matrix of normalized Hermite polynomial coefficients until n=60 and
then use the adapted recursion relation for multidimensional M-mode wavefunction for higher orders.
Parameters
Expand Down Expand Up @@ -567,7 +567,7 @@ def wavefunction(s_mode: bool = True, o_dimensional: bool = True, complex_bool:

"""
Main execution block to initialize the coefficient matrix and test the wavefunction computation.
This block checks for the existence of the precomputed Hermite polynomial coefficients matrix. If it doesn't exist,
This block checks for the existence of the precomputed normalized Hermite polynomial coefficients matrix. If it doesn't exist,
it computes the matrix and saves it for future use. Then, it performs a basic test to verify that the wavefunction
computation works as expected.
Expand Down

0 comments on commit 0c0c9d4

Please sign in to comment.