🌐 AI搜索 & 代理 主页
blob: 375c9c751adbf165c757832667102b761e1358c3 [file] [log] [blame]
Junio C Hamanoa84b7942013-04-13 18:56:411#ifndef COMMIT_SLAB_H
2#define COMMIT_SLAB_H
3
4/*
5 * define_commit_slab(slabname, elemtype) creates boilerplate code to define
6 * a new struct (struct slabname) that is used to associate a piece of data
7 * of elemtype to commits, and a few functions to use that struct.
8 *
9 * After including this header file, using:
10 *
11 * define_commit_slab(indegee, int);
12 *
13 * will let you call the following functions:
14 *
15 * - int *indegree_at(struct indegree *, struct commit *);
16 *
17 * This function locates the data associated with the given commit in
18 * the indegree slab, and returns the pointer to it.
19 *
20 * - void init_indegree(struct indegree *);
21 * void init_indegree_with_stride(struct indegree *, int);
22 *
23 * Initializes the indegree slab that associates an array of integers
24 * to each commit. 'stride' specifies how big each array is. The slab
Thomas Rastdcbbc8f2013-11-25 19:02:0025 * that is initialized by the variant without "_with_stride" associates
Junio C Hamanoa84b7942013-04-13 18:56:4126 * each commit with an array of one integer.
Thomas Rastdcbbc8f2013-11-25 19:02:0027 *
28 * - void clear_indegree(struct indegree *);
29 *
30 * Empties the slab. The slab can be reused with the same stride
31 * without calling init_indegree() again or can be reconfigured to a
32 * different stride by calling init_indegree_with_stride().
33 *
34 * Call this function before the slab falls out of scope to avoid
35 * leaking memory.
Junio C Hamanoa84b7942013-04-13 18:56:4136 */
37
38/* allocate ~512kB at once, allowing for malloc overhead */
39#ifndef COMMIT_SLAB_SIZE
40#define COMMIT_SLAB_SIZE (512*1024-32)
41#endif
42
Thomas Raste9e03a72013-11-25 20:04:0843#define MAYBE_UNUSED __attribute__((__unused__))
44
Junio C Hamanoa84b7942013-04-13 18:56:4145#define define_commit_slab(slabname, elemtype) \
46 \
47struct slabname { \
48 unsigned slab_size; \
49 unsigned stride; \
50 unsigned slab_count; \
51 elemtype **slab; \
52}; \
53static int stat_ ##slabname## realloc; \
54 \
Thomas Raste9e03a72013-11-25 20:04:0855static MAYBE_UNUSED void init_ ##slabname## _with_stride(struct slabname *s, \
56 unsigned stride) \
Junio C Hamanoa84b7942013-04-13 18:56:4157{ \
58 unsigned int elem_size; \
59 if (!stride) \
60 stride = 1; \
61 s->stride = stride; \
Ramsay Jonesd7a1d622013-07-27 19:00:0762 elem_size = sizeof(elemtype) * stride; \
Junio C Hamanoa84b7942013-04-13 18:56:4163 s->slab_size = COMMIT_SLAB_SIZE / elem_size; \
64 s->slab_count = 0; \
65 s->slab = NULL; \
66} \
67 \
Thomas Raste9e03a72013-11-25 20:04:0868static MAYBE_UNUSED void init_ ##slabname(struct slabname *s) \
Junio C Hamanoa84b7942013-04-13 18:56:4169{ \
70 init_ ##slabname## _with_stride(s, 1); \
71} \
72 \
Thomas Raste9e03a72013-11-25 20:04:0873static MAYBE_UNUSED void clear_ ##slabname(struct slabname *s) \
Junio C Hamanoa84b7942013-04-13 18:56:4174{ \
75 int i; \
76 for (i = 0; i < s->slab_count; i++) \
77 free(s->slab[i]); \
78 s->slab_count = 0; \
79 free(s->slab); \
80 s->slab = NULL; \
81} \
82 \
Thomas Raste9e03a72013-11-25 20:04:0883static MAYBE_UNUSED elemtype *slabname## _at(struct slabname *s, \
84 const struct commit *c) \
Junio C Hamanoa84b7942013-04-13 18:56:4185{ \
Ramsay Jonesd7a1d622013-07-27 19:00:0786 int nth_slab, nth_slot; \
Junio C Hamanoa84b7942013-04-13 18:56:4187 \
Ramsay Jonesd7a1d622013-07-27 19:00:0788 nth_slab = c->index / s->slab_size; \
89 nth_slot = c->index % s->slab_size; \
Junio C Hamanoa84b7942013-04-13 18:56:4190 \
91 if (s->slab_count <= nth_slab) { \
92 int i; \
93 s->slab = xrealloc(s->slab, \
Thomas Rastcde0a052013-12-01 20:41:5594 (nth_slab + 1) * sizeof(*s->slab)); \
Junio C Hamanoa84b7942013-04-13 18:56:4195 stat_ ##slabname## realloc++; \
96 for (i = s->slab_count; i <= nth_slab; i++) \
97 s->slab[i] = NULL; \
98 s->slab_count = nth_slab + 1; \
99 } \
100 if (!s->slab[nth_slab]) \
101 s->slab[nth_slab] = xcalloc(s->slab_size, \
Ramsay Jonesd7a1d622013-07-27 19:00:07102 sizeof(**s->slab) * s->stride); \
103 return &s->slab[nth_slab][nth_slot * s->stride]; \
Junio C Hamanoa84b7942013-04-13 18:56:41104} \
105 \
106static int stat_ ##slabname## realloc
107
Thomas Raste9e03a72013-11-25 20:04:08108/*
109 * Note that this seemingly redundant second declaration is required
110 * to allow a terminating semicolon, which makes instantiations look
111 * like function declarations. I.e., the expansion of
112 *
113 * define_commit_slab(indegree, int);
114 *
115 * ends in 'static int stat_indegreerealloc;'. This would otherwise
116 * be a syntax error according (at least) to ISO C. It's hard to
117 * catch because GCC silently parses it by default.
118 */
119
Jeff King80cdaba2014-06-10 21:42:51120/*
121 * Statically initialize a commit slab named "var". Note that this
122 * evaluates "stride" multiple times! Example:
123 *
124 * struct indegree indegrees = COMMIT_SLAB_INIT(1, indegrees);
125 *
126 */
127#define COMMIT_SLAB_INIT(stride, var) { \
128 COMMIT_SLAB_SIZE / sizeof(**((var).slab)) / (stride), \
129 (stride), 0, NULL \
130}
131
Junio C Hamanoa84b7942013-04-13 18:56:41132#endif /* COMMIT_SLAB_H */