examples/msg/kadeploy/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   //XBT_INFO("%d current\n", next);
  57   if (it->current >= it->length) {
  58     //XBT_INFO("Nothing to return!\n");
  59     return NULL;
  60   } else {
  61     next = xbt_dynar_get_ptr(it->indices_list, it->current);
  62     it->current++;
  63     return xbt_dynar_get_ptr(it->list, *next);
  64   }
  65 }
  66
  67 void xbt_dynar_iterator_delete(xbt_dynar_iterator_t it)
  68 {
  69   xbt_dynar_free_container(&(it->indices_list));
  70   xbt_free_ref(&it);
  71 }
  72
  73 xbt_dynar_t forward_indices_list(int size)
  74 {
  75   xbt_dynar_t indices_list = xbt_dynar_new(sizeof(int), NULL);
  76   int i;
  77   for (i = 0; i < size; i++)
  78     xbt_dynar_push_as(indices_list, int, i);
  79   return indices_list;
  80 }
  81
  82 xbt_dynar_t reverse_indices_list(int size)
  83 {
  84   xbt_dynar_t indices_list = xbt_dynar_new(sizeof(int), NULL);
  85   int i;
  86   for (i = size-1; i >= 0; i--)
  87     xbt_dynar_push_as(indices_list, int, i);
  88   return indices_list;
  89 }
  90
  91 xbt_dynar_t random_indices_list(int size)
  92 {
  93   xbt_dynar_t indices_list = forward_indices_list(size);
  94   xbt_dynar_shuffle_in_place(indices_list);
  95   return indices_list;
  96 }