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