initial import
[vuplus_webkit] / 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     {
41     }
42
43     void append(const UChar u)
44     {
45         m_okay &= buffer.tryAppend(&u, 1);
46     }
47
48     void append(const char* str)
49     {
50         append(str, strlen(str));
51     }
52
53     void append(const char* str, size_t len)
54     {
55         m_okay &= buffer.tryReserveCapacity(buffer.size() + len);
56         for (size_t i = 0; i < len; i++) {
57             UChar u = static_cast<unsigned char>(str[i]);
58             m_okay &= buffer.tryAppend(&u, 1);
59         }
60     }
61
62     void append(const UChar* str, size_t len)
63     {
64         m_okay &= buffer.tryAppend(str, len);
65     }
66
67     void append(const UString& str)
68     {
69         m_okay &= buffer.tryAppend(str.characters(), str.length());
70     }
71
72     JSValue build(ExecState* exec)
73     {
74         if (!m_okay)
75             return throwOutOfMemoryError(exec);
76         buffer.shrinkToFit();
77         if (!buffer.data())
78             return throwOutOfMemoryError(exec);
79         return jsString(exec, UString::adopt(buffer));
80     }
81
82 protected:
83     Vector<UChar, 64> buffer;
84     bool m_okay;
85 };
86
87 template<typename StringType1, typename StringType2>
88 inline JSValue jsMakeNontrivialString(ExecState* exec, StringType1 string1, StringType2 string2)
89 {
90     PassRefPtr<StringImpl> result = WTF::tryMakeString(string1, string2);
91     if (!result)
92         return throwOutOfMemoryError(exec);
93     return jsNontrivialString(exec, result);
94 }
95
96 template<typename StringType1, typename StringType2, typename StringType3>
97 inline JSValue jsMakeNontrivialString(ExecState* exec, StringType1 string1, StringType2 string2, StringType3 string3)
98 {
99     PassRefPtr<StringImpl> result = WTF::tryMakeString(string1, string2, string3);
100     if (!result)
101         return throwOutOfMemoryError(exec);
102     return jsNontrivialString(exec, result);
103 }
104
105 template<typename StringType1, typename StringType2, typename StringType3, typename StringType4>
106 inline JSValue jsMakeNontrivialString(ExecState* exec, StringType1 string1, StringType2 string2, StringType3 string3, StringType4 string4)
107 {
108     PassRefPtr<StringImpl> result = WTF::tryMakeString(string1, string2, string3, string4);
109     if (!result)
110         return throwOutOfMemoryError(exec);
111     return jsNontrivialString(exec, result);
112 }
113
114 template<typename StringType1, typename StringType2, typename StringType3, typename StringType4, typename StringType5>
115 inline JSValue jsMakeNontrivialString(ExecState* exec, StringType1 string1, StringType2 string2, StringType3 string3, StringType4 string4, StringType5 string5)
116 {
117     PassRefPtr<StringImpl> result = WTF::tryMakeString(string1, string2, string3, string4, string5);
118     if (!result)
119         return throwOutOfMemoryError(exec);
120     return jsNontrivialString(exec, result);
121 }
122
123 template<typename StringType1, typename StringType2, typename StringType3, typename StringType4, typename StringType5, typename StringType6>
124 inline JSValue jsMakeNontrivialString(ExecState* exec, StringType1 string1, StringType2 string2, StringType3 string3, StringType4 string4, StringType5 string5, StringType6 string6)
125 {
126     PassRefPtr<StringImpl> result = WTF::tryMakeString(string1, string2, string3, string4, string5, string6);
127     if (!result)
128         return throwOutOfMemoryError(exec);
129     return jsNontrivialString(exec, result);
130 }
131
132 }
133
134 #endif