Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
7b85efa9e5a6558dff8d379aecae3c6f5a699149
[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(dict_multi,dict, "Dictionaries of multiple keys");
13
14 static void _free_dict(void*d) {
15   VERB1("free dict %p",d);
16   xbt_dict_free((xbt_dict_t*)&d);
17 }
18
19 /** \brief Insert \e data under all the keys contained in \e keys, providing their sizes in \e lens.
20  *
21  * \arg mdict: the multi-dict
22  * \arg keys: dynar of (char *) containing all the keys
23  * \arg lens: length of each element of \e keys
24  * \arg data: what to store in the structure
25  * \arg free_ctn: function to use to free the pushed content on need
26  *
27  * Dynars are not modified during the operation.
28  */
29
30 void
31 xbt_multidict_set_ext(xbt_dict_t  mdict,
32                       xbt_dynar_t keys, xbt_dynar_t     lens,
33                       void       *data, void_f_pvoid_t *free_ctn) {
34       
35   xbt_ex_t e;
36   xbt_dict_t thislevel,nextlevel=NULL;
37   int i;
38   
39   unsigned long int thislen;
40   char *thiskey;
41   int keys_len=xbt_dynar_length(keys);
42
43   xbt_assert(xbt_dynar_length(keys) == xbt_dynar_length(lens));
44   xbt_assert0(keys_len, "Can't set a zero-long key set in a multidict");
45
46   DEBUG2("xbt_multidict_set(%p,%d)", mdict, keys_len);
47
48   for (i=0         , thislevel = mdict    ; 
49        i<keys_len-1                       ; 
50        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, thiskey, (int)thislen,thiskey);
56
57     /* search the dict of next level */
58     TRY {
59       nextlevel = xbt_dict_get_ext(thislevel, thiskey, thislen);
60     } CATCH(e) {
61       if (e.category == arg_error || e.category == mismatch_error) {
62         /* make sure the dict of next level exists */
63         nextlevel=xbt_dict_new();
64         VERB1("Create a dict (%p)",nextlevel);
65         xbt_dict_set_ext(thislevel, thiskey, thislen, nextlevel, &_free_dict);
66       } else {
67         RETHROW;
68       }
69     }
70   }
71
72   xbt_dynar_get_cpy(keys, i, &thiskey);
73   xbt_dynar_get_cpy(lens, i, &thislen);
74   
75   xbt_dict_set_ext(thislevel, thiskey, thislen, data, free_ctn);
76 }
77
78 /** \brief Insert \e data under all the keys contained in \e keys
79  *
80  * \arg head: the head of dict
81  * \arg keys: dynar of null-terminated strings containing all the keys
82  * \arg data: what to store in the structure
83  * \arg free_ctn: function to use to free the pushed content on need
84  */
85 void
86 xbt_multidict_set(xbt_dict_t  mdict,
87                   xbt_dynar_t keys,
88                   void       *data,  void_f_pvoid_t *free_ctn) {
89   xbt_dynar_t lens = xbt_dynar_new(sizeof(unsigned long int),NULL);
90   int i;
91
92   for (i = 0; i < xbt_dynar_length(keys); i++) {
93     char *thiskey = xbt_dynar_get_as(keys, i, char*);
94     unsigned long int thislen = (unsigned long int) strlen(thiskey);
95     DEBUG2("Push %ld as level %d length",thislen, i);
96     xbt_dynar_push(lens,&thislen);
97   }
98
99   xbt_multidict_set_ext(mdict, keys, lens, data, free_ctn);
100   xbt_dynar_free(&lens);         
101
102 }
103
104 /** \brief Insert \e data under all the keys contained in \e keys, providing their sizes in \e lens.
105  *
106  * \arg mdict: the multi-dict
107  * \arg keys: dynar of (char *) containing all the keys
108  * \arg lens: length of each element of \e keys
109  * \arg data: where to put what was found in structure
110  * \arg free_ctn: function to use to free the pushed content on need
111  *
112  * Dynars are not modified during the operation.
113  */
114 void *
115 xbt_multidict_get_ext(xbt_dict_t  mdict,
116                       xbt_dynar_t keys,   xbt_dynar_t lens) {
117   xbt_dict_t thislevel,nextlevel;
118   int i;
119
120   unsigned long int thislen;
121   char *thiskey;
122   int keys_len=xbt_dynar_length(keys);
123
124   xbt_assert(xbt_dynar_length(keys) == xbt_dynar_length(lens));
125   xbt_assert0(xbt_dynar_length(keys) >= 1, "Can't get a zero-long key set in a multidict");
126   
127   DEBUG2("xbt_multidict_get(%p, %ld)", mdict, xbt_dynar_length(keys));
128
129   for (i=0         , thislevel=mdict      ; 
130        i<keys_len-1                       ; 
131        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 *
150 xbt_multidict_get(xbt_dict_t mdict, xbt_dynar_t keys) {
151   xbt_dynar_t lens = xbt_dynar_new(sizeof(unsigned long int),NULL);
152   int 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),
162   xbt_dynar_free(&lens);         
163   return res;
164 }
165
166
167 /** \brief Remove the entry under all the keys contained in \e keys, providing their sizes in \e lens.
168  *
169  * \arg mdict: the multi-dict
170  * \arg keys: dynar of (char *) containing all the keys
171  * \arg lens: length of each element of \e keys
172  * \arg data: what to store in the structure
173  * \arg free_ctn: function to use to free the pushed content on need
174  *
175  * Dynars are not modified during the operation.
176  *
177  * Removing a non-existant key is ok.
178  */
179
180 xbt_error_t
181 xbt_multidict_remove_ext(xbt_dict_t mdict, xbt_dynar_t keys, xbt_dynar_t lens) {
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), "Can't remove a zero-long key set in a multidict");
192
193   for (i=0         , thislevel=mdict      ; 
194        i<keys_len-1                       ; 
195        i++         , thislevel = nextlevel) {
196        
197     xbt_dynar_get_cpy(keys, i, &thiskey);
198     xbt_dynar_get_cpy(lens, i, &thislen);
199
200     /* search the dict of next level */
201     TRY {
202       nextlevel = xbt_dict_get_ext(thislevel, thiskey, thislen);
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   return xbt_dict_remove_ext(thislevel, thiskey, thislen);
216 }
217
218 xbt_error_t
219 xbt_multidict_remove(xbt_dict_t mdict, xbt_dynar_t keys) {
220
221   xbt_error_t errcode;
222   xbt_dynar_t lens = xbt_dynar_new(sizeof(unsigned long int),NULL);
223   int 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   errcode = xbt_multidict_remove_ext(mdict, keys, lens);
232   xbt_dynar_free(&lens);
233   return errcode;
234 }