chiark - git - mdw - disorder/blob - lib/vector.h

   1 /*
   2  * This file is part of DisOrder.
   3  * Copyright (C) 2004, 2005 Richard Kettlewell
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  18  * USA
  19  */
  20 /** @file lib/vector.h @brief Dynamic array template */
  21
  22 #ifndef VECTOR_H
  23 #define VECTOR_H
  24
  25 /** @brief Dynamic array template
  26  * @param NAME type name
  27  * @param ETYPE element type
  28  * @param REALLOC realloc function
  29  *
  30  * Defines @c struct @p NAME as a dynamic array with element type @p
  31  * ETYPE.  @p REALLOC should have the same signature as realloc() and
  32  * will be used for all memory allocation.  Typically it would be
  33  * xrealloc() for pointer-containing element types and
  34  * xrealloc_noptr() for pointer-free element types.
  35  *
  36  * Clients are inspected to read the @p vec member of the structure,
  37  * which points to the first element, and the @p nvec member, which is
  38  * the number of elements.  It is safe to reduce @p nvec.  Do not
  39  * touch any other members.
  40  *
  41  * The functions defined are:
  42  * - NAME_init(struct NAME *v) which initializes @p v
  43  * - NAME_append(struct NAME *v, ETYPE value) which appends @p value to @p v
  44  * - NAME_terminate(struct NAME *v) which zeroes out the element beyond the last
  45  */
  46 #define VECTOR_TYPE(NAME,ETYPE,REALLOC)                         \
  47                                                                 \
  48 struct NAME {                                                   \
  49   /** @brief Pointer to elements */                             \
  50   ETYPE *vec;                                                   \
  51   /** @brief Number of elements */                              \
  52   int nvec;                                                     \
  53   /** @brief Number of slots */                                 \
  54   int nslots;                                                   \
  55 };                                                              \
  56                                                                 \
  57 static inline void NAME##_init(struct NAME *v) {                \
  58   memset(v, 0, sizeof *v);                                      \
  59 }                                                               \
  60                                                                 \
  61 static inline void NAME##_append(struct NAME *v, ETYPE val) {   \
  62   if(v->nvec >= v->nslots) {                                    \
  63     v->nslots = v->nslots ? 2 * v->nslots : 16;                 \
  64     v->vec = REALLOC(v->vec, v->nslots * sizeof(ETYPE));        \
  65   }                                                             \
  66   v->vec[v->nvec++] = val;                                      \
  67 }                                                               \
  68                                                                 \
  69 static inline void NAME##_terminate(struct NAME *v) {           \
  70   if(v->nvec >= v->nslots)                                      \
  71     v->vec = REALLOC(v->vec, ++v->nslots * sizeof(ETYPE));      \
  72   memset(&v->vec[v->nvec], 0, sizeof (ETYPE));                  \
  73 }                                                               \
  74                                                                 \
  75 struct vector_swallow_semicolon
  76
  77 /** @brief A dynamic array of pointers to strings */
  78 VECTOR_TYPE(vector, char *, xrealloc);
  79 /** @brief A dynamic string */
  80 VECTOR_TYPE(dynstr, char, xrealloc_noptr);
  81 /** @brief A dynamic unicode string */
  82 VECTOR_TYPE(dynstr_ucs4, uint32_t, xrealloc_noptr);
  83
  84 /** @brief Append many strings to a @ref vector */
  85 void vector_append_many(struct vector *v, char **vec, int nvec);
  86
  87 /** @brief Append @p n bytes to a @ref dynstr */
  88 void dynstr_append_bytes(struct dynstr *v, const char *ptr, size_t n);
  89
  90 /** @brief Append a string to a @ref dynstr */
  91 static inline void dynstr_append_string(struct dynstr *v, const char *ptr) {
  92   dynstr_append_bytes(v, ptr, strlen(ptr));
  93 }
  94
  95 #endif /* VECTOR_H */
  96
  97 /*
  98 Local Variables:
  99 c-basic-offset:2
 100 comment-column:40
 101 End:
 102 */