#include "mplutils.h"

#include "_image_resample.h"

#include "numpy_cpp.h"

#include "py_converters.h"

const char* image_resample__doc__ =

" The radius of the kernel, if method is SINC, LANCZOS or BLACKMAN.\n";

static PyArrayObject *

_get_transform_mesh(PyObject *py_affine, npy_intp *dims)

{

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

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

PyObject *py_inverse = NULL;

npy_intp out_dims[3];

out_dims[0] = dims[0] * dims[1];

out_dims[1] = 2;

py_inverse = PyObject_CallMethod(py_affine, "inverted", NULL);

if (py_inverse == NULL) {

return NULL;

}

numpy::array_view<double, 2> input_mesh(out_dims);

double *p = (double *)input_mesh.data();

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

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

*p++ = (double)x;

*p++ = (double)y;

}

}

PyObject *output_mesh = PyObject_CallMethod(

py_inverse, "transform", "O", input_mesh.pyobj_steal());

Py_DECREF(py_inverse);

if (output_mesh == NULL) {

return NULL;

}

PyArrayObject *output_mesh_array =

(PyArrayObject *)PyArray_ContiguousFromAny(

output_mesh, NPY_DOUBLE, 2, 2);

Py_DECREF(output_mesh);

if (output_mesh_array == NULL) {

return NULL;

}

return output_mesh_array;

}

static PyObject *

image_resample(PyObject *self, PyObject* args, PyObject *kwargs)

{

PyObject *py_input = NULL;

PyObject *py_output = NULL;

PyObject *py_transform = NULL;

resample_params_t params;

PyArrayObject *input = NULL;

PyArrayObject *output = NULL;

PyArrayObject *transform_mesh = NULL;

int ndim;

int type;

params.interpolation = NEAREST;

params.transform_mesh = NULL;

params.resample = false;

params.norm = false;

params.radius = 1.0;

params.alpha = 1.0;

const char *kwlist[] = {

"input_array", "output_array", "transform", "interpolation",

"resample", "alpha", "norm", "radius", NULL };

if (!PyArg_ParseTupleAndKeywords(

args, kwargs, "OOO|iO&dO&d:resample", (char **)kwlist,

&py_input, &py_output, &py_transform,

&params.interpolation, &convert_bool, &params.resample,

&params.alpha, &convert_bool, &params.norm, &params.radius)) {

return NULL;

}

if (params.interpolation < 0 || params.interpolation >= _n_interpolation) {

PyErr_Format(PyExc_ValueError, "Invalid interpolation value %d",

params.interpolation);

goto error;

}

input = (PyArrayObject *)PyArray_FromAny(

py_input, NULL, 2, 3, NPY_ARRAY_C_CONTIGUOUS, NULL);

if (!input) {

goto error;

}

ndim = PyArray_NDIM(input);

type = PyArray_TYPE(input);

if (!PyArray_Check(py_output)) {

PyErr_SetString(PyExc_ValueError, "Output array must be a NumPy array");

goto error;

}

output = (PyArrayObject *)py_output;

if (PyArray_NDIM(output) != ndim) {

PyErr_Format(

PyExc_ValueError,

"Input (%dD) and output (%dD) have different dimensionalities.",

ndim, PyArray_NDIM(output));

goto error;

}

// PyArray_FromAny above checks that input is 2D or 3D.

if (ndim == 3 && (PyArray_DIM(input, 2) != 4 || PyArray_DIM(output, 2) != 4)) {

PyErr_Format(

PyExc_ValueError,

"If 3D, input and output arrays must be RGBA with shape (M, N, 4); "

"got trailing dimensions of %" NPY_INTP_FMT " and %" NPY_INTP_FMT

" respectively", PyArray_DIM(input, 2), PyArray_DIM(output, 2));

goto error;

}

if (PyArray_TYPE(output) != type) {

PyErr_SetString(PyExc_ValueError, "Mismatched types");

goto error;

}

if (!PyArray_IS_C_CONTIGUOUS(output)) {

PyErr_SetString(PyExc_ValueError, "Output array must be C-contiguous");

goto error;

}

if (py_transform == NULL || py_transform == Py_None) {

params.is_affine = true;

} else {

PyObject *py_is_affine;

int py_is_affine2;

py_is_affine = PyObject_GetAttrString(py_transform, "is_affine");

if (!py_is_affine) {

goto error;

}

py_is_affine2 = PyObject_IsTrue(py_is_affine);

Py_DECREF(py_is_affine);

if (py_is_affine2 == -1) {

goto error;

} else if (py_is_affine2) {

if (!convert_trans_affine(py_transform, &params.affine)) {

goto error;

}

params.is_affine = true;

} else {

transform_mesh = _get_transform_mesh(

py_transform, PyArray_DIMS(output));

if (!transform_mesh) {

goto error;

}

params.transform_mesh = (double *)PyArray_DATA(transform_mesh);

params.is_affine = false;

}

}

if (auto resampler =

(ndim == 2) ? (

(type == NPY_UINT8) ? resample<agg::gray8> :

(type == NPY_INT8) ? resample<agg::gray8> :

(type == NPY_UINT16) ? resample<agg::gray16> :

(type == NPY_INT16) ? resample<agg::gray16> :

(type == NPY_FLOAT32) ? resample<agg::gray32> :

(type == NPY_FLOAT64) ? resample<agg::gray64> :

nullptr) : (

// ndim == 3

(type == NPY_UINT8) ? resample<agg::rgba8> :

(type == NPY_INT8) ? resample<agg::rgba8> :

(type == NPY_UINT16) ? resample<agg::rgba16> :

(type == NPY_INT16) ? resample<agg::rgba16> :

(type == NPY_FLOAT32) ? resample<agg::rgba32> :

(type == NPY_FLOAT64) ? resample<agg::rgba64> :

nullptr)) {

Py_BEGIN_ALLOW_THREADS

resampler(

PyArray_DATA(input), PyArray_DIM(input, 1), PyArray_DIM(input, 0),

PyArray_DATA(output), PyArray_DIM(output, 1), PyArray_DIM(output, 0),

params);

Py_END_ALLOW_THREADS

} else {

PyErr_SetString(

PyExc_ValueError,

"arrays must be of dtype byte, short, float32 or float64");

goto error;

}

Py_DECREF(input);

Py_XDECREF(transform_mesh);

Py_RETURN_NONE;

error:

Py_XDECREF(input);

Py_XDECREF(transform_mesh);

return NULL;

}

static PyMethodDef module_functions[] = {

{"resample", (PyCFunction)image_resample, METH_VARARGS|METH_KEYWORDS, image_resample__doc__},

{NULL}

};

static struct PyModuleDef moduledef = {

PyModuleDef_HEAD_INIT, "_image", NULL, 0, module_functions,

};

PyMODINIT_FUNC PyInit__image(void)

}