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 }