Mercurial > cpython-lru_cache-weakref
diff Modules/arraymodule.c @ 67537:2581a17b29af
Issue #5109: array.array constructor will now use fast code when
initial data is provided in an array object with correct type.
| author | Alexander Belopolsky <alexander.belopolsky@gmail.com> |
|---|---|
| date | Tue, 11 Jan 2011 21:44:00 +0000 |
| parents | 48c40709417b |
| children | 3c9232a684ae e2f96bd29252 |
line wrap: on
line diff
--- a/Modules/arraymodule.c Tue Jan 11 21:20:20 2011 +0000 +++ b/Modules/arraymodule.c Tue Jan 11 21:44:00 2011 +0000 @@ -2405,7 +2405,9 @@ || PyByteArray_Check(initial) || PyBytes_Check(initial) || PyTuple_Check(initial) - || ((c=='u') && PyUnicode_Check(initial)))) { + || ((c=='u') && PyUnicode_Check(initial)) + || (array_Check(initial) + && c == ((arrayobject*)initial)->ob_descr->typecode))) { it = PyObject_GetIter(initial); if (it == NULL) return NULL; @@ -2421,17 +2423,20 @@ PyObject *a; Py_ssize_t len; - if (initial == NULL || !(PyList_Check(initial) - || PyTuple_Check(initial))) + if (initial == NULL) len = 0; + else if (PyList_Check(initial)) + len = PyList_GET_SIZE(initial); + else if (PyTuple_Check(initial) || array_Check(initial)) + len = Py_SIZE(initial); else - len = PySequence_Size(initial); + len = 0; a = newarrayobject(type, len, descr); if (a == NULL) return NULL; - if (len > 0) { + if (len > 0 && !array_Check(initial)) { Py_ssize_t i; for (i = 0; i < len; i++) { PyObject *v = @@ -2482,6 +2487,11 @@ self->allocated = Py_SIZE(self); } } + else if (initial != NULL && array_Check(initial)) { + arrayobject *self = (arrayobject *)a; + arrayobject *other = (arrayobject *)initial; + memcpy(self->ob_item, other->ob_item, len * other->ob_descr->itemsize); + } if (it != NULL) { if (array_iter_extend((arrayobject *)a, it) == -1) { Py_DECREF(it);
