mirror of
https://github.com/git/git
synced 2024-08-27 11:39:22 +00:00
![Elijah Newren](/assets/img/avatar_default.png)
This allows us to replace includes of cache.h with includes of the much smaller alloc.h in many places. It does mean that we also need to add includes of alloc.h in a number of C files. Signed-off-by: Elijah Newren <newren@gmail.com> Signed-off-by: Junio C Hamano <gitster@pobox.com>
478 lines
11 KiB
C
478 lines
11 KiB
C
/**
|
|
* Copyright 2013, GitHub, Inc
|
|
* Copyright 2009-2013, Daniel Lemire, Cliff Moon,
|
|
* David McIntosh, Robert Becho, Google Inc. and Veronika Zenz
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version 2
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
#include "git-compat-util.h"
|
|
#include "alloc.h"
|
|
#include "ewok.h"
|
|
#include "ewok_rlw.h"
|
|
|
|
static inline size_t min_size(size_t a, size_t b)
|
|
{
|
|
return a < b ? a : b;
|
|
}
|
|
|
|
static inline size_t max_size(size_t a, size_t b)
|
|
{
|
|
return a > b ? a : b;
|
|
}
|
|
|
|
static inline void buffer_grow(struct ewah_bitmap *self, size_t new_size)
|
|
{
|
|
size_t rlw_offset = (uint8_t *)self->rlw - (uint8_t *)self->buffer;
|
|
ALLOC_GROW(self->buffer, new_size, self->alloc_size);
|
|
self->rlw = self->buffer + (rlw_offset / sizeof(eword_t));
|
|
}
|
|
|
|
static inline void buffer_push(struct ewah_bitmap *self, eword_t value)
|
|
{
|
|
buffer_grow(self, self->buffer_size + 1);
|
|
self->buffer[self->buffer_size++] = value;
|
|
}
|
|
|
|
static void buffer_push_rlw(struct ewah_bitmap *self, eword_t value)
|
|
{
|
|
buffer_push(self, value);
|
|
self->rlw = self->buffer + self->buffer_size - 1;
|
|
}
|
|
|
|
static size_t add_empty_words(struct ewah_bitmap *self, int v, size_t number)
|
|
{
|
|
size_t added = 0;
|
|
eword_t runlen, can_add;
|
|
|
|
if (rlw_get_run_bit(self->rlw) != v && rlw_size(self->rlw) == 0) {
|
|
rlw_set_run_bit(self->rlw, v);
|
|
} else if (rlw_get_literal_words(self->rlw) != 0 ||
|
|
rlw_get_run_bit(self->rlw) != v) {
|
|
buffer_push_rlw(self, 0);
|
|
if (v) rlw_set_run_bit(self->rlw, v);
|
|
added++;
|
|
}
|
|
|
|
runlen = rlw_get_running_len(self->rlw);
|
|
can_add = min_size(number, RLW_LARGEST_RUNNING_COUNT - runlen);
|
|
|
|
rlw_set_running_len(self->rlw, runlen + can_add);
|
|
number -= can_add;
|
|
|
|
while (number >= RLW_LARGEST_RUNNING_COUNT) {
|
|
buffer_push_rlw(self, 0);
|
|
added++;
|
|
if (v) rlw_set_run_bit(self->rlw, v);
|
|
rlw_set_running_len(self->rlw, RLW_LARGEST_RUNNING_COUNT);
|
|
number -= RLW_LARGEST_RUNNING_COUNT;
|
|
}
|
|
|
|
if (number > 0) {
|
|
buffer_push_rlw(self, 0);
|
|
added++;
|
|
|
|
if (v) rlw_set_run_bit(self->rlw, v);
|
|
rlw_set_running_len(self->rlw, number);
|
|
}
|
|
|
|
return added;
|
|
}
|
|
|
|
size_t ewah_add_empty_words(struct ewah_bitmap *self, int v, size_t number)
|
|
{
|
|
if (number == 0)
|
|
return 0;
|
|
|
|
self->bit_size += number * BITS_IN_EWORD;
|
|
return add_empty_words(self, v, number);
|
|
}
|
|
|
|
static size_t add_literal(struct ewah_bitmap *self, eword_t new_data)
|
|
{
|
|
eword_t current_num = rlw_get_literal_words(self->rlw);
|
|
|
|
if (current_num >= RLW_LARGEST_LITERAL_COUNT) {
|
|
buffer_push_rlw(self, 0);
|
|
|
|
rlw_set_literal_words(self->rlw, 1);
|
|
buffer_push(self, new_data);
|
|
return 2;
|
|
}
|
|
|
|
rlw_set_literal_words(self->rlw, current_num + 1);
|
|
|
|
/* sanity check */
|
|
assert(rlw_get_literal_words(self->rlw) == current_num + 1);
|
|
|
|
buffer_push(self, new_data);
|
|
return 1;
|
|
}
|
|
|
|
void ewah_add_dirty_words(
|
|
struct ewah_bitmap *self, const eword_t *buffer,
|
|
size_t number, int negate)
|
|
{
|
|
size_t literals, can_add;
|
|
|
|
while (1) {
|
|
literals = rlw_get_literal_words(self->rlw);
|
|
can_add = min_size(number, RLW_LARGEST_LITERAL_COUNT - literals);
|
|
|
|
rlw_set_literal_words(self->rlw, literals + can_add);
|
|
|
|
buffer_grow(self, self->buffer_size + can_add);
|
|
|
|
if (negate) {
|
|
size_t i;
|
|
for (i = 0; i < can_add; ++i)
|
|
self->buffer[self->buffer_size++] = ~buffer[i];
|
|
} else {
|
|
memcpy(self->buffer + self->buffer_size,
|
|
buffer, can_add * sizeof(eword_t));
|
|
self->buffer_size += can_add;
|
|
}
|
|
|
|
self->bit_size += can_add * BITS_IN_EWORD;
|
|
|
|
if (number - can_add == 0)
|
|
break;
|
|
|
|
buffer_push_rlw(self, 0);
|
|
buffer += can_add;
|
|
number -= can_add;
|
|
}
|
|
}
|
|
|
|
static size_t add_empty_word(struct ewah_bitmap *self, int v)
|
|
{
|
|
int no_literal = (rlw_get_literal_words(self->rlw) == 0);
|
|
eword_t run_len = rlw_get_running_len(self->rlw);
|
|
|
|
if (no_literal && run_len == 0) {
|
|
rlw_set_run_bit(self->rlw, v);
|
|
assert(rlw_get_run_bit(self->rlw) == v);
|
|
}
|
|
|
|
if (no_literal && rlw_get_run_bit(self->rlw) == v &&
|
|
run_len < RLW_LARGEST_RUNNING_COUNT) {
|
|
rlw_set_running_len(self->rlw, run_len + 1);
|
|
assert(rlw_get_running_len(self->rlw) == run_len + 1);
|
|
return 0;
|
|
} else {
|
|
buffer_push_rlw(self, 0);
|
|
|
|
assert(rlw_get_running_len(self->rlw) == 0);
|
|
assert(rlw_get_run_bit(self->rlw) == 0);
|
|
assert(rlw_get_literal_words(self->rlw) == 0);
|
|
|
|
rlw_set_run_bit(self->rlw, v);
|
|
assert(rlw_get_run_bit(self->rlw) == v);
|
|
|
|
rlw_set_running_len(self->rlw, 1);
|
|
assert(rlw_get_running_len(self->rlw) == 1);
|
|
assert(rlw_get_literal_words(self->rlw) == 0);
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
size_t ewah_add(struct ewah_bitmap *self, eword_t word)
|
|
{
|
|
self->bit_size += BITS_IN_EWORD;
|
|
|
|
if (word == 0)
|
|
return add_empty_word(self, 0);
|
|
|
|
if (word == (eword_t)(~0))
|
|
return add_empty_word(self, 1);
|
|
|
|
return add_literal(self, word);
|
|
}
|
|
|
|
void ewah_set(struct ewah_bitmap *self, size_t i)
|
|
{
|
|
const size_t dist =
|
|
DIV_ROUND_UP(i + 1, BITS_IN_EWORD) -
|
|
DIV_ROUND_UP(self->bit_size, BITS_IN_EWORD);
|
|
|
|
assert(i >= self->bit_size);
|
|
|
|
self->bit_size = i + 1;
|
|
|
|
if (dist > 0) {
|
|
if (dist > 1)
|
|
add_empty_words(self, 0, dist - 1);
|
|
|
|
add_literal(self, (eword_t)1 << (i % BITS_IN_EWORD));
|
|
return;
|
|
}
|
|
|
|
if (rlw_get_literal_words(self->rlw) == 0) {
|
|
rlw_set_running_len(self->rlw,
|
|
rlw_get_running_len(self->rlw) - 1);
|
|
add_literal(self, (eword_t)1 << (i % BITS_IN_EWORD));
|
|
return;
|
|
}
|
|
|
|
self->buffer[self->buffer_size - 1] |=
|
|
((eword_t)1 << (i % BITS_IN_EWORD));
|
|
|
|
/* check if we just completed a stream of 1s */
|
|
if (self->buffer[self->buffer_size - 1] == (eword_t)(~0)) {
|
|
self->buffer[--self->buffer_size] = 0;
|
|
rlw_set_literal_words(self->rlw,
|
|
rlw_get_literal_words(self->rlw) - 1);
|
|
add_empty_word(self, 1);
|
|
}
|
|
}
|
|
|
|
void ewah_each_bit(struct ewah_bitmap *self, void (*callback)(size_t, void*), void *payload)
|
|
{
|
|
size_t pos = 0;
|
|
size_t pointer = 0;
|
|
size_t k;
|
|
|
|
while (pointer < self->buffer_size) {
|
|
eword_t *word = &self->buffer[pointer];
|
|
|
|
if (rlw_get_run_bit(word)) {
|
|
size_t len = rlw_get_running_len(word) * BITS_IN_EWORD;
|
|
for (k = 0; k < len; ++k, ++pos)
|
|
callback(pos, payload);
|
|
} else {
|
|
pos += rlw_get_running_len(word) * BITS_IN_EWORD;
|
|
}
|
|
|
|
++pointer;
|
|
|
|
for (k = 0; k < rlw_get_literal_words(word); ++k) {
|
|
int c;
|
|
|
|
/* todo: zero count optimization */
|
|
for (c = 0; c < BITS_IN_EWORD; ++c, ++pos) {
|
|
if ((self->buffer[pointer] & ((eword_t)1 << c)) != 0)
|
|
callback(pos, payload);
|
|
}
|
|
|
|
++pointer;
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Clear all the bits in the bitmap. Does not free or resize
|
|
* memory.
|
|
*/
|
|
static void ewah_clear(struct ewah_bitmap *self)
|
|
{
|
|
self->buffer_size = 1;
|
|
self->buffer[0] = 0;
|
|
self->bit_size = 0;
|
|
self->rlw = self->buffer;
|
|
}
|
|
|
|
struct ewah_bitmap *ewah_new(void)
|
|
{
|
|
struct ewah_bitmap *self;
|
|
|
|
self = xmalloc(sizeof(struct ewah_bitmap));
|
|
self->alloc_size = 32;
|
|
ALLOC_ARRAY(self->buffer, self->alloc_size);
|
|
|
|
ewah_clear(self);
|
|
return self;
|
|
}
|
|
|
|
void ewah_free(struct ewah_bitmap *self)
|
|
{
|
|
if (!self)
|
|
return;
|
|
|
|
if (self->alloc_size)
|
|
free(self->buffer);
|
|
|
|
free(self);
|
|
}
|
|
|
|
static void read_new_rlw(struct ewah_iterator *it)
|
|
{
|
|
const eword_t *word = NULL;
|
|
|
|
it->literals = 0;
|
|
it->compressed = 0;
|
|
|
|
while (1) {
|
|
word = &it->buffer[it->pointer];
|
|
|
|
it->rl = rlw_get_running_len(word);
|
|
it->lw = rlw_get_literal_words(word);
|
|
it->b = rlw_get_run_bit(word);
|
|
|
|
if (it->rl || it->lw)
|
|
return;
|
|
|
|
if (it->pointer < it->buffer_size - 1) {
|
|
it->pointer++;
|
|
} else {
|
|
it->pointer = it->buffer_size;
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
int ewah_iterator_next(eword_t *next, struct ewah_iterator *it)
|
|
{
|
|
if (it->pointer >= it->buffer_size)
|
|
return 0;
|
|
|
|
if (it->compressed < it->rl) {
|
|
it->compressed++;
|
|
*next = it->b ? (eword_t)(~0) : 0;
|
|
} else {
|
|
assert(it->literals < it->lw);
|
|
|
|
it->literals++;
|
|
it->pointer++;
|
|
|
|
assert(it->pointer < it->buffer_size);
|
|
|
|
*next = it->buffer[it->pointer];
|
|
}
|
|
|
|
if (it->compressed == it->rl && it->literals == it->lw) {
|
|
if (++it->pointer < it->buffer_size)
|
|
read_new_rlw(it);
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
void ewah_iterator_init(struct ewah_iterator *it, struct ewah_bitmap *parent)
|
|
{
|
|
it->buffer = parent->buffer;
|
|
it->buffer_size = parent->buffer_size;
|
|
it->pointer = 0;
|
|
|
|
it->lw = 0;
|
|
it->rl = 0;
|
|
it->compressed = 0;
|
|
it->literals = 0;
|
|
it->b = 0;
|
|
|
|
if (it->pointer < it->buffer_size)
|
|
read_new_rlw(it);
|
|
}
|
|
|
|
void ewah_xor(
|
|
struct ewah_bitmap *ewah_i,
|
|
struct ewah_bitmap *ewah_j,
|
|
struct ewah_bitmap *out)
|
|
{
|
|
struct rlw_iterator rlw_i;
|
|
struct rlw_iterator rlw_j;
|
|
size_t literals;
|
|
|
|
rlwit_init(&rlw_i, ewah_i);
|
|
rlwit_init(&rlw_j, ewah_j);
|
|
|
|
while (rlwit_word_size(&rlw_i) > 0 && rlwit_word_size(&rlw_j) > 0) {
|
|
while (rlw_i.rlw.running_len > 0 || rlw_j.rlw.running_len > 0) {
|
|
struct rlw_iterator *prey, *predator;
|
|
size_t index;
|
|
int negate_words;
|
|
|
|
if (rlw_i.rlw.running_len < rlw_j.rlw.running_len) {
|
|
prey = &rlw_i;
|
|
predator = &rlw_j;
|
|
} else {
|
|
prey = &rlw_j;
|
|
predator = &rlw_i;
|
|
}
|
|
|
|
negate_words = !!predator->rlw.running_bit;
|
|
index = rlwit_discharge(prey, out,
|
|
predator->rlw.running_len, negate_words);
|
|
|
|
ewah_add_empty_words(out, negate_words,
|
|
predator->rlw.running_len - index);
|
|
|
|
rlwit_discard_first_words(predator,
|
|
predator->rlw.running_len);
|
|
}
|
|
|
|
literals = min_size(
|
|
rlw_i.rlw.literal_words,
|
|
rlw_j.rlw.literal_words);
|
|
|
|
if (literals) {
|
|
size_t k;
|
|
|
|
for (k = 0; k < literals; ++k) {
|
|
ewah_add(out,
|
|
rlw_i.buffer[rlw_i.literal_word_start + k] ^
|
|
rlw_j.buffer[rlw_j.literal_word_start + k]
|
|
);
|
|
}
|
|
|
|
rlwit_discard_first_words(&rlw_i, literals);
|
|
rlwit_discard_first_words(&rlw_j, literals);
|
|
}
|
|
}
|
|
|
|
if (rlwit_word_size(&rlw_i) > 0)
|
|
rlwit_discharge(&rlw_i, out, ~0, 0);
|
|
else
|
|
rlwit_discharge(&rlw_j, out, ~0, 0);
|
|
|
|
out->bit_size = max_size(ewah_i->bit_size, ewah_j->bit_size);
|
|
}
|
|
|
|
#define BITMAP_POOL_MAX 16
|
|
static struct ewah_bitmap *bitmap_pool[BITMAP_POOL_MAX];
|
|
static size_t bitmap_pool_size;
|
|
|
|
struct ewah_bitmap *ewah_pool_new(void)
|
|
{
|
|
if (bitmap_pool_size)
|
|
return bitmap_pool[--bitmap_pool_size];
|
|
|
|
return ewah_new();
|
|
}
|
|
|
|
void ewah_pool_free(struct ewah_bitmap *self)
|
|
{
|
|
if (!self)
|
|
return;
|
|
|
|
if (bitmap_pool_size == BITMAP_POOL_MAX ||
|
|
self->alloc_size == 0) {
|
|
ewah_free(self);
|
|
return;
|
|
}
|
|
|
|
ewah_clear(self);
|
|
bitmap_pool[bitmap_pool_size++] = self;
|
|
}
|
|
|
|
uint32_t ewah_checksum(struct ewah_bitmap *self)
|
|
{
|
|
const uint8_t *p = (uint8_t *)self->buffer;
|
|
uint32_t crc = (uint32_t)self->bit_size;
|
|
size_t size = self->buffer_size * sizeof(eword_t);
|
|
|
|
while (size--)
|
|
crc = (crc << 5) - crc + (uint32_t)*p++;
|
|
|
|
return crc;
|
|
}
|