Jeff King | 902bb36 | 2011-05-19 17:34:33 -0400 | [diff] [blame] | 1 | #include "cache.h" |
Jeff King | fe299ec | 2020-03-30 10:03:46 -0400 | [diff] [blame] | 2 | #include "oid-array.h" |
Martin Ågren | bc62692 | 2020-12-31 12:56:23 +0100 | [diff] [blame] | 3 | #include "hash-lookup.h" |
Jeff King | 902bb36 | 2011-05-19 17:34:33 -0400 | [diff] [blame] | 4 | |
brian m. carlson | 910650d | 2017-03-31 01:40:00 +0000 | [diff] [blame] | 5 | void oid_array_append(struct oid_array *array, const struct object_id *oid) |
Jeff King | 902bb36 | 2011-05-19 17:34:33 -0400 | [diff] [blame] | 6 | { |
brian m. carlson | ee3051b | 2017-03-26 16:01:37 +0000 | [diff] [blame] | 7 | ALLOC_GROW(array->oid, array->nr + 1, array->alloc); |
brian m. carlson | 98a72dd | 2017-03-31 01:39:56 +0000 | [diff] [blame] | 8 | oidcpy(&array->oid[array->nr++], oid); |
Jeff King | 902bb36 | 2011-05-19 17:34:33 -0400 | [diff] [blame] | 9 | array->sorted = 0; |
| 10 | } |
| 11 | |
| 12 | static int void_hashcmp(const void *a, const void *b) |
| 13 | { |
brian m. carlson | ee3051b | 2017-03-26 16:01:37 +0000 | [diff] [blame] | 14 | return oidcmp(a, b); |
Jeff King | 902bb36 | 2011-05-19 17:34:33 -0400 | [diff] [blame] | 15 | } |
| 16 | |
Jeff King | d0482b4 | 2020-12-04 13:52:07 -0500 | [diff] [blame] | 17 | void oid_array_sort(struct oid_array *array) |
Jeff King | 902bb36 | 2011-05-19 17:34:33 -0400 | [diff] [blame] | 18 | { |
Jeff King | d0482b4 | 2020-12-04 13:52:07 -0500 | [diff] [blame] | 19 | if (array->sorted) |
| 20 | return; |
brian m. carlson | ee3051b | 2017-03-26 16:01:37 +0000 | [diff] [blame] | 21 | QSORT(array->oid, array->nr, void_hashcmp); |
Jeff King | 902bb36 | 2011-05-19 17:34:33 -0400 | [diff] [blame] | 22 | array->sorted = 1; |
| 23 | } |
| 24 | |
Jeff King | 8380dcd | 2021-01-28 01:20:23 -0500 | [diff] [blame] | 25 | static const struct object_id *oid_access(size_t index, const void *table) |
Jeff King | 902bb36 | 2011-05-19 17:34:33 -0400 | [diff] [blame] | 26 | { |
Jeff King | 8380dcd | 2021-01-28 01:20:23 -0500 | [diff] [blame] | 27 | const struct object_id *array = table; |
Jeff King | 45ee13b | 2021-01-28 01:19:42 -0500 | [diff] [blame] | 28 | return &array[index]; |
Jeff King | 902bb36 | 2011-05-19 17:34:33 -0400 | [diff] [blame] | 29 | } |
| 30 | |
brian m. carlson | 910650d | 2017-03-31 01:40:00 +0000 | [diff] [blame] | 31 | int oid_array_lookup(struct oid_array *array, const struct object_id *oid) |
Jeff King | 902bb36 | 2011-05-19 17:34:33 -0400 | [diff] [blame] | 32 | { |
Jeff King | d0482b4 | 2020-12-04 13:52:07 -0500 | [diff] [blame] | 33 | oid_array_sort(array); |
Jeff King | 45ee13b | 2021-01-28 01:19:42 -0500 | [diff] [blame] | 34 | return oid_pos(oid, array->oid, array->nr, oid_access); |
Jeff King | 902bb36 | 2011-05-19 17:34:33 -0400 | [diff] [blame] | 35 | } |
| 36 | |
brian m. carlson | 910650d | 2017-03-31 01:40:00 +0000 | [diff] [blame] | 37 | void oid_array_clear(struct oid_array *array) |
Jeff King | 902bb36 | 2011-05-19 17:34:33 -0400 | [diff] [blame] | 38 | { |
Ævar Arnfjörð Bjarmason | 6a83d90 | 2017-06-15 23:15:46 +0000 | [diff] [blame] | 39 | FREE_AND_NULL(array->oid); |
Jeff King | 902bb36 | 2011-05-19 17:34:33 -0400 | [diff] [blame] | 40 | array->nr = 0; |
| 41 | array->alloc = 0; |
| 42 | array->sorted = 0; |
| 43 | } |
Jeff King | cff38a5 | 2011-05-19 17:34:46 -0400 | [diff] [blame] | 44 | |
Ævar Arnfjörð Bjarmason | 5cc044e | 2018-05-10 12:43:02 +0000 | [diff] [blame] | 45 | |
| 46 | int oid_array_for_each(struct oid_array *array, |
| 47 | for_each_oid_fn fn, |
| 48 | void *data) |
| 49 | { |
Jeff King | eccce52 | 2020-03-30 10:03:20 -0400 | [diff] [blame] | 50 | size_t i; |
Ævar Arnfjörð Bjarmason | 5cc044e | 2018-05-10 12:43:02 +0000 | [diff] [blame] | 51 | |
Jeff King | fe299ec | 2020-03-30 10:03:46 -0400 | [diff] [blame] | 52 | /* No oid_array_sort() here! See oid-array.h */ |
Ævar Arnfjörð Bjarmason | 5cc044e | 2018-05-10 12:43:02 +0000 | [diff] [blame] | 53 | |
| 54 | for (i = 0; i < array->nr; i++) { |
| 55 | int ret = fn(array->oid + i, data); |
| 56 | if (ret) |
| 57 | return ret; |
| 58 | } |
| 59 | return 0; |
| 60 | } |
| 61 | |
brian m. carlson | 910650d | 2017-03-31 01:40:00 +0000 | [diff] [blame] | 62 | int oid_array_for_each_unique(struct oid_array *array, |
Ævar Arnfjörð Bjarmason | 7248672 | 2018-05-10 12:42:59 +0000 | [diff] [blame] | 63 | for_each_oid_fn fn, |
| 64 | void *data) |
Jeff King | cff38a5 | 2011-05-19 17:34:46 -0400 | [diff] [blame] | 65 | { |
Jeff King | eccce52 | 2020-03-30 10:03:20 -0400 | [diff] [blame] | 66 | size_t i; |
Jeff King | cff38a5 | 2011-05-19 17:34:46 -0400 | [diff] [blame] | 67 | |
Jeff King | d0482b4 | 2020-12-04 13:52:07 -0500 | [diff] [blame] | 68 | oid_array_sort(array); |
Jeff King | cff38a5 | 2011-05-19 17:34:46 -0400 | [diff] [blame] | 69 | |
Jeff King | 12c4b4c | 2020-12-07 14:11:00 -0500 | [diff] [blame] | 70 | for (i = 0; i < array->nr; i = oid_array_next_unique(array, i)) { |
| 71 | int ret = fn(array->oid + i, data); |
Jeff King | 16ddcd4 | 2016-09-26 08:00:29 -0400 | [diff] [blame] | 72 | if (ret) |
| 73 | return ret; |
Jeff King | cff38a5 | 2011-05-19 17:34:46 -0400 | [diff] [blame] | 74 | } |
Jeff King | 16ddcd4 | 2016-09-26 08:00:29 -0400 | [diff] [blame] | 75 | return 0; |
Jeff King | cff38a5 | 2011-05-19 17:34:46 -0400 | [diff] [blame] | 76 | } |
Stefan Beller | 161b1cf | 2018-11-28 16:27:48 -0800 | [diff] [blame] | 77 | |
| 78 | void oid_array_filter(struct oid_array *array, |
| 79 | for_each_oid_fn want, |
| 80 | void *cb_data) |
| 81 | { |
Jeff King | eccce52 | 2020-03-30 10:03:20 -0400 | [diff] [blame] | 82 | size_t nr = array->nr, src, dst; |
Stefan Beller | 161b1cf | 2018-11-28 16:27:48 -0800 | [diff] [blame] | 83 | struct object_id *oids = array->oid; |
| 84 | |
| 85 | for (src = dst = 0; src < nr; src++) { |
| 86 | if (want(&oids[src], cb_data)) { |
| 87 | if (src != dst) |
| 88 | oidcpy(&oids[dst], &oids[src]); |
| 89 | dst++; |
| 90 | } |
| 91 | } |
| 92 | array->nr = dst; |
| 93 | } |