Add a vector of reference counted values.

This commit is contained in:
2018-03-31 20:44:01 -07:00
parent 35fc0e7fd1
commit e6801255fa
5 changed files with 140 additions and 5 deletions

66
src/ref_vec.c Normal file
View File

@@ -0,0 +1,66 @@
#include "ref_vec.h"
#include <stdlib.h>
libab_result libab_ref_vec_init(libab_ref_vec* vec) {
libab_result result = LIBAB_SUCCESS;
vec->capacity = LIBABACUS_REF_VEC_INITIAL_SIZE;
vec->size = 0;
vec->data =
malloc(sizeof(*vec->data) * LIBABACUS_REF_VEC_INITIAL_SIZE);
if(vec->data == NULL) {
result = LIBAB_MALLOC;
}
return result;
}
libab_result _libab_ref_vec_try_resize(libab_ref_vec* vec) {
libab_result result = LIBAB_SUCCESS;
if(vec->size == vec->capacity) {
libab_ref* new_memory = realloc(vec->data, (vec->capacity *= 2) * sizeof(*vec->data));
if(new_memory == NULL) {
free(vec->data);
result = LIBAB_MALLOC;
}
}
return result;
}
libab_result libab_ref_vec_insert(libab_ref_vec* vec, libab_ref* data) {
libab_result result = _libab_ref_vec_try_resize(vec);
if(result == LIBAB_SUCCESS) {
libab_ref_copy(data, &vec->data[vec->size++]);
}
return result;
}
libab_result libab_ref_vec_insert_value(libab_ref_vec* vec, void* data, void (*free_func)(void*)) {
libab_result result = _libab_ref_vec_try_resize(vec);
if(result == LIBAB_SUCCESS) {
result = libab_ref_new(&vec->data[vec->size], data, free_func);
}
if(result == LIBAB_SUCCESS) {
vec->size++;
}
return result;
}
const libab_ref* libab_ref_vec_index(libab_ref_vec* vec, size_t index) {
const libab_ref* to_return = NULL;
if(index < vec->size) {
to_return = &vec->data[index];
}
return to_return;
}
void libab_ref_vec_free(libab_ref_vec* vec) {
size_t i = 0;
for(; i < vec->size; i++) {
libab_ref_free(&vec->data[i]);
}
free(vec->data);
}

View File

@@ -42,13 +42,13 @@ void libab_ref_free(libab_ref* ref) {
_libab_ref_changed(ref);
}
void libab_ref_copy(libab_ref* ref, libab_ref* into) {
void libab_ref_copy(const libab_ref* ref, libab_ref* into) {
ref->count->strong++;
ref->count->weak++;
memcpy(into, ref, sizeof(*ref));
}
void* libab_ref_get(libab_ref* ref) {
void* libab_ref_get(const libab_ref* ref) {
void* to_return = NULL;
if(ref->count->strong > 0) {
to_return = ref->data;