672 lines
19 KiB
C
672 lines
19 KiB
C
/*
|
|
Written by Jim Hugunin and Chris Chase.
|
|
|
|
This includes both the singular ellipsis object and slice objects.
|
|
|
|
Guido, feel free to do whatever you want in the way of copyrights
|
|
for this file.
|
|
*/
|
|
|
|
/*
|
|
Py_Ellipsis encodes the '...' rubber index token. It is similar to
|
|
the Py_NoneStruct in that there is no way to create other objects of
|
|
this type and there is exactly one in existence.
|
|
*/
|
|
|
|
#include "Python.h"
|
|
#include "pycore_abstract.h" // _PyIndex_Check()
|
|
#include "pycore_long.h" // _PyLong_GetZero()
|
|
#include "pycore_object.h" // _PyObject_GC_TRACK()
|
|
#include "structmember.h" // PyMemberDef
|
|
|
|
static PyObject *
|
|
ellipsis_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
|
|
{
|
|
if (PyTuple_GET_SIZE(args) || (kwargs && PyDict_GET_SIZE(kwargs))) {
|
|
PyErr_SetString(PyExc_TypeError, "EllipsisType takes no arguments");
|
|
return NULL;
|
|
}
|
|
Py_INCREF(Py_Ellipsis);
|
|
return Py_Ellipsis;
|
|
}
|
|
|
|
static PyObject *
|
|
ellipsis_repr(PyObject *op)
|
|
{
|
|
return PyUnicode_FromString("Ellipsis");
|
|
}
|
|
|
|
static PyObject *
|
|
ellipsis_reduce(PyObject *op, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
return PyUnicode_FromString("Ellipsis");
|
|
}
|
|
|
|
static PyMethodDef ellipsis_methods[] = {
|
|
{"__reduce__", ellipsis_reduce, METH_NOARGS, NULL},
|
|
{NULL, NULL}
|
|
};
|
|
|
|
PyTypeObject PyEllipsis_Type = {
|
|
PyVarObject_HEAD_INIT(&PyType_Type, 0)
|
|
"ellipsis", /* tp_name */
|
|
0, /* tp_basicsize */
|
|
0, /* tp_itemsize */
|
|
0, /*never called*/ /* tp_dealloc */
|
|
0, /* tp_vectorcall_offset */
|
|
0, /* tp_getattr */
|
|
0, /* tp_setattr */
|
|
0, /* tp_as_async */
|
|
ellipsis_repr, /* tp_repr */
|
|
0, /* tp_as_number */
|
|
0, /* tp_as_sequence */
|
|
0, /* tp_as_mapping */
|
|
0, /* tp_hash */
|
|
0, /* tp_call */
|
|
0, /* tp_str */
|
|
PyObject_GenericGetAttr, /* tp_getattro */
|
|
0, /* tp_setattro */
|
|
0, /* tp_as_buffer */
|
|
Py_TPFLAGS_DEFAULT, /* tp_flags */
|
|
0, /* tp_doc */
|
|
0, /* tp_traverse */
|
|
0, /* tp_clear */
|
|
0, /* tp_richcompare */
|
|
0, /* tp_weaklistoffset */
|
|
0, /* tp_iter */
|
|
0, /* tp_iternext */
|
|
ellipsis_methods, /* tp_methods */
|
|
0, /* tp_members */
|
|
0, /* tp_getset */
|
|
0, /* tp_base */
|
|
0, /* tp_dict */
|
|
0, /* tp_descr_get */
|
|
0, /* tp_descr_set */
|
|
0, /* tp_dictoffset */
|
|
0, /* tp_init */
|
|
0, /* tp_alloc */
|
|
ellipsis_new, /* tp_new */
|
|
};
|
|
|
|
PyObject _Py_EllipsisObject = {
|
|
_PyObject_EXTRA_INIT
|
|
1, &PyEllipsis_Type
|
|
};
|
|
|
|
|
|
/* Slice object implementation */
|
|
|
|
|
|
void _PySlice_Fini(PyThreadState *tstate)
|
|
{
|
|
PyInterpreterState *interp = tstate->interp;
|
|
PySliceObject *obj = interp->slice_cache;
|
|
if (obj != NULL) {
|
|
interp->slice_cache = NULL;
|
|
PyObject_GC_Del(obj);
|
|
}
|
|
}
|
|
|
|
/* start, stop, and step are python objects with None indicating no
|
|
index is present.
|
|
*/
|
|
|
|
PyObject *
|
|
PySlice_New(PyObject *start, PyObject *stop, PyObject *step)
|
|
{
|
|
if (step == NULL) {
|
|
step = Py_None;
|
|
}
|
|
if (start == NULL) {
|
|
start = Py_None;
|
|
}
|
|
if (stop == NULL) {
|
|
stop = Py_None;
|
|
}
|
|
|
|
PyInterpreterState *interp = _PyInterpreterState_GET();
|
|
PySliceObject *obj;
|
|
if (interp->slice_cache != NULL) {
|
|
obj = interp->slice_cache;
|
|
interp->slice_cache = NULL;
|
|
_Py_NewReference((PyObject *)obj);
|
|
}
|
|
else {
|
|
obj = PyObject_GC_New(PySliceObject, &PySlice_Type);
|
|
if (obj == NULL) {
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
Py_INCREF(step);
|
|
obj->step = step;
|
|
Py_INCREF(start);
|
|
obj->start = start;
|
|
Py_INCREF(stop);
|
|
obj->stop = stop;
|
|
|
|
_PyObject_GC_TRACK(obj);
|
|
return (PyObject *) obj;
|
|
}
|
|
|
|
PyObject *
|
|
_PySlice_FromIndices(Py_ssize_t istart, Py_ssize_t istop)
|
|
{
|
|
PyObject *start, *end, *slice;
|
|
start = PyLong_FromSsize_t(istart);
|
|
if (!start)
|
|
return NULL;
|
|
end = PyLong_FromSsize_t(istop);
|
|
if (!end) {
|
|
Py_DECREF(start);
|
|
return NULL;
|
|
}
|
|
|
|
slice = PySlice_New(start, end, NULL);
|
|
Py_DECREF(start);
|
|
Py_DECREF(end);
|
|
return slice;
|
|
}
|
|
|
|
int
|
|
PySlice_GetIndices(PyObject *_r, Py_ssize_t length,
|
|
Py_ssize_t *start, Py_ssize_t *stop, Py_ssize_t *step)
|
|
{
|
|
PySliceObject *r = (PySliceObject*)_r;
|
|
/* XXX support long ints */
|
|
if (r->step == Py_None) {
|
|
*step = 1;
|
|
} else {
|
|
if (!PyLong_Check(r->step)) return -1;
|
|
*step = PyLong_AsSsize_t(r->step);
|
|
}
|
|
if (r->start == Py_None) {
|
|
*start = *step < 0 ? length-1 : 0;
|
|
} else {
|
|
if (!PyLong_Check(r->start)) return -1;
|
|
*start = PyLong_AsSsize_t(r->start);
|
|
if (*start < 0) *start += length;
|
|
}
|
|
if (r->stop == Py_None) {
|
|
*stop = *step < 0 ? -1 : length;
|
|
} else {
|
|
if (!PyLong_Check(r->stop)) return -1;
|
|
*stop = PyLong_AsSsize_t(r->stop);
|
|
if (*stop < 0) *stop += length;
|
|
}
|
|
if (*stop > length) return -1;
|
|
if (*start >= length) return -1;
|
|
if (*step == 0) return -1;
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
PySlice_Unpack(PyObject *_r,
|
|
Py_ssize_t *start, Py_ssize_t *stop, Py_ssize_t *step)
|
|
{
|
|
PySliceObject *r = (PySliceObject*)_r;
|
|
/* this is harder to get right than you might think */
|
|
|
|
Py_BUILD_ASSERT(PY_SSIZE_T_MIN + 1 <= -PY_SSIZE_T_MAX);
|
|
|
|
if (r->step == Py_None) {
|
|
*step = 1;
|
|
}
|
|
else {
|
|
if (!_PyEval_SliceIndex(r->step, step)) return -1;
|
|
if (*step == 0) {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"slice step cannot be zero");
|
|
return -1;
|
|
}
|
|
/* Here *step might be -PY_SSIZE_T_MAX-1; in this case we replace it
|
|
* with -PY_SSIZE_T_MAX. This doesn't affect the semantics, and it
|
|
* guards against later undefined behaviour resulting from code that
|
|
* does "step = -step" as part of a slice reversal.
|
|
*/
|
|
if (*step < -PY_SSIZE_T_MAX)
|
|
*step = -PY_SSIZE_T_MAX;
|
|
}
|
|
|
|
if (r->start == Py_None) {
|
|
*start = *step < 0 ? PY_SSIZE_T_MAX : 0;
|
|
}
|
|
else {
|
|
if (!_PyEval_SliceIndex(r->start, start)) return -1;
|
|
}
|
|
|
|
if (r->stop == Py_None) {
|
|
*stop = *step < 0 ? PY_SSIZE_T_MIN : PY_SSIZE_T_MAX;
|
|
}
|
|
else {
|
|
if (!_PyEval_SliceIndex(r->stop, stop)) return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
Py_ssize_t
|
|
PySlice_AdjustIndices(Py_ssize_t length,
|
|
Py_ssize_t *start, Py_ssize_t *stop, Py_ssize_t step)
|
|
{
|
|
/* this is harder to get right than you might think */
|
|
|
|
assert(step != 0);
|
|
assert(step >= -PY_SSIZE_T_MAX);
|
|
|
|
if (*start < 0) {
|
|
*start += length;
|
|
if (*start < 0) {
|
|
*start = (step < 0) ? -1 : 0;
|
|
}
|
|
}
|
|
else if (*start >= length) {
|
|
*start = (step < 0) ? length - 1 : length;
|
|
}
|
|
|
|
if (*stop < 0) {
|
|
*stop += length;
|
|
if (*stop < 0) {
|
|
*stop = (step < 0) ? -1 : 0;
|
|
}
|
|
}
|
|
else if (*stop >= length) {
|
|
*stop = (step < 0) ? length - 1 : length;
|
|
}
|
|
|
|
if (step < 0) {
|
|
if (*stop < *start) {
|
|
return (*start - *stop - 1) / (-step) + 1;
|
|
}
|
|
}
|
|
else {
|
|
if (*start < *stop) {
|
|
return (*stop - *start - 1) / step + 1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
#undef PySlice_GetIndicesEx
|
|
|
|
int
|
|
PySlice_GetIndicesEx(PyObject *_r, Py_ssize_t length,
|
|
Py_ssize_t *start, Py_ssize_t *stop, Py_ssize_t *step,
|
|
Py_ssize_t *slicelength)
|
|
{
|
|
if (PySlice_Unpack(_r, start, stop, step) < 0)
|
|
return -1;
|
|
*slicelength = PySlice_AdjustIndices(length, start, stop, *step);
|
|
return 0;
|
|
}
|
|
|
|
static PyObject *
|
|
slice_new(PyTypeObject *type, PyObject *args, PyObject *kw)
|
|
{
|
|
PyObject *start, *stop, *step;
|
|
|
|
start = stop = step = NULL;
|
|
|
|
if (!_PyArg_NoKeywords("slice", kw))
|
|
return NULL;
|
|
|
|
if (!PyArg_UnpackTuple(args, "slice", 1, 3, &start, &stop, &step))
|
|
return NULL;
|
|
|
|
/* This swapping of stop and start is to maintain similarity with
|
|
range(). */
|
|
if (stop == NULL) {
|
|
stop = start;
|
|
start = NULL;
|
|
}
|
|
return PySlice_New(start, stop, step);
|
|
}
|
|
|
|
PyDoc_STRVAR(slice_doc,
|
|
"slice(stop)\n\
|
|
slice(start, stop[, step])\n\
|
|
\n\
|
|
Create a slice object. This is used for extended slicing (e.g. a[0:10:2]).");
|
|
|
|
static void
|
|
slice_dealloc(PySliceObject *r)
|
|
{
|
|
PyInterpreterState *interp = _PyInterpreterState_GET();
|
|
_PyObject_GC_UNTRACK(r);
|
|
Py_DECREF(r->step);
|
|
Py_DECREF(r->start);
|
|
Py_DECREF(r->stop);
|
|
if (interp->slice_cache == NULL) {
|
|
interp->slice_cache = r;
|
|
}
|
|
else {
|
|
PyObject_GC_Del(r);
|
|
}
|
|
}
|
|
|
|
static PyObject *
|
|
slice_repr(PySliceObject *r)
|
|
{
|
|
return PyUnicode_FromFormat("slice(%R, %R, %R)", r->start, r->stop, r->step);
|
|
}
|
|
|
|
static PyMemberDef slice_members[] = {
|
|
{"start", T_OBJECT, offsetof(PySliceObject, start), READONLY},
|
|
{"stop", T_OBJECT, offsetof(PySliceObject, stop), READONLY},
|
|
{"step", T_OBJECT, offsetof(PySliceObject, step), READONLY},
|
|
{0}
|
|
};
|
|
|
|
/* Helper function to convert a slice argument to a PyLong, and raise TypeError
|
|
with a suitable message on failure. */
|
|
|
|
static PyObject*
|
|
evaluate_slice_index(PyObject *v)
|
|
{
|
|
if (_PyIndex_Check(v)) {
|
|
return PyNumber_Index(v);
|
|
}
|
|
else {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"slice indices must be integers or "
|
|
"None or have an __index__ method");
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
/* Compute slice indices given a slice and length. Return -1 on failure. Used
|
|
by slice.indices and rangeobject slicing. Assumes that `len` is a
|
|
nonnegative instance of PyLong. */
|
|
|
|
int
|
|
_PySlice_GetLongIndices(PySliceObject *self, PyObject *length,
|
|
PyObject **start_ptr, PyObject **stop_ptr,
|
|
PyObject **step_ptr)
|
|
{
|
|
PyObject *start=NULL, *stop=NULL, *step=NULL;
|
|
PyObject *upper=NULL, *lower=NULL;
|
|
int step_is_negative, cmp_result;
|
|
|
|
/* Convert step to an integer; raise for zero step. */
|
|
if (self->step == Py_None) {
|
|
step = _PyLong_GetOne();
|
|
Py_INCREF(step);
|
|
step_is_negative = 0;
|
|
}
|
|
else {
|
|
int step_sign;
|
|
step = evaluate_slice_index(self->step);
|
|
if (step == NULL)
|
|
goto error;
|
|
step_sign = _PyLong_Sign(step);
|
|
if (step_sign == 0) {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"slice step cannot be zero");
|
|
goto error;
|
|
}
|
|
step_is_negative = step_sign < 0;
|
|
}
|
|
|
|
/* Find lower and upper bounds for start and stop. */
|
|
if (step_is_negative) {
|
|
lower = PyLong_FromLong(-1L);
|
|
if (lower == NULL)
|
|
goto error;
|
|
|
|
upper = PyNumber_Add(length, lower);
|
|
if (upper == NULL)
|
|
goto error;
|
|
}
|
|
else {
|
|
lower = _PyLong_GetZero();
|
|
Py_INCREF(lower);
|
|
upper = length;
|
|
Py_INCREF(upper);
|
|
}
|
|
|
|
/* Compute start. */
|
|
if (self->start == Py_None) {
|
|
start = step_is_negative ? upper : lower;
|
|
Py_INCREF(start);
|
|
}
|
|
else {
|
|
start = evaluate_slice_index(self->start);
|
|
if (start == NULL)
|
|
goto error;
|
|
|
|
if (_PyLong_Sign(start) < 0) {
|
|
/* start += length */
|
|
PyObject *tmp = PyNumber_Add(start, length);
|
|
Py_DECREF(start);
|
|
start = tmp;
|
|
if (start == NULL)
|
|
goto error;
|
|
|
|
cmp_result = PyObject_RichCompareBool(start, lower, Py_LT);
|
|
if (cmp_result < 0)
|
|
goto error;
|
|
if (cmp_result) {
|
|
Py_INCREF(lower);
|
|
Py_DECREF(start);
|
|
start = lower;
|
|
}
|
|
}
|
|
else {
|
|
cmp_result = PyObject_RichCompareBool(start, upper, Py_GT);
|
|
if (cmp_result < 0)
|
|
goto error;
|
|
if (cmp_result) {
|
|
Py_INCREF(upper);
|
|
Py_DECREF(start);
|
|
start = upper;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Compute stop. */
|
|
if (self->stop == Py_None) {
|
|
stop = step_is_negative ? lower : upper;
|
|
Py_INCREF(stop);
|
|
}
|
|
else {
|
|
stop = evaluate_slice_index(self->stop);
|
|
if (stop == NULL)
|
|
goto error;
|
|
|
|
if (_PyLong_Sign(stop) < 0) {
|
|
/* stop += length */
|
|
PyObject *tmp = PyNumber_Add(stop, length);
|
|
Py_DECREF(stop);
|
|
stop = tmp;
|
|
if (stop == NULL)
|
|
goto error;
|
|
|
|
cmp_result = PyObject_RichCompareBool(stop, lower, Py_LT);
|
|
if (cmp_result < 0)
|
|
goto error;
|
|
if (cmp_result) {
|
|
Py_INCREF(lower);
|
|
Py_DECREF(stop);
|
|
stop = lower;
|
|
}
|
|
}
|
|
else {
|
|
cmp_result = PyObject_RichCompareBool(stop, upper, Py_GT);
|
|
if (cmp_result < 0)
|
|
goto error;
|
|
if (cmp_result) {
|
|
Py_INCREF(upper);
|
|
Py_DECREF(stop);
|
|
stop = upper;
|
|
}
|
|
}
|
|
}
|
|
|
|
*start_ptr = start;
|
|
*stop_ptr = stop;
|
|
*step_ptr = step;
|
|
Py_DECREF(upper);
|
|
Py_DECREF(lower);
|
|
return 0;
|
|
|
|
error:
|
|
*start_ptr = *stop_ptr = *step_ptr = NULL;
|
|
Py_XDECREF(start);
|
|
Py_XDECREF(stop);
|
|
Py_XDECREF(step);
|
|
Py_XDECREF(upper);
|
|
Py_XDECREF(lower);
|
|
return -1;
|
|
}
|
|
|
|
/* Implementation of slice.indices. */
|
|
|
|
static PyObject*
|
|
slice_indices(PySliceObject* self, PyObject* len)
|
|
{
|
|
PyObject *start, *stop, *step;
|
|
PyObject *length;
|
|
int error;
|
|
|
|
/* Convert length to an integer if necessary; raise for negative length. */
|
|
length = PyNumber_Index(len);
|
|
if (length == NULL)
|
|
return NULL;
|
|
|
|
if (_PyLong_Sign(length) < 0) {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"length should not be negative");
|
|
Py_DECREF(length);
|
|
return NULL;
|
|
}
|
|
|
|
error = _PySlice_GetLongIndices(self, length, &start, &stop, &step);
|
|
Py_DECREF(length);
|
|
if (error == -1)
|
|
return NULL;
|
|
else
|
|
return Py_BuildValue("(NNN)", start, stop, step);
|
|
}
|
|
|
|
PyDoc_STRVAR(slice_indices_doc,
|
|
"S.indices(len) -> (start, stop, stride)\n\
|
|
\n\
|
|
Assuming a sequence of length len, calculate the start and stop\n\
|
|
indices, and the stride length of the extended slice described by\n\
|
|
S. Out of bounds indices are clipped in a manner consistent with the\n\
|
|
handling of normal slices.");
|
|
|
|
static PyObject *
|
|
slice_reduce(PySliceObject* self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
return Py_BuildValue("O(OOO)", Py_TYPE(self), self->start, self->stop, self->step);
|
|
}
|
|
|
|
PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
|
|
|
|
static PyMethodDef slice_methods[] = {
|
|
{"indices", (PyCFunction)slice_indices,
|
|
METH_O, slice_indices_doc},
|
|
{"__reduce__", (PyCFunction)slice_reduce,
|
|
METH_NOARGS, reduce_doc},
|
|
{NULL, NULL}
|
|
};
|
|
|
|
static PyObject *
|
|
slice_richcompare(PyObject *v, PyObject *w, int op)
|
|
{
|
|
if (!PySlice_Check(v) || !PySlice_Check(w))
|
|
Py_RETURN_NOTIMPLEMENTED;
|
|
|
|
if (v == w) {
|
|
PyObject *res;
|
|
/* XXX Do we really need this shortcut?
|
|
There's a unit test for it, but is that fair? */
|
|
switch (op) {
|
|
case Py_EQ:
|
|
case Py_LE:
|
|
case Py_GE:
|
|
res = Py_True;
|
|
break;
|
|
default:
|
|
res = Py_False;
|
|
break;
|
|
}
|
|
Py_INCREF(res);
|
|
return res;
|
|
}
|
|
|
|
|
|
PyObject *t1 = PyTuple_Pack(3,
|
|
((PySliceObject *)v)->start,
|
|
((PySliceObject *)v)->stop,
|
|
((PySliceObject *)v)->step);
|
|
if (t1 == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
PyObject *t2 = PyTuple_Pack(3,
|
|
((PySliceObject *)w)->start,
|
|
((PySliceObject *)w)->stop,
|
|
((PySliceObject *)w)->step);
|
|
if (t2 == NULL) {
|
|
Py_DECREF(t1);
|
|
return NULL;
|
|
}
|
|
|
|
PyObject *res = PyObject_RichCompare(t1, t2, op);
|
|
Py_DECREF(t1);
|
|
Py_DECREF(t2);
|
|
return res;
|
|
}
|
|
|
|
static int
|
|
slice_traverse(PySliceObject *v, visitproc visit, void *arg)
|
|
{
|
|
Py_VISIT(v->start);
|
|
Py_VISIT(v->stop);
|
|
Py_VISIT(v->step);
|
|
return 0;
|
|
}
|
|
|
|
PyTypeObject PySlice_Type = {
|
|
PyVarObject_HEAD_INIT(&PyType_Type, 0)
|
|
"slice", /* Name of this type */
|
|
sizeof(PySliceObject), /* Basic object size */
|
|
0, /* Item size for varobject */
|
|
(destructor)slice_dealloc, /* tp_dealloc */
|
|
0, /* tp_vectorcall_offset */
|
|
0, /* tp_getattr */
|
|
0, /* tp_setattr */
|
|
0, /* tp_as_async */
|
|
(reprfunc)slice_repr, /* tp_repr */
|
|
0, /* tp_as_number */
|
|
0, /* tp_as_sequence */
|
|
0, /* tp_as_mapping */
|
|
PyObject_HashNotImplemented, /* tp_hash */
|
|
0, /* tp_call */
|
|
0, /* tp_str */
|
|
PyObject_GenericGetAttr, /* tp_getattro */
|
|
0, /* tp_setattro */
|
|
0, /* tp_as_buffer */
|
|
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC, /* tp_flags */
|
|
slice_doc, /* tp_doc */
|
|
(traverseproc)slice_traverse, /* tp_traverse */
|
|
0, /* tp_clear */
|
|
slice_richcompare, /* tp_richcompare */
|
|
0, /* tp_weaklistoffset */
|
|
0, /* tp_iter */
|
|
0, /* tp_iternext */
|
|
slice_methods, /* tp_methods */
|
|
slice_members, /* tp_members */
|
|
0, /* tp_getset */
|
|
0, /* tp_base */
|
|
0, /* tp_dict */
|
|
0, /* tp_descr_get */
|
|
0, /* tp_descr_set */
|
|
0, /* tp_dictoffset */
|
|
0, /* tp_init */
|
|
0, /* tp_alloc */
|
|
slice_new, /* tp_new */
|
|
};
|