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