src/xbt/dict_multi.c

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