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;
  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,
  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,
  65                        &_free_dict);
  66     }
  67   }
  68
  69   xbt_dynar_get_cpy(keys, i, &thiskey);
  70   xbt_dynar_get_cpy(lens, i, &thislen);
  71
  72   xbt_dict_set_ext(thislevel, thiskey, thislen, data, free_ctn);
  73 }
  74
  75 /** \brief Insert \e data under all the keys contained in \e keys
  76  *
  77  * \arg head: the head of dict
  78  * \arg keys: dynar of null-terminated strings containing all the keys
  79  * \arg data: what to store in the structure
  80  * \arg free_ctn: function to use to free the pushed content on need
  81  */
  82 void
  83 xbt_multidict_set(xbt_dict_t mdict,
  84                   xbt_dynar_t keys, void *data, void_f_pvoid_t free_ctn)
  85 {
  86   xbt_dynar_t lens = xbt_dynar_new(sizeof(unsigned long int), NULL);
  87   unsigned long i;
  88   xbt_ex_t e;
  89
  90   for (i = 0; i < xbt_dynar_length(keys); i++) {
  91     char *thiskey = xbt_dynar_get_as(keys, i, char *);
  92     unsigned long int thislen = (unsigned long int) strlen(thiskey);
  93     DEBUG2("Push %ld as level %lu length", thislen, i);
  94     xbt_dynar_push(lens, &thislen);
  95   }
  96
  97   TRY {
  98     xbt_multidict_set_ext(mdict, keys, lens, data, free_ctn);
  99   } TRY_CLEANUP {
 100     xbt_dynar_free(&lens);
 101   } CATCH(e) {
 102     RETHROW;
 103   }
 104 }
 105
 106 /** \brief Insert \e data under all the keys contained in \e keys, providing their sizes in \e lens.
 107  *
 108  * \arg mdict: the multi-dict
 109  * \arg keys: dynar of (char *) containing all the keys
 110  * \arg lens: length of each element of \e keys
 111  * \arg data: where to put what was found in structure
 112  * \arg free_ctn: function to use to free the pushed content on need
 113  *
 114  * Dynars are not modified during the operation.
 115  */
 116 void *xbt_multidict_get_ext(xbt_dict_t mdict,
 117                             xbt_dynar_t keys, xbt_dynar_t lens)
 118 {
 119   xbt_dict_t thislevel, nextlevel;
 120   int i;
 121
 122   unsigned long int thislen;
 123   char *thiskey;
 124   int keys_len = xbt_dynar_length(keys);
 125
 126   xbt_assert(xbt_dynar_length(keys) == xbt_dynar_length(lens));
 127   xbt_assert0(xbt_dynar_length(keys) >= 1,
 128               "Can't get a zero-long key set in a multidict");
 129
 130   DEBUG2("xbt_multidict_get(%p, %ld)", mdict, xbt_dynar_length(keys));
 131
 132   for (i = 0, thislevel = mdict; i < keys_len - 1;
 133        i++, thislevel = nextlevel) {
 134
 135     xbt_dynar_get_cpy(keys, i, &thiskey);
 136     xbt_dynar_get_cpy(lens, i, &thislen);
 137
 138     DEBUG6("multi_get: at level %d (%p), len=%ld, key=%p |%*s|",
 139            i, thislevel, thislen, thiskey, (int) thislen, thiskey);
 140
 141     /* search the dict of next level: let mismatch raise if not found */
 142     nextlevel = xbt_dict_get_ext(thislevel, thiskey, thislen);
 143   }
 144
 145   xbt_dynar_get_cpy(keys, i, &thiskey);
 146   xbt_dynar_get_cpy(lens, i, &thislen);
 147
 148   return xbt_dict_get_ext(thislevel, thiskey, thislen);
 149 }
 150
 151 void *xbt_multidict_get(xbt_dict_t mdict, xbt_dynar_t keys)
 152 {
 153   xbt_dynar_t lens = xbt_dynar_new(sizeof(unsigned long int), NULL);
 154   unsigned long i;
 155   void *res;
 156
 157   for (i = 0; i < xbt_dynar_length(keys); i++) {
 158     char *thiskey = xbt_dynar_get_as(keys, i, char *);
 159     unsigned long int thislen = (unsigned long int) strlen(thiskey);
 160     xbt_dynar_push(lens, &thislen);
 161   }
 162
 163   res = xbt_multidict_get_ext(mdict, keys, lens), xbt_dynar_free(&lens);
 164   return res;
 165 }
 166
 167
 168 /** \brief Remove the entry under all the keys contained in \e keys, providing their sizes in \e lens.
 169  *
 170  * \arg mdict: the multi-dict
 171  * \arg keys: dynar of (char *) containing all the keys
 172  * \arg lens: length of each element of \e keys
 173  * \arg data: what to store in the structure
 174  * \arg free_ctn: function to use to free the pushed content on need
 175  *
 176  * Dynars are not modified during the operation.
 177  *
 178  * Removing a non-existant key is ok.
 179  */
 180
 181 void
 182 xbt_multidict_remove_ext(xbt_dict_t mdict, xbt_dynar_t keys,
 183                          xbt_dynar_t lens)
 184 {
 185   xbt_dict_t thislevel, nextlevel = NULL;
 186   int i;
 187   xbt_ex_t e;
 188
 189   unsigned long int thislen;
 190   char *thiskey;
 191   int keys_len = xbt_dynar_length(keys);
 192
 193   xbt_assert(xbt_dynar_length(keys) == xbt_dynar_length(lens));
 194   xbt_assert0(xbt_dynar_length(keys),
 195               "Can't remove a zero-long key set in a multidict");
 196
 197   for (i = 0, thislevel = mdict; i < keys_len - 1;
 198        i++, thislevel = nextlevel) {
 199
 200     xbt_dynar_get_cpy(keys, i, &thiskey);
 201     xbt_dynar_get_cpy(lens, i, &thislen);
 202
 203     /* search the dict of next level */
 204     TRY {
 205       nextlevel = xbt_dict_get_ext(thislevel, thiskey, thislen);
 206     }
 207     CATCH(e) {
 208       /* If non-existant entry, nothing to do */
 209       if (e.category == arg_error)
 210         xbt_ex_free(e);
 211       else
 212         RETHROW;
 213     }
 214   }
 215
 216   xbt_dynar_get_cpy(keys, i, &thiskey);
 217   xbt_dynar_get_cpy(lens, i, &thislen);
 218
 219   xbt_dict_remove_ext(thislevel, thiskey, thislen);
 220 }
 221
 222 void xbt_multidict_remove(xbt_dict_t mdict, xbt_dynar_t keys)
 223 {
 224
 225   xbt_ex_t e;
 226   xbt_dynar_t lens = xbt_dynar_new(sizeof(unsigned long int), NULL);
 227   unsigned long i;
 228
 229   for (i = 0; i < xbt_dynar_length(keys); i++) {
 230     char *thiskey = xbt_dynar_get_as(keys, i, char *);
 231     unsigned long int thislen = strlen(thiskey);
 232     xbt_dynar_push(lens, &thislen);
 233   }
 234
 235   TRY {
 236     xbt_multidict_remove_ext(mdict, keys, lens);
 237   } TRY_CLEANUP {
 238     xbt_dynar_free(&lens);
 239   } CATCH(e) {
 240     RETHROW;
 241   }
 242 }