src/xbt/dict_multi.c

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