• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright 2006 The Android Open Source Project
3  *
4  * Use of this source code is governed by a BSD-style license that can be
5  * found in the LICENSE file.
6  */
7 
8 #ifndef SkString_DEFINED
9 #define SkString_DEFINED
10 
11 #include "include/core/SkRefCnt.h"
12 #include "include/core/SkScalar.h"
13 #include "include/core/SkTypes.h"
14 #include "include/private/SkMalloc.h"
15 #include "include/private/SkTArray.h"
16 #include "include/private/SkTo.h"
17 
18 #include <stdarg.h>
19 #include <string.h>
20 #include <atomic>
21 #include <string>
22 
23 namespace skstd {
24     class string_view;
25 }
26 
27 /*  Some helper functions for C strings */
SkStrStartsWith(const char string[],const char prefixStr[])28 static inline bool SkStrStartsWith(const char string[], const char prefixStr[]) {
29     SkASSERT(string);
30     SkASSERT(prefixStr);
31     return !strncmp(string, prefixStr, strlen(prefixStr));
32 }
SkStrStartsWith(const char string[],const char prefixChar)33 static inline bool SkStrStartsWith(const char string[], const char prefixChar) {
34     SkASSERT(string);
35     return (prefixChar == *string);
36 }
37 
38 bool SkStrEndsWith(const char string[], const char suffixStr[]);
39 bool SkStrEndsWith(const char string[], const char suffixChar);
40 
41 int SkStrStartsWithOneOf(const char string[], const char prefixes[]);
42 
SkStrFind(const char string[],const char substring[])43 static inline int SkStrFind(const char string[], const char substring[]) {
44     const char *first = strstr(string, substring);
45     if (nullptr == first) return -1;
46     return SkToInt(first - &string[0]);
47 }
48 
SkStrFindLastOf(const char string[],const char subchar)49 static inline int SkStrFindLastOf(const char string[], const char subchar) {
50     const char* last = strrchr(string, subchar);
51     if (nullptr == last) return -1;
52     return SkToInt(last - &string[0]);
53 }
54 
SkStrContains(const char string[],const char substring[])55 static inline bool SkStrContains(const char string[], const char substring[]) {
56     SkASSERT(string);
57     SkASSERT(substring);
58     return (-1 != SkStrFind(string, substring));
59 }
SkStrContains(const char string[],const char subchar)60 static inline bool SkStrContains(const char string[], const char subchar) {
61     SkASSERT(string);
62     char tmp[2];
63     tmp[0] = subchar;
64     tmp[1] = '\0';
65     return (-1 != SkStrFind(string, tmp));
66 }
67 
68 /*
69  *  The SkStrAppend... methods will write into the provided buffer, assuming it is large enough.
70  *  Each method has an associated const (e.g. kSkStrAppendU32_MaxSize) which will be the largest
71  *  value needed for that method's buffer.
72  *
73  *  char storage[kSkStrAppendU32_MaxSize];
74  *  SkStrAppendU32(storage, value);
75  *
76  *  Note : none of the SkStrAppend... methods write a terminating 0 to their buffers. Instead,
77  *  the methods return the ptr to the end of the written part of the buffer. This can be used
78  *  to compute the length, and/or know where to write a 0 if that is desired.
79  *
80  *  char storage[kSkStrAppendU32_MaxSize + 1];
81  *  char* stop = SkStrAppendU32(storage, value);
82  *  size_t len = stop - storage;
83  *  *stop = 0;   // valid, since storage was 1 byte larger than the max.
84  */
85 
86 static constexpr int kSkStrAppendU32_MaxSize = 10;
87 char*   SkStrAppendU32(char buffer[], uint32_t);
88 static constexpr int kSkStrAppendU64_MaxSize = 20;
89 char*   SkStrAppendU64(char buffer[], uint64_t, int minDigits);
90 
91 static constexpr int kSkStrAppendS32_MaxSize = kSkStrAppendU32_MaxSize + 1;
92 char*   SkStrAppendS32(char buffer[], int32_t);
93 static constexpr int kSkStrAppendS64_MaxSize = kSkStrAppendU64_MaxSize + 1;
94 char*   SkStrAppendS64(char buffer[], int64_t, int minDigits);
95 
96 /**
97  *  Floats have at most 8 significant digits, so we limit our %g to that.
98  *  However, the total string could be 15 characters: -1.2345678e-005
99  *
100  *  In theory we should only expect up to 2 digits for the exponent, but on
101  *  some platforms we have seen 3 (as in the example above).
102  */
103 static constexpr int kSkStrAppendScalar_MaxSize = 15;
104 
105 /**
106  *  Write the scalar in decimal format into buffer, and return a pointer to
107  *  the next char after the last one written. Note: a terminating 0 is not
108  *  written into buffer, which must be at least kSkStrAppendScalar_MaxSize.
109  *  Thus if the caller wants to add a 0 at the end, buffer must be at least
110  *  kSkStrAppendScalar_MaxSize + 1 bytes large.
111  */
112 char* SkStrAppendScalar(char buffer[], SkScalar);
113 
114 /** \class SkString
115 
116     Light weight class for managing strings. Uses reference
117     counting to make string assignments and copies very fast
118     with no extra RAM cost. Assumes UTF8 encoding.
119 */
120 class SK_API SkString {
121 public:
122                 SkString();
123     explicit    SkString(size_t len);
124     explicit    SkString(const char text[]);
125                 SkString(const char text[], size_t len);
126                 SkString(const SkString&);
127                 SkString(SkString&&);
128     explicit    SkString(const std::string&);
129     explicit    SkString(skstd::string_view);
130                 ~SkString();
131 
isEmpty()132     bool        isEmpty() const { return 0 == fRec->fLength; }
size()133     size_t      size() const { return (size_t) fRec->fLength; }
c_str()134     const char* c_str() const { return fRec->data(); }
135     char operator[](size_t n) const { return this->c_str()[n]; }
136 
137     bool equals(const SkString&) const;
138     bool equals(const char text[]) const;
139     bool equals(const char text[], size_t len) const;
140 
startsWith(const char prefixStr[])141     bool startsWith(const char prefixStr[]) const {
142         return SkStrStartsWith(fRec->data(), prefixStr);
143     }
startsWith(const char prefixChar)144     bool startsWith(const char prefixChar) const {
145         return SkStrStartsWith(fRec->data(), prefixChar);
146     }
endsWith(const char suffixStr[])147     bool endsWith(const char suffixStr[]) const {
148         return SkStrEndsWith(fRec->data(), suffixStr);
149     }
endsWith(const char suffixChar)150     bool endsWith(const char suffixChar) const {
151         return SkStrEndsWith(fRec->data(), suffixChar);
152     }
contains(const char substring[])153     bool contains(const char substring[]) const {
154         return SkStrContains(fRec->data(), substring);
155     }
contains(const char subchar)156     bool contains(const char subchar) const {
157         return SkStrContains(fRec->data(), subchar);
158     }
find(const char substring[])159     int find(const char substring[]) const {
160         return SkStrFind(fRec->data(), substring);
161     }
findLastOf(const char subchar)162     int findLastOf(const char subchar) const {
163         return SkStrFindLastOf(fRec->data(), subchar);
164     }
165 
166     friend bool operator==(const SkString& a, const SkString& b) {
167         return a.equals(b);
168     }
169     friend bool operator!=(const SkString& a, const SkString& b) {
170         return !a.equals(b);
171     }
172 
173     // these methods edit the string
174 
175     SkString& operator=(const SkString&);
176     SkString& operator=(SkString&&);
177     SkString& operator=(const char text[]);
178 
179     char* writable_str();
180     char& operator[](size_t n) { return this->writable_str()[n]; }
181 
182     void reset();
183     /** String contents are preserved on resize. (For destructive resize, `set(nullptr, length)`.)
184      * `resize` automatically reserves an extra byte at the end of the buffer for a null terminator.
185      */
186     void resize(size_t len);
set(const SkString & src)187     void set(const SkString& src) { *this = src; }
188     void set(const char text[]);
189     void set(const char text[], size_t len);
190 
insert(size_t offset,const SkString & src)191     void insert(size_t offset, const SkString& src) { this->insert(offset, src.c_str(), src.size()); }
192     void insert(size_t offset, const char text[]);
193     void insert(size_t offset, const char text[], size_t len);
194     void insertUnichar(size_t offset, SkUnichar);
195     void insertS32(size_t offset, int32_t value);
196     void insertS64(size_t offset, int64_t value, int minDigits = 0);
197     void insertU32(size_t offset, uint32_t value);
198     void insertU64(size_t offset, uint64_t value, int minDigits = 0);
199     void insertHex(size_t offset, uint32_t value, int minDigits = 0);
200     void insertScalar(size_t offset, SkScalar);
201 
append(const SkString & str)202     void append(const SkString& str) { this->insert((size_t)-1, str); }
append(const char text[])203     void append(const char text[]) { this->insert((size_t)-1, text); }
append(const char text[],size_t len)204     void append(const char text[], size_t len) { this->insert((size_t)-1, text, len); }
appendUnichar(SkUnichar uni)205     void appendUnichar(SkUnichar uni) { this->insertUnichar((size_t)-1, uni); }
appendS32(int32_t value)206     void appendS32(int32_t value) { this->insertS32((size_t)-1, value); }
207     void appendS64(int64_t value, int minDigits = 0) { this->insertS64((size_t)-1, value, minDigits); }
appendU32(uint32_t value)208     void appendU32(uint32_t value) { this->insertU32((size_t)-1, value); }
209     void appendU64(uint64_t value, int minDigits = 0) { this->insertU64((size_t)-1, value, minDigits); }
210     void appendHex(uint32_t value, int minDigits = 0) { this->insertHex((size_t)-1, value, minDigits); }
appendScalar(SkScalar value)211     void appendScalar(SkScalar value) { this->insertScalar((size_t)-1, value); }
212 
prepend(const SkString & str)213     void prepend(const SkString& str) { this->insert(0, str); }
prepend(const char text[])214     void prepend(const char text[]) { this->insert(0, text); }
prepend(const char text[],size_t len)215     void prepend(const char text[], size_t len) { this->insert(0, text, len); }
prependUnichar(SkUnichar uni)216     void prependUnichar(SkUnichar uni) { this->insertUnichar(0, uni); }
prependS32(int32_t value)217     void prependS32(int32_t value) { this->insertS32(0, value); }
218     void prependS64(int32_t value, int minDigits = 0) { this->insertS64(0, value, minDigits); }
219     void prependHex(uint32_t value, int minDigits = 0) { this->insertHex(0, value, minDigits); }
prependScalar(SkScalar value)220     void prependScalar(SkScalar value) { this->insertScalar((size_t)-1, value); }
221 
222     void printf(const char format[], ...) SK_PRINTF_LIKE(2, 3);
223     void printVAList(const char format[], va_list);
224     void appendf(const char format[], ...) SK_PRINTF_LIKE(2, 3);
225     void appendVAList(const char format[], va_list);
226     void prependf(const char format[], ...) SK_PRINTF_LIKE(2, 3);
227     void prependVAList(const char format[], va_list);
228 
229     void remove(size_t offset, size_t length);
230 
231     SkString& operator+=(const SkString& s) { this->append(s); return *this; }
232     SkString& operator+=(const char text[]) { this->append(text); return *this; }
233     SkString& operator+=(const char c) { this->append(&c, 1); return *this; }
234 
235     /**
236      *  Swap contents between this and other. This function is guaranteed
237      *  to never fail or throw.
238      */
239     void swap(SkString& other);
240 
241 private:
242     struct Rec {
243     public:
RecRec244         constexpr Rec(uint32_t len, int32_t refCnt) : fLength(len), fRefCnt(refCnt) {}
245         static sk_sp<Rec> Make(const char text[], size_t len);
dataRec246         char* data() { return fBeginningOfData; }
dataRec247         const char* data() const { return fBeginningOfData; }
248         void ref() const;
249         void unref() const;
250         bool unique() const;
251 #ifdef SK_DEBUG
252         int32_t getRefCnt() const;
253 #endif
254         uint32_t fLength; // logically size_t, but we want it to stay 32 bits
255 
256     private:
257         mutable std::atomic<int32_t> fRefCnt;
258         char fBeginningOfData[1] = {'\0'};
259 
260         // Ensure the unsized delete is called.
deleteRec261         void operator delete(void* p) { ::operator delete(p); }
262     };
263     sk_sp<Rec> fRec;
264 
265 #ifdef SK_DEBUG
266     const SkString& validate() const;
267 #else
validate()268     const SkString& validate() const { return *this; }
269 #endif
270 
271     static const Rec gEmptyRec;
272 };
273 
274 /// Creates a new string and writes into it using a printf()-style format.
275 SkString SkStringPrintf(const char* format, ...) SK_PRINTF_LIKE(1, 2);
276 /// This makes it easier to write a caller as a VAR_ARGS function where the format string is
277 /// optional.
SkStringPrintf()278 static inline SkString SkStringPrintf() { return SkString(); }
279 
swap(SkString & a,SkString & b)280 static inline void swap(SkString& a, SkString& b) {
281     a.swap(b);
282 }
283 
284 enum SkStrSplitMode {
285     // Strictly return all results. If the input is ",," and the separator is ',' this will return
286     // an array of three empty strings.
287     kStrict_SkStrSplitMode,
288 
289     // Only nonempty results will be added to the results. Multiple separators will be
290     // coalesced. Separators at the beginning and end of the input will be ignored.  If the input is
291     // ",," and the separator is ',', this will return an empty vector.
292     kCoalesce_SkStrSplitMode
293 };
294 
295 // Split str on any characters in delimiters into out.  (Think, strtok with a sane API.)
296 void SkStrSplit(const char* str, const char* delimiters, SkStrSplitMode splitMode,
297                 SkTArray<SkString>* out);
SkStrSplit(const char * str,const char * delimiters,SkTArray<SkString> * out)298 inline void SkStrSplit(const char* str, const char* delimiters, SkTArray<SkString>* out) {
299     SkStrSplit(str, delimiters, kCoalesce_SkStrSplitMode, out);
300 }
301 
302 #endif
303