[Python-checkins] cpython: I want a super fast 'a' * n!
victor.stinner
python-checkins at python.org
Sat Oct 1 02:49:31 CEST 2011
http://hg.python.org/cpython/rev/ba3e9f5bcbf6
changeset: 72555:ba3e9f5bcbf6
user: Victor Stinner <victor.stinner at haypocalc.com>
date: Sat Oct 01 02:47:29 2011 +0200
summary:
I want a super fast 'a' * n!
* Optimize unicode_repeat() for a special case with memset()
* Simplify integer overflow checking; remove the second check because
PyUnicode_New() already does it and uses a smaller limit (Py_ssize_t vs
size_t)
files:
Objects/unicodeobject.c | 25 ++++++++++---------------
1 files changed, 10 insertions(+), 15 deletions(-)
diff --git a/Objects/unicodeobject.c b/Objects/unicodeobject.c
--- a/Objects/unicodeobject.c
+++ b/Objects/unicodeobject.c
@@ -10583,7 +10583,6 @@
{
PyUnicodeObject *u;
Py_ssize_t nchars, n;
- size_t nbytes, char_size;
if (len < 1) {
Py_INCREF(unicode_empty);
@@ -10599,32 +10598,28 @@
if (PyUnicode_READY(str) == -1)
return NULL;
- /* ensure # of chars needed doesn't overflow int and # of bytes
- * needed doesn't overflow size_t
- */
- nchars = len * PyUnicode_GET_LENGTH(str);
- if (nchars / len != PyUnicode_GET_LENGTH(str)) {
+ if (len > PY_SSIZE_T_MAX / PyUnicode_GET_LENGTH(str)) {
PyErr_SetString(PyExc_OverflowError,
"repeated string is too long");
return NULL;
}
- char_size = PyUnicode_CHARACTER_SIZE(str);
- nbytes = (nchars + 1) * char_size;
- if (nbytes / char_size != (size_t)(nchars + 1)) {
- PyErr_SetString(PyExc_OverflowError,
- "repeated string is too long");
- return NULL;
- }
+ nchars = len * PyUnicode_GET_LENGTH(str);
+
u = (PyUnicodeObject *)PyUnicode_New(nchars, PyUnicode_MAX_CHAR_VALUE(str));
if (!u)
return NULL;
+ assert(PyUnicode_KIND(u) == PyUnicode_KIND(str));
if (PyUnicode_GET_LENGTH(str) == 1) {
const int kind = PyUnicode_KIND(str);
const Py_UCS4 fill_char = PyUnicode_READ(kind, PyUnicode_DATA(str), 0);
void *to = PyUnicode_DATA(u);
- for (n = 0; n < len; ++n)
- PyUnicode_WRITE(kind, to, n, fill_char);
+ if (kind == PyUnicode_1BYTE_KIND)
+ memset(to, (unsigned char)fill_char, len);
+ else {
+ for (n = 0; n < len; ++n)
+ PyUnicode_WRITE(kind, to, n, fill_char);
+ }
}
else {
/* number of characters copied this far */
--
Repository URL: http://hg.python.org/cpython
More information about the Python-checkins
mailing list