{

/* TODO: Could we get away with float, rather than double, arrays here? */

/* Given a non-affine transform object, create a mesh that maps

// If attribute doesn't exist, raises Python AttributeError

auto inverse = transform.attr("inverted")();

py::ssize_t mesh_dims[2] = {dims[0]*dims[1], 2};

py::array_t<double> input_mesh(mesh_dims);

auto p = input_mesh.mutable_data();

for (auto y = 0; y < dims[0]; ++y) {

for (auto x = 0; x < dims[1]; ++x) {

*p++ = (double)x;

*p++ = (double)y;

}

}

auto output_mesh = inverse.attr("transform")(input_mesh);

auto output_mesh_array =

py::array_t<double, py::array::c_style | py::array::forcecast>(output_mesh);

if (output_mesh_array.ndim() != 2) {

throw std::runtime_error(

"Inverse transformed mesh array should be 2D not {}D"_s.format(

output_mesh_array.ndim()));

}

return output_mesh_array;

py::array& output_array,

const py::object& transform,

interpolation_e interpolation,

bool resample_, // Avoid name clash with resample() function

float alpha,

bool norm,

float radius)

{

// Validate input_array

auto dtype = input_array.dtype(); // Validated when determine resampler below

auto ndim = input_array.ndim();

if (ndim != 2 && ndim != 3) {

throw std::invalid_argument("Input array must be a 2D or 3D array");

}

if (ndim == 3 && input_array.shape(2) != 4) {

throw std::invalid_argument(

"3D input array must be RGBA with shape (M, N, 4), has trailing dimension of {}"_s.format(

input_array.shape(2)));

}

// Ensure input array is contiguous, regardless of dtype

input_array = py::array::ensure(input_array, py::array::c_style);

// Validate output array

auto out_ndim = output_array.ndim();

if (out_ndim != ndim) {

throw std::invalid_argument(

"Input ({}D) and output ({}D) arrays have different dimensionalities"_s.format(

ndim, out_ndim));

}

if (out_ndim == 3 && output_array.shape(2) != 4) {

throw std::invalid_argument(

"3D output array must be RGBA with shape (M, N, 4), has trailing dimension of {}"_s.format(

output_array.shape(2)));

}

if (!output_array.dtype().is(dtype)) {

throw std::invalid_argument("Input and output arrays have mismatched types");

}

if ((output_array.flags() & py::array::c_style) == 0) {

throw std::invalid_argument("Output array must be C-contiguous");

}

if (!output_array.writeable()) {

throw std::invalid_argument("Output array must be writeable");

}

resample_params_t params;

params.interpolation = interpolation;

params.transform_mesh = nullptr;

params.resample = resample_;

params.norm = norm;

params.radius = radius;

params.alpha = alpha;

// Only used if transform is not affine.

// Need to keep it in scope for the duration of this function.

py::array_t<double> transform_mesh;

// Validate transform

if (transform.is_none()) {

params.is_affine = true;

} else {

// Raises Python AttributeError if no such attribute or TypeError if cast fails

bool is_affine = py::cast<bool>(transform.attr("is_affine"));

if (is_affine) {

convert_trans_affine(transform, params.affine);

params.is_affine = true;

} else {

transform_mesh = _get_transform_mesh(transform, output_array.shape());

params.transform_mesh = transform_mesh.data();

params.is_affine = false;

}

}

if (auto resampler =

(ndim == 2) ? (

(dtype.equal(py::dtype::of<std::uint8_t>())) ? resample<agg::gray8> :

(dtype.equal(py::dtype::of<std::int8_t>())) ? resample<agg::gray8> :

(dtype.equal(py::dtype::of<std::uint16_t>())) ? resample<agg::gray16> :

(dtype.equal(py::dtype::of<std::int16_t>())) ? resample<agg::gray16> :

(dtype.equal(py::dtype::of<float>())) ? resample<agg::gray32> :

(dtype.equal(py::dtype::of<double>())) ? resample<agg::gray64> :

nullptr) : (

// ndim == 3

(dtype.equal(py::dtype::of<std::uint8_t>())) ? resample<agg::rgba8> :

(dtype.equal(py::dtype::of<std::int8_t>())) ? resample<agg::rgba8> :

(dtype.equal(py::dtype::of<std::uint16_t>())) ? resample<agg::rgba16> :

(dtype.equal(py::dtype::of<std::int16_t>())) ? resample<agg::rgba16> :

(dtype.equal(py::dtype::of<float>())) ? resample<agg::rgba32> :

(dtype.equal(py::dtype::of<double>())) ? resample<agg::rgba64> :

nullptr)) {

Py_BEGIN_ALLOW_THREADS

resampler(

input_array.data(), input_array.shape(1), input_array.shape(0),

output_array.mutable_data(), output_array.shape(1), output_array.shape(0),

params);

Py_END_ALLOW_THREADS

} else {

throw std::invalid_argument("arrays must be of dtype byte, short, float32 or float64");

}

{

py::enum_<interpolation_e>(m, "_InterpolationType")

.value("NEAREST", NEAREST)

.value("BILINEAR", BILINEAR)

.value("BICUBIC", BICUBIC)

.value("SPLINE16", SPLINE16)

.value("SPLINE36", SPLINE36)

.value("HANNING", HANNING)

.value("HAMMING", HAMMING)

.value("HERMITE", HERMITE)

.value("KAISER", KAISER)

.value("QUADRIC", QUADRIC)

.value("CATROM", CATROM)

.value("GAUSSIAN", GAUSSIAN)

.value("BESSEL", BESSEL)

.value("MITCHELL", MITCHELL)

.value("SINC", SINC)

.value("LANCZOS", LANCZOS)

.value("BLACKMAN", BLACKMAN)

.export_values();

m.def("resample", &image_resample,

"input_array"_a,

"output_array"_a,

"transform"_a,

"interpolation"_a = interpolation_e::NEAREST,

"resample"_a = false,

"alpha"_a = 1,

"norm"_a = false,

"radius"_a = 1,

image_resample__doc__);