kvec.h
Go to the documentation of this file.
1 // The MIT License
2 //
3 // Copyright (c) 2008, by Attractive Chaos <[email protected]>
4 //
5 // Permission is hereby granted, free of charge, to any person obtaining
6 // a copy of this software and associated documentation files (the
7 // "Software"), to deal in the Software without restriction, including
8 // without limitation the rights to use, copy, modify, merge, publish,
9 // distribute, sublicense, and/or sell copies of the Software, and to
10 // permit persons to whom the Software is furnished to do so, subject to
11 // the following conditions:
12 //
13 // The above copyright notice and this permission notice shall be
14 // included in all copies or substantial portions of the Software.
15 //
16 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
17 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
18 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
19 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
20 // BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
21 // ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
22 // CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
23 // SOFTWARE.
24 
25 // An example:
26 //
27 // #include "kvec.h"
28 // int main() {
29 // kvec_t(int) array = KV_INITIAL_VALUE;
30 // kv_push(array, 10); // append
31 // kv_a(array, 20) = 5; // dynamic
32 // kv_A(array, 20) = 4; // static
33 // kv_destroy(array);
34 // return 0;
35 // }
36 
37 #ifndef NVIM_LIB_KVEC_H
38 #define NVIM_LIB_KVEC_H
39 
40 #include <stdlib.h>
41 #include <string.h>
42 
43 #include "nvim/memory.h"
44 #include "nvim/os/os_defs.h"
45 
46 #define kv_roundup32(x) \
47  ((--(x)), \
48  ((x)|=(x)>>1, (x)|=(x)>>2, (x)|=(x)>>4, (x)|=(x)>>8, (x)|=(x)>>16), \
49  (++(x)))
50 
51 #define KV_INITIAL_VALUE { .size = 0, .capacity = 0, .items = NULL }
52 
53 #define kvec_t(type) \
54  struct { \
55  size_t size; \
56  size_t capacity; \
57  type *items; \
58  }
59 
60 #define kv_init(v) ((v).size = (v).capacity = 0, (v).items = 0)
61 #define kv_destroy(v) \
62  do { \
63  xfree((v).items); \
64  kv_init(v); \
65  } while (0)
66 #define kv_A(v, i) ((v).items[(i)])
67 #define kv_pop(v) ((v).items[--(v).size])
68 #define kv_size(v) ((v).size)
69 #define kv_max(v) ((v).capacity)
70 #define kv_Z(v, i) kv_A(v, kv_size(v) - (i) - 1)
71 #define kv_last(v) kv_Z(v, 0)
72 
79 #define kv_drop(v, n) ((v).size -= (n))
80 
81 #define kv_resize(v, s) \
82  ((v).capacity = (s), \
83  (v).items = xrealloc((v).items, sizeof((v).items[0]) * (v).capacity))
84 
85 #define kv_resize_full(v) \
86  kv_resize(v, (v).capacity ? (v).capacity << 1 : 8)
87 
88 #define kv_copy(v1, v0) \
89  do { \
90  if ((v1).capacity < (v0).size) { \
91  kv_resize(v1, (v0).size); \
92  } \
93  (v1).size = (v0).size; \
94  memcpy((v1).items, (v0).items, sizeof((v1).items[0]) * (v0).size); \
95  } while (0)
96 
98 #define kv_ensure_space(v, len) \
99  do { \
100  if ((v).capacity < (v).size + len) { \
101  (v).capacity = (v).size + len; \
102  kv_roundup32((v).capacity); \
103  kv_resize((v), (v).capacity); \
104  } \
105  } while (0)
106 
107 #define kv_concat_len(v, data, len) \
108  do { \
109  kv_ensure_space(v, len); \
110  memcpy((v).items + (v).size, data, sizeof((v).items[0]) * len); \
111  (v).size = (v).size + len; \
112  } while (0)
113 
114 #define kv_concat(v, str) kv_concat_len(v, str, STRLEN(str))
115 #define kv_splice(v1, v0) kv_concat_len(v1, (v0).items, (v0).size)
116 
117 #define kv_pushp(v) \
118  ((((v).size == (v).capacity) ? (kv_resize_full(v), 0) : 0), \
119  ((v).items + ((v).size++)))
120 
121 #define kv_push(v, x) \
122  (*kv_pushp(v) = (x))
123 
124 
125 #define kv_a(v, i) \
126  (*(((v).capacity <= (size_t)(i) \
127  ? ((v).capacity = (v).size = (i) + 1, \
128  kv_roundup32((v).capacity), \
129  kv_resize((v), (v).capacity), 0UL) \
130  : ((v).size <= (size_t)(i) \
131  ? (v).size = (i) + 1 \
132  : 0UL)), \
133  &(v).items[(i)]))
134 
135 #define kv_printf(v, ...) kv_do_printf(&(v), __VA_ARGS__)
136 
145 #define kvec_withinit_t(type, INIT_SIZE) \
146  struct { \
147  size_t size; \
148  size_t capacity; \
149  type *items; \
150  type init_array[INIT_SIZE]; \
151  }
152 
156 #define kvi_init(v) \
157  ((v).capacity = ARRAY_SIZE((v).init_array), \
158  (v).size = 0, \
159  (v).items = (v).init_array)
160 
162 static inline void *_memcpy_free(void *const restrict dest, void *const restrict src,
163  const size_t size)
165 {
166  memcpy(dest, src, size);
167  XFREE_CLEAR(src);
168  return dest;
169 }
170 
171 // -V:kvi_push:512
172 
180 #define kvi_resize(v, s) \
181  ((v).capacity = ((s) > ARRAY_SIZE((v).init_array) \
182  ? (s) \
183  : ARRAY_SIZE((v).init_array)), \
184  (v).items = ((v).capacity == ARRAY_SIZE((v).init_array) \
185  ? ((v).items == (v).init_array \
186  ? (v).items \
187  : _memcpy_free((v).init_array, (v).items, \
188  (v).size * sizeof((v).items[0]))) \
189  : ((v).items == (v).init_array \
190  ? memcpy(xmalloc((v).capacity * sizeof((v).items[0])), \
191  (v).items, \
192  (v).size * sizeof((v).items[0])) \
193  : xrealloc((v).items, \
194  (v).capacity * sizeof((v).items[0])))))
195 
199 #define kvi_resize_full(v) \
200  /* ARRAY_SIZE((v).init_array) is the minimal capacity of this vector. */ \
201  /* Thus when vector is full capacity may not be zero and it is safe */ \
202  /* not to bother with checking whether (v).capacity is 0. But now */ \
203  /* capacity is not guaranteed to have size that is a power of 2, it is */ \
204  /* hard to fix this here and is not very necessary if users will use */ \
205  /* 2^x initial array size. */ \
206  kvi_resize(v, (v).capacity << 1)
207 
213 #define kvi_pushp(v) \
214  ((((v).size == (v).capacity) ? (kvi_resize_full(v), 0) : 0), \
215  ((v).items + ((v).size++)))
216 
221 #define kvi_push(v, x) \
222  (*kvi_pushp(v) = (x))
223 
227 #define kvi_destroy(v) \
228  do { \
229  if ((v).items != (v).init_array) { \
230  XFREE_CLEAR((v).items); \
231  } \
232  } while (0)
233 
234 #endif // NVIM_LIB_KVEC_H
src
const char_u * src
Definition: keycodes.c:625
size
size_t size
Definition: regexp_nfa.c:5986
FUNC_ATTR_NONNULL_RET
void FUNC_ATTR_NONNULL_RET
Definition: memory.c:133
FUNC_ATTR_NONNULL_ALL
#define FUNC_ATTR_NONNULL_ALL
Definition: func_attr.h:247
os_defs.h
XFREE_CLEAR
#define XFREE_CLEAR(ptr)
Definition: memory.h:44
stdlib.h
FUNC_ATTR_ALWAYS_INLINE
#define FUNC_ATTR_ALWAYS_INLINE
Definition: func_attr.h:245
memory.h