examples/msg/chainsend/iterator.c

   1 #include "iterator.h"
   2
   3 /* http://stackoverflow.com/a/3348142 */
   4 static int rand_int(int n)
   5 {
   6   int limit = RAND_MAX - RAND_MAX % n;
   7   int rnd;
   8
   9   do {
  10     rnd = rand();
  11   } while (rnd >= limit);
  12
  13   return rnd % n;
  14 }
  15
  16 void xbt_dynar_shuffle_in_place(xbt_dynar_t indices_list)
  17 {
  18   int i, j;
  19
  20   for (i = xbt_dynar_length(indices_list) - 1; i > 0; i--) {
  21     j = rand_int(i + 1);
  22     xbt_dynar_swap_elements(indices_list, int, i, j);
  23   }
  24 }
  25 /**************************************/
  26
  27 /* Allocates and initializes a new xbt_dynar iterator for list, using criteria_fn as iteration criteria
  28    criteria_fn: given an array size, it must generate a list containing the indices of every item in some order */
  29 xbt_dynar_iterator_t xbt_dynar_iterator_new(xbt_dynar_t list, xbt_dynar_t (*criteria_fn)(int))
  30 {
  31   xbt_dynar_iterator_t it = xbt_new(xbt_dynar_iterator_s, 1);
  32
  33   it->list = list;
  34   it->length = xbt_dynar_length(list);
  35   it->indices_list = criteria_fn(it->length); //xbt_dynar_new(sizeof(int), NULL);
  36   it->criteria_fn = criteria_fn;
  37   it->current = 0;
  38 }
  39
  40 void xbt_dynar_iterator_reset(xbt_dynar_iterator_t it)
  41 {
  42   xbt_dynar_free_container(&(it->indices_list));
  43   it->indices_list = it->criteria_fn(it->length);
  44   it->current = 0;
  45 }
  46
  47 void xbt_dynar_iterator_seek(xbt_dynar_iterator_t it, int pos)
  48 {
  49   it->current = pos;
  50 }
  51
  52 /* Returns the next element iterated by iterator it, NULL if there are no more elements */
  53 void *xbt_dynar_iterator_next(xbt_dynar_iterator_t it)
  54 {
  55   int *next;
  56   if (it->current >= it->length) {
  57     return NULL;
  58   } else {
  59     next = xbt_dynar_get_ptr(it->indices_list, it->current);
  60     it->current++;
  61     return xbt_dynar_get_ptr(it->list, *next);
  62   }
  63 }
  64
  65 void xbt_dynar_iterator_delete(xbt_dynar_iterator_t it)
  66 {
  67   xbt_dynar_free_container(&(it->indices_list));
  68   xbt_free_ref(&it);
  69 }
  70
  71 xbt_dynar_t forward_indices_list(int size)
  72 {
  73   xbt_dynar_t indices_list = xbt_dynar_new(sizeof(int), NULL);
  74   int i;
  75   for (i = 0; i < size; i++)
  76     xbt_dynar_push_as(indices_list, int, i);
  77   return indices_list;
  78 }
  79
  80 xbt_dynar_t reverse_indices_list(int size)
  81 {
  82   xbt_dynar_t indices_list = xbt_dynar_new(sizeof(int), NULL);
  83   int i;
  84   for (i = size-1; i >= 0; i--)
  85     xbt_dynar_push_as(indices_list, int, i);
  86   return indices_list;
  87 }
  88
  89 xbt_dynar_t random_indices_list(int size)
  90 {
  91   xbt_dynar_t indices_list = forward_indices_list(size);
  92   xbt_dynar_shuffle_in_place(indices_list);
  93   return indices_list;
  94 }