blob: fbf0955af49489c05c42fbd92692d89b32af0d60 [file] [log] [blame]
Guido van Rossumf2d125b1996-07-30 16:45:48 +00001/*
2Written by Jim Hugunin and Chris Chase.
3
Guido van Rossume449af71996-10-11 16:25:41 +00004This includes both the singular ellipsis object and slice objects.
Guido van Rossumf2d125b1996-07-30 16:45:48 +00005
6Guido, feel free to do whatever you want in the way of copyrights
7for this file.
8*/
9
10/*
Guido van Rossume449af71996-10-11 16:25:41 +000011Py_Ellipsis encodes the '...' rubber index token. It is similar to
Guido van Rossumf2d125b1996-07-30 16:45:48 +000012the Py_NoneStruct in that there is no way to create other objects of
13this type and there is exactly one in existence.
14*/
15
16#include "Python.h"
Tim Peters6d6c1a32001-08-02 04:15:00 +000017#include "structmember.h"
Guido van Rossumf2d125b1996-07-30 16:45:48 +000018
19static PyObject *
Fred Drake45cfbcc2000-07-09 06:21:27 +000020ellipsis_repr(PyObject *op)
Guido van Rossumf2d125b1996-07-30 16:45:48 +000021{
Walter Dörwald1ab83302007-05-18 17:15:44 +000022 return PyUnicode_FromString("Ellipsis");
Guido van Rossumf2d125b1996-07-30 16:45:48 +000023}
24
Guido van Rossume449af71996-10-11 16:25:41 +000025static PyTypeObject PyEllipsis_Type = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +000026 PyVarObject_HEAD_INIT(&PyType_Type, 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000027 "ellipsis", /* tp_name */
28 0, /* tp_basicsize */
29 0, /* tp_itemsize */
30 0, /*never called*/ /* tp_dealloc */
31 0, /* tp_print */
32 0, /* tp_getattr */
33 0, /* tp_setattr */
34 0, /* tp_compare */
35 ellipsis_repr, /* tp_repr */
36 0, /* tp_as_number */
37 0, /* tp_as_sequence */
38 0, /* tp_as_mapping */
39 0, /* tp_hash */
40 0, /* tp_call */
41 0, /* tp_str */
42 PyObject_GenericGetAttr, /* tp_getattro */
43 0, /* tp_setattro */
44 0, /* tp_as_buffer */
45 Py_TPFLAGS_DEFAULT, /* tp_flags */
Guido van Rossumf2d125b1996-07-30 16:45:48 +000046};
47
Guido van Rossume449af71996-10-11 16:25:41 +000048PyObject _Py_EllipsisObject = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +000049 _PyObject_EXTRA_INIT
50 1, &PyEllipsis_Type
Guido van Rossumf2d125b1996-07-30 16:45:48 +000051};
52
53
54/* Slice object implementation
55
56 start, stop, and step are python objects with None indicating no
57 index is present.
58*/
59
60PyObject *
Fred Drake45cfbcc2000-07-09 06:21:27 +000061PySlice_New(PyObject *start, PyObject *stop, PyObject *step)
Guido van Rossumf2d125b1996-07-30 16:45:48 +000062{
Neil Schemenauer7465ad22002-04-12 03:05:37 +000063 PySliceObject *obj = PyObject_New(PySliceObject, &PySlice_Type);
Guido van Rossumf2d125b1996-07-30 16:45:48 +000064
Guido van Rossumadf54102000-12-14 15:09:46 +000065 if (obj == NULL)
66 return NULL;
67
Guido van Rossumf2d125b1996-07-30 16:45:48 +000068 if (step == NULL) step = Py_None;
69 Py_INCREF(step);
70 if (start == NULL) start = Py_None;
71 Py_INCREF(start);
72 if (stop == NULL) stop = Py_None;
73 Py_INCREF(stop);
74
75 obj->step = step;
76 obj->start = start;
77 obj->stop = stop;
78
79 return (PyObject *) obj;
80}
81
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000082PyObject *
83_PySlice_FromIndices(Py_ssize_t istart, Py_ssize_t istop)
84{
85 PyObject *start, *end, *slice;
Christian Heimes217cfd12007-12-02 14:31:20 +000086 start = PyLong_FromSsize_t(istart);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000087 if (!start)
88 return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +000089 end = PyLong_FromSsize_t(istop);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000090 if (!end) {
91 Py_DECREF(start);
92 return NULL;
93 }
94
95 slice = PySlice_New(start, end, NULL);
96 Py_DECREF(start);
97 Py_DECREF(end);
98 return slice;
99}
100
Guido van Rossumf2d125b1996-07-30 16:45:48 +0000101int
Martin v. Löwis18e16552006-02-15 17:27:45 +0000102PySlice_GetIndices(PySliceObject *r, Py_ssize_t length,
103 Py_ssize_t *start, Py_ssize_t *stop, Py_ssize_t *step)
Guido van Rossumf2d125b1996-07-30 16:45:48 +0000104{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000105 /* XXX support long ints */
Guido van Rossumf2d125b1996-07-30 16:45:48 +0000106 if (r->step == Py_None) {
107 *step = 1;
108 } else {
Neal Norwitz1fe5f382007-08-31 04:32:55 +0000109 if (!PyLong_Check(r->step)) return -1;
Christian Heimes217cfd12007-12-02 14:31:20 +0000110 *step = PyLong_AsSsize_t(r->step);
Guido van Rossumf2d125b1996-07-30 16:45:48 +0000111 }
112 if (r->start == Py_None) {
113 *start = *step < 0 ? length-1 : 0;
114 } else {
Christian Heimes217cfd12007-12-02 14:31:20 +0000115 if (!PyLong_Check(r->start)) return -1;
116 *start = PyLong_AsSsize_t(r->start);
Guido van Rossumf2d125b1996-07-30 16:45:48 +0000117 if (*start < 0) *start += length;
118 }
119 if (r->stop == Py_None) {
120 *stop = *step < 0 ? -1 : length;
121 } else {
Christian Heimes217cfd12007-12-02 14:31:20 +0000122 if (!PyLong_Check(r->stop)) return -1;
123 *stop = PyLong_AsSsize_t(r->stop);
Guido van Rossumf2d125b1996-07-30 16:45:48 +0000124 if (*stop < 0) *stop += length;
125 }
126 if (*stop > length) return -1;
127 if (*start >= length) return -1;
128 if (*step == 0) return -1;
129 return 0;
130}
131
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000132int
Martin v. Löwis18e16552006-02-15 17:27:45 +0000133PySlice_GetIndicesEx(PySliceObject *r, Py_ssize_t length,
134 Py_ssize_t *start, Py_ssize_t *stop, Py_ssize_t *step, Py_ssize_t *slicelength)
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000135{
136 /* this is harder to get right than you might think */
Michael W. Hudson173f11d2002-11-05 15:28:51 +0000137
Martin v. Löwis18e16552006-02-15 17:27:45 +0000138 Py_ssize_t defstart, defstop;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000139
140 if (r->step == Py_None) {
141 *step = 1;
Michael W. Hudson173f11d2002-11-05 15:28:51 +0000142 }
143 else {
Michael W. Hudsoncbd6fb92002-11-06 15:17:32 +0000144 if (!_PyEval_SliceIndex(r->step, step)) return -1;
145 if (*step == 0) {
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000146 PyErr_SetString(PyExc_ValueError,
147 "slice step cannot be zero");
148 return -1;
149 }
150 }
151
152 defstart = *step < 0 ? length-1 : 0;
153 defstop = *step < 0 ? -1 : length;
154
155 if (r->start == Py_None) {
156 *start = defstart;
Michael W. Hudson173f11d2002-11-05 15:28:51 +0000157 }
158 else {
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000159 if (!_PyEval_SliceIndex(r->start, start)) return -1;
160 if (*start < 0) *start += length;
161 if (*start < 0) *start = (*step < 0) ? -1 : 0;
162 if (*start >= length)
163 *start = (*step < 0) ? length - 1 : length;
164 }
165
166 if (r->stop == Py_None) {
167 *stop = defstop;
Michael W. Hudson173f11d2002-11-05 15:28:51 +0000168 }
169 else {
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000170 if (!_PyEval_SliceIndex(r->stop, stop)) return -1;
171 if (*stop < 0) *stop += length;
Benjamin Petersonfea6a942008-07-02 16:11:42 +0000172 if (*stop < 0) *stop = (*step < 0) ? -1 : 0;
173 if (*stop >= length)
174 *stop = (*step < 0) ? length - 1 : length;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000175 }
Michael W. Hudson173f11d2002-11-05 15:28:51 +0000176
177 if ((*step < 0 && *stop >= *start)
178 || (*step > 0 && *start >= *stop)) {
Michael W. Hudson589dc932002-06-11 13:38:42 +0000179 *slicelength = 0;
180 }
181 else if (*step < 0) {
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000182 *slicelength = (*stop-*start+1)/(*step)+1;
Michael W. Hudson173f11d2002-11-05 15:28:51 +0000183 }
184 else {
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000185 *slicelength = (*stop-*start-1)/(*step)+1;
186 }
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000187
188 return 0;
189}
190
Guido van Rossumbea18cc2002-06-14 20:41:17 +0000191static PyObject *
192slice_new(PyTypeObject *type, PyObject *args, PyObject *kw)
193{
194 PyObject *start, *stop, *step;
195
196 start = stop = step = NULL;
197
Georg Brandl02c42872005-08-26 06:42:30 +0000198 if (!_PyArg_NoKeywords("slice()", kw))
199 return NULL;
200
Raymond Hettingerea3fdf42002-12-29 16:33:45 +0000201 if (!PyArg_UnpackTuple(args, "slice", 1, 3, &start, &stop, &step))
Guido van Rossumbea18cc2002-06-14 20:41:17 +0000202 return NULL;
203
204 /* This swapping of stop and start is to maintain similarity with
205 range(). */
206 if (stop == NULL) {
207 stop = start;
208 start = NULL;
209 }
210 return PySlice_New(start, stop, step);
211}
212
213PyDoc_STRVAR(slice_doc,
214"slice([start,] stop[, step])\n\
215\n\
216Create a slice object. This is used for extended slicing (e.g. a[0:10:2]).");
217
Guido van Rossumf2d125b1996-07-30 16:45:48 +0000218static void
Fred Drake45cfbcc2000-07-09 06:21:27 +0000219slice_dealloc(PySliceObject *r)
Guido van Rossumf2d125b1996-07-30 16:45:48 +0000220{
221 Py_DECREF(r->step);
222 Py_DECREF(r->start);
223 Py_DECREF(r->stop);
Neil Schemenauer7465ad22002-04-12 03:05:37 +0000224 PyObject_Del(r);
Guido van Rossumf2d125b1996-07-30 16:45:48 +0000225}
226
227static PyObject *
Fred Drake45cfbcc2000-07-09 06:21:27 +0000228slice_repr(PySliceObject *r)
Guido van Rossumf2d125b1996-07-30 16:45:48 +0000229{
Walter Dörwald7569dfe2007-05-19 21:49:49 +0000230 return PyUnicode_FromFormat("slice(%R, %R, %R)", r->start, r->stop, r->step);
Guido van Rossumf2d125b1996-07-30 16:45:48 +0000231}
232
Guido van Rossum6f799372001-09-20 20:46:19 +0000233static PyMemberDef slice_members[] = {
Tim Peters6d6c1a32001-08-02 04:15:00 +0000234 {"start", T_OBJECT, offsetof(PySliceObject, start), READONLY},
235 {"stop", T_OBJECT, offsetof(PySliceObject, stop), READONLY},
236 {"step", T_OBJECT, offsetof(PySliceObject, step), READONLY},
237 {0}
238};
Guido van Rossumf2d125b1996-07-30 16:45:48 +0000239
Michael W. Hudsonf0d777c2002-07-19 15:47:06 +0000240static PyObject*
241slice_indices(PySliceObject* self, PyObject* len)
242{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000243 Py_ssize_t ilen, start, stop, step, slicelength;
Michael W. Hudsonf0d777c2002-07-19 15:47:06 +0000244
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000245 ilen = PyNumber_AsSsize_t(len, PyExc_OverflowError);
Michael W. Hudsonf0d777c2002-07-19 15:47:06 +0000246
247 if (ilen == -1 && PyErr_Occurred()) {
248 return NULL;
249 }
250
251 if (PySlice_GetIndicesEx(self, ilen, &start, &stop,
252 &step, &slicelength) < 0) {
253 return NULL;
254 }
255
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000256 return Py_BuildValue("(nnn)", start, stop, step);
Michael W. Hudsonf0d777c2002-07-19 15:47:06 +0000257}
258
259PyDoc_STRVAR(slice_indices_doc,
260"S.indices(len) -> (start, stop, stride)\n\
261\n\
262Assuming a sequence of length len, calculate the start and stop\n\
263indices, and the stride length of the extended slice described by\n\
264S. Out of bounds indices are clipped in a manner consistent with the\n\
265handling of normal slices.");
266
Guido van Rossumd8faa362007-04-27 19:54:29 +0000267static PyObject *
268slice_reduce(PySliceObject* self)
269{
Christian Heimes90aa7642007-12-19 02:45:37 +0000270 return Py_BuildValue("O(OOO)", Py_TYPE(self), self->start, self->stop, self->step);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000271}
272
273PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
274
Michael W. Hudsonf0d777c2002-07-19 15:47:06 +0000275static PyMethodDef slice_methods[] = {
Michael W. Hudson206d8f82002-07-19 15:52:38 +0000276 {"indices", (PyCFunction)slice_indices,
Michael W. Hudsonf0d777c2002-07-19 15:47:06 +0000277 METH_O, slice_indices_doc},
Guido van Rossumd8faa362007-04-27 19:54:29 +0000278 {"__reduce__", (PyCFunction)slice_reduce,
279 METH_NOARGS, reduce_doc},
Michael W. Hudsonf0d777c2002-07-19 15:47:06 +0000280 {NULL, NULL}
281};
282
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000283static PyObject *
284slice_richcompare(PyObject *v, PyObject *w, int op)
Guido van Rossuma1351fb2001-03-20 12:41:34 +0000285{
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000286 PyObject *t1;
287 PyObject *t2;
288 PyObject *res;
Guido van Rossuma1351fb2001-03-20 12:41:34 +0000289
Thomas Wouters3e57b522007-08-28 23:07:26 +0000290 if (!PySlice_Check(v) || !PySlice_Check(w)) {
291 Py_INCREF(Py_NotImplemented);
292 return Py_NotImplemented;
293 }
294
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000295 if (v == w) {
296 /* XXX Do we really need this shortcut?
297 There's a unit test for it, but is that fair? */
298 switch (op) {
299 case Py_EQ:
300 case Py_LE:
301 case Py_GE:
302 res = Py_True;
303 break;
304 default:
305 res = Py_False;
306 break;
307 }
308 Py_INCREF(res);
309 return res;
310 }
Guido van Rossuma1351fb2001-03-20 12:41:34 +0000311
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000312 t1 = PyTuple_New(3);
313 t2 = PyTuple_New(3);
314 if (t1 == NULL || t2 == NULL)
315 return NULL;
316
317 PyTuple_SET_ITEM(t1, 0, ((PySliceObject *)v)->start);
318 PyTuple_SET_ITEM(t1, 1, ((PySliceObject *)v)->stop);
319 PyTuple_SET_ITEM(t1, 2, ((PySliceObject *)v)->step);
320 PyTuple_SET_ITEM(t2, 0, ((PySliceObject *)w)->start);
321 PyTuple_SET_ITEM(t2, 1, ((PySliceObject *)w)->stop);
322 PyTuple_SET_ITEM(t2, 2, ((PySliceObject *)w)->step);
323
324 res = PyObject_RichCompare(t1, t2, op);
325
326 PyTuple_SET_ITEM(t1, 0, NULL);
327 PyTuple_SET_ITEM(t1, 1, NULL);
328 PyTuple_SET_ITEM(t1, 2, NULL);
329 PyTuple_SET_ITEM(t2, 0, NULL);
330 PyTuple_SET_ITEM(t2, 1, NULL);
331 PyTuple_SET_ITEM(t2, 2, NULL);
332
333 Py_DECREF(t1);
334 Py_DECREF(t2);
335
336 return res;
Guido van Rossuma1351fb2001-03-20 12:41:34 +0000337}
Guido van Rossumf2d125b1996-07-30 16:45:48 +0000338
Raymond Hettingerb859c072003-09-05 14:27:30 +0000339static long
340slice_hash(PySliceObject *v)
341{
342 PyErr_SetString(PyExc_TypeError, "unhashable type");
343 return -1L;
344}
345
Guido van Rossumf2d125b1996-07-30 16:45:48 +0000346PyTypeObject PySlice_Type = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +0000347 PyVarObject_HEAD_INIT(&PyType_Type, 0)
Guido van Rossumf2d125b1996-07-30 16:45:48 +0000348 "slice", /* Name of this type */
349 sizeof(PySliceObject), /* Basic object size */
350 0, /* Item size for varobject */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000351 (destructor)slice_dealloc, /* tp_dealloc */
352 0, /* tp_print */
353 0, /* tp_getattr */
354 0, /* tp_setattr */
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000355 0, /* tp_compare */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000356 (reprfunc)slice_repr, /* tp_repr */
357 0, /* tp_as_number */
358 0, /* tp_as_sequence */
359 0, /* tp_as_mapping */
Raymond Hettingerb859c072003-09-05 14:27:30 +0000360 (hashfunc)slice_hash, /* tp_hash */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000361 0, /* tp_call */
362 0, /* tp_str */
363 PyObject_GenericGetAttr, /* tp_getattro */
364 0, /* tp_setattro */
365 0, /* tp_as_buffer */
366 Py_TPFLAGS_DEFAULT, /* tp_flags */
Guido van Rossumbea18cc2002-06-14 20:41:17 +0000367 slice_doc, /* tp_doc */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000368 0, /* tp_traverse */
369 0, /* tp_clear */
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000370 slice_richcompare, /* tp_richcompare */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000371 0, /* tp_weaklistoffset */
372 0, /* tp_iter */
373 0, /* tp_iternext */
Michael W. Hudsonf0d777c2002-07-19 15:47:06 +0000374 slice_methods, /* tp_methods */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000375 slice_members, /* tp_members */
376 0, /* tp_getset */
377 0, /* tp_base */
378 0, /* tp_dict */
Guido van Rossumbea18cc2002-06-14 20:41:17 +0000379 0, /* tp_descr_get */
380 0, /* tp_descr_set */
381 0, /* tp_dictoffset */
382 0, /* tp_init */
383 0, /* tp_alloc */
384 slice_new, /* tp_new */
Guido van Rossumf2d125b1996-07-30 16:45:48 +0000385};