blob: dc1bca9c9aea6cdbf001b8d43d2e43b99c034d81 [file] [log] [blame]
Jeff King902bb362011-05-19 17:34:33 -04001#ifndef SHA1_ARRAY_H
2#define SHA1_ARRAY_H
3
Heba Waly405c6b12019-11-17 21:04:44 +00004/**
5 * The API provides storage and manipulation of sets of object identifiers.
6 * The emphasis is on storage and processing efficiency, making them suitable
7 * for large lists. Note that the ordering of items is not preserved over some
8 * operations.
9 *
10 * Examples
11 * --------
12 * -----------------------------------------
13 * int print_callback(const struct object_id *oid,
14 * void *data)
15 * {
16 * printf("%s\n", oid_to_hex(oid));
17 * return 0; // always continue
18 * }
19 *
20 * void some_func(void)
21 * {
22 * struct sha1_array hashes = OID_ARRAY_INIT;
23 * struct object_id oid;
24 *
25 * // Read objects into our set
26 * while (read_object_from_stdin(oid.hash))
27 * oid_array_append(&hashes, &oid);
28 *
29 * // Check if some objects are in our set
30 * while (read_object_from_stdin(oid.hash)) {
31 * if (oid_array_lookup(&hashes, &oid) >= 0)
32 * printf("it's in there!\n");
33 *
34 * // Print the unique set of objects. We could also have
35 * // avoided adding duplicate objects in the first place,
36 * // but we would end up re-sorting the array repeatedly.
37 * // Instead, this will sort once and then skip duplicates
38 * // in linear time.
39 *
40 * oid_array_for_each_unique(&hashes, print_callback, NULL);
41 * }
42 */
43
44/**
45 * A single array of object IDs. This should be initialized by assignment from
46 * `OID_ARRAY_INIT`. The `oid` member contains the actual data. The `nr` member
47 * contains the number of items in the set. The `alloc` and `sorted` members
48 * are used internally, and should not be needed by API callers.
49 */
brian m. carlson910650d2017-03-31 01:40:00 +000050struct oid_array {
brian m. carlsonee3051b2017-03-26 16:01:37 +000051 struct object_id *oid;
Jeff King902bb362011-05-19 17:34:33 -040052 int nr;
53 int alloc;
54 int sorted;
55};
56
brian m. carlson910650d2017-03-31 01:40:00 +000057#define OID_ARRAY_INIT { NULL, 0, 0, 0 }
Jeff King902bb362011-05-19 17:34:33 -040058
Heba Waly405c6b12019-11-17 21:04:44 +000059/**
60 * Add an item to the set. The object ID will be placed at the end of the array
61 * (but note that some operations below may lose this ordering).
62 */
brian m. carlson910650d2017-03-31 01:40:00 +000063void oid_array_append(struct oid_array *array, const struct object_id *oid);
Heba Waly405c6b12019-11-17 21:04:44 +000064
65/**
66 * Perform a binary search of the array for a specific object ID. If found,
67 * returns the offset (in number of elements) of the object ID. If not found,
68 * returns a negative integer. If the array is not sorted, this function has
69 * the side effect of sorting it.
70 */
brian m. carlson910650d2017-03-31 01:40:00 +000071int oid_array_lookup(struct oid_array *array, const struct object_id *oid);
Heba Waly405c6b12019-11-17 21:04:44 +000072
73/**
74 * Free all memory associated with the array and return it to the initial,
75 * empty state.
76 */
brian m. carlson910650d2017-03-31 01:40:00 +000077void oid_array_clear(struct oid_array *array);
Jeff King902bb362011-05-19 17:34:33 -040078
brian m. carlson1b7ba792017-03-31 01:39:59 +000079typedef int (*for_each_oid_fn)(const struct object_id *oid,
80 void *data);
Heba Waly405c6b12019-11-17 21:04:44 +000081/**
82 * Iterate over each element of the list, executing the callback function for
83 * each one. Does not sort the list, so any custom hash order is retained.
84 * If the callback returns a non-zero value, the iteration ends immediately
85 * and the callback's return is propagated; otherwise, 0 is returned.
86 */
Ævar Arnfjörð Bjarmason5cc044e2018-05-10 12:43:02 +000087int oid_array_for_each(struct oid_array *array,
88 for_each_oid_fn fn,
89 void *data);
Heba Waly405c6b12019-11-17 21:04:44 +000090
91/**
92 * Iterate over each unique element of the list in sorted order, but otherwise
93 * behave like `oid_array_for_each`. If the array is not sorted, this function
94 * has the side effect of sorting it.
95 */
brian m. carlson910650d2017-03-31 01:40:00 +000096int oid_array_for_each_unique(struct oid_array *array,
Ævar Arnfjörð Bjarmason72486722018-05-10 12:42:59 +000097 for_each_oid_fn fn,
98 void *data);
Heba Waly405c6b12019-11-17 21:04:44 +000099
100/**
101 * Apply the callback function `want` to each entry in the array, retaining
102 * only the entries for which the function returns true. Preserve the order
103 * of the entries that are retained.
104 */
Stefan Beller161b1cf2018-11-28 16:27:48 -0800105void oid_array_filter(struct oid_array *array,
106 for_each_oid_fn want,
107 void *cbdata);
Jeff Kingcff38a52011-05-19 17:34:46 -0400108
Jeff King902bb362011-05-19 17:34:33 -0400109#endif /* SHA1_ARRAY_H */