tupperware: vector: add filter & map

This commit is contained in:
Bruno BELANYI 2020-11-28 14:23:54 +01:00
parent d94d848669
commit e296b4711c
2 changed files with 34 additions and 0 deletions

View file

@ -46,4 +46,11 @@ bool vector_heap_sort(struct vector *v, vector_cmp_f cmp, void *cookie);
bool vector_merge_sort(struct vector *v, vector_cmp_f cmp, void *cookie); bool vector_merge_sort(struct vector *v, vector_cmp_f cmp, void *cookie);
bool vector_sort(struct vector *v, vector_cmp_f cmp, void *cookie); bool vector_sort(struct vector *v, vector_cmp_f cmp, void *cookie);
typedef void (*vector_map_f)(void *v, void *cookie);
typedef bool (*vector_filter_f)(void *v, void *cookie);
bool vector_filter(struct vector *res,
struct vector *v, vector_filter_f filter, void *cookie);
void vector_map(struct vector *v, vector_map_f map, void *cookie);
#endif /* !TUPPERWARE_VECTOR_H */ #endif /* !TUPPERWARE_VECTOR_H */

View file

@ -569,3 +569,30 @@ bool vector_sort(struct vector *v, vector_cmp_f cmp, void *cookie) {
return true; return true;
} }
bool vector_filter(struct vector *res,
struct vector *v, vector_filter_f filter, void *cookie) {
if (!v || !res)
return false;
for (size_t i = 0; i < v->nmemb; ++i) {
void *elem = VEC_AT(v, i);
if (!filter(elem, cookie))
continue;
if (!vector_push_back(res, elem) || !vector_pop_at(v, NULL, i))
return false; // Undefined state at this point...
--i; // We need to see the new element at the same index
}
return true;
}
void vector_map(struct vector *v, vector_map_f map, void *cookie) {
if (!v)
return;
for (size_t i = 0; i < v->nmemb; ++i) {
void *elem = VEC_AT(v, i);
map(elem, cookie);
}
}