src/zbset.c


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46

/* See LICENSE file for copyright and license details. */
#include "internals.h"


#define PROLOGUE(MAY_INCREASE)\
	zahl_char_t mask = 1;\
	size_t chars = FLOOR_BITS_TO_CHARS(bit);\
	if (MAY_INCREASE) {\
		if (zzero(a)) {\
			a->used = 0;\
			SET_SIGNUM(a, 1);\
		}\
		if (unlikely(chars >= a->used)) {\
			ENSURE_SIZE(a, chars + 1);\
			zmemset(a->chars + a->used, 0, chars + 1 - a->used);\
			        a->used = chars + 1;\
		}\
	} else if (unlikely(chars >= a->used)) {\
		return;\
	}\
	bit = BITS_IN_LAST_CHAR(bit);\
	mask <<= bit


void
zbset_impl_set(z_t a, size_t bit)
{
	PROLOGUE(1);
	a->chars[chars] |= mask;
}

void
zbset_impl_clear(z_t a, size_t bit)
{
	PROLOGUE(0);
	a->chars[chars] &= ~mask;
	TRIM_AND_ZERO(a);
}

void
zbset_impl_flip(z_t a, size_t bit)
{
	PROLOGUE(1);
	a->chars[chars] ^= mask;
	TRIM_AND_ZERO(a);
}