cvector_str.c

Go to the documentation of this file.

#include "cvector_str.h"
 
 
 
#if defined(CVEC_MALLOC) && defined(CVEC_FREE) && defined(CVEC_REALLOC)
/* ok */
#elif !defined(CVEC_MALLOC) && !defined(CVEC_FREE) && !defined(CVEC_REALLOC)
/* ok */
#else
#error "Must define all or none of CVEC_MALLOC, CVEC_FREE, and CVEC_REALLOC."
#endif
 
#ifndef CVEC_MALLOC
#define CVEC_MALLOC(sz)      malloc(sz)
#define CVEC_REALLOC(p, sz)  realloc(p, sz)
#define CVEC_FREE(p)         free(p)
#endif
 
#ifndef CVEC_MEMMOVE
#include <string.h>
#define CVEC_MEMMOVE(dst, src, sz)  memmove(dst, src, sz)
#endif
 
#ifndef CVEC_ASSERT
#include <assert.h>
#define CVEC_ASSERT(x)       assert(x)
#endif
 
cvec_sz CVEC_STR_START_SZ = 20;
 
#define CVEC_STR_ALLOCATOR(x) ((x+1) * 2)
 
#if CVEC_STRDUP == cvec_strdup
char* cvec_strdup(const char* str)
{
    cvec_sz len;
    char* temp;
    if (!str)
        return NULL;
 
    len = strlen(str);
    temp = (char*)CVEC_MALLOC(len+1);
    if (!temp) {
        CVEC_ASSERT(temp != NULL);
        return NULL;
    }
    temp[len] = 0;
    
    return (char*)CVEC_MEMMOVE(temp, str, len);  /* CVEC_MEMMOVE returns to */
}
#endif
 
cvector_str* cvec_str_heap(cvec_sz size, cvec_sz capacity)
{
    cvector_str* vec;
    if (!(vec = (cvector_str*)CVEC_MALLOC(sizeof(cvector_str)))) {
        CVEC_ASSERT(vec != NULL);
        return NULL;
    }
 
    vec->size = size;
    vec->capacity = (capacity > vec->size || (vec->size && capacity == vec->size)) ? capacity : vec->size + CVEC_STR_START_SZ;
 
    if (!(vec->a = (char**)CVEC_MALLOC(vec->capacity * sizeof(char*)))) {
        CVEC_ASSERT(vec->a != NULL);
        CVEC_FREE(vec);
        return NULL;
    }
    /* clearing to 0 here because if the user gave a non-zero initial size, popping/
     * erasing will crash unless they're NULL.  Really the user should never do that.
     * They should use cvec_init or otherwise immediately assign to the size elements they
     * started with.  */
    memset(vec->a, 0, vec->capacity*sizeof(char*));
 
    return vec;
}
 
cvector_str* cvec_init_str_heap(char** vals, cvec_sz num)
{
    cvector_str* vec;
    cvec_sz i;
    
    if (!(vec = (cvector_str*)CVEC_MALLOC(sizeof(cvector_str)))) {
        CVEC_ASSERT(vec != NULL);
        return NULL;
    }
 
    vec->capacity = num + CVEC_STR_START_SZ;
    vec->size = num;
    if (!(vec->a = (char**)CVEC_MALLOC(vec->capacity*sizeof(char*)))) {
        CVEC_ASSERT(vec->a != NULL);
        CVEC_FREE(vec);
        return NULL;
    }
 
    for(i=0; i<num; i++) {
        vec->a[i] = CVEC_STRDUP(vals[i]);
    }
    
    return vec;
}
 
int cvec_str(cvector_str* vec, cvec_sz size, cvec_sz capacity)
{
    vec->size = size;
    vec->capacity = (capacity > vec->size || (vec->size && capacity == vec->size)) ? capacity : vec->size + CVEC_STR_START_SZ;
 
    if (!(vec->a = (char**)CVEC_MALLOC(vec->capacity * sizeof(char*)))) {
        CVEC_ASSERT(vec->a != NULL);
        vec->size = vec->capacity = 0;
        return 0;
    }
    /* clearing to 0 here because if the user gave a non-zero initial size, popping/
     * erasing will crash unless they're NULL */
    memset(vec->a, 0, vec->capacity*sizeof(char*));
 
    return 1;
}
 
int cvec_init_str(cvector_str* vec, char** vals, cvec_sz num)
{
    cvec_sz i;
    
    vec->capacity = num + CVEC_STR_START_SZ;
    vec->size = num;
    if (!(vec->a = (char**)CVEC_MALLOC(vec->capacity*sizeof(char*)))) {
        CVEC_ASSERT(vec->a != NULL);
        vec->size = vec->capacity = 0;
        return 0;
    }
 
    for(i=0; i<num; i++) {
        vec->a[i] = CVEC_STRDUP(vals[i]);
    }
    
    return 1;
}
 
