prio-queue.c

   1 #include "git-compat-util.h"
   2 #include "alloc.h"
   3 #include "prio-queue.h"
   4
   5 static inline int compare(struct prio_queue *queue, int i, int j)
   6 {
   7         int cmp = queue->compare(queue->array[i].data, queue->array[j].data,
   8                                  queue->cb_data);
   9         if (!cmp)
  10                 cmp = queue->array[i].ctr - queue->array[j].ctr;
  11         return cmp;
  12 }
  13
  14 static inline void swap(struct prio_queue *queue, int i, int j)
  15 {
  16         SWAP(queue->array[i], queue->array[j]);
  17 }
  18
  19 void prio_queue_reverse(struct prio_queue *queue)
  20 {
  21         int i, j;
  22
  23         if (queue->compare)
  24                 BUG("prio_queue_reverse() on non-LIFO queue");
  25         for (i = 0; i < (j = (queue->nr - 1) - i); i++)
  26                 swap(queue, i, j);
  27 }
  28
  29 void clear_prio_queue(struct prio_queue *queue)
  30 {
  31         FREE_AND_NULL(queue->array);
  32         queue->nr = 0;
  33         queue->alloc = 0;
  34         queue->insertion_ctr = 0;
  35 }
  36
  37 void prio_queue_put(struct prio_queue *queue, void *thing)
  38 {
  39         int ix, parent;
  40
  41         /* Append at the end */
  42         ALLOC_GROW(queue->array, queue->nr + 1, queue->alloc);
  43         queue->array[queue->nr].ctr = queue->insertion_ctr++;
  44         queue->array[queue->nr].data = thing;
  45         queue->nr++;
  46         if (!queue->compare)
  47                 return; /* LIFO */
  48
  49         /* Bubble up the new one */
  50         for (ix = queue->nr - 1; ix; ix = parent) {
  51                 parent = (ix - 1) / 2;
  52                 if (compare(queue, parent, ix) <= 0)
  53                         break;
  54
  55                 swap(queue, parent, ix);
  56         }
  57 }
  58
  59 void *prio_queue_get(struct prio_queue *queue)
  60 {
  61         void *result;
  62         int ix, child;
  63
  64         if (!queue->nr)
  65                 return NULL;
  66         if (!queue->compare)
  67                 return queue->array[--queue->nr].data; /* LIFO */
  68
  69         result = queue->array[0].data;
  70         if (!--queue->nr)
  71                 return result;
  72
  73         queue->array[0] = queue->array[queue->nr];
  74
  75         /* Push down the one at the root */
  76         for (ix = 0; ix * 2 + 1 < queue->nr; ix = child) {
  77                 child = ix * 2 + 1; /* left */
  78                 if (child + 1 < queue->nr &&
  79                     compare(queue, child, child + 1) >= 0)
  80                         child++; /* use right child */
  81
  82                 if (compare(queue, ix, child) <= 0)
  83                         break;
  84
  85                 swap(queue, child, ix);
  86         }
  87         return result;
  88 }
  89
  90 void *prio_queue_peek(struct prio_queue *queue)
  91 {
  92         if (!queue->nr)
  93                 return NULL;
  94         if (!queue->compare)
  95                 return queue->array[queue->nr - 1].data;
  96         return queue->array[0].data;
  97 }