Left: | ||
Right: |
OLD | NEW |
---|---|
(Empty) | |
1 #pragma once | |
2 | |
3 #include <cstddef> | |
4 #include <cstring> | |
5 #include <algorithm> | |
6 | |
7 #include <emscripten.h> | |
8 | |
9 #include "debug.h" | |
10 | |
11 inline void String_assert_readonly(bool readOnly); | |
12 | |
13 class String | |
14 { | |
15 friend class DependentString; | |
16 friend class OwnedString; | |
17 | |
18 public: | |
19 typedef char16_t value_type; | |
20 typedef size_t size_type; | |
21 | |
22 // Type flags, stored in the top 2 bits of the mLen member | |
23 static constexpr size_type INVALID = 0xC0000000; | |
24 static constexpr size_type DELETED = 0x80000000; | |
25 static constexpr size_type READ_ONLY = 0x40000000; | |
26 static constexpr size_type READ_WRITE = 0x00000000; | |
27 | |
28 static constexpr size_type FLAGS_MASK = 0xC0000000; | |
29 static constexpr size_type LENGTH_MASK = 0x3FFFFFFF; | |
30 | |
31 static constexpr size_type npos = -1; | |
32 | |
33 protected: | |
34 value_type* mBuf; | |
35 size_type mLen; | |
36 | |
37 explicit String(value_type* buf, size_type len, size_type flags) | |
38 : mBuf(buf), mLen((len & LENGTH_MASK) | flags) | |
39 { | |
40 } | |
41 | |
42 ~String() | |
43 { | |
44 } | |
45 | |
46 void reset(value_type* buf, size_type len, size_type flags) | |
47 { | |
48 mBuf = buf; | |
49 mLen = (len & LENGTH_MASK) | flags; | |
50 } | |
51 | |
52 public: | |
53 size_type length() const | |
54 { | |
55 return mLen & LENGTH_MASK; | |
56 } | |
57 | |
58 bool empty() const | |
59 { | |
60 return !(mLen & LENGTH_MASK); | |
61 } | |
62 | |
63 const value_type* data() const | |
64 { | |
65 return mBuf; | |
66 } | |
67 | |
68 value_type* data() | |
69 { | |
70 String_assert_readonly(is_readOnly()); | |
71 return mBuf; | |
72 } | |
73 | |
74 const value_type& operator[](size_type pos) const | |
75 { | |
76 return mBuf[pos]; | |
77 } | |
78 | |
79 value_type& operator[](size_type pos) | |
80 { | |
81 String_assert_readonly(is_readOnly()); | |
82 return mBuf[pos]; | |
83 } | |
84 | |
85 bool is_readOnly() const | |
86 { | |
87 return (mLen & FLAGS_MASK) != READ_WRITE; | |
88 } | |
89 | |
90 bool equals(const String& other) const | |
91 { | |
92 if (length() != other.length()) | |
93 return false; | |
94 | |
95 return std::memcmp(mBuf, other.mBuf, sizeof(value_type) * length()) == 0; | |
96 } | |
97 | |
98 size_type find(value_type c, size_type pos = 0) const | |
99 { | |
100 for (size_type i = pos; i < length(); ++i) | |
101 if (mBuf[i] == c) | |
102 return i; | |
103 return npos; | |
104 } | |
105 | |
106 size_type find(const String& str, size_type pos = 0) const | |
107 { | |
108 if (pos > LENGTH_MASK || pos + str.length() > length()) | |
109 return npos; | |
110 | |
111 if (!str.length()) | |
112 return pos; | |
113 | |
114 for (; pos + str.length() <= length(); ++pos) | |
115 { | |
116 if (mBuf[pos] == str[0] && | |
117 std::memcmp(mBuf + pos, str.mBuf, sizeof(value_type) * str.length()) = = 0) | |
118 { | |
119 return pos; | |
120 } | |
121 } | |
122 | |
123 return npos; | |
124 } | |
125 | |
126 size_type rfind(value_type c, size_type pos = npos) const | |
127 { | |
128 if (length() == 0) | |
129 return npos; | |
130 | |
131 if (pos >= length()) | |
132 pos = length() - 1; | |
133 | |
134 for (int i = pos; i >= 0; --i) | |
135 if (mBuf[i] == c) | |
136 return i; | |
137 return npos; | |
138 } | |
139 | |
140 bool is_invalid() const | |
141 { | |
142 return (mLen & FLAGS_MASK) == INVALID; | |
143 } | |
144 | |
145 bool is_deleted() const | |
146 { | |
147 return (mLen & FLAGS_MASK) == DELETED; | |
148 } | |
149 | |
150 void toLower() | |
151 { | |
152 size_type len = length(); | |
153 for (size_type i = 0; i < len; ++i) | |
154 { | |
155 value_type currChar = mBuf[i]; | |
156 | |
157 // This should be more efficient with a lookup table but I couldn't measur e | |
158 // any performance difference. | |
159 if (currChar >= u'A' && currChar <= u'Z') | |
160 mBuf[i] = currChar + u'a' - u'A'; | |
161 else if (currChar >= 128) | |
162 { | |
163 // It seems that calling JS is the easiest solution for lowercasing | |
164 // Unicode characters. | |
165 mBuf[i] = EM_ASM_INT({ | |
166 return String.fromCharCode($0).toLowerCase().charCodeAt(0); | |
167 }, currChar); | |
168 } | |
169 } | |
170 } | |
171 }; | |
172 | |
173 class DependentString : public String | |
174 { | |
175 public: | |
176 explicit DependentString() | |
177 : String(nullptr, 0, INVALID) | |
178 { | |
179 } | |
180 | |
181 explicit DependentString(value_type* buf, size_type len) | |
182 : String(buf, len, READ_WRITE) | |
183 { | |
184 } | |
185 | |
186 explicit DependentString(const value_type* buf, size_type len) | |
187 : String(const_cast<value_type*>(buf), len, READ_ONLY) | |
188 { | |
189 } | |
190 | |
191 DependentString(String& str, size_type pos = 0, size_type len = npos) | |
sergei
2017/03/20 17:07:06
Why have explicit keywords been removed here?
Wladimir Palant
2017/03/21 10:10:04
Done.
| |
192 : String( | |
193 str.mBuf + std::min(pos, str.length()), | |
194 std::min(len, str.length() - std::min(pos, str.length())), | |
195 str.is_readOnly() ? READ_ONLY : READ_WRITE | |
196 ) | |
197 { | |
198 } | |
199 | |
200 DependentString(const String& str, size_type pos = 0, | |
201 size_type len = npos) | |
202 : String( | |
203 str.mBuf + std::min(pos, str.length()), | |
204 std::min(len, str.length() - std::min(pos, str.length())), | |
205 READ_ONLY | |
206 ) | |
207 { | |
208 } | |
209 | |
210 void reset(value_type* buf, size_type len) | |
211 { | |
212 *this = DependentString(buf, len); | |
213 } | |
214 | |
215 void reset(const value_type* buf, size_type len) | |
216 { | |
217 *this = DependentString(buf, len); | |
218 } | |
219 | |
220 void reset(String& str, size_type pos = 0, size_type len = npos) | |
221 { | |
222 *this = DependentString(str, pos, len); | |
223 } | |
224 | |
225 void reset(const String& str, size_type pos = 0, size_type len = npos) | |
226 { | |
227 *this = DependentString(str, pos, len); | |
228 } | |
229 | |
230 void erase() | |
231 { | |
232 *this = DependentString(); | |
233 mLen = DELETED; | |
234 } | |
235 }; | |
236 | |
237 inline DependentString operator "" _str(const String::value_type* str, | |
238 String::size_type len) | |
239 { | |
240 return DependentString(str, len); | |
241 } | |
242 | |
243 inline void String_assert_readonly(bool readOnly) | |
244 { | |
245 assert(!readOnly, u"Writing access to a read-only string"_str); | |
246 } | |
247 | |
248 class OwnedString : public String | |
249 { | |
250 private: | |
251 void grow(size_type additionalSize) | |
252 { | |
253 auto newValue = OwnedString(length() + additionalSize); | |
sergei
2017/03/20 17:07:04
I know, it's a copy from my code but
OwnedString n
Wladimir Palant
2017/03/21 10:10:01
Done.
| |
254 if (length() > 0) | |
255 std::memcpy(newValue.mBuf, mBuf, sizeof(value_type) * length()); | |
256 *this = std::move(newValue); | |
257 } | |
258 | |
259 public: | |
260 explicit OwnedString(size_type len = 0) | |
261 : String(nullptr, len, READ_WRITE) | |
262 { | |
263 if (len) | |
264 { | |
265 mBuf = new value_type[length()]; | |
266 annotate_address(mBuf, "String"); | |
267 } | |
268 else | |
269 mBuf = nullptr; | |
270 } | |
271 | |
272 explicit OwnedString(const String& str) | |
273 : OwnedString(str.length()) | |
274 { | |
275 if (length()) | |
276 std::memcpy(mBuf, str.mBuf, sizeof(value_type) * length()); | |
277 } | |
278 | |
279 OwnedString(const OwnedString& str) | |
280 : OwnedString(static_cast<const String&>(str)) | |
281 { | |
282 } | |
283 | |
284 explicit OwnedString(const value_type* str, size_type len) | |
285 : OwnedString(DependentString(str, len)) | |
286 { | |
287 } | |
288 | |
289 explicit OwnedString(OwnedString&& str) | |
290 : OwnedString(0) | |
291 { | |
292 mBuf = str.mBuf; | |
293 mLen = str.mLen; | |
294 str.mBuf = nullptr; | |
295 str.mLen = READ_WRITE | 0; | |
296 } | |
297 | |
298 ~OwnedString() | |
299 { | |
300 if (mBuf) | |
301 delete[] mBuf; | |
302 } | |
303 | |
304 OwnedString& operator=(const String& str) | |
305 { | |
306 *this = std::move(OwnedString(str)); | |
307 return *this; | |
308 } | |
309 | |
310 OwnedString& operator=(const OwnedString& str) | |
311 { | |
312 *this = std::move(OwnedString(str)); | |
313 return *this; | |
314 } | |
315 | |
316 OwnedString& operator=(OwnedString&& str) | |
317 { | |
318 std::swap(mBuf, str.mBuf); | |
319 std::swap(mLen, str.mLen); | |
320 return *this; | |
321 } | |
322 | |
323 void append(const value_type* source, size_type sourceLen) | |
324 { | |
325 if (!sourceLen) | |
326 return; | |
327 | |
328 assert(source, u"Null buffer passed to OwnedString.append()"_str); | |
329 size_t oldLength = length(); | |
330 grow(sourceLen); | |
331 std::memcpy(mBuf + oldLength, source, sizeof(value_type) * sourceLen); | |
332 } | |
333 | |
334 void append(const String& str) | |
335 { | |
336 append(str.mBuf, str.length()); | |
337 } | |
338 | |
339 void append(value_type c) | |
340 { | |
341 append(&c, 1); | |
342 } | |
343 }; | |
OLD | NEW |