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(dict_multi,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 xbt_error_t
  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_error_t errcode;
  36   xbt_dict_t thislevel,nextlevel;
  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     TRYCATCH(xbt_dict_get_ext(thislevel, thiskey, thislen, (void*)&nextlevel), mismatch_error);
  59
  60     /* make sure the dict of next level exists */
  61     if (errcode == mismatch_error) {
  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   return no_error;
  74 }
  75
  76 /** \brief Insert \e data under all the keys contained in \e keys
  77  *
  78  * \arg head: the head of dict
  79  * \arg keys: dynar of null-terminated strings containing all the keys
  80  * \arg data: what to store in the structure
  81  * \arg free_ctn: function to use to free the pushed content on need
  82  */
  83 xbt_error_t
  84 xbt_multidict_set(xbt_dict_t  mdict,
  85                   xbt_dynar_t keys,
  86                   void       *data,  void_f_pvoid_t *free_ctn) {
  87   xbt_error_t errcode;
  88   xbt_dynar_t lens = xbt_dynar_new(sizeof(unsigned long int),NULL);
  89   int i;
  90
  91   for (i = 0; i < xbt_dynar_length(keys); i++) {
  92     char *thiskey = xbt_dynar_get_as(keys, i, char*);
  93     unsigned long int thislen = (unsigned long int) strlen(thiskey);
  94     DEBUG2("Push %ld as level %d length",thislen, i);
  95     xbt_dynar_push(lens,&thislen);
  96   }
  97
  98   errcode = xbt_multidict_set_ext(mdict, keys, lens, data, free_ctn);
  99   xbt_dynar_free(&lens);
 100   return errcode;
 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 xbt_error_t
 114 xbt_multidict_get_ext(xbt_dict_t  mdict,
 115                       xbt_dynar_t keys,   xbt_dynar_t lens,
 116                       /*OUT*/void **data) {
 117   xbt_error_t errcode;
 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, "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      ;
 131        i<keys_len-1                       ;
 132        i++         , thislevel = nextlevel) {
 133
 134     xbt_dynar_get_cpy(keys, i, &thiskey);
 135     xbt_dynar_get_cpy(lens, i, &thislen);
 136
 137     DEBUG6("multi_get: at level %d (%p), len=%ld, key=%p |%*s|",
 138       i, thislevel, thislen, thiskey, (int)thislen,thiskey);
 139
 140     /* search the dict of next level: let mismatch raise if not found */
 141     TRY(xbt_dict_get_ext(thislevel, thiskey, thislen, (void*)&nextlevel));
 142   }
 143
 144   xbt_dynar_get_cpy(keys, i, &thiskey);
 145   xbt_dynar_get_cpy(lens, i, &thislen);
 146
 147   return xbt_dict_get_ext(thislevel, thiskey, thislen, data);
 148 }
 149
 150 xbt_error_t
 151 xbt_multidict_get(xbt_dict_t mdict, xbt_dynar_t keys, /*OUT*/void **data) {
 152   xbt_error_t errcode;
 153   xbt_dynar_t lens = xbt_dynar_new(sizeof(unsigned long int),NULL);
 154   int i;
 155
 156   for (i = 0; i < xbt_dynar_length(keys); i++) {
 157     char *thiskey = xbt_dynar_get_as(keys, i, char*);
 158     unsigned long int thislen = (unsigned long int) strlen(thiskey);
 159     xbt_dynar_push(lens,&thislen);
 160   }
 161
 162   errcode = xbt_multidict_get_ext(mdict, keys, lens, data),
 163   xbt_dynar_free(&lens);
 164   return errcode;
 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 xbt_error_t
 182 xbt_multidict_remove_ext(xbt_dict_t mdict, xbt_dynar_t keys, xbt_dynar_t lens) {
 183   xbt_error_t errcode;
 184   xbt_dict_t thislevel,nextlevel;
 185   int i;
 186
 187   xbt_assert(xbt_dynar_length(keys) == xbt_dynar_length(lens));
 188   xbt_assert0(xbt_dynar_length(keys), "Can't remove a zero-long key set in a multidict");
 189
 190   unsigned long int thislen;
 191   char *thiskey;
 192   int keys_len=xbt_dynar_length(keys);
 193
 194   xbt_assert(xbt_dynar_length(keys) == xbt_dynar_length(lens));
 195   xbt_assert0(keys_len, "Can't set a zero-long key set in a multidict");
 196
 197   for (i=0         , thislevel=mdict      ;
 198        i<keys_len-1                       ;
 199        i++         , thislevel = nextlevel) {
 200
 201     xbt_dynar_get_cpy(keys, i, &thiskey);
 202     xbt_dynar_get_cpy(lens, i, &thislen);
 203
 204     /* search the dict of next level */
 205     TRYCATCH(xbt_dict_get_ext(thislevel, thiskey, thislen, (void*)&nextlevel), mismatch_error);
 206
 207     /* If non-existant entry, nothing to do */
 208     if (errcode == mismatch_error)
 209       return no_error;
 210   }
 211
 212   xbt_dynar_get_cpy(keys, i, &thiskey);
 213   xbt_dynar_get_cpy(lens, i, &thislen);
 214
 215   return xbt_dict_remove_ext(thislevel, thiskey, thislen);
 216 }
 217
 218 xbt_error_t
 219 xbt_multidict_remove(xbt_dict_t mdict, xbt_dynar_t keys) {
 220
 221   xbt_error_t errcode;
 222   xbt_dynar_t lens = xbt_dynar_new(sizeof(unsigned long int),NULL);
 223   int 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   errcode = xbt_multidict_remove_ext(mdict, keys, lens);
 232   xbt_dynar_free(&lens);
 233   return errcode;
 234 }