1 | |
2 | // vim:sw=2:ai |
3 | |
4 | /* |
5 | * Copyright (C) 2010-2011 DeNA Co.,Ltd.. All rights reserved. |
6 | * Copyright (C) 2011 Kentoku SHIBA |
7 | * See COPYRIGHT.txt for details. |
8 | */ |
9 | |
10 | #ifndef DENA_STRING_REF_HPP |
11 | #define DENA_STRING_REF_HPP |
12 | |
13 | namespace dena { |
14 | |
15 | struct string_wref { |
16 | typedef char value_type; |
17 | char *begin() const { return start; } |
18 | char *end() const { return start + length; } |
19 | size_t size() const { return length; } |
20 | private: |
21 | char *start; |
22 | size_t length; |
23 | public: |
24 | string_wref(char *s = 0, size_t len = 0) : start(s), length(len) { } |
25 | }; |
26 | |
27 | struct string_ref { |
28 | typedef const char value_type; |
29 | const char *begin() const { return start; } |
30 | const char *end() const { return start + length; } |
31 | size_t size() const { return length; } |
32 | void set(const char *s, size_t len) { start = s; length = len; } |
33 | void set(const char *s, const char *f) { start = s; length = f - s; } |
34 | private: |
35 | const char *start; |
36 | size_t length; |
37 | public: |
38 | string_ref(const char *s = 0, size_t len = 0) : start(s), length(len) { } |
39 | string_ref(const char *s, const char *f) : start(s), length(f - s) { } |
40 | string_ref(const string_wref& w) : start(w.begin()), length(w.size()) { } |
41 | }; |
42 | |
43 | template <size_t N> inline bool |
44 | operator ==(const string_ref& x, const char (& y)[N]) { |
45 | return (x.size() == N - 1) && (::memcmp(x.begin(), y, N - 1) == 0); |
46 | } |
47 | |
48 | inline bool |
49 | operator ==(const string_ref& x, const string_ref& y) { |
50 | return (x.size() == y.size()) && |
51 | (::memcmp(x.begin(), y.begin(), x.size()) == 0); |
52 | } |
53 | |
54 | inline bool |
55 | operator !=(const string_ref& x, const string_ref& y) { |
56 | return (x.size() != y.size()) || |
57 | (::memcmp(x.begin(), y.begin(), x.size()) != 0); |
58 | } |
59 | |
60 | struct string_ref_list_wrap { |
61 | string_ref_list_wrap() { |
62 | if (SPD_INIT_DYNAMIC_ARRAY2(&string_ref_list, sizeof(string_ref), |
63 | NULL, 16, 16, MYF(MY_WME))) |
64 | string_ref_list_init = FALSE; |
65 | else |
66 | string_ref_list_init = TRUE; |
67 | } |
68 | virtual ~string_ref_list_wrap() { |
69 | if (string_ref_list_init) delete_dynamic(&string_ref_list); } |
70 | void clear() { |
71 | if (string_ref_list_init) string_ref_list.elements = 0; } |
72 | void push_back(string_ref &e) { |
73 | if (string_ref_list_init) insert_dynamic(&string_ref_list, (uchar*) &e); |
74 | return; } |
75 | size_t size() { |
76 | return string_ref_list_init ? string_ref_list.elements : 0; } |
77 | bool resize(size_t new_size) { |
78 | if (string_ref_list_init) { |
79 | if (string_ref_list.max_element < new_size && allocate_dynamic( |
80 | &string_ref_list, new_size)) return TRUE; |
81 | string_ref_list.elements = new_size; |
82 | return FALSE; |
83 | } |
84 | return TRUE; |
85 | } |
86 | bool empty() { |
87 | return string_ref_list_init ? string_ref_list.elements ? |
88 | FALSE : TRUE : TRUE; } |
89 | string_ref &operator [](size_t n) { |
90 | return ((string_ref *) (string_ref_list.buffer + |
91 | string_ref_list.size_of_element * n))[0]; } |
92 | bool string_ref_list_init; |
93 | DYNAMIC_ARRAY string_ref_list; |
94 | }; |
95 | |
96 | inline String * |
97 | q_append_str(String *str, const char *p) { |
98 | uint32 p_len = strlen(p); |
99 | if (str->reserve(p_len)) return NULL; |
100 | str->q_append(p, p_len); return str; |
101 | } |
102 | |
103 | }; |
104 | |
105 | #endif |
106 | |
107 | |