Changed to heap-based implementation.

This commit is contained in:
Scott Anderson 2017-05-10 01:50:54 +12:00
parent d404c4ab28
commit 464bc5ff8e
3 changed files with 71 additions and 47 deletions

View file

@ -3,7 +3,7 @@ cmake_minimum_required(VERSION 3.1.0)
project(sway C)
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -g")
set(CMAKE_C_STANDARD 11)
set(CMAKE_C_STANDARD 99)
set(CMAKE_C_EXTENSIONS OFF)
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/bin)

View file

@ -8,7 +8,7 @@ list_t *list_new(size_t memb_size, size_t capacity) {
if (capacity == 0)
capacity = 8;
list_t *l = malloc(sizeof(*l) + memb_size * capacity);
list_t *l = malloc(sizeof(*l));
if (!l) {
return NULL;
}
@ -17,48 +17,62 @@ list_t *list_new(size_t memb_size, size_t capacity) {
l->length = 0;
l->memb_size = memb_size;
l->data = malloc(memb_size * capacity);
if (!l->data) {
free(l);
return NULL;
}
return l;
}
static bool resize(list_t **list) {
list_t *l = *list;
void list_free(list_t *list) {
if (!list) {
return;
}
if (l->length < l->capacity) {
free(list->data);
free(list);
}
static bool resize(list_t *list) {
if (list->length < list->capacity) {
return true;
}
size_t new_cap = l->capacity * 2;
l = realloc(l, sizeof(*l) + l->memb_size * new_cap);
if (!l) {
size_t new_cap = list->capacity * 2;
void *data = realloc(list->data, list->memb_size * new_cap);
if (!data) {
return false;
}
*list = l;
l->capacity = new_cap;
list->data = data;
list->capacity = new_cap;
return true;
}
void list_add(list_t **list, const void *data) {
if (!data || !list || !*list || !resize(list)) {
void list_add(list_t *list, const void *data) {
if (!data || !list || !resize(list)) {
return;
}
list_t *l = *list;
memcpy(l->data + l->memb_size * l->length, data, l->memb_size);
++l->length;
size_t size = list->memb_size;
unsigned char (*array)[size] = list->data;
memcpy(&array[list->length], data, size);
++list->length;
}
void list_insert(list_t **list, size_t index, const void *data) {
if (!data || !list || !*list || index > (*list)->length || !resize(list)) {
void list_insert(list_t *list, size_t index, const void *data) {
if (!data || !list || index > list->length || !resize(list)) {
return;
}
list_t *l = *list;
size_t size = l->memb_size;
unsigned char (*array)[size] = (void *)l->data;
memmove(&array[index + 1], &array[index], size * (l->length - index));
size_t size = list->memb_size;
unsigned char (*array)[size] = list->data;
memmove(&array[index + 1], &array[index], size * (list->length - index));
memcpy(&array[index], data, size);
++l->length;
++list->length;
}
void list_delete(list_t *list, size_t index) {
@ -67,7 +81,7 @@ void list_delete(list_t *list, size_t index) {
}
size_t size = list->memb_size;
unsigned char (*array)[size] = (void *)list->data;
unsigned char (*array)[size] = list->data;
memmove(&array[index], &array[index + 1], size * (list->length - index));
--list->length;
@ -79,7 +93,7 @@ void list_swap(list_t *list, size_t i1, size_t i2) {
}
size_t size = list->memb_size;
unsigned char (*array)[size] = (void *)list->data;
unsigned char (*array)[size] = list->data;
unsigned char tmp[size];
memcpy(tmp, &array[i1], size);
@ -92,7 +106,10 @@ void *list_get(list_t *list, size_t index) {
return NULL;
}
return list->data + list->memb_size * index;
size_t size = list->memb_size;
unsigned char (*array)[size] = list->data;
return &array[index];
}
void list_qsort(list_t *list, int compare(const void *, const void *)) {
@ -109,7 +126,7 @@ void list_isort(list_t *list, int compare(const void *, const void *)) {
}
size_t size = list->memb_size;
unsigned char (*array)[size] = (void *)list->data;
unsigned char (*array)[size] = list->data;
for (size_t i = 1; i < list->length; ++i) {
unsigned char tmp[size];
@ -138,7 +155,7 @@ ssize_t list_bsearch(const list_t *list, int compare(const void *, const void *)
if (ret) {
memcpy(ret, ptr, list->memb_size);
}
return (ptr - list->data) / list->memb_size;
return (ptr - (unsigned char *)list->data) / list->memb_size;
}
}
@ -149,7 +166,7 @@ ssize_t list_lsearch(const list_t *list, int compare(const void *, const void *)
}
size_t size = list->memb_size;
unsigned char (*array)[size] = (void *)list->data;
unsigned char (*array)[size] = list->data;
for (size_t i = 0; i < list->length; ++i) {
if (compare(&array[i], key) == 0) {
@ -163,15 +180,23 @@ ssize_t list_lsearch(const list_t *list, int compare(const void *, const void *)
return -1;
}
void list_foreach_cb(list_t *list, void callback(void *)) {
void list_foreach(list_t *list, void callback(void *)) {
if (!list || !callback) {
return;
}
size_t size = list->memb_size;
unsigned char (*array)[size] = (void *)list->data;
unsigned char (*array)[size] = list->data;
for (size_t i = 0; i < list->length; ++i) {
callback(&array[i]);
}
}
void *list_end(list_t *list) {
if (!list) {
return NULL;
}
return (unsigned char *)list->data + list->memb_size * list->length;
}

View file

@ -1,7 +1,6 @@
#ifndef _SWAY_LIST_H
#define _SWAY_LIST_H
#include <stdalign.h>
#include <stddef.h>
#include <sys/types.h>
@ -9,7 +8,7 @@ typedef struct {
size_t capacity;
size_t length;
size_t memb_size;
alignas(max_align_t) unsigned char data[];
void *data;
} list_t;
/*
@ -19,22 +18,25 @@ typedef struct {
* memb_size must be the size of the type stored in this list.
* If an element is added to this list which is not the same type
* as the elements of the list, the behavior is undefined.
*
* This list should be freed with the stdlib free() function.
*/
list_t *list_new(size_t memb_size, size_t capacity);
/*
* Frees a list.
*/
void list_free(list_t *list);
/*
* Adds an element to the end of the list.
*/
void list_add(list_t **list, const void *data);
void list_add(list_t *list, const void *data);
/*
* Adds an element at an arbitrary position in the list, moving
* the elements past index to make space.
* index must be less than or equal to the length of the list.
*/
void list_insert(list_t **list, size_t index, const void *data);
void list_insert(list_t *list, size_t index, const void *data);
/*
* Deletes the element at an arbitrary position in the list,
@ -87,22 +89,19 @@ ssize_t list_lsearch(const list_t *list, int compare(const void *, const void *)
/*
* Calls a function on every item in the list.
*/
void list_foreach_cb(list_t *list, void callback(void *));
void list_foreach(list_t *list, void callback(void *));
/*
* This is for loop helper, iterating over every element in list.
* iter must be a pointer to the list element's type.
* You should not add, remove, or sort elements from the list while inside this loop.
* Returns a pointer to just past the end of the list.
* This can be used to write for loops more easily.
*
* Example:
* char **ptr;
* list_foreach(list_of_strings, ptr) {
* Example (for a list of char *):
* char **end = list_end(list);
* for (char **ptr = list->items; ptr < end; ++ptr) {
* printf("%s\n", *ptr);
* }
*/
#define list_foreach(list, iter) \
for (iter = (void *)list->data; \
(unsigned char *)iter < list->data + list->memb_size * list->length; \
++iter)
// Consider making this inline or __attribute__((pure))__
void *list_end(list_t *list);
#endif