Source/JavaScriptCore/runtime/JSStringBuilder.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 JSStringBuilder_h
  27 #define JSStringBuilder_h
  28
  29 #include "ExceptionHelpers.h"
  30 #include "JSString.h"
  31 #include "UStringConcatenate.h"
  32 #include "Vector.h"
  33
  34 namespace JSC {
  35
  36 class JSStringBuilder {
  37 public:
  38     JSStringBuilder()
  39         : m_okay(true)
  40         , m_is8Bit(true)
  41     {
  42     }
  43
  44     void append(const UChar u)
  45     {
  46         if (m_is8Bit) {
  47             if (u < 0xff) {
  48                 LChar c = u;
  49                 m_okay &= buffer8.tryAppend(&c, 1);
  50                 return;
  51             }
  52             upConvert();
  53         }
  54         m_okay &= buffer16.tryAppend(&u, 1);
  55     }
  56
  57     void append(const char* str)
  58     {
  59         append(str, strlen(str));
  60     }
  61
  62     void append(const char* str, size_t len)
  63     {
  64         if (m_is8Bit) {
  65             m_okay &= buffer8.tryAppend(reinterpret_cast<const LChar*>(str), len);
  66             return;
  67         }
  68         m_okay &= buffer8.tryReserveCapacity(buffer16.size() + len);
  69         for (size_t i = 0; i < len; i++) {
  70             UChar u = static_cast<unsigned char>(str[i]);
  71             m_okay &= buffer16.tryAppend(&u, 1);
  72         }
  73     }
  74
  75     void append(const UChar* str, size_t len)
  76     {
  77         if (m_is8Bit)
  78             upConvert(); // FIXME: We could check character by character its size.
  79         m_okay &= buffer16.tryAppend(str, len);
  80     }
  81
  82     void append(const UString& str)
  83     {
  84         unsigned length = str.length();
  85
  86         if (!length)
  87             return;
  88
  89         if (m_is8Bit) {
  90             if (str.is8Bit()) {
  91                 m_okay &= buffer8.tryAppend(str.characters8(), length);
  92                 return;
  93             }
  94             upConvert();
  95         }
  96         m_okay &= buffer16.tryAppend(str.characters(), length);
  97     }
  98
  99     void upConvert()
 100     {
 101         ASSERT(m_is8Bit);
 102         size_t len = buffer8.size();
 103
 104         for (size_t i = 0; i < len; i++)
 105             buffer16.append(buffer8[i]);
 106
 107         buffer8.clear();
 108         m_is8Bit = false;
 109     }
 110
 111     JSValue build(ExecState* exec)
 112     {
 113         if (!m_okay)
 114             return throwOutOfMemoryError(exec);
 115         if (m_is8Bit) {
 116             buffer8.shrinkToFit();
 117             if (!buffer8.data())
 118                 return throwOutOfMemoryError(exec);
 119             return jsString(exec, UString::adopt(buffer8));
 120         }
 121         buffer16.shrinkToFit();
 122         if (!buffer16.data())
 123             return throwOutOfMemoryError(exec);
 124         return jsString(exec, UString::adopt(buffer16));
 125     }
 126
 127 protected:
 128     Vector<LChar, 64> buffer8;
 129     Vector<UChar, 64> buffer16;
 130     bool m_okay;
 131     bool m_is8Bit;
 132 };
 133
 134 template<typename StringType1, typename StringType2>
 135 inline JSValue jsMakeNontrivialString(ExecState* exec, StringType1 string1, StringType2 string2)
 136 {
 137     PassRefPtr<StringImpl> result = WTF::tryMakeString(string1, string2);
 138     if (!result)
 139         return throwOutOfMemoryError(exec);
 140     return jsNontrivialString(exec, result);
 141 }
 142
 143 template<typename StringType1, typename StringType2, typename StringType3>
 144 inline JSValue jsMakeNontrivialString(ExecState* exec, StringType1 string1, StringType2 string2, StringType3 string3)
 145 {
 146     PassRefPtr<StringImpl> result = WTF::tryMakeString(string1, string2, string3);
 147     if (!result)
 148         return throwOutOfMemoryError(exec);
 149     return jsNontrivialString(exec, result);
 150 }
 151
 152 template<typename StringType1, typename StringType2, typename StringType3, typename StringType4>
 153 inline JSValue jsMakeNontrivialString(ExecState* exec, StringType1 string1, StringType2 string2, StringType3 string3, StringType4 string4)
 154 {
 155     PassRefPtr<StringImpl> result = WTF::tryMakeString(string1, string2, string3, string4);
 156     if (!result)
 157         return throwOutOfMemoryError(exec);
 158     return jsNontrivialString(exec, result);
 159 }
 160
 161 template<typename StringType1, typename StringType2, typename StringType3, typename StringType4, typename StringType5>
 162 inline JSValue jsMakeNontrivialString(ExecState* exec, StringType1 string1, StringType2 string2, StringType3 string3, StringType4 string4, StringType5 string5)
 163 {
 164     PassRefPtr<StringImpl> result = WTF::tryMakeString(string1, string2, string3, string4, string5);
 165     if (!result)
 166         return throwOutOfMemoryError(exec);
 167     return jsNontrivialString(exec, result);
 168 }
 169
 170 template<typename StringType1, typename StringType2, typename StringType3, typename StringType4, typename StringType5, typename StringType6>
 171 inline JSValue jsMakeNontrivialString(ExecState* exec, StringType1 string1, StringType2 string2, StringType3 string3, StringType4 string4, StringType5 string5, StringType6 string6)
 172 {
 173     PassRefPtr<StringImpl> result = WTF::tryMakeString(string1, string2, string3, string4, string5, string6);
 174     if (!result)
 175         return throwOutOfMemoryError(exec);
 176     return jsNontrivialString(exec, result);
 177 }
 178
 179 }
 180
 181 #endif