aboutsummaryrefslogtreecommitdiffstats
path: root/libs/bitfield-c/src/bitfield/bitfield.h
blob: 8dd7499f8bdc12a199dff0dea8856892030bbdc0 (plain)
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
#ifndef __BITFIELD_H__
#define __BITFIELD_H__

#include <stdint.h>
#include <stdbool.h>

#define NIBBLE_SIZE (CHAR_BIT / 2)

#ifdef __cplusplus
extern "C" {
#endif

/* Public: Reads a subset of bits into a uint64_t, right aligned so they may be
 * interpreted as a number.
 *
 * source - the bytes in question.
 * source_size - the number of bytes in the source.
 * offset - the starting index of the bit field (beginning from 0).
 * bit_count - the width of the bit field to extract. This must be less than or
 *      equal to 64.
 *
 * Bit fields are positioned according to big-endian bit layout and the data is
 * swapped automatically as necessary depending on the compiled architecture.
 *
 * For example, the bit layout of the value "42" (i.e. 00101010 set at position
 * 14 with length 6 is:
 *
 *     000000000000001010100000000000000000000000000000000000000000000
 *
 * and the same value and position but with length 8 is:
 *
 *     000000000000000010101000000000000000000000000000000000000000000
 *
 * Examples
 *
 *  uint64_t value = get_bitfield(data, data_size, 2, 4);
 *
 * Returns the value of the requested bit field, right aligned in a uint64_t.
 */
uint64_t get_bitfield(const uint8_t source[], const unsigned int source_length,
                const unsigned int offset, const unsigned int bit_count);

/* Public: Return a single nibble from the byte array, with range checking.
 *
 * source - the source byte array.
 * source_length - the total length of the source array.
 * nibble_index - the index of the nibble to retreive. The leftmost nibble is
 *      index 0.
 *
 * Returns the retreived nibble, right aligned in a unsigned int.
 */
unsigned int get_nibble(const uint8_t source[], const unsigned int source_length,
                const unsigned int nibble_index);

/* Public: Return a single byte from the byte array, with range checking.
 *
 * source - the source byte array.
 * source_length - the total length of the source array.
 * byte_index - the index of the byte to retreive. The leftmost byte is index 0.
 *
 * Returns the retreived byte.
 */
unsigned int get_byte(const uint8_t source[], const unsigned int source_length,
        const unsigned int byte_index);

/* Public: Copy a range of bits from one bit array to another.
 *
 * The range does not need to be byte aligned, and the source and destination do
 * not have to be the same size (as long as the desitnation has enough room to
 * fit the range).
 *
 * A bit array with regards to this function always has the leftmost bit in byte
 * 0, i.e. bit index is the leftmost bit of byte 0. Endianness does not matter.
 *
 * For example:
 *
 *      unsigned int source[4] = {0x11, 0x22, 0x33, 0x44};
 *      unsigned int destination[4] = {0};
 *      copy_bits(source, sizeof(source), 8, 8, destination,
 *              sizeof(destination), 0);
 *      // destination[0] == 0x22
 *      // destination[1] == 0x0
 *      // destination[2] == 0x0
 *      // destination[3] == 0x0
 *
 * Thanks to
 * http://stackoverflow.com/questions/3534535/whats-a-time-efficient-algorithm-to-copy-unaligned-bit-arrays
 * for the implementation of the algorithm.
 *
 * source_origin - the source array.
 * source_length - the total length of the source array in bytes,
 *      for range checking.
 * source_offset - an offset in bits to start the copy from the source array.
 *      Specify 0 to start from source_origin.
 * bit_count - the number of bits to copy.
 * destination_origin - the destination array.
 * desitnation_length - the total length of the destination array in bytes,
 *      for range checking.
 * destination_offset - an offset in bits to start placing the copied range into
 *      the destination array. Specify 0 to start from the beginning of the
 *      destination. If you are copying a range not aligned on a byte, you
 *      probably want to set this to a positive offset to right the resulting
 *      bits in the destination.
 *
 * Returns true if the copy was successful and false if the range exceeded the
 * size of the source or destination, or if the range size negative or 0.
 */
bool copy_bits(const uint8_t* source_origin, const unsigned int source_length,
        const unsigned int source_offset, unsigned int bit_count,
        uint8_t* destination_origin, const unsigned int destination_length,
        const unsigned int destination_offset);

/* Public: Copy a range of bits from one array to another, right aligning the
 * result.
 *
 * This is mostly useful if you want to cast the result to an integer type
 * instead of a byte array.
 *
 * For example:
 *
 *      unsigned int source[4] = {0x11, 0x22, 0x33, 0x44};
 *      unsigned int destination[4] = {0};
 *      copy_bits_right_aligned(source, sizeof(source), 8, 8, destination,
 *              sizeof(destination));
 *      // destination[0] == 0x0
 *      // destination[1] == 0x0
 *      // destination[2] == 0x0
 *      // destination[3] == 0x22
 *
 *      int value = (int)destination;
 *      // value == 0x22 == 32
 *
 * The arguments are the same as copy_bits, but without the destination_offset
 * option - that's set automatically to right align the result.
 *
 * Returns true if the copy was successful and false if the range exceeded the
 * size of the source or destination, or if the range size negative or 0.
 */
bool copy_bits_right_aligned(const uint8_t source[], const unsigned int source_length,
                const unsigned int offset, const unsigned int bit_count,
                uint8_t* destination, const unsigned int destination_length);

/* Public: Copy a range of bytes from one byte array to another.
 *
 * The source and destination do not have to be the same size (as long as the
 * desitnation has enough room to fit the range).
 *
 * source_origin - the source array.
 * source_length - the total length of the source array in bytes,
 *      for range checking.
 * source_offset - a byte offset to start the copy from the source array.
 *      Specify 0 to start from source_origin.
 * byte_count - the number of bytes to copy.
 * destination_origin - the destination array.
 * desitnation_length - the total length of the destination array in bytes,
 *      for range checking.
 * destination_offset - an offset in bytes to start placing the copied range into
 *      the destination array. Specify 0 to start from the beginning of the
 *      destination.
 *
 * Returns true if the copy was successful and false if the range exceeded the
 * size of the source or destination, or if the range size negative or 0.
 */
bool copy_bytes_right_aligned(const uint8_t source[], const unsigned int source_length,
                const unsigned int offset, const unsigned int byte_count,
                uint8_t* destination, const unsigned int destination_length);

/* Public: Set the a nibble in the given data array to the new value.
 *
 * nibble_index - the index of the nibble to retreive. The leftmost nibble is
 *      index 0.
 * value - the value to set in the bit field.
 * destination - the destination array.
 * destination_length - the total length of the destination array in bytes,
 *      for range checking.
 *
 * Returns true if the bit_count is enough to fully represent the value, and
 *      false if it will not fit.
 */
bool set_nibble(const unsigned int nibble_index, const unsigned int value,
                uint8_t* destination, const unsigned int destination_length);

/* Public: Set the bit field in the given data array to the new value.
 *
 * value - the value to set in the bit field.
 * offset - the starting index of the bit field (beginning from 0).
 * bit_count - the number of bits to set in the data.
 * destination - the destination array.
 * destination_length - the total length of the destination array in bytes,
 *      for range checking.
 *
 * Returns true if the bit_count is enough to fully represent the value, and
 *      false if it will not fit.
 */
bool set_bitfield(const uint64_t value, const unsigned int offset,
        const unsigned int bit_count, uint8_t destination[],
        unsigned int destination_length);

/* Public: Return a right aligned bitmask for a uint64_t.
 *
 * bit_count - the number of bits to mask, right aligned.
 */
uint64_t bitmask(const unsigned int bit_count);

/* Private:
 */
unsigned int bits_to_bytes(uint32_t bits);

/* Private: A union to assist swapping between uint64_t and a unsigned int array.
 */
typedef union {
    uint64_t whole;
    uint8_t bytes[sizeof(uint64_t)];
} ArrayOrBytes;

#ifdef __cplusplus
}
#endif

#endif // __BITFIELD_H__