vector.h 3.73 KB
Newer Older
Vicent Marti committed
1
/*
Edward Thomson committed
2
 * Copyright (C) the libgit2 contributors. All rights reserved.
Vicent Marti committed
3 4 5 6
 *
 * This file is part of libgit2, distributed under the GNU GPL v2 with
 * a Linking Exception. For full terms see the included COPYING file.
 */
7 8 9
#ifndef INCLUDE_vector_h__
#define INCLUDE_vector_h__

Russell Belfer committed
10
#include "common.h"
11 12 13

typedef int (*git_vector_cmp)(const void *, const void *);

14 15 16 17 18
enum {
	GIT_VECTOR_SORTED = (1u << 0),
	GIT_VECTOR_FLAG_MAX = (1u << 1),
};

19
typedef struct git_vector {
20
	size_t _alloc_size;
21 22
	git_vector_cmp _cmp;
	void **contents;
23
	size_t length;
24
	uint32_t flags;
25 26
} git_vector;

27 28
#define GIT_VECTOR_INIT {0}

29
int git_vector_init(git_vector *v, size_t initial_size, git_vector_cmp cmp);
30
void git_vector_free(git_vector *v);
31
void git_vector_free_deep(git_vector *v); /* free each entry and self */
32
void git_vector_clear(git_vector *v);
33
int git_vector_dup(git_vector *v, const git_vector *src, git_vector_cmp cmp);
34
void git_vector_swap(git_vector *a, git_vector *b);
35
int git_vector_size_hint(git_vector *v, size_t size_hint);
36

37 38
void **git_vector_detach(size_t *size, size_t *asize, git_vector *v);

39 40
void git_vector_sort(git_vector *v);

41
/** Linear search for matching entry using internal comparison function */
42
int git_vector_search(size_t *at_pos, const git_vector *v, const void *entry);
43 44

/** Linear search for matching entry using explicit comparison function */
45
int git_vector_search2(size_t *at_pos, const git_vector *v, git_vector_cmp cmp, const void *key);
46

47 48 49 50
/**
 * Binary search for matching entry using explicit comparison function that
 * returns position where item would go if not found.
 */
51
int git_vector_bsearch2(
52
	size_t *at_pos, git_vector *v, git_vector_cmp cmp, const void *key);
53

54
/** Binary search for matching entry using internal comparison function */
55
GIT_INLINE(int) git_vector_bsearch(size_t *at_pos, git_vector *v, const void *key)
56
{
57
	return git_vector_bsearch2(at_pos, v, v->_cmp, key);
58
}
59

Ben Straub committed
60
GIT_INLINE(void *) git_vector_get(const git_vector *v, size_t position)
61 62 63
{
	return (position < v->length) ? v->contents[position] : NULL;
}
64

65 66
#define GIT_VECTOR_GET(V,I) ((I) < (V)->length ? (V)->contents[(I)] : NULL)

67 68 69 70 71
GIT_INLINE(size_t) git_vector_length(const git_vector *v)
{
	return v->length;
}

72
GIT_INLINE(void *) git_vector_last(const git_vector *v)
73 74 75 76
{
	return (v->length > 0) ? git_vector_get(v, v->length - 1) : NULL;
}

77 78 79
#define git_vector_foreach(v, iter, elem)	\
	for ((iter) = 0; (iter) < (v)->length && ((elem) = (v)->contents[(iter)], 1); (iter)++ )

80
#define git_vector_rforeach(v, iter, elem)	\
81
	for ((iter) = (v)->length - 1; (iter) < SIZE_MAX && ((elem) = (v)->contents[(iter)], 1); (iter)-- )
82

83
int git_vector_insert(git_vector *v, void *element);
84 85
int git_vector_insert_sorted(git_vector *v, void *element,
	int (*on_dup)(void **old, void *new));
86
int git_vector_remove(git_vector *v, size_t idx);
87
void git_vector_pop(git_vector *v);
88
void git_vector_uniq(git_vector *v, void  (*git_free_cb)(void *));
89

90
void git_vector_remove_matching(
91 92 93
	git_vector *v,
	int (*match)(const git_vector *v, size_t idx, void *payload),
	void *payload);
94

95 96 97
int git_vector_resize_to(git_vector *v, size_t new_length);
int git_vector_set(void **old, git_vector *v, size_t position, void *value);

98 99 100 101 102 103 104 105
/** Check if vector is sorted */
#define git_vector_is_sorted(V) (((V)->flags & GIT_VECTOR_SORTED) != 0)

/** Directly set sorted state of vector */
#define git_vector_set_sorted(V,S) do { \
	(V)->flags = (S) ? ((V)->flags | GIT_VECTOR_SORTED) : \
		((V)->flags & ~GIT_VECTOR_SORTED); } while (0)

106 107 108 109 110
/** Set the comparison function used for sorting the vector */
GIT_INLINE(void) git_vector_set_cmp(git_vector *v, git_vector_cmp cmp)
{
	if (cmp != v->_cmp) {
		v->_cmp = cmp;
111
		git_vector_set_sorted(v, 0);
112 113 114
	}
}

115 116 117
/* Just use this in tests, not for realz. returns -1 if not sorted */
int git_vector_verify_sorted(const git_vector *v);

118
#endif