garray.c

   1 /* GLIB - Library of useful routines for C programming
   2  * Copyright (C) 1995-1997  Peter Mattis, Spencer Kimball and Josh MacDonald
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Library General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Library General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Library General Public
  15  * License along with this library; if not, write to the
  16  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  17  * Boston, MA 02111-1307, USA.
  18  */
  19 #include <string.h>
  20 #include "glib.h"
  21
  22
  23 #define MIN_ARRAY_SIZE  16
  24
  25
  26 typedef struct _GRealArray  GRealArray;
  27
  28 struct _GRealArray
  29 {
  30   guint8 *data;
  31   guint   len;
  32   guint   alloc;
  33   guint   zero_terminated;
  34 };
  35
  36
  37 static gint g_nearest_pow        (gint        num);
  38 static void g_array_maybe_expand (GRealArray *array,
  39                                   gint        len);
  40
  41
  42 static GMemChunk *array_mem_chunk = NULL;
  43
  44
  45 GArray*
  46 g_array_new (gboolean zero_terminated)
  47 {
  48   GRealArray *array;
  49
  50   if (!array_mem_chunk)
  51     array_mem_chunk = g_mem_chunk_new ("array mem chunk",
  52                                        sizeof (GRealArray),
  53                                        1024, G_ALLOC_AND_FREE);
  54
  55   array = g_chunk_new (GRealArray, array_mem_chunk);
  56
  57   array->data = NULL;
  58   array->len = 0;
  59   array->alloc = 0;
  60   array->zero_terminated = (zero_terminated ? 1 : 0);
  61
  62   return (GArray*) array;
  63 }
  64
  65 void
  66 g_array_free (GArray  *array,
  67               gboolean free_segment)
  68 {
  69   if (free_segment)
  70     g_free (array->data);
  71
  72   g_mem_chunk_free (array_mem_chunk, array);
  73 }
  74
  75 GArray*
  76 g_rarray_append (GArray   *array,
  77                  gpointer  data,
  78                  gint      size)
  79 {
  80   g_array_maybe_expand ((GRealArray*) array, size);
  81
  82   memcpy (array->data + array->len, data, size);
  83
  84   array->len += size;
  85
  86   return array;
  87 }
  88
  89 GArray*
  90 g_rarray_prepend (GArray   *array,
  91                   gpointer  data,
  92                   gint      size)
  93 {
  94   g_array_maybe_expand ((GRealArray*) array, size);
  95
  96   g_memmove (array->data + size, array->data, array->len);
  97   memcpy (array->data, data, size);
  98
  99   array->len += size;
 100
 101   return array;
 102 }
 103
 104 GArray*
 105 g_rarray_truncate (GArray *array,
 106                    gint    length,
 107                    gint    size)
 108 {
 109   if (array->data)
 110     memset (array->data + length * size, 0, size);
 111   array->len = length * size;
 112   return array;
 113 }
 114
 115
 116 static gint
 117 g_nearest_pow (gint num)
 118 {
 119   gint n = 1;
 120
 121   while (n < num)
 122     n <<= 1;
 123
 124   return n;
 125 }
 126
 127 static void
 128 g_array_maybe_expand (GRealArray *array,
 129                       gint        len)
 130 {
 131   guint old_alloc;
 132
 133   if ((array->len + len) > array->alloc)
 134     {
 135       old_alloc = array->alloc;
 136
 137       array->alloc = g_nearest_pow (array->len + array->zero_terminated + len);
 138       array->alloc = MAX (array->alloc, MIN_ARRAY_SIZE);
 139       array->data = g_realloc (array->data, array->alloc);
 140
 141       memset (array->data + old_alloc, 0, array->alloc - old_alloc);
 142     }
 143 }
 144
 145 /* Pointer Array
 146  */
 147
 148 typedef struct _GRealPtrArray  GRealPtrArray;
 149
 150 struct _GRealPtrArray
 151 {
 152   gpointer *pdata;
 153   guint     len;
 154   guint     alloc;
 155 };
 156
 157 static void g_ptr_array_maybe_expand (GRealPtrArray *array,
 158                                       gint           len);
 159
 160
 161 static GMemChunk *ptr_array_mem_chunk = NULL;
 162
 163
 164
 165 GPtrArray*
 166 g_ptr_array_new ()
 167 {
 168   GRealPtrArray *array;
 169
 170   if (!ptr_array_mem_chunk)
 171     ptr_array_mem_chunk = g_mem_chunk_new ("array mem chunk",
 172                                            sizeof (GRealPtrArray),
 173                                            1024, G_ALLOC_AND_FREE);
 174
 175   array = g_chunk_new (GRealPtrArray, ptr_array_mem_chunk);
 176
 177   array->pdata = NULL;
 178   array->len = 0;
 179   array->alloc = 0;
 180
 181   return (GPtrArray*) array;
 182 }
 183
 184 void
 185 g_ptr_array_free (GPtrArray   *array,
 186                   gboolean  free_segment)
 187 {
 188   g_return_if_fail (array);
 189
 190   if (free_segment)
 191     g_free (array->pdata);
 192
 193   g_mem_chunk_free (ptr_array_mem_chunk, array);
 194 }
 195
 196 static void
 197 g_ptr_array_maybe_expand (GRealPtrArray *array,
 198                           gint        len)
 199 {
 200   guint old_alloc;
 201
 202   if ((array->len + len) > array->alloc)
 203     {
 204       old_alloc = array->alloc;
 205
 206       array->alloc = g_nearest_pow (array->len + len);
 207       array->alloc = MAX (array->alloc, MIN_ARRAY_SIZE);
 208       if (array->pdata)
 209         array->pdata = g_realloc (array->pdata, sizeof(gpointer) * array->alloc);
 210       else
 211         array->pdata = g_new0 (gpointer, array->alloc);
 212
 213       memset (array->pdata + old_alloc, 0, array->alloc - old_alloc);
 214     }
 215 }
 216
 217 void
 218 g_ptr_array_set_size  (GPtrArray   *farray,
 219                        gint          length)
 220 {
 221   GRealPtrArray* array = (GRealPtrArray*) farray;
 222
 223   g_return_if_fail (array);
 224
 225   if (length > array->len)
 226     g_ptr_array_maybe_expand (array, (length - array->len));
 227
 228   array->len = length;
 229 }
 230
 231 void
 232 g_ptr_array_remove_index (GPtrArray* farray,
 233                           gint index)
 234 {
 235   GRealPtrArray* array = (GRealPtrArray*) farray;
 236
 237   g_return_if_fail (array);
 238
 239   g_return_if_fail (index < array->len);
 240
 241   array->pdata[index] = array->pdata[array->len - 1];
 242
 243   array->pdata[array->len - 1] = NULL;
 244
 245   array->len -= 1;
 246 }
 247
 248 gboolean
 249 g_ptr_array_remove (GPtrArray* farray,
 250                     gpointer data)
 251 {
 252   GRealPtrArray* array = (GRealPtrArray*) farray;
 253   int i;
 254
 255   g_return_val_if_fail (array, FALSE);
 256
 257   for (i = 0; i < array->len; i += 1)
 258     {
 259       if (array->pdata[i] == data)
 260         {
 261           g_ptr_array_remove_index (farray, i);
 262           return TRUE;
 263         }
 264     }
 265
 266   return FALSE;
 267 }
 268
 269 void
 270 g_ptr_array_add (GPtrArray* farray,
 271                  gpointer data)
 272 {
 273   GRealPtrArray* array = (GRealPtrArray*) farray;
 274
 275   g_return_if_fail (array);
 276
 277   g_ptr_array_maybe_expand (array, 1);
 278
 279   array->pdata[array->len++] = data;
 280 }
 281
 282 /* Byte arrays
 283  */
 284
 285 GByteArray* g_byte_array_new      (void)
 286 {
 287   return (GByteArray*) g_array_new (FALSE);
 288 }
 289
 290 void        g_byte_array_free     (GByteArray *array,
 291                                    gint        free_segment)
 292 {
 293   g_array_free ((GArray*) array, free_segment);
 294 }
 295
 296 GByteArray* g_byte_array_append   (GByteArray *array,
 297                                    const guint8 *data,
 298                                    guint       len)
 299 {
 300   g_rarray_append ((GArray*) array, (guint8*)data, len);
 301
 302   return array;
 303 }
 304
 305 GByteArray* g_byte_array_prepend  (GByteArray *array,
 306                                    const guint8 *data,
 307                                    guint       len)
 308 {
 309   g_rarray_prepend ((GArray*) array, (guint8*)data, len);
 310
 311   return array;
 312 }
 313
 314 GByteArray* g_byte_array_truncate (GByteArray *array,
 315                                    gint        length)
 316 {
 317   g_rarray_truncate ((GArray*) array, length, 1);
 318
 319   return array;
 320 }