add .gitignore

[tedtools.git] / tmalloc.c

diff --git a/tmalloc.c b/tmalloc.c
index 0d51e75..e93a46c 100644 (file)
--- a/tmalloc.c
+++ b/tmalloc.c
@@ -298,16 +298,52 @@ mcnstrdup(MemoryContext *cntx, char *src, int len) {
 }
 
 /*********StringBuffer********/
-StringBuffer*
-initStringBuffer(StringBuffer* state, MemoryContext *mc, int initsize) {
+
+static inline void*
+StringBufferAlloc(StringBuffer* state, void *ptr, size_t size) {
+       if (state->mc)
+               return (ptr) ? mcrealloc(ptr, size) : mcalloc(state->mc, size);
+       else if (state->memalloc)
+               return state->memalloc(ptr, size);
+       else
+               return (ptr) ? trealloc(ptr, size) : tmalloc(size);
+}
+
+static StringBuffer*
+initStringBufferInternal(StringBuffer* state, int initsize) {
     state->len = (initsize>0) ? initsize : 1024;
-       state->mc = mc;
-       state->ptr = state->buf = (char*)mcalloc(state->mc, state->len);
+       state->ptr = state->buf = (char*)StringBufferAlloc(state, NULL, initsize);
+       if (!state->ptr)
+               return NULL;
+
        *(state->ptr) ='\0';
 
        return state;
 }
 
+StringBuffer*
+initStringBuffer(StringBuffer* state, int initsize) {
+       memset(state, 0, sizeof(*state));
+       
+       return initStringBufferInternal(state, initsize);
+}
+
+StringBuffer*
+initStringBufferMC(StringBuffer* state, MemoryContext *mc, int initsize) {
+       memset(state, 0, sizeof(*state));
+       state->mc = mc;
+
+       return initStringBufferInternal(state, initsize);
+}
+
+StringBuffer*
+initStringBufferMA(StringBuffer* state, MemAlloc memalloc, int initsize) {
+       memset(state, 0, sizeof(*state));
+       state->memalloc = memalloc;
+
+       return initStringBufferInternal(state, initsize);
+}
+
 StringBuffer*
 appendStringBuffer( StringBuffer *state, char *string, int stringlen) {
     if ( string ) {
@@ -320,8 +356,11 @@ appendStringBuffer( StringBuffer *state, char *string, int stringlen) {
                return state;
                                                                                                          
        while ( state->len - ( state->ptr - state->buf ) < stringlen + 1 ) {
+                       int     diff = state->ptr - state->buf;
+
                        state->len *= 2;
-                       state->buf = (char*)mcrealloc( (void*)state->buf, state->len );
+                       state->buf = (char*)StringBufferAlloc(state, state->buf, state->len);
+                       state->ptr = state->buf + diff;
        }
 
        memcpy(state->ptr, string, stringlen);
@@ -351,7 +390,7 @@ printStringBuffer( StringBuffer *state, const char *format, ...) {
                        state->len *= 2;
                } while( state->ptr - state->buf + printedlen >= state->len );
 
-               state->buf = (char*)mcrealloc( (void*)state->buf, state->len );
+               state->buf = (char*)StringBufferAlloc(state, state->buf, state->len);
                state->ptr = state->buf + curlen;
                va_start(args, format);
                printedlen = vsnprintf(state->ptr, printedlen+1, format, args);