mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-04 11:21:07 +08:00
Gregory R. Lee
32 lines
1.1 KiB
Cython
32 lines
1.1 KiB
Cython
from interpolation cimport coord_map as _coord_map
|
|
from interpolation cimport get_pixel2d
|
|
import numpy as np
|
|
cimport numpy as cnp
|
|
|
|
|
|
def coord_map(Py_ssize_t dim, long coord, mode):
|
|
""" interpolation.coord_map python wrapper """
|
|
cdef char mode_c = ord(mode[0].upper())
|
|
return _coord_map(dim, coord, mode_c)
|
|
|
|
|
|
def extend_image(image, pad=10, mode='C', cval=0):
|
|
""" can be used to verify proper get_pixel2d behavior. """
|
|
cdef:
|
|
Py_ssize_t rows = image.shape[0]
|
|
Py_ssize_t cols = image.shape[1]
|
|
long ro, co
|
|
char mode_c = ord(mode[0].upper())
|
|
|
|
image = np.ascontiguousarray(image.astype(np.float64))
|
|
output_shape = np.asarray(image.shape) + 2*pad
|
|
image_out = np.zeros(output_shape, dtype=image.dtype)
|
|
for r in range(-pad, rows+pad):
|
|
for c in range(-pad, cols+pad):
|
|
ro = r + pad
|
|
co = c + pad
|
|
image_out[ro, co] = get_pixel2d(<double*> cnp.PyArray_DATA(image),
|
|
rows, cols, <long> r, <long> c,
|
|
mode_c, <double> cval)
|
|
return image_out
|