traverseda
/
cpython
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

needforspeed: stringlib refactoring: use stringlib/find for string find

This commit is contained in:
Fredrik Lundh 2006-05-26 19:48:07 +00:00
parent cc7570fd90
commit e6e43c867d
3 changed files with 26 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
Objects/stringlib/find.h
View File

@ -9,29 +9,38 @@
Py_LOCAL(Py_ssize_t)
stringlib_find(const STRINGLIB_CHAR* str, Py_ssize_t str_len,
const STRINGLIB_CHAR* sub, Py_ssize_t sub_len)
const STRINGLIB_CHAR* sub, Py_ssize_t sub_len,
Py_ssize_t offset)
{
if (sub_len == 0)
return 0;
Py_ssize_t pos;
return fastsearch(str, str_len, sub, sub_len, FAST_SEARCH);
if (sub_len == 0)
return offset;
pos = fastsearch(str, str_len, sub, sub_len, FAST_SEARCH);
if (pos >= 0)
pos += offset;
return pos;
}
Py_LOCAL(Py_ssize_t)
stringlib_rfind(const STRINGLIB_CHAR* str, Py_ssize_t str_len,
const STRINGLIB_CHAR* sub, Py_ssize_t sub_len)
const STRINGLIB_CHAR* sub, Py_ssize_t sub_len,
Py_ssize_t offset)
{
Py_ssize_t pos;
/* XXX - create reversefastsearch helper! */
if (sub_len == 0)
pos = str_len;
pos = str_len + offset;
else {
Py_ssize_t j;
pos = -1;
for (j = str_len - sub_len; j >= 0; --j)
if (STRINGLIB_CMP(str+j, sub, sub_len) == 0) {
pos = j;
pos = j + offset;
break;
}
}

25
Objects/stringobject.c
View File

@ -775,6 +775,8 @@ PyString_AsStringAndSize(register PyObject *obj,
#define STRINGLIB_EMPTY nullstring
#include "stringlib/fastsearch.h"
#include "stringlib/find.h"
#include "stringlib/partition.h"
/* -------------------------------------------------------------------- */
@ -1876,25 +1878,10 @@ string_find_internal(PyStringObject *self, PyObject *args, int dir)
string_adjust_indices(&i, &last, len);
if (n == 0)
return (dir > 0) ? i : last;
if (dir > 0) {
Py_ssize_t pos = fastsearch(s + i, last - i, sub, n,
FAST_SEARCH);
if (pos < 0)
return pos;
return pos + i;
} else {
Py_ssize_t j;
if (n == 0 && i <= last)
return last;
for (j = last-n; j >= i; --j)
if (s[j] == sub[0] && memcmp(&s[j], sub, n) == 0)
return j;
}
return -1;
if (dir > 0)
return stringlib_find(s+i, last-i, sub, n, i);
else
return stringlib_rfind(s+i, last-i, sub, n, i);
}

9
Objects/unicodeobject.c
View File

@ -3967,16 +3967,15 @@ Py_ssize_t PyUnicode_Find(PyObject *str,
if (direction > 0)
result = stringlib_find(
str_obj->str + start, end - start, sub_obj->str, sub_obj->length
str_obj->str + start, end - start, sub_obj->str, sub_obj->length,
start
);
else
result = stringlib_rfind(
str_obj->str + start, end - start, sub_obj->str, sub_obj->length
str_obj->str + start, end - start, sub_obj->str, sub_obj->length,
start
);
if (result >= 0)
result += start;
Py_DECREF(str_obj);
Py_DECREF(sub_obj);
return result;