Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
"new ruby host method"
[simgrid.git] / src / xbt / dict_multi.c
1 /* $Id$ */
2
3 /* dict_multi - dictionnaries of dictionnaries of ... of data               */
4
5 /* Copyright (c) 2003-2005 Martin Quinson. All rights reserved.             */
6
7 /* This program is free software; you can redistribute it and/or modify it
8  * under the terms of the license (GNU LGPL) which comes with this package. */
9
10 #include "dict_private.h"
11
12 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(xbt_dict_multi, xbt_dict,
13                                 "Dictionaries of multiple keys");
14
15 static void _free_dict(void *d)
16 {
17   VERB1("free dict %p", d);
18   xbt_dict_free((xbt_dict_t *) & d);
19 }
20
21 /** \brief Insert \e data under all the keys contained in \e keys, providing their sizes in \e lens.
22  *
23  * \arg mdict: the multi-dict
24  * \arg keys: dynar of (char *) containing all the keys
25  * \arg lens: length of each element of \e keys
26  * \arg data: what to store in the structure
27  * \arg free_ctn: function to use to free the pushed content on need
28  *
29  * Dynars are not modified during the operation.
30  */
31
32 void
33 xbt_multidict_set_ext(xbt_dict_t mdict,
34                       xbt_dynar_t keys, xbt_dynar_t lens,
35                       void *data, void_f_pvoid_t free_ctn)
36 {
37
38   xbt_dict_t thislevel, nextlevel = NULL;
39   int i;
40
41   unsigned long int thislen;
42   char *thiskey;
43   int keys_len = xbt_dynar_length(keys);
44
45   xbt_assert(xbt_dynar_length(keys) == xbt_dynar_length(lens));
46   xbt_assert0(keys_len, "Can't set a zero-long key set in a multidict");
47
48   DEBUG2("xbt_multidict_set(%p,%d)", mdict, keys_len);
49
50   for (i = 0, thislevel = mdict; i < keys_len - 1; i++, thislevel = nextlevel) {
51
52     xbt_dynar_get_cpy(keys, i, &thiskey);
53     xbt_dynar_get_cpy(lens, i, &thislen);
54
55     DEBUG5("multi_set: at level %d, len=%ld, key=%p |%*s|", i, thislen,
56            thiskey, (int) thislen, thiskey);
57
58     /* search the dict of next level */
59     nextlevel = xbt_dict_get_or_null_ext(thislevel, thiskey, thislen);
60     if (nextlevel == NULL) {
61       /* make sure the dict of next level exists */
62       nextlevel = xbt_dict_new();
63       VERB1("Create a dict (%p)", nextlevel);
64       xbt_dict_set_ext(thislevel, thiskey, thislen, nextlevel, &_free_dict);
65     }
66   }
67
68   xbt_dynar_get_cpy(keys, i, &thiskey);
69   xbt_dynar_get_cpy(lens, i, &thislen);
70
71   xbt_dict_set_ext(thislevel, thiskey, thislen, data, free_ctn);
72 }
73
74 /** \brief Insert \e data under all the keys contained in \e keys
75  *
76  * \arg head: the head of dict
77  * \arg keys: dynar of null-terminated strings containing all the keys
78  * \arg data: what to store in the structure
79  * \arg free_ctn: function to use to free the pushed content on need
80  */
81 void
82 xbt_multidict_set(xbt_dict_t mdict,
83                   xbt_dynar_t keys, void *data, void_f_pvoid_t free_ctn)
84 {
85   xbt_dynar_t lens = xbt_dynar_new(sizeof(unsigned long int), NULL);
86   unsigned long i;
87   xbt_ex_t e;
88
89   for (i = 0; i < xbt_dynar_length(keys); i++) {
90     char *thiskey = xbt_dynar_get_as(keys, i, char *);
91     unsigned long int thislen = (unsigned long int) strlen(thiskey);
92     DEBUG2("Push %ld as level %lu length", thislen, i);
93     xbt_dynar_push(lens, &thislen);
94   }
95
96   TRY {
97     xbt_multidict_set_ext(mdict, keys, lens, data, free_ctn);
98   } CLEANUP {
99     xbt_dynar_free(&lens);
100   } CATCH(e) {
101     RETHROW;
102   }
103 }
104
105 /** \brief Insert \e data under all the keys contained in \e keys, providing their sizes in \e lens.
106  *
107  * \arg mdict: the multi-dict
108  * \arg keys: dynar of (char *) containing all the keys
109  * \arg lens: length of each element of \e keys
110  * \arg data: where to put what was found in structure
111  * \arg free_ctn: function to use to free the pushed content on need
112  *
113  * Dynars are not modified during the operation.
114  */
115 void *xbt_multidict_get_ext(xbt_dict_t mdict,
116                             xbt_dynar_t keys, xbt_dynar_t lens)
117 {
118   xbt_dict_t thislevel, nextlevel;
119   int i;
120
121   unsigned long int thislen;
122   char *thiskey;
123   int keys_len = xbt_dynar_length(keys);
124
125   xbt_assert(xbt_dynar_length(keys) == xbt_dynar_length(lens));
126   xbt_assert0(xbt_dynar_length(keys) >= 1,
127               "Can't get a zero-long key set in a multidict");
128
129   DEBUG2("xbt_multidict_get(%p, %ld)", mdict, xbt_dynar_length(keys));
130
131   for (i = 0, thislevel = mdict; i < keys_len - 1; i++, thislevel = nextlevel) {
132
133     xbt_dynar_get_cpy(keys, i, &thiskey);
134     xbt_dynar_get_cpy(lens, i, &thislen);
135
136     DEBUG6("multi_get: at level %d (%p), len=%ld, key=%p |%*s|",
137            i, thislevel, thislen, thiskey, (int) thislen, thiskey);
138
139     /* search the dict of next level: let mismatch raise if not found */
140     nextlevel = xbt_dict_get_ext(thislevel, thiskey, thislen);
141   }
142
143   xbt_dynar_get_cpy(keys, i, &thiskey);
144   xbt_dynar_get_cpy(lens, i, &thislen);
145
146   return xbt_dict_get_ext(thislevel, thiskey, thislen);
147 }
148
149 void *xbt_multidict_get(xbt_dict_t mdict, xbt_dynar_t keys)
150 {
151   xbt_dynar_t lens = xbt_dynar_new(sizeof(unsigned long int), NULL);
152   unsigned long i;
153   void *res;
154
155   for (i = 0; i < xbt_dynar_length(keys); i++) {
156     char *thiskey = xbt_dynar_get_as(keys, i, char *);
157     unsigned long int thislen = (unsigned long int) strlen(thiskey);
158     xbt_dynar_push(lens, &thislen);
159   }
160
161   res = xbt_multidict_get_ext(mdict, keys, lens), xbt_dynar_free(&lens);
162   return res;
163 }
164
165
166 /** \brief Remove the entry under all the keys contained in \e keys, providing their sizes in \e lens.
167  *
168  * \arg mdict: the multi-dict
169  * \arg keys: dynar of (char *) containing all the keys
170  * \arg lens: length of each element of \e keys
171  * \arg data: what to store in the structure
172  * \arg free_ctn: function to use to free the pushed content on need
173  *
174  * Dynars are not modified during the operation.
175  *
176  * Removing a non-existant key is ok.
177  */
178
179 void
180 xbt_multidict_remove_ext(xbt_dict_t mdict, xbt_dynar_t keys, xbt_dynar_t lens)
181 {
182   xbt_dict_t thislevel, nextlevel = NULL;
183   int i;
184   xbt_ex_t e;
185
186   unsigned long int thislen;
187   char *thiskey;
188   int keys_len = xbt_dynar_length(keys);
189
190   xbt_assert(xbt_dynar_length(keys) == xbt_dynar_length(lens));
191   xbt_assert0(xbt_dynar_length(keys),
192               "Can't remove a zero-long key set in a multidict");
193
194   for (i = 0, thislevel = mdict; i < keys_len - 1; i++, thislevel = nextlevel) {
195
196     xbt_dynar_get_cpy(keys, i, &thiskey);
197     xbt_dynar_get_cpy(lens, i, &thislen);
198
199     /* search the dict of next level */
200     TRY {
201       nextlevel = xbt_dict_get_ext(thislevel, thiskey, thislen);
202     }
203     CATCH(e) {
204       /* If non-existant entry, nothing to do */
205       if (e.category == arg_error)
206         xbt_ex_free(e);
207       else
208         RETHROW;
209     }
210   }
211
212   xbt_dynar_get_cpy(keys, i, &thiskey);
213   xbt_dynar_get_cpy(lens, i, &thislen);
214
215   xbt_dict_remove_ext(thislevel, thiskey, thislen);
216 }
217
218 void xbt_multidict_remove(xbt_dict_t mdict, xbt_dynar_t keys)
219 {
220
221   xbt_ex_t e;
222   xbt_dynar_t lens = xbt_dynar_new(sizeof(unsigned long int), NULL);
223   unsigned long i;
224
225   for (i = 0; i < xbt_dynar_length(keys); i++) {
226     char *thiskey = xbt_dynar_get_as(keys, i, char *);
227     unsigned long int thislen = strlen(thiskey);
228     xbt_dynar_push(lens, &thislen);
229   }
230
231   TRY {
232     xbt_multidict_remove_ext(mdict, keys, lens);
233   } CLEANUP {
234     xbt_dynar_free(&lens);
235   } CATCH(e) {
236     RETHROW;
237   }
238 }