MAINT: PEP8 fixes and _interpolation_test.pyx -> interpolation.pyx rename

This commit is contained in:
Gregory R. Lee
2015-07-09 13:02:37 -04:00
parent fb0b258deb
commit cef22e4234
8 changed files with 69 additions and 47 deletions
+1 -1
View File
@@ -34,4 +34,4 @@ for n, mode in enumerate(modes):
plt.tight_layout()
plt.show()
plt.show()
-31
View File
@@ -1,31 +0,0 @@
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
+56
View File
@@ -0,0 +1,56 @@
from interpolation cimport coord_map, get_pixel2d
import numpy as np
cimport numpy as cnp
def coord_map_py(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='constant', cval=0):
""" Pad a 2D image by ``pad`` pixels on each side.
Parameters
----------
image : ndarray
Input image.
pad : int, optional
The number of pixels to pad around the border
mode : {'constant', 'nearest', 'reflect', 'mirror', 'wrap'}, optional
Points outside the boundaries of the input are filled according
to the given mode.
cval : float, optional
Used in conjunction with mode 'constant', the value outside
the image boundaries.
Returns
-------
extended : ndarray
The extended version of the input image.
Note
----
For image padding, ``skimage.util.pad`` should be used instead. This
function is intended only for testing get_pixel2d and demonstrating the
coordinate mapping modes implemented in ``coord_map``.
"""
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
extended = 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
extended[ro, co] = get_pixel2d(<double*> cnp.PyArray_DATA(image),
rows, cols, <long> r, <long> c,
mode_c, <double> cval)
return extended
+2 -2
View File
@@ -15,12 +15,12 @@ def configuration(parent_package='', top_path=None):
cython(['geometry.pyx'], working_path=base_path)
cython(['transform.pyx'], working_path=base_path)
cython(['_interpolation_test.pyx'], working_path=base_path)
cython(['interpolation.pyx'], working_path=base_path)
config.add_extension('geometry', sources=['geometry.c'])
config.add_extension('transform', sources=['transform.c'],
include_dirs=[get_numpy_include_dirs()])
config.add_extension('_interpolation_test', sources=['_interpolation_test.c'])
config.add_extension('interpolation', sources=['interpolation.c'])
return config
+6 -7
View File
@@ -1,24 +1,23 @@
from skimage._shared._interpolation_test import coord_map
from skimage._shared.interpolation import coord_map_py
from numpy.testing import assert_array_equal
def test_coord_map():
reflect = [coord_map(4, n, 'R') for n in range(-6, 6)]
reflect = [coord_map_py(4, n, 'R') for n in range(-6, 6)]
expected_reflect = [2, 3, 3, 2, 1, 0, 0, 1, 2, 3, 3, 2]
assert_array_equal(reflect, expected_reflect)
wrap = [coord_map(4, n, 'W') for n in range(-6, 6)]
wrap = [coord_map_py(4, n, 'W') for n in range(-6, 6)]
expected_wrap = [2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1]
assert_array_equal(wrap, expected_wrap)
nearest = [coord_map(4, n, 'N') for n in range(-6, 6)]
nearest = [coord_map_py(4, n, 'N') for n in range(-6, 6)]
expected_neareset = [0, 0, 0, 0, 0, 0, 0, 1, 2, 3, 3, 3]
assert_array_equal(nearest, expected_neareset)
mirror = [coord_map(4, n, 'M') for n in range(-6, 6)]
mirror = [coord_map_py(4, n, 'M') for n in range(-6, 6)]
expected_mirror = [0, 1, 2, 3, 2, 1, 0, 1, 2, 3, 2, 1]
assert_array_equal(mirror, expected_mirror)
other = [coord_map(4, n, 'undefined') for n in range(-6, 6)]
other = [coord_map_py(4, n, 'undefined') for n in range(-6, 6)]
assert_array_equal(other, list(range(-6, 6)))
+2 -2
View File
@@ -49,8 +49,8 @@ def resize(image, output_shape, order=1, mode='constant', cval=0, clip=True,
Whether to keep the original range of values. Otherwise, the input
image is converted according to the conventions of `img_as_float`.
Notes
-----
Note
----
Modes 'mirror' and 'reflect' are similar, but differ in whether the edge
voxels are duplicated during the reflection. As an example, if an array
has values [0, 1, 2] and was padded to the right by four values using
+2 -2
View File
@@ -76,8 +76,8 @@ def _warp_fast(cnp.ndarray image, cnp.ndarray H, output_shape=None,
Used in conjunction with mode 'C' (constant), the value
outside the image boundaries.
Notes
-----
Note
----
Modes 'mirror' and 'reflect' are similar, but differ in whether the edge
voxels are duplicated during the reflection. As an example, if an array
has values [0, 1, 2] and was padded to the right by four values using
-2
View File
@@ -181,7 +181,6 @@ def test_resize3d_resize():
assert_almost_equal(resized, ref)
def test_resize3d_2din_3dout():
# 3D output with 2D input
x = np.zeros((5, 5), dtype=np.double)
@@ -192,7 +191,6 @@ def test_resize3d_2din_3dout():
assert_almost_equal(resized, ref)
def test_resize3d_bilinear():
# bilinear 3rd dimension
x = np.zeros((5, 5, 2), dtype=np.double)