2004-01-05 18:13:35 +08:00
|
|
|
/* Bisection algorithms. Drop in replacement for bisect.py
|
|
|
|
|
|
|
|
Converted to C by Dmitry Vasiliev (dima at hlabs.spb.ru).
|
|
|
|
*/
|
|
|
|
|
2012-05-16 20:37:54 +08:00
|
|
|
#define PY_SSIZE_T_CLEAN
|
2004-01-05 18:13:35 +08:00
|
|
|
#include "Python.h"
|
|
|
|
|
Merged revisions 65209-65216,65225-65226,65233,65239,65246-65247,65255-65256 via svnmerge from
svn+ssh://pythondev@svn.python.org/python/trunk
........
r65209 | raymond.hettinger | 2008-07-23 19:08:18 -0500 (Wed, 23 Jul 2008) | 1 line
Finish-up the partial conversion from int to Py_ssize_t for deque indices and length.
........
r65210 | raymond.hettinger | 2008-07-23 19:53:49 -0500 (Wed, 23 Jul 2008) | 1 line
Parse to the correct datatype.
........
r65211 | benjamin.peterson | 2008-07-23 21:27:46 -0500 (Wed, 23 Jul 2008) | 1 line
fix spacing
........
r65212 | benjamin.peterson | 2008-07-23 21:31:28 -0500 (Wed, 23 Jul 2008) | 1 line
fix markup
........
r65213 | benjamin.peterson | 2008-07-23 21:45:37 -0500 (Wed, 23 Jul 2008) | 1 line
add some documentation for 2to3
........
r65214 | raymond.hettinger | 2008-07-24 00:38:48 -0500 (Thu, 24 Jul 2008) | 1 line
Finish conversion from int to Py_ssize_t.
........
r65215 | raymond.hettinger | 2008-07-24 02:04:55 -0500 (Thu, 24 Jul 2008) | 1 line
Convert from long to Py_ssize_t.
........
r65216 | georg.brandl | 2008-07-24 02:09:21 -0500 (Thu, 24 Jul 2008) | 2 lines
Fix indentation.
........
r65225 | benjamin.peterson | 2008-07-25 11:55:37 -0500 (Fri, 25 Jul 2008) | 1 line
teach .bzrignore about doc tools
........
r65226 | benjamin.peterson | 2008-07-25 12:02:11 -0500 (Fri, 25 Jul 2008) | 1 line
document default value for fillvalue
........
r65233 | raymond.hettinger | 2008-07-25 13:43:33 -0500 (Fri, 25 Jul 2008) | 1 line
Issue 1592: Better error reporting for operations on closed shelves.
........
r65239 | benjamin.peterson | 2008-07-25 16:59:53 -0500 (Fri, 25 Jul 2008) | 1 line
fix indentation
........
r65246 | andrew.kuchling | 2008-07-26 08:08:19 -0500 (Sat, 26 Jul 2008) | 1 line
This sentence continues to bug me; rewrite it for the second time
........
r65247 | andrew.kuchling | 2008-07-26 08:09:06 -0500 (Sat, 26 Jul 2008) | 1 line
Remove extra words
........
r65255 | skip.montanaro | 2008-07-26 19:49:02 -0500 (Sat, 26 Jul 2008) | 3 lines
Close issue 3437 - missing state change when Allow lines are processed.
Adds test cases which use Allow: as well.
........
r65256 | skip.montanaro | 2008-07-26 19:50:41 -0500 (Sat, 26 Jul 2008) | 2 lines
note robotparser bug fix.
........
2008-08-01 00:23:04 +08:00
|
|
|
static Py_ssize_t
|
2006-02-16 22:30:23 +08:00
|
|
|
internal_bisect_right(PyObject *list, PyObject *item, Py_ssize_t lo, Py_ssize_t hi)
|
2004-01-05 18:13:35 +08:00
|
|
|
{
|
2010-05-09 23:52:27 +08:00
|
|
|
PyObject *litem;
|
|
|
|
Py_ssize_t mid, res;
|
|
|
|
|
|
|
|
if (lo < 0) {
|
|
|
|
PyErr_SetString(PyExc_ValueError, "lo must be non-negative");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
if (hi == -1) {
|
|
|
|
hi = PySequence_Size(list);
|
|
|
|
if (hi < 0)
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
while (lo < hi) {
|
2012-04-15 23:30:35 +08:00
|
|
|
/* The (size_t)cast ensures that the addition and subsequent division
|
|
|
|
are performed as unsigned operations, avoiding difficulties from
|
|
|
|
signed overflow. (See issue 13496.) */
|
|
|
|
mid = ((size_t)lo + hi) / 2;
|
2010-05-09 23:52:27 +08:00
|
|
|
litem = PySequence_GetItem(list, mid);
|
|
|
|
if (litem == NULL)
|
|
|
|
return -1;
|
|
|
|
res = PyObject_RichCompareBool(item, litem, Py_LT);
|
|
|
|
Py_DECREF(litem);
|
|
|
|
if (res < 0)
|
|
|
|
return -1;
|
|
|
|
if (res)
|
|
|
|
hi = mid;
|
|
|
|
else
|
|
|
|
lo = mid + 1;
|
|
|
|
}
|
|
|
|
return lo;
|
2004-01-05 18:13:35 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *
|
2005-10-05 19:39:12 +08:00
|
|
|
bisect_right(PyObject *self, PyObject *args, PyObject *kw)
|
2004-01-05 18:13:35 +08:00
|
|
|
{
|
2010-05-09 23:52:27 +08:00
|
|
|
PyObject *list, *item;
|
|
|
|
Py_ssize_t lo = 0;
|
|
|
|
Py_ssize_t hi = -1;
|
|
|
|
Py_ssize_t index;
|
|
|
|
static char *keywords[] = {"a", "x", "lo", "hi", NULL};
|
|
|
|
|
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kw, "OO|nn:bisect_right",
|
|
|
|
keywords, &list, &item, &lo, &hi))
|
|
|
|
return NULL;
|
|
|
|
index = internal_bisect_right(list, item, lo, hi);
|
|
|
|
if (index < 0)
|
|
|
|
return NULL;
|
|
|
|
return PyLong_FromSsize_t(index);
|
2004-01-05 18:13:35 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
PyDoc_STRVAR(bisect_right_doc,
|
2005-10-05 19:39:12 +08:00
|
|
|
"bisect_right(a, x[, lo[, hi]]) -> index\n\
|
2004-01-05 18:13:35 +08:00
|
|
|
\n\
|
|
|
|
Return the index where to insert item x in list a, assuming a is sorted.\n\
|
|
|
|
\n\
|
|
|
|
The return value i is such that all e in a[:i] have e <= x, and all e in\n\
|
|
|
|
a[i:] have e > x. So if x already appears in the list, i points just\n\
|
|
|
|
beyond the rightmost x already there\n\
|
|
|
|
\n\
|
|
|
|
Optional args lo (default 0) and hi (default len(a)) bound the\n\
|
|
|
|
slice of a to be searched.\n");
|
|
|
|
|
|
|
|
static PyObject *
|
2005-10-05 19:39:12 +08:00
|
|
|
insort_right(PyObject *self, PyObject *args, PyObject *kw)
|
2004-01-05 18:13:35 +08:00
|
|
|
{
|
2010-05-09 23:52:27 +08:00
|
|
|
PyObject *list, *item, *result;
|
|
|
|
Py_ssize_t lo = 0;
|
|
|
|
Py_ssize_t hi = -1;
|
|
|
|
Py_ssize_t index;
|
|
|
|
static char *keywords[] = {"a", "x", "lo", "hi", NULL};
|
|
|
|
|
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kw, "OO|nn:insort_right",
|
|
|
|
keywords, &list, &item, &lo, &hi))
|
|
|
|
return NULL;
|
|
|
|
index = internal_bisect_right(list, item, lo, hi);
|
|
|
|
if (index < 0)
|
|
|
|
return NULL;
|
|
|
|
if (PyList_CheckExact(list)) {
|
|
|
|
if (PyList_Insert(list, index, item) < 0)
|
|
|
|
return NULL;
|
|
|
|
} else {
|
2010-09-01 16:56:10 +08:00
|
|
|
result = PyObject_CallMethod(list, "insert", "nO", index, item);
|
2010-05-09 23:52:27 +08:00
|
|
|
if (result == NULL)
|
|
|
|
return NULL;
|
|
|
|
Py_DECREF(result);
|
|
|
|
}
|
|
|
|
|
|
|
|
Py_RETURN_NONE;
|
2004-01-05 18:13:35 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
PyDoc_STRVAR(insort_right_doc,
|
2005-10-05 19:39:12 +08:00
|
|
|
"insort_right(a, x[, lo[, hi]])\n\
|
2004-01-05 18:13:35 +08:00
|
|
|
\n\
|
|
|
|
Insert item x in list a, and keep it sorted assuming a is sorted.\n\
|
|
|
|
\n\
|
|
|
|
If x is already in a, insert it to the right of the rightmost x.\n\
|
|
|
|
\n\
|
|
|
|
Optional args lo (default 0) and hi (default len(a)) bound the\n\
|
|
|
|
slice of a to be searched.\n");
|
|
|
|
|
Merged revisions 65209-65216,65225-65226,65233,65239,65246-65247,65255-65256 via svnmerge from
svn+ssh://pythondev@svn.python.org/python/trunk
........
r65209 | raymond.hettinger | 2008-07-23 19:08:18 -0500 (Wed, 23 Jul 2008) | 1 line
Finish-up the partial conversion from int to Py_ssize_t for deque indices and length.
........
r65210 | raymond.hettinger | 2008-07-23 19:53:49 -0500 (Wed, 23 Jul 2008) | 1 line
Parse to the correct datatype.
........
r65211 | benjamin.peterson | 2008-07-23 21:27:46 -0500 (Wed, 23 Jul 2008) | 1 line
fix spacing
........
r65212 | benjamin.peterson | 2008-07-23 21:31:28 -0500 (Wed, 23 Jul 2008) | 1 line
fix markup
........
r65213 | benjamin.peterson | 2008-07-23 21:45:37 -0500 (Wed, 23 Jul 2008) | 1 line
add some documentation for 2to3
........
r65214 | raymond.hettinger | 2008-07-24 00:38:48 -0500 (Thu, 24 Jul 2008) | 1 line
Finish conversion from int to Py_ssize_t.
........
r65215 | raymond.hettinger | 2008-07-24 02:04:55 -0500 (Thu, 24 Jul 2008) | 1 line
Convert from long to Py_ssize_t.
........
r65216 | georg.brandl | 2008-07-24 02:09:21 -0500 (Thu, 24 Jul 2008) | 2 lines
Fix indentation.
........
r65225 | benjamin.peterson | 2008-07-25 11:55:37 -0500 (Fri, 25 Jul 2008) | 1 line
teach .bzrignore about doc tools
........
r65226 | benjamin.peterson | 2008-07-25 12:02:11 -0500 (Fri, 25 Jul 2008) | 1 line
document default value for fillvalue
........
r65233 | raymond.hettinger | 2008-07-25 13:43:33 -0500 (Fri, 25 Jul 2008) | 1 line
Issue 1592: Better error reporting for operations on closed shelves.
........
r65239 | benjamin.peterson | 2008-07-25 16:59:53 -0500 (Fri, 25 Jul 2008) | 1 line
fix indentation
........
r65246 | andrew.kuchling | 2008-07-26 08:08:19 -0500 (Sat, 26 Jul 2008) | 1 line
This sentence continues to bug me; rewrite it for the second time
........
r65247 | andrew.kuchling | 2008-07-26 08:09:06 -0500 (Sat, 26 Jul 2008) | 1 line
Remove extra words
........
r65255 | skip.montanaro | 2008-07-26 19:49:02 -0500 (Sat, 26 Jul 2008) | 3 lines
Close issue 3437 - missing state change when Allow lines are processed.
Adds test cases which use Allow: as well.
........
r65256 | skip.montanaro | 2008-07-26 19:50:41 -0500 (Sat, 26 Jul 2008) | 2 lines
note robotparser bug fix.
........
2008-08-01 00:23:04 +08:00
|
|
|
static Py_ssize_t
|
|
|
|
internal_bisect_left(PyObject *list, PyObject *item, Py_ssize_t lo, Py_ssize_t hi)
|
2004-01-05 18:13:35 +08:00
|
|
|
{
|
2010-05-09 23:52:27 +08:00
|
|
|
PyObject *litem;
|
|
|
|
Py_ssize_t mid, res;
|
|
|
|
|
|
|
|
if (lo < 0) {
|
|
|
|
PyErr_SetString(PyExc_ValueError, "lo must be non-negative");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
if (hi == -1) {
|
|
|
|
hi = PySequence_Size(list);
|
|
|
|
if (hi < 0)
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
while (lo < hi) {
|
2012-04-15 23:30:35 +08:00
|
|
|
/* The (size_t)cast ensures that the addition and subsequent division
|
|
|
|
are performed as unsigned operations, avoiding difficulties from
|
|
|
|
signed overflow. (See issue 13496.) */
|
|
|
|
mid = ((size_t)lo + hi) / 2;
|
2010-05-09 23:52:27 +08:00
|
|
|
litem = PySequence_GetItem(list, mid);
|
|
|
|
if (litem == NULL)
|
|
|
|
return -1;
|
|
|
|
res = PyObject_RichCompareBool(litem, item, Py_LT);
|
|
|
|
Py_DECREF(litem);
|
|
|
|
if (res < 0)
|
|
|
|
return -1;
|
|
|
|
if (res)
|
|
|
|
lo = mid + 1;
|
|
|
|
else
|
|
|
|
hi = mid;
|
|
|
|
}
|
|
|
|
return lo;
|
2004-01-05 18:13:35 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *
|
2005-10-05 19:39:12 +08:00
|
|
|
bisect_left(PyObject *self, PyObject *args, PyObject *kw)
|
2004-01-05 18:13:35 +08:00
|
|
|
{
|
2010-05-09 23:52:27 +08:00
|
|
|
PyObject *list, *item;
|
|
|
|
Py_ssize_t lo = 0;
|
|
|
|
Py_ssize_t hi = -1;
|
|
|
|
Py_ssize_t index;
|
|
|
|
static char *keywords[] = {"a", "x", "lo", "hi", NULL};
|
|
|
|
|
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kw, "OO|nn:bisect_left",
|
|
|
|
keywords, &list, &item, &lo, &hi))
|
|
|
|
return NULL;
|
|
|
|
index = internal_bisect_left(list, item, lo, hi);
|
|
|
|
if (index < 0)
|
|
|
|
return NULL;
|
|
|
|
return PyLong_FromSsize_t(index);
|
2004-01-05 18:13:35 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
PyDoc_STRVAR(bisect_left_doc,
|
2005-10-05 19:39:12 +08:00
|
|
|
"bisect_left(a, x[, lo[, hi]]) -> index\n\
|
2004-01-05 18:13:35 +08:00
|
|
|
\n\
|
|
|
|
Return the index where to insert item x in list a, assuming a is sorted.\n\
|
|
|
|
\n\
|
|
|
|
The return value i is such that all e in a[:i] have e < x, and all e in\n\
|
|
|
|
a[i:] have e >= x. So if x already appears in the list, i points just\n\
|
|
|
|
before the leftmost x already there.\n\
|
|
|
|
\n\
|
|
|
|
Optional args lo (default 0) and hi (default len(a)) bound the\n\
|
|
|
|
slice of a to be searched.\n");
|
|
|
|
|
|
|
|
static PyObject *
|
2005-10-05 19:39:12 +08:00
|
|
|
insort_left(PyObject *self, PyObject *args, PyObject *kw)
|
2004-01-05 18:13:35 +08:00
|
|
|
{
|
2010-05-09 23:52:27 +08:00
|
|
|
PyObject *list, *item, *result;
|
|
|
|
Py_ssize_t lo = 0;
|
|
|
|
Py_ssize_t hi = -1;
|
|
|
|
Py_ssize_t index;
|
|
|
|
static char *keywords[] = {"a", "x", "lo", "hi", NULL};
|
|
|
|
|
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kw, "OO|nn:insort_left",
|
|
|
|
keywords, &list, &item, &lo, &hi))
|
|
|
|
return NULL;
|
|
|
|
index = internal_bisect_left(list, item, lo, hi);
|
|
|
|
if (index < 0)
|
|
|
|
return NULL;
|
|
|
|
if (PyList_CheckExact(list)) {
|
|
|
|
if (PyList_Insert(list, index, item) < 0)
|
|
|
|
return NULL;
|
|
|
|
} else {
|
2012-05-16 20:37:54 +08:00
|
|
|
result = PyObject_CallMethod(list, "insert", "nO", index, item);
|
2010-05-09 23:52:27 +08:00
|
|
|
if (result == NULL)
|
|
|
|
return NULL;
|
|
|
|
Py_DECREF(result);
|
|
|
|
}
|
|
|
|
|
|
|
|
Py_RETURN_NONE;
|
2004-01-05 18:13:35 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
PyDoc_STRVAR(insort_left_doc,
|
2005-10-05 19:39:12 +08:00
|
|
|
"insort_left(a, x[, lo[, hi]])\n\
|
2004-01-05 18:13:35 +08:00
|
|
|
\n\
|
|
|
|
Insert item x in list a, and keep it sorted assuming a is sorted.\n\
|
|
|
|
\n\
|
|
|
|
If x is already in a, insert it to the left of the leftmost x.\n\
|
|
|
|
\n\
|
|
|
|
Optional args lo (default 0) and hi (default len(a)) bound the\n\
|
|
|
|
slice of a to be searched.\n");
|
|
|
|
|
|
|
|
PyDoc_STRVAR(bisect_doc, "Alias for bisect_right().\n");
|
|
|
|
PyDoc_STRVAR(insort_doc, "Alias for insort_right().\n");
|
|
|
|
|
|
|
|
static PyMethodDef bisect_methods[] = {
|
2010-05-09 23:52:27 +08:00
|
|
|
{"bisect_right", (PyCFunction)bisect_right,
|
|
|
|
METH_VARARGS|METH_KEYWORDS, bisect_right_doc},
|
|
|
|
{"bisect", (PyCFunction)bisect_right,
|
|
|
|
METH_VARARGS|METH_KEYWORDS, bisect_doc},
|
|
|
|
{"insort_right", (PyCFunction)insort_right,
|
|
|
|
METH_VARARGS|METH_KEYWORDS, insort_right_doc},
|
|
|
|
{"insort", (PyCFunction)insort_right,
|
|
|
|
METH_VARARGS|METH_KEYWORDS, insort_doc},
|
|
|
|
{"bisect_left", (PyCFunction)bisect_left,
|
|
|
|
METH_VARARGS|METH_KEYWORDS, bisect_left_doc},
|
|
|
|
{"insort_left", (PyCFunction)insort_left,
|
|
|
|
METH_VARARGS|METH_KEYWORDS, insort_left_doc},
|
|
|
|
{NULL, NULL} /* sentinel */
|
2004-01-05 18:13:35 +08:00
|
|
|
};
|
|
|
|
|
|
|
|
PyDoc_STRVAR(module_doc,
|
|
|
|
"Bisection algorithms.\n\
|
|
|
|
\n\
|
|
|
|
This module provides support for maintaining a list in sorted order without\n\
|
|
|
|
having to sort the list after each insertion. For long lists of items with\n\
|
|
|
|
expensive comparison operations, this can be an improvement over the more\n\
|
|
|
|
common approach.\n");
|
|
|
|
|
2008-06-11 13:26:20 +08:00
|
|
|
|
|
|
|
static struct PyModuleDef _bisectmodule = {
|
2010-05-09 23:52:27 +08:00
|
|
|
PyModuleDef_HEAD_INIT,
|
|
|
|
"_bisect",
|
|
|
|
module_doc,
|
|
|
|
-1,
|
|
|
|
bisect_methods,
|
|
|
|
NULL,
|
|
|
|
NULL,
|
|
|
|
NULL,
|
|
|
|
NULL
|
2008-06-11 13:26:20 +08:00
|
|
|
};
|
|
|
|
|
2004-01-05 18:13:35 +08:00
|
|
|
PyMODINIT_FUNC
|
2008-06-11 13:26:20 +08:00
|
|
|
PyInit__bisect(void)
|
2004-01-05 18:13:35 +08:00
|
|
|
{
|
2010-05-09 23:52:27 +08:00
|
|
|
return PyModule_Create(&_bisectmodule);
|
2004-01-05 18:13:35 +08:00
|
|
|
}
|