X-Git-Url: http://mj.ucw.cz/gitweb/?a=blobdiff_plain;ds=inline;f=lib%2Fgbuf.h;h=f0e618bf616d40e7a688b58461cb67d05fe28bc0;hb=6a26177ec0df11f55bff590d63ea02e7c9884993;hp=0775f20cb3738bb976176a72af94918322261f9d;hpb=e5f5e72c77dfa5bf4dcaf4b52c5dc42981abb631;p=libucw.git

diff --git a/lib/gbuf.h b/lib/gbuf.h
index 0775f20c..f0e618bf 100644
--- a/lib/gbuf.h
+++ b/lib/gbuf.h
@@ -2,18 +2,18 @@
  *	UCW Library -- A simple growing buffer
  *
  *	(c) 2004, Robert Spalek <robert@ucw.cz>
+ *	(c) 2005, Martin Mares <mj@ucw.cz>
  *
  *	Define the following macros:
  *
  *	GBUF_TYPE	data type of records stored in the buffer
  *	GBUF_PREFIX(x)	add a name prefix to all global symbols
+ *	GBUF_TRACE(msg...) log growing of buffer [optional]
  *
  *	This software may be freely distributed and used according to the terms
  *	of the GNU Lesser General Public License.
  */
 
-#include <stdlib.h>
-
 #define	BUF_T	GBUF_PREFIX(t)
 
 typedef struct
@@ -30,7 +30,7 @@ GBUF_PREFIX(init)(BUF_T *b)
   b->len = 0;
 }
 
-static inline void
+static void UNUSED
 GBUF_PREFIX(done)(BUF_T *b)
 {
   if (b->ptr)
@@ -39,23 +39,33 @@ GBUF_PREFIX(done)(BUF_T *b)
   b->len = 0;
 }
 
-static inline void
-GBUF_PREFIX(realloc)(BUF_T *b, uns len)
+static void UNUSED
+GBUF_PREFIX(set_size)(BUF_T *b, uns len)
 {
   b->len = len;
   b->ptr = xrealloc(b->ptr, len * sizeof(GBUF_TYPE));
+#ifdef GBUF_TRACE
+  GBUF_TRACE(STRINGIFY_EXPANDED(BUF_T) " growing to %u items", len);
+#endif
 }
 
-static inline void
-GBUF_PREFIX(grow)(BUF_T *b, uns len)
+static void UNUSED
+GBUF_PREFIX(do_grow)(BUF_T *b, uns len)
 {
-  if (likely(len <= b->len))
-    return;
   if (len < 2*b->len)			// to ensure logarithmic cost
     len = 2*b->len;
-  GBUF_PREFIX(realloc)(b, len);
+  GBUF_PREFIX(set_size)(b, len);
+}
+
+static inline GBUF_TYPE *
+GBUF_PREFIX(grow)(BUF_T *b, uns len)
+{
+  if (unlikely(len > b->len))
+    GBUF_PREFIX(do_grow)(b, len);
+  return b->ptr;
 }
 
 #undef	GBUF_TYPE
 #undef	GBUF_PREFIX
+#undef  GBUF_TRACE
 #undef	BUF_T