38 lines
1.1 KiB
Cython
38 lines
1.1 KiB
Cython
|
# cython: language_level=3
|
||
|
|
|
||
|
|
cimport numpy as np
|
||
|
|
from libcpp cimport bool
|
||
|
|
|
||
|
|
np.import_array()
|
||
|
|
|
||
|
|
|
||
|
|
cdef extern from "src/convolve.h":
|
||
|
|
void convolveNd_c(np.float64_t * const result,
|
||
|
|
const np.float64_t * const f,
|
||
|
|
const unsigned n_dim,
|
||
|
|
const size_t * const image_shape,
|
||
|
|
const np.float64_t * const g,
|
||
|
|
const size_t * const kernel_shape,
|
||
|
|
const bool nan_interpolate,
|
||
|
|
const bool embed_result_within_padded_region,
|
||
|
|
const unsigned n_threads) nogil
|
||
|
|
|
||
|
|
|
||
|
|
def _convolveNd_c(np.ndarray result,
|
||
|
|
np.ndarray array_to_convolve,
|
||
|
|
np.ndarray kernel,
|
||
|
|
bool nan_interpolate,
|
||
|
|
bool embed_result_within_padded_region,
|
||
|
|
int n_threads):
|
||
|
|
convolveNd_c(
|
||
|
|
<np.float64_t*>np.PyArray_DATA(result),
|
||
|
|
<np.float64_t*>np.PyArray_DATA(array_to_convolve),
|
||
|
|
array_to_convolve.ndim,
|
||
|
|
<size_t*>array_to_convolve.shape,
|
||
|
|
<np.float64_t*>np.PyArray_DATA(kernel),
|
||
|
|
<size_t*>kernel.shape,
|
||
|
|
nan_interpolate,
|
||
|
|
embed_result_within_padded_region,
|
||
|
|
n_threads,
|
||
|
|
)
|