cpython/Objects/sliceobject.c

195 lines
4.4 KiB
C
Raw Normal View History

1996-07-30 13:45:48 -03:00
/*
Written by Jim Hugunin and Chris Chase.
This includes both the singular ellipsis object and slice objects.
1996-07-30 13:45:48 -03:00
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
1996-07-30 13:45:48 -03:00
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"
2001-08-02 01:15:00 -03:00
#include "structmember.h"
1996-07-30 13:45:48 -03:00
static PyObject *
2000-07-09 03:21:27 -03:00
ellipsis_repr(PyObject *op)
1996-07-30 13:45:48 -03:00
{
return PyString_FromString("Ellipsis");
1996-07-30 13:45:48 -03:00
}
static PyTypeObject PyEllipsis_Type = {
1996-07-30 13:45:48 -03:00
PyObject_HEAD_INIT(&PyType_Type)
0,
"ellipsis",
1996-07-30 13:45:48 -03:00
0,
0,
0, /*tp_dealloc*/ /*never called*/
0, /*tp_print*/
0, /*tp_getattr*/
0, /*tp_setattr*/
0, /*tp_compare*/
(reprfunc)ellipsis_repr, /*tp_repr*/
1996-07-30 13:45:48 -03:00
0, /*tp_as_number*/
0, /*tp_as_sequence*/
0, /*tp_as_mapping*/
0, /*tp_hash */
};
PyObject _Py_EllipsisObject = {
PyObject_HEAD_INIT(&PyEllipsis_Type)
1996-07-30 13:45:48 -03:00
};
/* Slice object implementation
start, stop, and step are python objects with None indicating no
index is present.
*/
PyObject *
2000-07-09 03:21:27 -03:00
PySlice_New(PyObject *start, PyObject *stop, PyObject *step)
1996-07-30 13:45:48 -03:00
{
PySliceObject *obj = PyObject_NEW(PySliceObject, &PySlice_Type);
1996-07-30 13:45:48 -03:00
if (obj == NULL)
return NULL;
1996-07-30 13:45:48 -03:00
if (step == NULL) step = Py_None;
Py_INCREF(step);
if (start == NULL) start = Py_None;
Py_INCREF(start);
if (stop == NULL) stop = Py_None;
Py_INCREF(stop);
obj->step = step;
obj->start = start;
obj->stop = stop;
return (PyObject *) obj;
}
int
2000-07-09 03:21:27 -03:00
PySlice_GetIndices(PySliceObject *r, int length,
int *start, int *stop, int *step)
1996-07-30 13:45:48 -03:00
{
if (r->step == Py_None) {
*step = 1;
} else {
if (!PyInt_Check(r->step)) return -1;
*step = PyInt_AsLong(r->step);
}
if (r->start == Py_None) {
*start = *step < 0 ? length-1 : 0;
} else {
if (!PyInt_Check(r->start)) return -1;
*start = PyInt_AsLong(r->start);
if (*start < 0) *start += length;
}
if (r->stop == Py_None) {
*stop = *step < 0 ? -1 : length;
} else {
if (!PyInt_Check(r->stop)) return -1;
*stop = PyInt_AsLong(r->stop);
if (*stop < 0) *stop += length;
}
if (*stop > length) return -1;
if (*start >= length) return -1;
if (*step == 0) return -1;
return 0;
}
static void
2000-07-09 03:21:27 -03:00
slice_dealloc(PySliceObject *r)
1996-07-30 13:45:48 -03:00
{
Py_DECREF(r->step);
Py_DECREF(r->start);
Py_DECREF(r->stop);
PyObject_DEL(r);
1996-07-30 13:45:48 -03:00
}
static PyObject *
2000-07-09 03:21:27 -03:00
slice_repr(PySliceObject *r)
1996-07-30 13:45:48 -03:00
{
PyObject *s, *comma;
s = PyString_FromString("slice(");
comma = PyString_FromString(", ");
PyString_ConcatAndDel(&s, PyObject_Repr(r->start));
PyString_Concat(&s, comma);
PyString_ConcatAndDel(&s, PyObject_Repr(r->stop));
PyString_Concat(&s, comma);
PyString_ConcatAndDel(&s, PyObject_Repr(r->step));
PyString_ConcatAndDel(&s, PyString_FromString(")"));
Py_DECREF(comma);
return s;
}
static PyMemberDef slice_members[] = {
2001-08-02 01:15:00 -03:00
{"start", T_OBJECT, offsetof(PySliceObject, start), READONLY},
{"stop", T_OBJECT, offsetof(PySliceObject, stop), READONLY},
{"step", T_OBJECT, offsetof(PySliceObject, step), READONLY},
{0}
};
1996-07-30 13:45:48 -03:00
static int
slice_compare(PySliceObject *v, PySliceObject *w)
{
int result = 0;
if (v == w)
return 0;
if (PyObject_Cmp(v->start, w->start, &result) < 0)
return -2;
if (result != 0)
return result;
if (PyObject_Cmp(v->stop, w->stop, &result) < 0)
return -2;
if (result != 0)
return result;
if (PyObject_Cmp(v->step, w->step, &result) < 0)
return -2;
return result;
}
1996-07-30 13:45:48 -03:00
PyTypeObject PySlice_Type = {
PyObject_HEAD_INIT(&PyType_Type)
0, /* Number of items for varobject */
"slice", /* Name of this type */
sizeof(PySliceObject), /* Basic object size */
0, /* Item size for varobject */
2001-08-02 01:15:00 -03:00
(destructor)slice_dealloc, /* tp_dealloc */
0, /* tp_print */
0, /* tp_getattr */
0, /* tp_setattr */
(cmpfunc)slice_compare, /* tp_compare */
(reprfunc)slice_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 */
0, /* tp_methods */
slice_members, /* tp_members */
0, /* tp_getset */
0, /* tp_base */
0, /* tp_dict */
1996-07-30 13:45:48 -03:00
};