src/xbt/swag.c

   1 /*      $Id$     */
   2
   3 /* Copyright (c) 2004 Arnaud Legrand. All rights reserved.                  */
   4
   5 /* This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the license (GNU LGPL) which comes with this package. */
   7
   8 /* Warning, this module is done to be efficient and performs tons of
   9    cast and dirty things. So avoid using it unless you really know
  10    what you are doing. */
  11
  12 /* This type should be added to a type that is to be used in such a swag */
  13
  14 #include "xbt/sysdep.h"
  15 #include "xbt/error.h"
  16 #include "xbt/swag.h"
  17
  18 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(swag,xbt,"Swag : O(1) set library");
  19
  20 #define PREV(obj,offset) xbt_swag_getPrev(obj,offset)
  21 #define NEXT(obj,offset) xbt_swag_getNext(obj,offset)
  22
  23 /*
  24   Usage : xbt_swag_new(&obj.setA-&obj);
  25  */
  26
  27 xbt_swag_t xbt_swag_new(size_t offset)
  28 {
  29   xbt_swag_t swag = xbt_new0(s_xbt_swag_t, 1);
  30
  31   swag->offset = offset;
  32
  33   return swag;
  34 }
  35
  36
  37 void xbt_swag_free(xbt_swag_t swag)
  38 {
  39   xbt_free(swag);
  40 }
  41
  42 void xbt_swag_init(xbt_swag_t swag, size_t offset)
  43 {
  44   swag->offset = offset;
  45 }
  46
  47 void xbt_swag_insert(void *obj, xbt_swag_t swag)
  48 {
  49
  50   if (xbt_swag_belongs(obj, swag))
  51     return;
  52
  53   (swag->count)++;
  54   if (swag->head == NULL) {
  55     swag->head = obj;
  56     swag->tail = obj;
  57     return;
  58   }
  59
  60   PREV(obj, swag->offset) = swag->tail;
  61   NEXT(PREV(obj, swag->offset), swag->offset) = obj;
  62
  63   swag->tail = obj;
  64 }
  65
  66 void *xbt_swag_remove(void *obj, xbt_swag_t swag)
  67 {
  68   size_t offset = swag->offset;
  69
  70   if ((!obj) || (!swag))
  71     return NULL;
  72   if(!xbt_swag_belongs(obj, swag)) /* Trying to remove an object that
  73                                       was not in this swag */
  74       return NULL;
  75
  76   if (swag->head == swag->tail) {       /* special case */
  77     if (swag->head != obj)      /* Trying to remove an object that was not in this swag */
  78       return NULL;
  79     swag->head = NULL;
  80     swag->tail = NULL;
  81   } else if (obj == swag->head) {       /* It's the head */
  82     swag->head = NEXT(obj, offset);
  83     PREV(swag->head, offset) = NULL;
  84     NEXT(obj, offset) = NULL;
  85   } else if (obj == swag->tail) {       /* It's the tail */
  86     swag->tail = PREV(obj, offset);
  87     NEXT(swag->tail, offset) = NULL;
  88     PREV(obj, offset) = NULL;
  89   } else {                      /* It's in the middle */
  90     NEXT(PREV(obj, offset), offset) = NEXT(obj, offset);
  91     PREV(NEXT(obj, offset), offset) = PREV(obj, offset);
  92     PREV(obj, offset) = NEXT(obj, offset) = NULL;
  93   }
  94   (swag->count)--;
  95   return obj;
  96 }
  97
  98 void *xbt_swag_extract(xbt_swag_t swag)
  99 {
 100   size_t offset = swag->offset;
 101   void *obj = NULL;
 102
 103   if ((!swag) || (!(swag->head)))
 104     return NULL;
 105
 106   obj = swag->head;
 107
 108   if (swag->head == swag->tail) {       /* special case */
 109     swag->head = swag->tail = NULL;
 110   } else {
 111     swag->head = NEXT(obj, offset);
 112     PREV(swag->head, offset) = NULL;
 113     NEXT(obj, offset) = NULL;
 114   }
 115   (swag->count)--;
 116
 117   return obj;
 118 }
 119
 120 int xbt_swag_size(xbt_swag_t swag)
 121 {
 122   return (swag->count);
 123 }
 124
 125 int xbt_swag_belongs(void *obj, xbt_swag_t swag)
 126 {
 127   return ((NEXT(obj, swag->offset)) || (PREV(obj, swag->offset))
 128           || (swag->head == obj));
 129 }