| 1 |
717 |
jeremybenn |
/* objc-map.h -- Implementation of map data structures for ObjC compiler
|
| 2 |
|
|
Copyright 2011 Free Software Foundation, Inc.
|
| 3 |
|
|
Written by Nicola Pero <nicola.pero@meta-innovation.com>
|
| 4 |
|
|
|
| 5 |
|
|
This program is free software; you can redistribute it and/or modify it
|
| 6 |
|
|
under the terms of the GNU Lesser Public License as published by the
|
| 7 |
|
|
Free Software Foundation; either version 3, or (at your option) any
|
| 8 |
|
|
later version.
|
| 9 |
|
|
|
| 10 |
|
|
This program is distributed in the hope that it will be useful,
|
| 11 |
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
| 12 |
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
| 13 |
|
|
GNU Lesser Public License for more details.
|
| 14 |
|
|
|
| 15 |
|
|
You should have received a copy of the GNU Lesser Public License
|
| 16 |
|
|
along with this program; if not, write to the Free Software
|
| 17 |
|
|
Foundation, 51 Franklin Street - Fifth Floor,
|
| 18 |
|
|
Boston, MA 02110-1301, USA. */
|
| 19 |
|
|
|
| 20 |
|
|
#ifndef OBJC_MAP_H
|
| 21 |
|
|
#define OBJC_MAP_H
|
| 22 |
|
|
|
| 23 |
|
|
/* A map is a data structure that maps a key to a value. In this file
|
| 24 |
|
|
we currently have maps that can map a GCC identifier (a tree) to
|
| 25 |
|
|
some other GCC tree. This is what the ObjC frontend mostly needs:
|
| 26 |
|
|
being able to look up an identifier into an ObjC data structure. A
|
| 27 |
|
|
typical usage is mapping ObjC class names (as identifiers) to a
|
| 28 |
|
|
tree representing the class.
|
| 29 |
|
|
|
| 30 |
|
|
This implementation is fast. :-) */
|
| 31 |
|
|
|
| 32 |
|
|
/**
|
| 33 |
|
|
** Private definitions.
|
| 34 |
|
|
**/
|
| 35 |
|
|
|
| 36 |
|
|
/* We include private declaration and definitions that are required to
|
| 37 |
|
|
provide the implementation of inline functions. You should ignore
|
| 38 |
|
|
these definitions (and the implementation of the inline functions)
|
| 39 |
|
|
as they are not part of the public API and may change. */
|
| 40 |
|
|
typedef unsigned int objc_map_private_hash_t;
|
| 41 |
|
|
|
| 42 |
|
|
/* This is used as sentinel. */
|
| 43 |
|
|
#define OBJC_MAP_PRIVATE_EMPTY_SLOT (tree)0
|
| 44 |
|
|
|
| 45 |
|
|
struct GTY(()) objc_map_private {
|
| 46 |
|
|
/* Total number of slots. This is the maximum number of elements
|
| 47 |
|
|
that can be currently stored in the map before resizing. This is
|
| 48 |
|
|
the number of slots in the C array. Important: this is
|
| 49 |
|
|
guaranteed to be a power of 2. When we create (or resize) the
|
| 50 |
|
|
map, we round up the size to the next power of 2. This allows us
|
| 51 |
|
|
to convert a hash to a position in the hashtable by simply doing
|
| 52 |
|
|
"position = hash & mask", where mask is number_of_slots - 1
|
| 53 |
|
|
instead of using a modulo (which requires a division). */
|
| 54 |
|
|
size_t number_of_slots;
|
| 55 |
|
|
|
| 56 |
|
|
/* This is number_of_slots - 1, precomputed. */
|
| 57 |
|
|
size_t mask;
|
| 58 |
|
|
|
| 59 |
|
|
/* Number of slots that are not empty (ie, that are active). We
|
| 60 |
|
|
keep counts using this variable which can easily be checked
|
| 61 |
|
|
against max_number_of_non_empty_slots. */
|
| 62 |
|
|
size_t number_of_non_empty_slots;
|
| 63 |
|
|
|
| 64 |
|
|
/* This is the load factor limit. When the number of non empty
|
| 65 |
|
|
slots equals this number, we need to resize the array. This is
|
| 66 |
|
|
calculated once, when the slots are resized, and then kept cached
|
| 67 |
|
|
so it can be compared quickly when elements are added. */
|
| 68 |
|
|
size_t max_number_of_non_empty_slots;
|
| 69 |
|
|
|
| 70 |
|
|
/* The maximum load factor. */
|
| 71 |
|
|
int maximum_load_factor;
|
| 72 |
|
|
|
| 73 |
|
|
/* These are the keys. */
|
| 74 |
|
|
tree * GTY ((length ("%h.number_of_slots"))) slots;
|
| 75 |
|
|
|
| 76 |
|
|
/* These are the values. values[i] is the the value corresponding
|
| 77 |
|
|
to slots[i]. */
|
| 78 |
|
|
tree * GTY ((length ("%h.number_of_slots"))) values;
|
| 79 |
|
|
};
|
| 80 |
|
|
|
| 81 |
|
|
/* Private functions used to resize the map. They may be called by
|
| 82 |
|
|
the inline functions when adding elements. */
|
| 83 |
|
|
extern void
|
| 84 |
|
|
objc_map_private_grow (struct objc_map_private *map);
|
| 85 |
|
|
|
| 86 |
|
|
|
| 87 |
|
|
/**
|
| 88 |
|
|
** The definition of a map.
|
| 89 |
|
|
**/
|
| 90 |
|
|
typedef struct objc_map_private *objc_map_t;
|
| 91 |
|
|
|
| 92 |
|
|
|
| 93 |
|
|
/**
|
| 94 |
|
|
** Creating a map.
|
| 95 |
|
|
**/
|
| 96 |
|
|
|
| 97 |
|
|
/* objc_map_alloc_ggc() creates a new map which is under GGC. The initial
|
| 98 |
|
|
capacity must be specified as an argument; this is used to size the map
|
| 99 |
|
|
when it is created. */
|
| 100 |
|
|
objc_map_t objc_map_alloc_ggc (size_t initial_capacity);
|
| 101 |
|
|
|
| 102 |
|
|
/**
|
| 103 |
|
|
** Performance tuning.
|
| 104 |
|
|
**/
|
| 105 |
|
|
|
| 106 |
|
|
/* Set a maximum load factor for the data structure. This is the main
|
| 107 |
|
|
tuning parameter to improve performance (at the expense of
|
| 108 |
|
|
memory). */
|
| 109 |
|
|
void objc_map_set_maximum_load_factor (objc_map_t map, int number_between_zero_and_one_hundred);
|
| 110 |
|
|
|
| 111 |
|
|
/* Read the maximum load factor. */
|
| 112 |
|
|
int objc_map_maximum_load_factor (objc_map_t map);
|
| 113 |
|
|
|
| 114 |
|
|
|
| 115 |
|
|
/**
|
| 116 |
|
|
** Getting the value corresponding to a key.
|
| 117 |
|
|
**/
|
| 118 |
|
|
|
| 119 |
|
|
/* This is the value returned by objc_map_get() when the value
|
| 120 |
|
|
corresponding to a key is not found. */
|
| 121 |
|
|
#define OBJC_MAP_NOT_FOUND (tree)1
|
| 122 |
|
|
|
| 123 |
|
|
/* objc_map_get() returns the value associated with a certain key,
|
| 124 |
|
|
or OBJC_MAP_NOT_FOUND if there is no value associated with that key.
|
| 125 |
|
|
Note that you can also use it to simply check if the map contains a
|
| 126 |
|
|
pair with a certain key; just compare the result of calling
|
| 127 |
|
|
objc_map_get() to OBJC_MAP_NOT_FOUND.
|
| 128 |
|
|
|
| 129 |
|
|
It is essential to always check the results of the call to make
|
| 130 |
|
|
sure it is not OBJC_MAP_NOT_FOUND.
|
| 131 |
|
|
|
| 132 |
|
|
NULL is a valid value, so a key can be inserted into a map with
|
| 133 |
|
|
value NULL, and objc_map_get() will return NULL in that case.
|
| 134 |
|
|
So a result of NULL means that they key *was* found, and the value
|
| 135 |
|
|
associated with it was NULL. */
|
| 136 |
|
|
static inline tree
|
| 137 |
|
|
objc_map_get (objc_map_t map, /* struct tree_identifier * */tree key)
|
| 138 |
|
|
{
|
| 139 |
|
|
/* The inline implementation is private and may change without notice. */
|
| 140 |
|
|
objc_map_private_hash_t hash = IDENTIFIER_HASH_VALUE (key);
|
| 141 |
|
|
size_t i = hash & map->mask;
|
| 142 |
|
|
size_t j = 1;
|
| 143 |
|
|
|
| 144 |
|
|
if (map->slots[i] == OBJC_MAP_PRIVATE_EMPTY_SLOT)
|
| 145 |
|
|
return OBJC_MAP_NOT_FOUND;
|
| 146 |
|
|
|
| 147 |
|
|
if (map->slots[i] == key)
|
| 148 |
|
|
return map->values[i];
|
| 149 |
|
|
|
| 150 |
|
|
while (1)
|
| 151 |
|
|
{
|
| 152 |
|
|
i = (i + j) & map->mask;
|
| 153 |
|
|
|
| 154 |
|
|
if (map->slots[i] == OBJC_MAP_PRIVATE_EMPTY_SLOT)
|
| 155 |
|
|
return OBJC_MAP_NOT_FOUND;
|
| 156 |
|
|
|
| 157 |
|
|
if (map->slots[i] == key)
|
| 158 |
|
|
return map->values[i];
|
| 159 |
|
|
|
| 160 |
|
|
j++;
|
| 161 |
|
|
}
|
| 162 |
|
|
}
|
| 163 |
|
|
|
| 164 |
|
|
/* objc_map_put() puts a key/value pair into the map. If the map does
|
| 165 |
|
|
not contain the key, it is added to it with the specified value.
|
| 166 |
|
|
If the map already contains the key, the previous value is replaced
|
| 167 |
|
|
with the new one.
|
| 168 |
|
|
|
| 169 |
|
|
You can use any identifier as key, with the exception of NULL.
|
| 170 |
|
|
|
| 171 |
|
|
You can use any tree as value, including NULL. */
|
| 172 |
|
|
static inline
|
| 173 |
|
|
void objc_map_put (objc_map_t map, /*struct tree_identifier * */tree key, tree value)
|
| 174 |
|
|
{
|
| 175 |
|
|
/* The inline implementation is private and may change without notice. */
|
| 176 |
|
|
objc_map_private_hash_t hash = IDENTIFIER_HASH_VALUE (key);
|
| 177 |
|
|
size_t i, j = 0;
|
| 178 |
|
|
|
| 179 |
|
|
if (map->number_of_non_empty_slots == map->max_number_of_non_empty_slots)
|
| 180 |
|
|
objc_map_private_grow (map);
|
| 181 |
|
|
|
| 182 |
|
|
i = hash & map->mask;
|
| 183 |
|
|
|
| 184 |
|
|
while (1)
|
| 185 |
|
|
{
|
| 186 |
|
|
if (map->slots[i] == OBJC_MAP_PRIVATE_EMPTY_SLOT)
|
| 187 |
|
|
{
|
| 188 |
|
|
map->number_of_non_empty_slots++;
|
| 189 |
|
|
map->slots[i] = key;
|
| 190 |
|
|
map->values[i] = value;
|
| 191 |
|
|
return;
|
| 192 |
|
|
}
|
| 193 |
|
|
if (map->slots[i] == key)
|
| 194 |
|
|
{
|
| 195 |
|
|
map->values[i] = value;
|
| 196 |
|
|
return;
|
| 197 |
|
|
}
|
| 198 |
|
|
|
| 199 |
|
|
j++;
|
| 200 |
|
|
i = (i + j) & map->mask;
|
| 201 |
|
|
}
|
| 202 |
|
|
}
|
| 203 |
|
|
|
| 204 |
|
|
/**
|
| 205 |
|
|
** Iterating over a map using an iterator.
|
| 206 |
|
|
**/
|
| 207 |
|
|
|
| 208 |
|
|
/* When using iterators you can iterate directly on the elements in
|
| 209 |
|
|
the map, and take an action over each one.
|
| 210 |
|
|
|
| 211 |
|
|
Here is how you iterate over a hmap_pointer using iterators:
|
| 212 |
|
|
|
| 213 |
|
|
objc_map_iterator_t i;
|
| 214 |
|
|
|
| 215 |
|
|
objc_map_iterator_initialize (map, &i);
|
| 216 |
|
|
|
| 217 |
|
|
while (objc_map_iterator_move_to_next (map, &i))
|
| 218 |
|
|
{
|
| 219 |
|
|
tree p = objc_map_iterator_current_key (map, i);
|
| 220 |
|
|
tree q = objc_map_iterator_current_value (map, i);
|
| 221 |
|
|
|
| 222 |
|
|
... do something with p and q ...
|
| 223 |
|
|
}
|
| 224 |
|
|
|
| 225 |
|
|
You'll notice that the functions that modify the iterator (to
|
| 226 |
|
|
initialize it, or move it to the next element) take a pointer to it
|
| 227 |
|
|
as argument (as in "&i"), while the functions that only read its
|
| 228 |
|
|
state (to read the current key/value, or remove the current
|
| 229 |
|
|
key/value from the map) take it as a direct argument (as in "i").
|
| 230 |
|
|
|
| 231 |
|
|
Note that all the objc_map_iterator_*() functions are inline and if
|
| 232 |
|
|
you follow the pattern above, the compiler should be able to inline
|
| 233 |
|
|
everything into a very efficient loop, roughly equivalent to
|
| 234 |
|
|
hand-writing a C loop that iterates directly onto the hmap_pointer
|
| 235 |
|
|
internal data structures. */
|
| 236 |
|
|
|
| 237 |
|
|
/* A objc_map_iterator_t variable encapsulates the state of an
|
| 238 |
|
|
iteration. The fact that this is actually a size_t (pointing to
|
| 239 |
|
|
the index of the slot that we return next) is an internal, private
|
| 240 |
|
|
detail of the implementation and may change without notice. */
|
| 241 |
|
|
typedef size_t objc_map_iterator_t;
|
| 242 |
|
|
|
| 243 |
|
|
/* Initialize an iterator to iterate over the specified objc_map. You
|
| 244 |
|
|
must use this before starting the iteration, to get a working
|
| 245 |
|
|
iterator. */
|
| 246 |
|
|
static inline
|
| 247 |
|
|
void
|
| 248 |
|
|
objc_map_iterator_initialize (objc_map_t map ATTRIBUTE_UNUSED, objc_map_iterator_t *i)
|
| 249 |
|
|
{
|
| 250 |
|
|
/* The inline implementation is private and may change without notice. */
|
| 251 |
|
|
/* This is trivial, but the same API would work to initialize more
|
| 252 |
|
|
complicated iterators. */
|
| 253 |
|
|
*i = 0;
|
| 254 |
|
|
}
|
| 255 |
|
|
|
| 256 |
|
|
#define OBJC_MAP_FAILURE 0
|
| 257 |
|
|
#define OBJC_MAP_SUCCESS 1
|
| 258 |
|
|
|
| 259 |
|
|
/* Move the iterator to the next key/value pair, and return
|
| 260 |
|
|
OBJC_MAP_SUCCESS if there is such a key/value pair, and
|
| 261 |
|
|
OBJC_MAP_FAILURE if there are no more ones. The iterator must have
|
| 262 |
|
|
been initialized using objc_map_iterator_initialize(). Note that
|
| 263 |
|
|
because this function is modifying the iterator, you need to pass a
|
| 264 |
|
|
pointer to it. */
|
| 265 |
|
|
static inline
|
| 266 |
|
|
int
|
| 267 |
|
|
objc_map_iterator_move_to_next (objc_map_t map, objc_map_iterator_t *i)
|
| 268 |
|
|
{
|
| 269 |
|
|
/* The inline implementation is private and may change without notice. */
|
| 270 |
|
|
while (1)
|
| 271 |
|
|
{
|
| 272 |
|
|
void *slot;
|
| 273 |
|
|
if (*i == map->number_of_slots)
|
| 274 |
|
|
return OBJC_MAP_FAILURE;
|
| 275 |
|
|
|
| 276 |
|
|
slot = map->slots[*i];
|
| 277 |
|
|
*i = *i + 1;
|
| 278 |
|
|
if (slot != OBJC_MAP_PRIVATE_EMPTY_SLOT)
|
| 279 |
|
|
return OBJC_MAP_SUCCESS;
|
| 280 |
|
|
}
|
| 281 |
|
|
}
|
| 282 |
|
|
|
| 283 |
|
|
/* Return the current key. You can only call it after you have called
|
| 284 |
|
|
objc_map_iterator_move_to_next() at least once (to move to the
|
| 285 |
|
|
first element), and only if the last call returned
|
| 286 |
|
|
OBJC_MAP_SUCCESS. The behaviour is otherwise undefined, probably a
|
| 287 |
|
|
segmentation fault. */
|
| 288 |
|
|
static inline
|
| 289 |
|
|
tree
|
| 290 |
|
|
objc_map_iterator_current_key (objc_map_t map, objc_map_iterator_t i)
|
| 291 |
|
|
{
|
| 292 |
|
|
/* The inline implementation is private and may change without notice. */
|
| 293 |
|
|
return map->slots[i - 1];
|
| 294 |
|
|
}
|
| 295 |
|
|
|
| 296 |
|
|
/* Return the current value. You can only call it after you have
|
| 297 |
|
|
called objc_map_iterator_move_to_next() at least once (to move to
|
| 298 |
|
|
the first element), and only if the last call returned
|
| 299 |
|
|
OBJC_MAP_SUCCESS. The behaviour is otherwise undefined, probably a
|
| 300 |
|
|
segmentation fault. */
|
| 301 |
|
|
static inline
|
| 302 |
|
|
tree
|
| 303 |
|
|
objc_map_iterator_current_value (objc_map_t map, objc_map_iterator_t i)
|
| 304 |
|
|
{
|
| 305 |
|
|
/* The inline implementation is private and may change without notice. */
|
| 306 |
|
|
return map->values[i - 1];
|
| 307 |
|
|
}
|
| 308 |
|
|
|
| 309 |
|
|
#endif /* OBJC_MAP_H */
|