[utils] Removing kvec and using similar implementation with pointers that can be passed around
This commit is contained in:
114
src/vector.h
114
src/vector.h
@@ -1,66 +1,70 @@
|
||||
#ifndef VECTOR_H
|
||||
#define VECTOR_H
|
||||
|
||||
|
||||
|
||||
#include <stdlib.h>
|
||||
|
||||
#include "klib/kvec.h"
|
||||
|
||||
#define DEFAULT_VECTOR_SIZE 8
|
||||
|
||||
// Wrapper around kvec.h to provide dynamically allocated vectors
|
||||
#define __VECTOR_BASE(name, type) typedef kvec_t(type) name; \
|
||||
static inline name *name##_new(void) { \
|
||||
name *array = malloc(sizeof(name)); \
|
||||
if (array == NULL) return NULL; \
|
||||
kv_init(*array); \
|
||||
kv_resize(type, *array, DEFAULT_VECTOR_SIZE); \
|
||||
return array; \
|
||||
} \
|
||||
static inline name *name##_new_size(size_t size) { \
|
||||
name *array = name##_new(); \
|
||||
if (size > DEFAULT_VECTOR_SIZE) { \
|
||||
kv_resize(type, *array, size); \
|
||||
} \
|
||||
return array; \
|
||||
} \
|
||||
static inline void name##_push(name *array, type value) { \
|
||||
kv_push(type, *array, value); \
|
||||
} \
|
||||
static inline void name##_extend(name *array, name *other) { \
|
||||
for (int i = 0; i < other->n; i++) { \
|
||||
kv_push(type, *array, *(other->a + i)); \
|
||||
} \
|
||||
} \
|
||||
static inline type name##_pop(name *array) { \
|
||||
return kv_pop(*array); \
|
||||
} \
|
||||
static inline void name##_clear(name *array) { \
|
||||
kv_clear(*array); \
|
||||
} \
|
||||
static inline void name##_resize(name *array, size_t size) { \
|
||||
kv_resize(type, *array, size); \
|
||||
} \
|
||||
\
|
||||
static inline void name##_copy(name *dst, name *src, size_t n) { \
|
||||
if (dst->m < n) name##_resize(dst, n); \
|
||||
memcpy(dst->a, src->a, n * sizeof(type)); \
|
||||
dst->n = n; \
|
||||
} \
|
||||
\
|
||||
static inline name *name##_new_copy(name *vector, size_t n) { \
|
||||
name *cpy = name##_new_size(n); \
|
||||
name##_copy(cpy, vector, n); \
|
||||
return cpy; \
|
||||
} \
|
||||
\
|
||||
// Based kvec.h, dynamic vectors of any type
|
||||
#define __VECTOR_BASE(name, type) typedef struct { size_t n, m; type *a; } name; \
|
||||
static inline name *name##_new_size(size_t size) { \
|
||||
name *array = malloc(sizeof(name)); \
|
||||
if (array == NULL) return NULL; \
|
||||
array->n = array->m = 0; \
|
||||
array->a = NULL; \
|
||||
array->a = malloc(size * sizeof(type)); \
|
||||
if (array->a == NULL) return NULL; \
|
||||
array->m = size; \
|
||||
return array; \
|
||||
} \
|
||||
static inline name *name##_new(void) { \
|
||||
return name##_new_size(DEFAULT_VECTOR_SIZE); \
|
||||
} \
|
||||
static inline void name##_resize(name *array, size_t size) { \
|
||||
if (size <= array->m) return; \
|
||||
type *ptr = realloc(array->a, sizeof(type) * size); \
|
||||
if (ptr == NULL) return; \
|
||||
array->a = ptr; \
|
||||
array->m = size; \
|
||||
} \
|
||||
static inline void name##_push(name *array, type value) { \
|
||||
if (array->n == array->m) { \
|
||||
size_t size = array->m ? array->m << 1 : 2; \
|
||||
type *ptr = realloc(array->a, sizeof(type) * size); \
|
||||
if (ptr == NULL) return; \
|
||||
array->a = ptr; \
|
||||
array->m = size; \
|
||||
} \
|
||||
array->a[array->n++] = value; \
|
||||
} \
|
||||
static inline void name##_extend(name *array, name *other) { \
|
||||
size_t new_size = array->n + other->n; \
|
||||
if (new_size >= array->m) name##_resize(array, new_size); \
|
||||
memcpy(array->a + array->n, other->a, other->n * sizeof(type)); \
|
||||
array->n = new_size; \
|
||||
} \
|
||||
static inline type name##_pop(name *array) { \
|
||||
return array->a[--array->n]; \
|
||||
} \
|
||||
static inline void name##_clear(name *array) { \
|
||||
array->n = 0; \
|
||||
} \
|
||||
static inline void name##_copy(name *dst, name *src, size_t n) { \
|
||||
if (dst->m < n) name##_resize(dst, n); \
|
||||
memcpy(dst->a, src->a, n * sizeof(type)); \
|
||||
dst->n = n; \
|
||||
} \
|
||||
static inline name *name##_new_copy(name *vector, size_t n) { \
|
||||
name *cpy = name##_new_size(n); \
|
||||
name##_copy(cpy, vector, n); \
|
||||
return cpy; \
|
||||
}
|
||||
|
||||
|
||||
#define __VECTOR_DESTROY(name, type) \
|
||||
static inline void name##_destroy(name *array) { \
|
||||
if (array == NULL) return; \
|
||||
kv_destroy(*array); \
|
||||
if (array->a != NULL) free(array->a); \
|
||||
free(array); \
|
||||
}
|
||||
|
||||
@@ -68,10 +72,12 @@
|
||||
#define __VECTOR_DESTROY_FREE_DATA(name, type, free_func) \
|
||||
static inline void name##_destroy(name *array) { \
|
||||
if (array == NULL) return; \
|
||||
for (int i = 0; i < array->n; i++) { \
|
||||
free_func(array->a[i]); \
|
||||
if (array->a != NULL) { \
|
||||
for (int i = 0; i < array->n; i++) { \
|
||||
free_func(array->a[i]); \
|
||||
} \
|
||||
} \
|
||||
kv_destroy(*array); \
|
||||
free(array->a); \
|
||||
free(array); \
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user