[Python-checkins] cpython: Issue #14687: Cleanup unicode_writer_prepare()

[victor.stinner]

http://hg.python.org/cpython/rev/0a9143d7b097
changeset:   76731:0a9143d7b097
user:        Victor Stinner <victor.stinner at gmail.com>
date:        Thu May 03 13:43:07 2012 +0200
summary:
  Issue #14687: Cleanup unicode_writer_prepare()

"Inline" PyUnicode_Resize(): call directly resize_compact()

files:
  Objects/unicodeobject.c |  44 +++++++++++++---------------
  1 files changed, 20 insertions(+), 24 deletions(-)


diff --git a/Objects/unicodeobject.c b/Objects/unicodeobject.c
--- a/Objects/unicodeobject.c
+++ b/Objects/unicodeobject.c
@@ -654,6 +654,7 @@
     Py_ssize_t new_size;
     int share_wstr;
     PyObject *new_unicode;
+    assert(unicode_modifiable(unicode));
     assert(PyUnicode_IS_READY(unicode));
     assert(PyUnicode_IS_COMPACT(unicode));
 
@@ -690,6 +691,7 @@
     }
     PyUnicode_WRITE(PyUnicode_KIND(unicode), PyUnicode_DATA(unicode),
                     length, 0);
+    assert(_PyUnicode_CheckConsistency(unicode, 0));
     return unicode;
 }
 
@@ -1603,7 +1605,6 @@
         if (new_unicode == NULL)
             return -1;
         *p_unicode = new_unicode;
-        assert(_PyUnicode_CheckConsistency(*p_unicode, 0));
         return 0;
     }
     return resize_inplace(unicode, length);
@@ -13690,6 +13691,7 @@
                        Py_ssize_t length, Py_UCS4 maxchar)
 {
     Py_ssize_t newlen;
+    PyObject *newbuffer;
 
     if (length > PY_SSIZE_T_MAX - writer->pos) {
         PyErr_NoMemory();
@@ -13697,37 +13699,31 @@
     }
     newlen = writer->pos + length;
 
-    if (newlen > writer->length && maxchar > writer->maxchar) {
-        PyObject *newbuffer;
-
-        /* overallocate 25% to limit the number of resize */
-        if (newlen > PY_SSIZE_T_MAX - newlen / 4)
-            writer->length = newlen;
-        else
-            writer->length = newlen + newlen / 4;
-
-        /* resize + widen */
-        newbuffer = PyUnicode_New(writer->length, maxchar);
-        if (newbuffer == NULL)
-            return -1;
-        PyUnicode_CopyCharacters(newbuffer, 0,
-                                 writer->buffer, 0, writer->pos);
-        Py_DECREF(writer->buffer);
-        writer->buffer = newbuffer;
-        unicode_writer_update(writer);
-        return 0;
-    }
     if (newlen > writer->length) {
         /* overallocate 25% to limit the number of resize */
         if (newlen > PY_SSIZE_T_MAX - newlen / 4)
             writer->length = newlen;
         else
             writer->length = newlen + newlen / 4;
-        if (PyUnicode_Resize(&writer->buffer, writer->length) < 0)
-            return -1;
+
+        if (maxchar > writer->maxchar) {
+            /* resize + widen */
+            newbuffer = PyUnicode_New(writer->length, maxchar);
+            if (newbuffer == NULL)
+                return -1;
+            PyUnicode_CopyCharacters(newbuffer, 0,
+                                     writer->buffer, 0, writer->pos);
+            Py_DECREF(writer->buffer);
+        }
+        else {
+            newbuffer = resize_compact(writer->buffer, writer->length);
+            if (newbuffer == NULL)
+                return -1;
+        }
+        writer->buffer = newbuffer;
         unicode_writer_update(writer);
     }
-    if (maxchar > writer->maxchar) {
+    else if (maxchar > writer->maxchar) {
         if (unicode_widen(&writer->buffer, writer->pos, maxchar) < 0)
             return -1;
         unicode_writer_update(writer);

-- 
Repository URL: http://hg.python.org/cpython