Magisk/native/jni/utils/vector.c

92 lines
2.1 KiB
C
Raw Normal View History

2017-04-04 21:44:13 +02:00
/* vector.c - A simple vector implementation in c
*/
#include <stdlib.h>
2017-05-04 22:16:00 +02:00
#include <string.h>
#include <stdarg.h>
2017-04-04 21:44:13 +02:00
#include "vector.h"
void vec_init(struct vector *v) {
2017-04-21 19:40:07 +02:00
if (v == NULL) return;
2017-04-04 21:44:13 +02:00
vec_size(v) = 0;
vec_cap(v) = 1;
vec_entry(v) = malloc(sizeof(void*));
}
void vec_push_back(struct vector *v, void *p) {
if (v == NULL) return;
if (vec_size(v) == vec_cap(v)) {
vec_cap(v) *= 2;
vec_entry(v) = realloc(vec_entry(v), sizeof(void*) * vec_cap(v));
}
vec_entry(v)[vec_size(v)] = p;
++vec_size(v);
}
void vec_push_back_all(struct vector *v, void *p, ...) {
va_list argv;
va_start(argv, p);
vec_push_back(v, p);
for (void *arg = va_arg(argv, char*); arg; arg = va_arg(argv, char*))
vec_push_back(v, arg);
va_end(argv);
}
void *vec_pop_back(struct vector *v) {
void *ret = vec_entry(v)[vec_size(v) - 1];
--vec_size(v);
return ret;
}
2017-12-20 20:36:18 +01:00
static int (*cmp)(const void *, const void *);
static int vec_comp(const void *a, const void *b) {
void *aa = *((void **)a), *bb = *((void **)b);
if (aa == NULL && bb == NULL) return 0;
else if (aa == NULL) return 1;
else if (bb == NULL) return -1;
else return cmp ? cmp(aa, bb) : 0;
}
2017-04-04 21:44:13 +02:00
void vec_sort(struct vector *v, int (*compar)(const void *, const void *)) {
2017-04-21 19:40:07 +02:00
if (v == NULL) return;
2017-12-20 20:36:18 +01:00
cmp = compar;
qsort(vec_entry(v), vec_size(v), sizeof(void*), vec_comp);
void *e;
vec_for_each_r(v, e) {
if (e) break;
--vec_size(v);
}
2017-04-04 21:44:13 +02:00
}
2017-04-06 00:12:29 +02:00
/* Will cleanup only the vector itself
* use in cases when each element requires special cleanup
*/
2017-04-04 21:44:13 +02:00
void vec_destroy(struct vector *v) {
2017-04-21 19:40:07 +02:00
if (v == NULL) return;
2017-04-04 21:44:13 +02:00
vec_size(v) = 0;
vec_cap(v) = 0;
2017-04-06 00:12:29 +02:00
free(vec_entry(v));
vec_entry(v) = NULL; // Prevent double destroy segfault
}
/* Will cleanup each element AND the vector itself
* Shall be the general case
*/
void vec_deep_destroy(struct vector *v) {
2017-04-21 19:40:07 +02:00
if (v == NULL) return;
2017-04-06 00:12:29 +02:00
void *e;
vec_for_each(v, e) {
free(e);
}
vec_destroy(v);
}
2017-05-04 22:16:00 +02:00
void vec_dup(struct vector *v, struct vector *vv) {
vec_size(vv) = vec_size(v);
vec_cap(vv) = vec_cap(v);
vec_entry(vv) = malloc(sizeof(void*) * vec_cap(v));
memcpy(vec_entry(vv), vec_entry(v), sizeof(void*) * vec_cap(v));
2017-05-04 22:16:00 +02:00
}