/*NUMPY_API
* Convert an object to NPY_RAISE / NPY_CLIP / NPY_WRAP
*/
NPY_NO_EXPORT int
PyArray_ClipmodeConverter(PyObject *object, NPY_CLIPMODE *val)
{
if (object == NULL || object == Py_None) {
*val = NPY_RAISE;
}
else if (PyBytes_Check(object)) {
char *str;
str = PyBytes_AS_STRING(object);
if (str[0] == 'C' || str[0] == 'c') {
*val = NPY_CLIP;
}
else if (str[0] == 'W' || str[0] == 'w') {
*val = NPY_WRAP;
}
else if (str[0] == 'R' || str[0] == 'r') {
*val = NPY_RAISE;
}
else {
PyErr_SetString(PyExc_TypeError,
"clipmode not understood");
return NPY_FAIL;
}
}
else if (PyUnicode_Check(object)) {
PyObject *tmp;
int ret;
tmp = PyUnicode_AsASCIIString(object);
if (tmp == NULL) {
return NPY_FAIL;
}
ret = PyArray_ClipmodeConverter(tmp, val);
Py_DECREF(tmp);
return ret;
}
else {
int number = PyArray_PyIntAsInt(object);
if (error_converting(number)) {
goto fail;
}
if (number <= (int) NPY_RAISE
&& number >= (int) NPY_CLIP) {
*val = (NPY_CLIPMODE) number;
}
else {
goto fail;
}
}
return NPY_SUCCEED;
fail:
PyErr_SetString(PyExc_TypeError,
"clipmode not understood");
return NPY_FAIL;
}
/*NUMPY_API
* Get axis from an object (possibly None) -- a converter function,
*
* See also PyArray_ConvertMultiAxis, which also handles a tuple of axes.
*/
NPY_NO_EXPORT int
PyArray_AxisConverter(PyObject *obj, int *axis)
{
if (obj == Py_None) {
*axis = NPY_MAXDIMS;
}
else {
*axis = PyArray_PyIntAsInt(obj);
if (PyErr_Occurred()) {
return NPY_FAIL;
}
}
return NPY_SUCCEED;
}
/*
* Converts an axis parameter into an ndim-length C-array of
* boolean flags, True for each axis specified.
*
* If obj is None or NULL, everything is set to True. If obj is a tuple,
* each axis within the tuple is set to True. If obj is an integer,
* just that axis is set to True.
*/
NPY_NO_EXPORT int
PyArray_ConvertMultiAxis(PyObject *axis_in, int ndim, npy_bool *out_axis_flags)
{
/* None means all of the axes */
if (axis_in == Py_None || axis_in == NULL) {
memset(out_axis_flags, 1, ndim);
return NPY_SUCCEED;
}
/* A tuple of which axes */
else if (PyTuple_Check(axis_in)) {
int i, naxes;
memset(out_axis_flags, 0, ndim);
naxes = PyTuple_Size(axis_in);
if (naxes < 0) {
return NPY_FAIL;
}
for (i = 0; i < naxes; ++i) {
PyObject *tmp = PyTuple_GET_ITEM(axis_in, i);
int axis = PyArray_PyIntAsInt(tmp);
int axis_orig = axis;
if (error_converting(axis)) {
return NPY_FAIL;
}
if (axis < 0) {
axis += ndim;
}
if (axis < 0 || axis >= ndim) {
PyErr_Format(PyExc_ValueError,
"'axis' entry %d is out of bounds [-%d, %d)",
axis_orig, ndim, ndim);
return NPY_FAIL;
}
if (out_axis_flags[axis]) {
PyErr_SetString(PyExc_ValueError,
"duplicate value in 'axis'");
return NPY_FAIL;
}
out_axis_flags[axis] = 1;
}
return NPY_SUCCEED;
}
/* Try to interpret axis as an integer */
else {
int axis, axis_orig;
memset(out_axis_flags, 0, ndim);
axis = PyArray_PyIntAsInt(axis_in);
axis_orig = axis;
if (error_converting(axis)) {
return NPY_FAIL;
}
if (axis < 0) {
axis += ndim;
}
/*
* Special case letting axis={-1,0} slip through for scalars,
* for backwards compatibility reasons.
*/
if (ndim == 0 && (axis == 0 || axis == -1)) {
return NPY_SUCCEED;
}
if (axis < 0 || axis >= ndim) {
PyErr_Format(PyExc_ValueError,
"'axis' entry %d is out of bounds [-%d, %d)",
axis_orig, ndim, ndim);
return NPY_FAIL;
}
out_axis_flags[axis] = 1;
return NPY_SUCCEED;
}
}
