vector.h 4.01 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 30
GIT_WARN_UNUSED_RESULT int git_vector_init(
	git_vector *v, size_t initial_size, git_vector_cmp cmp);
31
void git_vector_free(git_vector *v);
32
void git_vector_free_deep(git_vector *v); /* free each entry and self */
33
void git_vector_clear(git_vector *v);
34 35
GIT_WARN_UNUSED_RESULT int git_vector_dup(
	git_vector *v, const git_vector *src, git_vector_cmp cmp);
36
void git_vector_swap(git_vector *a, git_vector *b);
37
int git_vector_size_hint(git_vector *v, size_t size_hint);
38

39 40
void **git_vector_detach(size_t *size, size_t *asize, git_vector *v);

41 42
void git_vector_sort(git_vector *v);

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

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

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

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

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

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

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

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

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

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

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

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

97
int git_vector_resize_to(git_vector *v, size_t new_length);
98 99
int git_vector_insert_null(git_vector *v, size_t idx, size_t insert_len);
int git_vector_remove_range(git_vector *v, size_t idx, size_t remove_len);
100

101 102
int git_vector_set(void **old, git_vector *v, size_t position, void *value);

103 104 105 106 107 108 109 110
/** 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)

111 112 113 114 115
/** 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;
116
		git_vector_set_sorted(v, 0);
117 118 119
	}
}

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

123 124 125 126 127
/**
 * Reverse the vector in-place.
 */
void git_vector_reverse(git_vector *v);

128
#endif