int cvec_copyc_str(void* dest, void* src)
{
    cvector_str* vec1 = (cvector_str*)dest;
    cvector_str* vec2 = (cvector_str*)src;
 
    vec1->a = NULL;
    vec1->size = 0;
    vec1->capacity = 0;
 
    return cvec_copy_str(vec1, vec2);
}
 
int cvec_copy_str(cvector_str* dest, cvector_str* src)
{
    int i;
    char** tmp = NULL;
    if (!(tmp = (char**)CVEC_REALLOC(dest->a, src->capacity*sizeof(char*)))) {
        CVEC_ASSERT(tmp != NULL);
        return 0;
    }
    dest->a = tmp;
    
    for (i=0; i<src->size; ++i) {
        dest->a[i] = CVEC_STRDUP(src->a[i]);
    }
    dest->size = src->size;
    dest->capacity = src->capacity;
    return 1;
}
 
int cvec_push_str(cvector_str* vec, char* a)
{
    char** tmp;
    cvec_sz tmp_sz;
    if (vec->capacity == vec->size) {
        tmp_sz = CVEC_STR_ALLOCATOR(vec->capacity);
        if (!(tmp = (char**)CVEC_REALLOC(vec->a, sizeof(char*)*tmp_sz))) {
            CVEC_ASSERT(tmp != NULL);
            return 0;
        }
        vec->a = tmp;
        vec->capacity = tmp_sz;
    }
    
    vec->a[vec->size++] = CVEC_STRDUP(a);
    return 1;
}
 
int cvec_pushm_str(cvector_str* vec, char* a)
{
    char** tmp;
    cvec_sz tmp_sz;
    if (vec->capacity == vec->size) {
        tmp_sz = CVEC_STR_ALLOCATOR(vec->capacity);
        if (!(tmp = (char**)CVEC_REALLOC(vec->a, sizeof(char*)*tmp_sz))) {
            CVEC_ASSERT(tmp != NULL);
            return 0;
        }
        vec->a = tmp;
        vec->capacity = tmp_sz;
    }
    
    vec->a[vec->size++] = a;
    return 1;
}
 
void cvec_pop_str(cvector_str* vec, char* ret)
{
    vec->size--;
    if (ret)
        strcpy(ret, vec->a[vec->size]);
    CVEC_FREE(vec->a[vec->size]);
}
 
char** cvec_back_str(cvector_str* vec)
{
    return &vec->a[vec->size-1];
}
 
int cvec_extend_str(cvector_str* vec, cvec_sz num)
{
    char** tmp;
    cvec_sz tmp_sz;
    if (vec->capacity < vec->size + num) {
        tmp_sz = vec->capacity + num + CVEC_STR_START_SZ;
        if (!(tmp = (char**)CVEC_REALLOC(vec->a, sizeof(char*)*tmp_sz))) {
            CVEC_ASSERT(tmp != NULL);
            return 0;
        }
        vec->a = tmp;
        vec->capacity = tmp_sz;
    }
 
    memset(&vec->a[vec->size], 0, num*sizeof(char*));
    vec->size += num;
    return 1;
}
 
int cvec_insert_str(cvector_str* vec, cvec_sz i, char* a)
{
    char** tmp;
    cvec_sz tmp_sz;
    if (vec->capacity == vec->size) {
        tmp_sz = CVEC_STR_ALLOCATOR(vec->capacity);
        if (!(tmp = (char**)CVEC_REALLOC(vec->a, sizeof(char*)*tmp_sz))) {
            CVEC_ASSERT(tmp != NULL);
            return 0;
        }
        vec->a = tmp;
        vec->capacity = tmp_sz;
    }
 
    CVEC_MEMMOVE(&vec->a[i+1], &vec->a[i], (vec->size-i)*sizeof(char*));
    vec->a[i] = CVEC_STRDUP(a);
    vec->size++;
    return 1;
}
 
int cvec_insertm_str(cvector_str* vec, cvec_sz i, char* a)
{
    char** tmp;
    cvec_sz tmp_sz;
    if (vec->capacity == vec->size) {
        tmp_sz = CVEC_STR_ALLOCATOR(vec->capacity);
        if (!(tmp = (char**)CVEC_REALLOC(vec->a, sizeof(char*)*tmp_sz))) {
            CVEC_ASSERT(tmp != NULL);
            return 0;
        }
        vec->a = tmp;
        vec->capacity = tmp_sz;
    }
 
    CVEC_MEMMOVE(&vec->a[i+1], &vec->a[i], (vec->size-i)*sizeof(char*));
    vec->a[i] = a;
    vec->size++;
    return 1;
}
 
int cvec_insert_array_str(cvector_str* vec, cvec_sz i, char** a, cvec_sz num)
{
    char** tmp;
    cvec_sz tmp_sz, j;
    if (vec->capacity < vec->size + num) {
        tmp_sz = vec->capacity + num + CVEC_STR_START_SZ;
        if (!(tmp = (char**)CVEC_REALLOC(vec->a, sizeof(char*)*tmp_sz))) {
            CVEC_ASSERT(tmp != NULL);
            return 0;
        }
        vec->a = tmp;
        vec->capacity = tmp_sz;
    }
 
    CVEC_MEMMOVE(&vec->a[i+num], &vec->a[i], (vec->size-i)*sizeof(char*));
    for (j=0; j<num; ++j) {
        vec->a[j+i] = CVEC_STRDUP(a[j]);
    }
    
    vec->size += num;
    return 1;
}
 
