blob: 73ba76e9e9a223306a03c3d04123c0c4af4aeda2 [file] [log] [blame]
Jeff King902bb362011-05-19 17:34:33 -04001#include "cache.h"
Jeff Kingfe299ec2020-03-30 10:03:46 -04002#include "oid-array.h"
Martin Ågrenbc626922020-12-31 12:56:23 +01003#include "hash-lookup.h"
Jeff King902bb362011-05-19 17:34:33 -04004
brian m. carlson910650d2017-03-31 01:40:00 +00005void oid_array_append(struct oid_array *array, const struct object_id *oid)
Jeff King902bb362011-05-19 17:34:33 -04006{
brian m. carlsonee3051b2017-03-26 16:01:37 +00007 ALLOC_GROW(array->oid, array->nr + 1, array->alloc);
brian m. carlson98a72dd2017-03-31 01:39:56 +00008 oidcpy(&array->oid[array->nr++], oid);
Jeff King902bb362011-05-19 17:34:33 -04009 array->sorted = 0;
10}
11
12static int void_hashcmp(const void *a, const void *b)
13{
brian m. carlsonee3051b2017-03-26 16:01:37 +000014 return oidcmp(a, b);
Jeff King902bb362011-05-19 17:34:33 -040015}
16
Jeff Kingd0482b42020-12-04 13:52:07 -050017void oid_array_sort(struct oid_array *array)
Jeff King902bb362011-05-19 17:34:33 -040018{
Jeff Kingd0482b42020-12-04 13:52:07 -050019 if (array->sorted)
20 return;
brian m. carlsonee3051b2017-03-26 16:01:37 +000021 QSORT(array->oid, array->nr, void_hashcmp);
Jeff King902bb362011-05-19 17:34:33 -040022 array->sorted = 1;
23}
24
Jeff King8380dcd2021-01-28 01:20:23 -050025static const struct object_id *oid_access(size_t index, const void *table)
Jeff King902bb362011-05-19 17:34:33 -040026{
Jeff King8380dcd2021-01-28 01:20:23 -050027 const struct object_id *array = table;
Jeff King45ee13b2021-01-28 01:19:42 -050028 return &array[index];
Jeff King902bb362011-05-19 17:34:33 -040029}
30
brian m. carlson910650d2017-03-31 01:40:00 +000031int oid_array_lookup(struct oid_array *array, const struct object_id *oid)
Jeff King902bb362011-05-19 17:34:33 -040032{
Jeff Kingd0482b42020-12-04 13:52:07 -050033 oid_array_sort(array);
Jeff King45ee13b2021-01-28 01:19:42 -050034 return oid_pos(oid, array->oid, array->nr, oid_access);
Jeff King902bb362011-05-19 17:34:33 -040035}
36
brian m. carlson910650d2017-03-31 01:40:00 +000037void oid_array_clear(struct oid_array *array)
Jeff King902bb362011-05-19 17:34:33 -040038{
Ævar Arnfjörð Bjarmason6a83d902017-06-15 23:15:46 +000039 FREE_AND_NULL(array->oid);
Jeff King902bb362011-05-19 17:34:33 -040040 array->nr = 0;
41 array->alloc = 0;
42 array->sorted = 0;
43}
Jeff Kingcff38a52011-05-19 17:34:46 -040044
Ævar Arnfjörð Bjarmason5cc044e2018-05-10 12:43:02 +000045
46int oid_array_for_each(struct oid_array *array,
47 for_each_oid_fn fn,
48 void *data)
49{
Jeff Kingeccce522020-03-30 10:03:20 -040050 size_t i;
Ævar Arnfjörð Bjarmason5cc044e2018-05-10 12:43:02 +000051
Jeff Kingfe299ec2020-03-30 10:03:46 -040052 /* No oid_array_sort() here! See oid-array.h */
Ævar Arnfjörð Bjarmason5cc044e2018-05-10 12:43:02 +000053
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. carlson910650d2017-03-31 01:40:00 +000062int oid_array_for_each_unique(struct oid_array *array,
Ævar Arnfjörð Bjarmason72486722018-05-10 12:42:59 +000063 for_each_oid_fn fn,
64 void *data)
Jeff Kingcff38a52011-05-19 17:34:46 -040065{
Jeff Kingeccce522020-03-30 10:03:20 -040066 size_t i;
Jeff Kingcff38a52011-05-19 17:34:46 -040067
Jeff Kingd0482b42020-12-04 13:52:07 -050068 oid_array_sort(array);
Jeff Kingcff38a52011-05-19 17:34:46 -040069
Jeff King12c4b4c2020-12-07 14:11:00 -050070 for (i = 0; i < array->nr; i = oid_array_next_unique(array, i)) {
71 int ret = fn(array->oid + i, data);
Jeff King16ddcd42016-09-26 08:00:29 -040072 if (ret)
73 return ret;
Jeff Kingcff38a52011-05-19 17:34:46 -040074 }
Jeff King16ddcd42016-09-26 08:00:29 -040075 return 0;
Jeff Kingcff38a52011-05-19 17:34:46 -040076}
Stefan Beller161b1cf2018-11-28 16:27:48 -080077
78void oid_array_filter(struct oid_array *array,
79 for_each_oid_fn want,
80 void *cb_data)
81{
Jeff Kingeccce522020-03-30 10:03:20 -040082 size_t nr = array->nr, src, dst;
Stefan Beller161b1cf2018-11-28 16:27:48 -080083 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}