43 lines
967 B
Cython
43 lines
967 B
Cython
|
|
||
|
|
import numpy as np
|
||
|
|
cimport numpy as np
|
||
|
|
from libcpp cimport bool
|
||
|
|
|
||
|
|
DEF ASCEE_FLOAT = "@ASCEE_FLOAT@"
|
||
|
|
|
||
|
|
IF ASCEE_FLOAT == "double":
|
||
|
|
ctypedef double d
|
||
|
|
ctypedef double complex c
|
||
|
|
NUMPY_FLOAT_TYPE = np.float64
|
||
|
|
NUMPY_COMPLEX_TYPE = np.complex128
|
||
|
|
ELSE:
|
||
|
|
ctypedef float d
|
||
|
|
ctypedef float complex c
|
||
|
|
NUMPY_FLOAT_TYPE = np.float32
|
||
|
|
NUMPY_COMPLEX_TYPE = np.complex128
|
||
|
|
|
||
|
|
ctypedef size_t us
|
||
|
|
|
||
|
|
cdef extern from "math.h":
|
||
|
|
ctypedef struct dmat:
|
||
|
|
d* data
|
||
|
|
us n_cols,n_rows
|
||
|
|
ctypedef struct cmat:
|
||
|
|
c* data
|
||
|
|
us n_cols,n_rows
|
||
|
|
|
||
|
|
# cdef np.ndarray empty(us nrows,us ncols,dtype):
|
||
|
|
cdef dmat dmat_from_array(d[::1,:] arr):
|
||
|
|
cdef dmat result
|
||
|
|
result.data = &arr[0,0]
|
||
|
|
result.n_rows = arr.shape[0]
|
||
|
|
result.n_cols = arr.shape[1]
|
||
|
|
return result
|
||
|
|
|
||
|
|
cdef cmat cmat_from_array(c[::1,:] arr):
|
||
|
|
cdef cmat result
|
||
|
|
result.data = &arr[0,0]
|
||
|
|
result.n_rows = arr.shape[0]
|
||
|
|
result.n_cols = arr.shape[1]
|
||
|
|
return result
|