int cvec_insert_arraym_str(cvector_str* vec, cvec_sz i, char** a, cvec_sz num)
{
    char** tmp;
    cvec_sz tmp_sz;
    if (vec->capacity < vec->size + num) {
        tmp_sz = vec->capacity + num + CVEC_STR_START_SZ;
        if (!(tmp = (char**)CVEC_REALLOC(vec->a, sizeof(char*)*tmp_sz))) {
            CVEC_ASSERT(tmp != NULL);
            return 0;
        }
        vec->a = tmp;
        vec->capacity = tmp_sz;
    }
 
    CVEC_MEMMOVE(&vec->a[i+num], &vec->a[i], (vec->size-i)*sizeof(char*));
 
    CVEC_MEMMOVE(&vec->a[i], a, num*sizeof(char*));
    vec->size += num;
    return 1;
}
 
void cvec_replace_str(cvector_str* vec, cvec_sz i, char* a, char* ret)
{
    if (ret)
        strcpy(ret, vec->a[i]);
    CVEC_FREE(vec->a[i]);
    vec->a[i] = CVEC_STRDUP(a);
}
 
void cvec_erase_str(cvector_str* vec, cvec_sz start, cvec_sz end)
{
    cvec_sz i;
    cvec_sz d = end - start + 1;
    for (i=start; i<=end; i++) {
        CVEC_FREE(vec->a[i]);
    }
    
    CVEC_MEMMOVE(&vec->a[start], &vec->a[end+1], (vec->size-1-end)*sizeof(char*));
    vec->size -= d;
}
 
void cvec_remove_str(cvector_str* vec, cvec_sz start, cvec_sz end)
{
    cvec_sz d = end - start + 1;
    CVEC_MEMMOVE(&vec->a[start], &vec->a[end+1], (vec->size-1-end)*sizeof(char*));
    vec->size -= d;
}
 
int cvec_reserve_str(cvector_str* vec, cvec_sz size)
{
    char** tmp;
    if (vec->capacity < size) {
        if (!(tmp = (char**)CVEC_REALLOC(vec->a, sizeof(char*)*(size+CVEC_STR_START_SZ)))) {
            CVEC_ASSERT(tmp != NULL);
            return 0;
        }
        vec->a = tmp;
        vec->capacity = size + CVEC_STR_START_SZ;
    }
    return 1;
}
 
int cvec_set_cap_str(cvector_str* vec, cvec_sz size)
{
    cvec_sz i;
    char** tmp;
    if (size < vec->size) {
        for(i=vec->size-1; i>size-1; i--) {
            CVEC_FREE(vec->a[i]);
        }
 
        vec->size = size;
    }
 
    if (!(tmp = (char**)CVEC_REALLOC(vec->a, sizeof(char*)*size))) {
        CVEC_ASSERT(tmp != NULL);
        return 0;
    }
    vec->a = tmp;
    vec->capacity = size;
    return 1;
}
 
void cvec_set_val_sz_str(cvector_str* vec, char* val)
{
    cvec_sz i;
    for(i=0; i<vec->size; i++) {
        CVEC_FREE(vec->a[i]);
 
        /* not worth checking to me see commit msg aa0c5cf */
        vec->a[i] = CVEC_STRDUP(val);
    }
}
 
void cvec_set_val_cap_str(cvector_str* vec, char* val)
{
    cvec_sz i;
    for (i=0; i<vec->capacity; i++) {
        if (i<vec->size) {
            CVEC_FREE(vec->a[i]);
        }
        
        vec->a[i] = CVEC_STRDUP(val);
    }
    vec->size = vec->capacity;
}
 
void cvec_clear_str(cvector_str* vec)
{
    int i;
    for (i=0; i<vec->size; i++) {
        CVEC_FREE(vec->a[i]);
    }
    
    vec->size = 0;
}
 
void cvec_free_str_heap(void* vec)
{
    cvec_sz i;
    cvector_str* tmp = (cvector_str*)vec;
    if (!tmp) return;
    for (i=0; i<tmp->size; i++) {
        CVEC_FREE(tmp->a[i]);
    }
    
    CVEC_FREE(tmp->a);
    CVEC_FREE(tmp);
}
 
void cvec_free_str(void* vec)
{
    cvec_sz i;
    cvector_str* tmp = (cvector_str*)vec;
    for (i=0; i<tmp->size; i++) {
        CVEC_FREE(tmp->a[i]);
    }
    
    CVEC_FREE(tmp->a);
    tmp->size = 0;
    tmp->capacity = 0;
}