Speed tuple comparisons in two ways:
1. Omit the early-out EQ/NE "lengths different?" test. Was unable to find any real code where it triggered, but it always costs. The same is not true of list richcmps, where different-size lists appeared to get compared about half the time. 2. Because tuples are immutable, there's no need to refetch the lengths of both tuples from memory again on each loop trip. BUG ALERT: The tuple (and list) richcmp algorithm is arguably wrong, because it won't believe there's any difference unless Py_EQ returns false for some corresponding elements: >>> class C: ... def __lt__(x, y): return 1 ... __eq__ = __lt__ ... >>> C() < C() 1 >>> (C(),) < (C(),) 0 >>> That doesn't make sense -- provided you believe the defn. of C makes sense.
This commit is contained in:
parent
fab96cc2ff
commit
d7ed3bf552
|
@ -376,6 +376,7 @@ tuplerichcompare(PyObject *v, PyObject *w, int op)
|
|||
{
|
||||
PyTupleObject *vt, *wt;
|
||||
int i;
|
||||
int vlen, wlen;
|
||||
|
||||
if (!PyTuple_Check(v) || !PyTuple_Check(w)) {
|
||||
Py_INCREF(Py_NotImplemented);
|
||||
|
@ -385,19 +386,21 @@ tuplerichcompare(PyObject *v, PyObject *w, int op)
|
|||
vt = (PyTupleObject *)v;
|
||||
wt = (PyTupleObject *)w;
|
||||
|
||||
if (vt->ob_size != wt->ob_size && (op == Py_EQ || op == Py_NE)) {
|
||||
/* Shortcut: if the lengths differ, the tuples differ */
|
||||
PyObject *res;
|
||||
if (op == Py_EQ)
|
||||
res = Py_False;
|
||||
else
|
||||
res = Py_True;
|
||||
Py_INCREF(res);
|
||||
return res;
|
||||
}
|
||||
vlen = vt->ob_size;
|
||||
wlen = wt->ob_size;
|
||||
|
||||
/* Search for the first index where items are different */
|
||||
for (i = 0; i < vt->ob_size && i < wt->ob_size; i++) {
|
||||
/* Note: the corresponding code for lists has an "early out" test
|
||||
* here when op is EQ or NE and the lengths differ. That pays there,
|
||||
* but Tim was unable to find any real code where EQ/NE tuple
|
||||
* compares don't have the same length, so testing for it here would
|
||||
* have cost without benefit.
|
||||
*/
|
||||
|
||||
/* Search for the first index where items are different.
|
||||
* Note that because tuples are immutable, it's safe to reuse
|
||||
* vlen and wlen across the comparison calls.
|
||||
*/
|
||||
for (i = 0; i < vlen && i < wlen; i++) {
|
||||
int k = PyObject_RichCompareBool(vt->ob_item[i],
|
||||
wt->ob_item[i], Py_EQ);
|
||||
if (k < 0)
|
||||
|
@ -406,19 +409,17 @@ tuplerichcompare(PyObject *v, PyObject *w, int op)
|
|||
break;
|
||||
}
|
||||
|
||||
if (i >= vt->ob_size || i >= wt->ob_size) {
|
||||
if (i >= vlen || i >= wlen) {
|
||||
/* No more items to compare -- compare sizes */
|
||||
int vs = vt->ob_size;
|
||||
int ws = wt->ob_size;
|
||||
int cmp;
|
||||
PyObject *res;
|
||||
switch (op) {
|
||||
case Py_LT: cmp = vs < ws; break;
|
||||
case Py_LE: cmp = ws <= ws; break;
|
||||
case Py_EQ: cmp = vs == ws; break;
|
||||
case Py_NE: cmp = vs != ws; break;
|
||||
case Py_GT: cmp = vs > ws; break;
|
||||
case Py_GE: cmp = vs >= ws; break;
|
||||
case Py_LT: cmp = vlen < wlen; break;
|
||||
case Py_LE: cmp = vlen <= wlen; break;
|
||||
case Py_EQ: cmp = vlen == wlen; break;
|
||||
case Py_NE: cmp = vlen != wlen; break;
|
||||
case Py_GT: cmp = vlen > wlen; break;
|
||||
case Py_GE: cmp = vlen >= wlen; break;
|
||||
default: return NULL; /* cannot happen */
|
||||
}
|
||||
if (cmp)
|
||||
|
|
Loading…
Reference in New Issue