/*NUMPY_API
* Get intp chunk from sequence
*
* This function takes a Python sequence object and allocates and
* fills in an intp array with the converted values.
*
* Remember to free the pointer seq.ptr when done using
* PyDimMem_FREE(seq.ptr)**
*/
NPY_NO_EXPORT int
PyArray_IntpConverter(PyObject *obj, PyArray_Dims *seq)
{
Py_ssize_t len;
int nd;
seq->ptr = NULL;
seq->len = 0;
if (obj == Py_None) {
return NPY_SUCCEED;
}
len = PySequence_Size(obj);
if (len == -1) {
/* Check to see if it is an integer number */
if (PyNumber_Check(obj)) {
/*
* After the deprecation the PyNumber_Check could be replaced
* by PyIndex_Check.
* FIXME 1.9 ?
*/
len = 1;
}
}
if (len < 0) {
PyErr_SetString(PyExc_TypeError,
"expected sequence object with len >= 0 or a single integer");
return NPY_FAIL;
}
if (len > NPY_MAXDIMS) {
PyErr_Format(PyExc_ValueError, "sequence too large; "
"cannot be greater than %d", NPY_MAXDIMS);
return NPY_FAIL;
}
if (len > 0) {
seq->ptr = PyDimMem_NEW(len);
if (seq->ptr == NULL) {
PyErr_NoMemory();
return NPY_FAIL;
}
}
seq->len = len;
nd = PyArray_IntpFromIndexSequence(obj, (npy_intp *)seq->ptr, len);
if (nd == -1 || nd != len) {
PyDimMem_FREE(seq->ptr);
seq->ptr = NULL;
return NPY_FAIL;
}
return NPY_SUCCEED;
}
/*NUMPY_API
* Get intp chunk from sequence
*
* This function takes a Python sequence object and allocates and
* fills in an intp array with the converted values.
*
* Remember to free the pointer seq.ptr when done using
* PyDimMem_FREE(seq.ptr)**
*/
NPY_NO_EXPORT int
PyArray_IntpConverter(PyObject *obj, PyArray_Dims *seq)
{
int len;
int nd;
seq->ptr = NULL;
seq->len = 0;
if (obj == Py_None) {
return PY_SUCCEED;
}
len = PySequence_Size(obj);
if (len == -1) {
/* Check to see if it is a number */
if (PyNumber_Check(obj)) {
len = 1;
}
}
if (len < 0) {
PyErr_SetString(PyExc_TypeError,
"expected sequence object with len >= 0");
return PY_FAIL;
}
if (len > MAX_DIMS) {
PyErr_Format(PyExc_ValueError, "sequence too large; " \
"must be smaller than %d", MAX_DIMS);
return PY_FAIL;
}
if (len > 0) {
seq->ptr = PyDimMem_NEW(len);
if (seq->ptr == NULL) {
PyErr_NoMemory();
return PY_FAIL;
}
}
seq->len = len;
nd = PyArray_IntpFromSequence(obj, (intp *)seq->ptr, len);
if (nd == -1 || nd != len) {
PyDimMem_FREE(seq->ptr);
seq->ptr = NULL;
return PY_FAIL;
}
return PY_SUCCEED;
}
