runtime/StringBuilder.h

   1 /*
   2  * Copyright (C) 2009 Apple Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
  14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
  17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #ifndef StringBuilder_h
  27 #define StringBuilder_h
  28
  29 #include <wtf/Vector.h>
  30
  31 namespace JSC {
  32
  33 class StringBuilder {
  34 public:
  35     void append(const UChar u)
  36     {
  37         buffer.append(u);
  38     }
  39
  40     void append(const char* str)
  41     {
  42         append(str, strlen(str));
  43     }
  44
  45     void append(const char* str, size_t len)
  46     {
  47         reserveCapacity(buffer.size() + len);
  48         for (size_t i = 0; i < len; i++)
  49             buffer.append(static_cast<unsigned char>(str[i]));
  50     }
  51
  52     void append(const UChar* str, size_t len)
  53     {
  54         buffer.append(str, len);
  55     }
  56
  57     void append(const UString& str)
  58     {
  59         buffer.append(str.data(), str.size());
  60     }
  61
  62     bool isEmpty() { return buffer.isEmpty(); }
  63     void reserveCapacity(size_t newCapacity)
  64     {
  65         if (newCapacity < buffer.capacity())
  66             return;
  67         buffer.reserveCapacity(std::max(newCapacity, buffer.capacity() + buffer.capacity() / 4 + 1));
  68     }
  69     void resize(size_t size) { buffer.resize(size); }
  70     size_t size() const { return buffer.size(); }
  71
  72     UChar operator[](size_t i) const { return buffer.at(i); }
  73
  74     UString build()
  75     {
  76         buffer.shrinkToFit();
  77         ASSERT(buffer.data() || !buffer.size());
  78         return UString::adopt(buffer);
  79     }
  80
  81 protected:
  82     Vector<UChar, 64> buffer;
  83 };
  84
  85 }
  86
  87 #endif