Initial version.
This commit is contained in:
@@ -0,0 +1,148 @@
|
||||
/*
|
||||
bar.h
|
||||
project: bit array C library
|
||||
url: https://github.com/noporpoise/BitArray/
|
||||
maintainer: Isaac Turner <[email protected]>
|
||||
license: Public Domain, no warranty
|
||||
date: Sept 2014
|
||||
*/
|
||||
|
||||
// shorten the names of some of the bit_array functions to be more
|
||||
// like the str* function names. The prefix "bar" is used to represent
|
||||
// bit_array and is analogous to "str".
|
||||
|
||||
#ifndef BAR_HEADER_SEEN
|
||||
#define BAR_HEADER_SEEN
|
||||
|
||||
#include "bit_array.h"
|
||||
|
||||
#define bar BIT_ARRAY
|
||||
|
||||
#define barcreate bit_array_create
|
||||
#define bardestroy bit_array_free
|
||||
#define baralloc bit_array_alloc
|
||||
#define barfree bit_array_dealloc
|
||||
#define barlen bit_array_length
|
||||
|
||||
#define barsize bit_array_resize
|
||||
#define barcap bit_array_ensure_size
|
||||
|
||||
// These five are MACROs
|
||||
#define barget bit_array_get
|
||||
#define barset bit_array_set
|
||||
#define barclr bit_array_clear
|
||||
#define barflip bit_array_toggle
|
||||
#define barmake bit_array_assign
|
||||
|
||||
/* Functions instead of macros bars* => s for safe */
|
||||
#define barsget bit_array_get_bit
|
||||
#define barsset bit_array_set_bit
|
||||
#define barsclr bit_array_clear_bit
|
||||
#define barsflip bit_array_toggle_bit
|
||||
#define barsmake bit_array_assign_bit
|
||||
|
||||
/* "resize" functions barr*: automatically enlarge array if needed */
|
||||
#define barrget bit_array_rget
|
||||
#define barrset bit_array_rset
|
||||
#define barrclr bit_array_rclear
|
||||
#define barrflip bit_array_rtoggle
|
||||
#define barrmake bit_array_rassign
|
||||
|
||||
#define barsetn bit_array_set_bits
|
||||
#define barclrn bit_array_clear_bits
|
||||
#define barflipn bit_array_toggle_bits
|
||||
|
||||
#define barsetr bit_array_set_region
|
||||
#define barclrr bit_array_clear_region
|
||||
#define barflipr bit_array_toggle_region
|
||||
|
||||
#define barfill bit_array_set_all
|
||||
#define barzero bit_array_clear_all
|
||||
#define bartogl bit_array_toggle_all
|
||||
|
||||
/* gw "get word" */
|
||||
#define bargw64 bit_array_get_word64
|
||||
#define bargw32 bit_array_get_word32
|
||||
#define bargw16 bit_array_get_word16
|
||||
#define bargw8 bit_array_get_word8
|
||||
#define bargwn bit_array_get_wordn
|
||||
|
||||
/* sw "set word" */
|
||||
#define barsw64 bit_array_set_word64
|
||||
#define barsw32 bit_array_set_word32
|
||||
#define barsw16 bit_array_set_word16
|
||||
#define barswn bit_array_set_wordn
|
||||
|
||||
#define barncpy bit_array_copy
|
||||
#define barcpy bit_array_copy_all
|
||||
#define bardup bit_array_clone
|
||||
|
||||
#define barpopc bit_array_num_bits_set
|
||||
#define barzeros bit_array_num_bits_cleared
|
||||
#define bardist bit_array_hamming_distance
|
||||
#define barparity bit_array_parity
|
||||
|
||||
#define barfns bit_array_find_next_set_bit
|
||||
#define barfps bit_array_find_prev_set_bit
|
||||
#define barffs bit_array_find_first_set_bit
|
||||
#define barfls bit_array_find_last_set_bit
|
||||
|
||||
#define barfnz bit_array_find_next_clear_bit
|
||||
#define barfpz bit_array_find_prev_clear_bit
|
||||
#define barffz bit_array_find_first_clear_bit
|
||||
#define barflz bit_array_find_last_clear_bit
|
||||
|
||||
#define barsort bit_array_sort_bits
|
||||
#define barsortr bit_array_sort_bits_rev
|
||||
|
||||
#define barand bit_array_and
|
||||
#define baror bit_array_or
|
||||
#define barxor bit_array_xor
|
||||
#define barnot bit_array_not
|
||||
|
||||
#define barcmp bit_array_cmp
|
||||
#define barcmpbe bit_array_cmp_big_endian
|
||||
#define barcmpw bit_array_cmp_words
|
||||
#define barcmp64 bit_array_cmp_uint64
|
||||
|
||||
#define barshr bit_array_shift_right
|
||||
#define barshl bit_array_shift_left
|
||||
#define bareshl bit_array_shift_left_extend
|
||||
|
||||
#define barcycr bit_array_cycle_right
|
||||
#define barcycl bit_array_cycle_left
|
||||
|
||||
#define barmix bit_array_interleave
|
||||
|
||||
#define barrev bit_array_reverse
|
||||
#define barrevr bit_array_reverse_region
|
||||
|
||||
#define bar2num bit_array_as_num
|
||||
|
||||
/* Add/sub/mult/div a bit array with: */
|
||||
/* _i unsigned integer, _si shifted integer, _sb shifted bitarray */
|
||||
#define baraddi bit_array_add_uint64
|
||||
#define baraddsi bit_array_add_word
|
||||
#define baraddsb bit_array_add_words
|
||||
#define barsubi bit_array_sub_uint64
|
||||
#define barsubsi bit_array_sub_word
|
||||
#define barsubsb bit_array_sub_words
|
||||
#define barmuli bit_array_mul_uint64
|
||||
#define bardivi bit_array_div_uint64
|
||||
|
||||
/* arguments are both bit arrays */
|
||||
#define baradd bit_array_add
|
||||
#define barsub bit_array_subtract
|
||||
#define barmul bit_array_multiply
|
||||
#define bardiv bit_array_divide
|
||||
|
||||
#define barsave bit_array_save
|
||||
#define barload bit_array_load
|
||||
|
||||
#define barhash bit_array_hash
|
||||
|
||||
#define barrand bit_array_random
|
||||
#define barshfl bit_array_shuffle
|
||||
#define barperm bit_array_next_permutation
|
||||
|
||||
#endif /* BAR_HEADER_SEEN */
|
||||
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,548 @@
|
||||
/*
|
||||
bit_array.h
|
||||
project: bit array C library
|
||||
url: https://github.com/noporpoise/BitArray/
|
||||
maintainer: Isaac Turner <[email protected]>
|
||||
license: Public Domain, no warranty
|
||||
date: Sep 2014
|
||||
*/
|
||||
|
||||
#ifndef BIT_ARRAY_HEADER_SEEN
|
||||
#define BIT_ARRAY_HEADER_SEEN
|
||||
|
||||
#include <stdio.h>
|
||||
#include <inttypes.h>
|
||||
|
||||
#include "bit_macros.h"
|
||||
|
||||
typedef struct BIT_ARRAY BIT_ARRAY;
|
||||
|
||||
// 64 bit words
|
||||
typedef uint64_t word_t, word_addr_t, bit_index_t;
|
||||
typedef uint8_t word_offset_t; // Offset within a 64 bit word
|
||||
|
||||
#define BIT_INDEX_MIN 0
|
||||
#define BIT_INDEX_MAX (~(bit_index_t)0)
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
//
|
||||
// Structs
|
||||
//
|
||||
|
||||
struct BIT_ARRAY
|
||||
{
|
||||
word_t* words;
|
||||
bit_index_t num_of_bits;
|
||||
// Number of words used -- this is just round_up(num_of_bits / 64)
|
||||
// if num_of_bits == 0, this is 0
|
||||
word_addr_t num_of_words;
|
||||
// For more efficient allocation we use realloc only to double size --
|
||||
// not for adding every word. Initial size is INIT_CAPACITY_WORDS.
|
||||
word_addr_t capacity_in_words;
|
||||
};
|
||||
|
||||
//
|
||||
// Basics: Constructor, destructor, get length, resize
|
||||
//
|
||||
|
||||
// Constructor - create a new bit array of length nbits
|
||||
BIT_ARRAY* bit_array_create(bit_index_t nbits);
|
||||
|
||||
// Destructor - free the memory used for a bit array
|
||||
void bit_array_free(BIT_ARRAY* bitarray);
|
||||
|
||||
// Allocate using existing struct
|
||||
BIT_ARRAY* bit_array_alloc(BIT_ARRAY* bitarr, bit_index_t nbits);
|
||||
void bit_array_dealloc(BIT_ARRAY* bitarr);
|
||||
|
||||
// Get length of bit array
|
||||
bit_index_t bit_array_length(const BIT_ARRAY* bit_arr);
|
||||
|
||||
// Change the size of a bit array. Enlarging an array will add zeros
|
||||
// to the end of it. Returns 1 on success, 0 on failure (e.g. not enough memory)
|
||||
char bit_array_resize(BIT_ARRAY* bitarr, bit_index_t new_num_of_bits);
|
||||
|
||||
// If bitarr length < num_bits, resizes to num_bits
|
||||
char bit_array_ensure_size(BIT_ARRAY* bitarr, bit_index_t ensure_num_of_bits);
|
||||
|
||||
// Same as above but exit with an error message if out of memory
|
||||
void bit_array_resize_critical(BIT_ARRAY* bitarr, bit_index_t num_of_bits);
|
||||
void bit_array_ensure_size_critical(BIT_ARRAY* bitarr, bit_index_t num_of_bits);
|
||||
|
||||
|
||||
//
|
||||
// Macros
|
||||
//
|
||||
|
||||
//
|
||||
// Get, set, clear, assign and toggle individual bits
|
||||
// Macros for fast access -- beware: no bounds checking
|
||||
//
|
||||
|
||||
#define bit_array_get(arr,i) bitset_get((arr)->words, i)
|
||||
#define bit_array_set(arr,i) bitset_set((arr)->words, i)
|
||||
#define bit_array_clear(arr,i) bitset_del((arr)->words, i)
|
||||
#define bit_array_toggle(arr,i) bitset_tgl((arr)->words, i)
|
||||
// c must be 0 or 1
|
||||
#define bit_array_assign(arr,i,c) bitset_cpy((arr)->words,i,c)
|
||||
|
||||
//
|
||||
// Get, set, clear, assign and toggle individual bits
|
||||
// "Safe": use assert() to check bounds
|
||||
//
|
||||
|
||||
// Get the value of a bit (returns 0 or 1)
|
||||
char bit_array_get_bit(const BIT_ARRAY* bitarr, bit_index_t b);
|
||||
void bit_array_set_bit(BIT_ARRAY* bitarr, bit_index_t b);
|
||||
void bit_array_clear_bit(BIT_ARRAY* bitarr, bit_index_t b);
|
||||
void bit_array_toggle_bit(BIT_ARRAY* bitarr, bit_index_t b);
|
||||
// If char c != 0, set bit; otherwise clear bit
|
||||
void bit_array_assign_bit(BIT_ARRAY* bitarr, bit_index_t b, char c);
|
||||
|
||||
//
|
||||
// "Resizing": enlarge array if needed
|
||||
//
|
||||
|
||||
char bit_array_rget(BIT_ARRAY* bitarr, bit_index_t b);
|
||||
void bit_array_rset(BIT_ARRAY* bitarr, bit_index_t b);
|
||||
void bit_array_rclear(BIT_ARRAY* bitarr, bit_index_t b);
|
||||
void bit_array_rtoggle(BIT_ARRAY* bitarr, bit_index_t b);
|
||||
void bit_array_rassign(BIT_ARRAY* bitarr, bit_index_t b, char c);
|
||||
|
||||
//
|
||||
// Set, clear and toggle several bits at once
|
||||
//
|
||||
|
||||
// Set multiple bits at once.
|
||||
// e.g. set bits 1, 20 & 31: bit_array_set_bits(bitarr, 3, 1,20,31);
|
||||
// Note: variable args are of type unsigned int
|
||||
void bit_array_set_bits(BIT_ARRAY* bitarr, size_t n, ...);
|
||||
|
||||
// Clear multiple bits at once.
|
||||
// e.g. clear bits 1, 20 & 31: bit_array_clear_bits(bitarr, 3, 1,20,31);
|
||||
// Note: variable args are of type unsigned int
|
||||
void bit_array_clear_bits(BIT_ARRAY* bitarr, size_t n, ...);
|
||||
|
||||
// Toggle multiple bits at once
|
||||
// e.g. toggle bits 1, 20 & 31: bit_array_toggle_bits(bitarr, 3, 1,20,31);
|
||||
// Note: variable args are of type unsigned int
|
||||
void bit_array_toggle_bits(BIT_ARRAY* bitarr, size_t n, ...);
|
||||
|
||||
//
|
||||
// Set, clear and toggle all bits in a region
|
||||
//
|
||||
|
||||
// Set all the bits in a region
|
||||
void bit_array_set_region(BIT_ARRAY* bitarr, bit_index_t start, bit_index_t len);
|
||||
|
||||
// Clear all the bits in a region
|
||||
void bit_array_clear_region(BIT_ARRAY* bitarr, bit_index_t start, bit_index_t len);
|
||||
|
||||
// Toggle all the bits in a region
|
||||
void bit_array_toggle_region(BIT_ARRAY* bitarr, bit_index_t start, bit_index_t len);
|
||||
|
||||
//
|
||||
// Set, clear and toggle all bits at once
|
||||
//
|
||||
|
||||
// Set all bits in this array to 1
|
||||
void bit_array_set_all(BIT_ARRAY* bitarr);
|
||||
|
||||
// Set all bits in this array to 0
|
||||
void bit_array_clear_all(BIT_ARRAY* bitarr);
|
||||
|
||||
// Set all 1 bits to 0, and all 0 bits to 1
|
||||
void bit_array_toggle_all(BIT_ARRAY* bitarr);
|
||||
|
||||
//
|
||||
// Get / set a word of a given size
|
||||
//
|
||||
|
||||
// First bit is in the least significant bit position
|
||||
// start index must be within the range of the bit array (0 <= x < length)
|
||||
uint64_t bit_array_get_word64(const BIT_ARRAY* bitarr, bit_index_t start);
|
||||
uint32_t bit_array_get_word32(const BIT_ARRAY* bitarr, bit_index_t start);
|
||||
uint16_t bit_array_get_word16(const BIT_ARRAY* bitarr, bit_index_t start);
|
||||
uint8_t bit_array_get_word8(const BIT_ARRAY* bitarr, bit_index_t start);
|
||||
uint64_t bit_array_get_wordn(const BIT_ARRAY* bitarr, bit_index_t start, int n);
|
||||
|
||||
// Set 64 bits at once from a particular start position
|
||||
// Doesn't extend bit array. However it is safe to TRY to set bits beyond the
|
||||
// end of the array, as long as: `start` is < `bit_array_length(arr)`
|
||||
void bit_array_set_word64(BIT_ARRAY* bitarr, bit_index_t start, uint64_t word);
|
||||
void bit_array_set_word32(BIT_ARRAY* bitarr, bit_index_t start, uint32_t word);
|
||||
void bit_array_set_word16(BIT_ARRAY* bitarr, bit_index_t start, uint16_t word);
|
||||
void bit_array_set_word8(BIT_ARRAY* bitarr, bit_index_t start, uint8_t byte);
|
||||
void bit_array_set_wordn(BIT_ARRAY* bitarr, bit_index_t start, uint64_t word, int n);
|
||||
|
||||
//
|
||||
// Number of bits set
|
||||
//
|
||||
|
||||
// Get the number of bits set (hamming weight)
|
||||
bit_index_t bit_array_num_bits_set(const BIT_ARRAY* bitarr);
|
||||
|
||||
// Get the number of bits not set (length - hamming weight)
|
||||
bit_index_t bit_array_num_bits_cleared(const BIT_ARRAY* bitarr);
|
||||
|
||||
// Get the number of bits set in on array and not the other. This is equivalent
|
||||
// to hamming weight of the XOR when the two arrays are the same length.
|
||||
// e.g. 10101 vs 00111 => hamming distance 2 (XOR is 10010)
|
||||
bit_index_t bit_array_hamming_distance(const BIT_ARRAY* arr1,
|
||||
const BIT_ARRAY* arr2);
|
||||
|
||||
// Parity - returns 1 if odd number of bits set, 0 if even
|
||||
char bit_array_parity(const BIT_ARRAY* bitarr);
|
||||
|
||||
//
|
||||
// Find indices of set/clear bits
|
||||
//
|
||||
|
||||
// Find the index of the next bit that is set, at or after `offset`
|
||||
// Returns 1 if a bit is set, otherwise 0
|
||||
// Index of next set bit is stored in the integer pointed to by result
|
||||
// If no next bit is set result is not changed
|
||||
char bit_array_find_next_set_bit(const BIT_ARRAY* bitarr, bit_index_t offset,
|
||||
bit_index_t* result);
|
||||
|
||||
// Find the index of the next bit that is NOT set, at or after `offset`
|
||||
// Returns 1 if a bit is NOT set, otherwise 0
|
||||
// Index of next zero bit is stored in the integer pointed to by `result`
|
||||
// If no next bit is zero, value at `result` is not changed
|
||||
char bit_array_find_next_clear_bit(const BIT_ARRAY* bitarr, bit_index_t offset,
|
||||
bit_index_t* result);
|
||||
|
||||
// Find the index of the previous bit that is set, before offset.
|
||||
// Returns 1 if a bit is set, otherwise 0
|
||||
// Index of previous set bit is stored in the integer pointed to by `result`
|
||||
// If no previous bit is set result is not changed
|
||||
char bit_array_find_prev_set_bit(const BIT_ARRAY* bitarr, bit_index_t offset,
|
||||
bit_index_t* result);
|
||||
|
||||
// Find the index of the previous bit that is NOT set, before offset.
|
||||
// Returns 1 if a bit is clear, otherwise 0
|
||||
// Index of previous zero bit is stored in the integer pointed to by `result`
|
||||
// If no previous bit is zero result is not changed
|
||||
char bit_array_find_prev_clear_bit(const BIT_ARRAY* bitarr, bit_index_t offset,
|
||||
bit_index_t* result);
|
||||
|
||||
// Find the index of the first bit that is set.
|
||||
// Returns 1 if a bit is set, otherwise 0
|
||||
// Index of first set bit is stored in the integer pointed to by `result`
|
||||
// If no bit is set result is not changed
|
||||
char bit_array_find_first_set_bit(const BIT_ARRAY* bitarr, bit_index_t* result);
|
||||
|
||||
// Find the index of the first bit that is NOT set.
|
||||
// Returns 1 if a bit is clear, otherwise 0
|
||||
// Index of first zero bit is stored in the integer pointed to by `result`
|
||||
// If no bit is zero result is not changed
|
||||
char bit_array_find_first_clear_bit(const BIT_ARRAY* bitarr, bit_index_t* result);
|
||||
|
||||
// Find the index of the last bit that is set.
|
||||
// Returns 1 if a bit is set, otherwise 0
|
||||
// Index of last set bit is stored in the integer pointed to by `result`
|
||||
// If no bit is set result is not changed
|
||||
char bit_array_find_last_set_bit(const BIT_ARRAY* bitarr, bit_index_t* result);
|
||||
|
||||
// Find the index of the last bit that is NOT set.
|
||||
// Returns 1 if a bit is clear, otherwise 0
|
||||
// Index of last zero bit is stored in the integer pointed to by `result`
|
||||
// If no bit is zero result is not changed
|
||||
char bit_array_find_last_clear_bit(const BIT_ARRAY* bitarr, bit_index_t* result);
|
||||
|
||||
|
||||
//
|
||||
// Sorting
|
||||
//
|
||||
|
||||
// Put all the 0s before all the 1s
|
||||
void bit_array_sort_bits(BIT_ARRAY* bitarr);
|
||||
|
||||
// Put all the 1s before all the 0s
|
||||
void bit_array_sort_bits_rev(BIT_ARRAY* bitarr);
|
||||
|
||||
|
||||
//
|
||||
// String and printing methods
|
||||
//
|
||||
|
||||
// Construct a BIT_ARRAY from a string.
|
||||
void bit_array_from_str(BIT_ARRAY* bitarr, const char* bitstr);
|
||||
|
||||
// Construct a BIT_ARRAY from a substring with given on and off characters.
|
||||
void bit_array_from_substr(BIT_ARRAY* bitarr, bit_index_t offset,
|
||||
const char* str, size_t len,
|
||||
const char *on, const char *off, char left_to_right);
|
||||
|
||||
// Takes a char array to write to. `str` must be bitarr->num_of_bits+1 in
|
||||
// length. Terminates string with '\0'
|
||||
char* bit_array_to_str(const BIT_ARRAY* bitarr, char* str);
|
||||
char* bit_array_to_str_rev(const BIT_ARRAY* bitarr, char* str);
|
||||
|
||||
// Get a string representations for a given region, using given on/off
|
||||
// characters.
|
||||
// Note: does not null-terminate
|
||||
void bit_array_to_substr(const BIT_ARRAY* bitarr,
|
||||
bit_index_t start, bit_index_t length,
|
||||
char* str, char on, char off, char left_to_right);
|
||||
|
||||
// Print this array to a file stream. Prints '0's and '1'. Doesn't print
|
||||
// newline.
|
||||
void bit_array_print(const BIT_ARRAY* bitarr, FILE* fout);
|
||||
|
||||
// Print a string representations for a given region, using given on/off
|
||||
// characters. Reverse prints from highest to lowest -- this is useful for
|
||||
// printing binary numbers
|
||||
void bit_array_print_substr(const BIT_ARRAY* bitarr,
|
||||
bit_index_t start, bit_index_t length,
|
||||
FILE* fout, char on, char off, char left_to_right);
|
||||
|
||||
//
|
||||
// Decimal
|
||||
//
|
||||
|
||||
// Get bit array as decimal str (e.g. 0b1101 -> "13")
|
||||
size_t bit_array_to_decimal(const BIT_ARRAY *bitarr, char *str, size_t len);
|
||||
|
||||
// Return number of characters used
|
||||
size_t bit_array_from_decimal(BIT_ARRAY *bitarr, const char* decimal);
|
||||
|
||||
//
|
||||
// Hexidecimal
|
||||
//
|
||||
|
||||
// Loads array from hex string
|
||||
// Returns the number of bits loaded (will be chars rounded up to multiple of 8)
|
||||
// (0 on failure)
|
||||
bit_index_t bit_array_from_hex(BIT_ARRAY* bitarr, bit_index_t offset,
|
||||
const char* str, size_t len);
|
||||
|
||||
// Returns number of characters written
|
||||
size_t bit_array_to_hex(const BIT_ARRAY* bitarr,
|
||||
bit_index_t start, bit_index_t length,
|
||||
char* str, char uppercase);
|
||||
|
||||
// Print bit array as hex
|
||||
size_t bit_array_print_hex(const BIT_ARRAY* bitarr,
|
||||
bit_index_t start, bit_index_t length,
|
||||
FILE* fout, char uppercase);
|
||||
|
||||
//
|
||||
// Clone and copy
|
||||
//
|
||||
|
||||
// Copy a BIT_ARRAY struct and the data it holds - returns pointer to new object
|
||||
#define bit_array_dup bit_array_clone
|
||||
BIT_ARRAY* bit_array_clone(const BIT_ARRAY* bitarr);
|
||||
|
||||
// Copy bits from one array to another
|
||||
// Note: use MACRO bit_array_copy
|
||||
// Destination and source can be the same bit_array and
|
||||
// src/dst regions can overlap
|
||||
void bit_array_copy(BIT_ARRAY* dst, bit_index_t dstindx,
|
||||
const BIT_ARRAY* src, bit_index_t srcindx,
|
||||
bit_index_t length);
|
||||
|
||||
// copy all of src to dst. dst is resized to match src.
|
||||
void bit_array_copy_all(BIT_ARRAY* dst, const BIT_ARRAY* src);
|
||||
|
||||
//
|
||||
// Logic operators
|
||||
//
|
||||
|
||||
// BIT_ARRAYs can all be different or the same object
|
||||
// dest array will be resized if it is too short
|
||||
//
|
||||
void bit_array_and(BIT_ARRAY* dest, const BIT_ARRAY* src1, const BIT_ARRAY* src2);
|
||||
void bit_array_or (BIT_ARRAY* dest, const BIT_ARRAY* src1, const BIT_ARRAY* src2);
|
||||
void bit_array_xor(BIT_ARRAY* dest, const BIT_ARRAY* src1, const BIT_ARRAY* src2);
|
||||
void bit_array_not(BIT_ARRAY* dest, const BIT_ARRAY* src);
|
||||
|
||||
//
|
||||
// Comparisons
|
||||
//
|
||||
|
||||
// Note: (bit_array_cmp(a,b) == 0) <=> (bit_array_cmp_big_endian(a,b) == 0)
|
||||
|
||||
// comparison functions return:
|
||||
// 1 iff bitarr1 > bitarr2
|
||||
// 0 iff bitarr1 == bitarr2
|
||||
// -1 iff bitarr1 < bitarr2
|
||||
|
||||
// Compare two bit arrays by value stored, with index 0 being the Least
|
||||
// Significant Bit (LSB). Arrays do not have to be the same length.
|
||||
// Example: ..0101 (5) > ...0011 (3) [index 0 is LSB at right hand side]
|
||||
int bit_array_cmp(const BIT_ARRAY* bitarr1, const BIT_ARRAY* bitarr2);
|
||||
|
||||
// Compare two bit arrays by value stored, with index 0 being the Most
|
||||
// Significant Bit (MSB). Arrays do not have to be the same length.
|
||||
// Example: 10.. > 01.. [index 0 is MSB at left hand side]
|
||||
int bit_array_cmp_big_endian(const BIT_ARRAY* bitarr1, const BIT_ARRAY* bitarr2);
|
||||
|
||||
// compare bitarr with (bitarr2 << pos)
|
||||
int bit_array_cmp_words(const BIT_ARRAY *bitarr,
|
||||
bit_index_t pos, const BIT_ARRAY *bitarr2);
|
||||
|
||||
//
|
||||
// Shift, interleave, reverse
|
||||
//
|
||||
|
||||
// Shift array left/right. If fill is zero, filled with 0, otherwise 1
|
||||
void bit_array_shift_right(BIT_ARRAY* bitarr, bit_index_t shift_dist, char fill);
|
||||
void bit_array_shift_left (BIT_ARRAY* bitarr, bit_index_t shift_dist, char fill);
|
||||
|
||||
// shift left without losing any bits. Resizes bitarr.
|
||||
void bit_array_shift_left_extend(BIT_ARRAY* bitarr, bit_index_t shift_dist,
|
||||
char fill);
|
||||
|
||||
// Cyclic shift
|
||||
void bit_array_cycle_right(BIT_ARRAY* bitarr, bit_index_t dist);
|
||||
void bit_array_cycle_left (BIT_ARRAY* bitarr, bit_index_t dist);
|
||||
|
||||
// Interleave
|
||||
// dst cannot point to the same bit array as src1 or src2
|
||||
// src1, src2 may point to the same bit array
|
||||
// abcd 1234 -> a1b2c3d4
|
||||
// 0011 0000 -> 00001010
|
||||
// 1111 0000 -> 10101010
|
||||
// 0101 1010 -> 01100110
|
||||
// Extends dst if it is too short, but does not shrink it if it is too long
|
||||
// if dst is longer than length(src1)+length(src2), the end bits are not altered
|
||||
void bit_array_interleave(BIT_ARRAY* dst,
|
||||
const BIT_ARRAY* src1,
|
||||
const BIT_ARRAY* src2);
|
||||
|
||||
// Reverse the whole array or part of it
|
||||
void bit_array_reverse(BIT_ARRAY* bitarr);
|
||||
void bit_array_reverse_region(BIT_ARRAY* bitarr, bit_index_t start, bit_index_t len);
|
||||
|
||||
//
|
||||
// Numeric
|
||||
//
|
||||
|
||||
// Returns 1 on sucess, 0 if value in array is too big
|
||||
char bit_array_as_num(const BIT_ARRAY* bitarr, uint64_t* result);
|
||||
|
||||
// 1 iff bitarr > value
|
||||
// 0 iff bitarr == value
|
||||
// -1 iff bitarr < value
|
||||
int bit_array_cmp_uint64(const BIT_ARRAY* bitarr, uint64_t value);
|
||||
|
||||
//
|
||||
// Arithmetic
|
||||
//
|
||||
|
||||
// bitarr will be extended if needed
|
||||
void bit_array_add_uint64(BIT_ARRAY* bitarr, uint64_t value);
|
||||
|
||||
// Add `add` to `bitarr` at `pos` -- same as:
|
||||
// bitarr + (add << pos)
|
||||
// where pos can be bigger than the length of the array (bitarr will be resized)
|
||||
void bit_array_add_word(BIT_ARRAY *bitarr, bit_index_t pos, uint64_t add);
|
||||
|
||||
// Add `add` to `bitarr` at `pos`
|
||||
void bit_array_add_words(BIT_ARRAY *bitarr, bit_index_t pos, const BIT_ARRAY *add);
|
||||
|
||||
// If value is greater than bitarr, bitarr is not changed and 0 is returned
|
||||
// Returns 1 on success, 0 if value > bitarr
|
||||
char bit_array_sub_uint64(BIT_ARRAY* bitarr, uint64_t value);
|
||||
|
||||
// minus `minus` from `bitarr` at `pos` -- same as:
|
||||
// bitarr + (minus << pos)
|
||||
// Returns 1 on success, 0 if value > bitarr
|
||||
char bit_array_sub_word(BIT_ARRAY *bitarr, bit_index_t pos, word_t minus);
|
||||
|
||||
// minus `minus` from `bitarr` at `pos`
|
||||
// Returns 1 on success, 0 if value > bitarr
|
||||
char bit_array_sub_words(BIT_ARRAY* bitarr, bit_index_t pos, BIT_ARRAY* minus);
|
||||
|
||||
// Multiply by some value
|
||||
void bit_array_mul_uint64(BIT_ARRAY *bitarr, uint64_t multiplier);
|
||||
|
||||
// bitarr = round_down(bitarr / divisor)
|
||||
// rem = bitarr % divisor
|
||||
void bit_array_div_uint64(BIT_ARRAY *bitarr, uint64_t divisor, uint64_t *rem);
|
||||
|
||||
//
|
||||
// Arithmetic between arrays
|
||||
//
|
||||
|
||||
// dst = src1 + src2
|
||||
// src1, src2 and dst can all be the same BIT_ARRAY
|
||||
// If dst is shorter than either of src1, src2, it is enlarged
|
||||
void bit_array_add(BIT_ARRAY* dst, const BIT_ARRAY* src1, const BIT_ARRAY* src2);
|
||||
|
||||
// dst = src1 - src2
|
||||
// src1, src2 and dst can all be the same BIT_ARRAY
|
||||
// If dst is shorter than src1, it will be extended to be as long as src1
|
||||
// src1 must be greater than or equal to src2 (src1 >= src2)
|
||||
void bit_array_subtract(BIT_ARRAY* dst,
|
||||
const BIT_ARRAY* src1, const BIT_ARRAY* src2);
|
||||
|
||||
// dst = src1 * src2
|
||||
// Pointers cannot all point to the same BIT_ARRAY
|
||||
void bit_array_multiply(BIT_ARRAY *dst, BIT_ARRAY *src1, BIT_ARRAY *src2);
|
||||
|
||||
// Results in:
|
||||
// quotient = dividend / divisor
|
||||
// dividend = dividend % divisor
|
||||
// (dividend is used to return the remainder)
|
||||
void bit_array_divide(BIT_ARRAY *dividend, BIT_ARRAY *quotient, BIT_ARRAY *divisor);
|
||||
|
||||
//
|
||||
// Read/Write bit_array to a file
|
||||
//
|
||||
// File format is [8 bytes: for number of elements in array][data]
|
||||
// Number of bytes of data is: (int)((num_of_bits + 7) / 8)
|
||||
//
|
||||
|
||||
// Saves bit array to a file
|
||||
// returns the number of bytes written
|
||||
bit_index_t bit_array_save(const BIT_ARRAY* bitarr, FILE* f);
|
||||
|
||||
// Reads bit array from a file. bitarr is resized and filled.
|
||||
// Returns 1 on success, 0 on failure
|
||||
char bit_array_load(BIT_ARRAY* bitarr, FILE* f);
|
||||
|
||||
|
||||
//
|
||||
// Hash function
|
||||
//
|
||||
|
||||
// Pass seed as 0 on first call, pass previous hash value if rehashing due
|
||||
// to a collision
|
||||
// Using bob jenkins hash lookup3
|
||||
uint64_t bit_array_hash(const BIT_ARRAY* bitarr, uint64_t seed);
|
||||
|
||||
//
|
||||
// Randomness
|
||||
//
|
||||
|
||||
// Get the next permutation of an array with a fixed size and given number of
|
||||
// bits set. Also known as next lexicographic permutation.
|
||||
// Given a bit array find the next lexicographic orginisation of the bits
|
||||
// Number of possible combinations given by (size choose bits_set) i.e. nCk
|
||||
// 00011 -> 00101 -> 00110 -> 01001 -> 01010 ->
|
||||
// 01100 -> 10001 -> 10010 -> 10100 -> 11000 -> 00011 (back to start)
|
||||
void bit_array_next_permutation(BIT_ARRAY* bitarr);
|
||||
|
||||
//
|
||||
// Generally useful functions
|
||||
//
|
||||
|
||||
// Generalised 'binary to string' function
|
||||
// Adds bits to the string in order of lsb to msb
|
||||
// e.g. 0b11010 (26 in decimal) would come out as "01011"
|
||||
char* bit_array_word2str(const void *ptr, size_t num_of_bits, char *str);
|
||||
|
||||
// Same as above but in reverse
|
||||
char* bit_array_word2str_rev(const void *ptr, size_t num_of_bits, char *str);
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
||||
#endif
|
||||
@@ -0,0 +1,205 @@
|
||||
/*
|
||||
bit_macros.h
|
||||
project: bit array C library
|
||||
url: https://github.com/noporpoise/BitArray/
|
||||
author: Isaac Turner <[email protected]>
|
||||
license: Public Domain, no warranty
|
||||
date: Dec 2013
|
||||
*/
|
||||
|
||||
#ifndef BITSET_H_
|
||||
#define BITSET_H_
|
||||
|
||||
#include <inttypes.h>
|
||||
#include <sched.h>
|
||||
|
||||
// trailing_zeros is number of least significant zeros
|
||||
// leading_zeros is number of most significant zeros
|
||||
#if defined(_WIN32)
|
||||
#define trailing_zeros(x) ({ __typeof(x) _r; _BitScanReverse64(&_r, x); _r; })
|
||||
#define leading_zeros(x) ({ __typeof(x) _r; _BitScanForward64(&_r, x); _r; })
|
||||
#else
|
||||
#define trailing_zeros(x) ((x) ? (__typeof(x))__builtin_ctzll(x) : (__typeof(x))sizeof(x)*8)
|
||||
#define leading_zeros(x) ((x) ? (__typeof(x))__builtin_clzll(x) : (__typeof(x))sizeof(x)*8)
|
||||
#endif
|
||||
|
||||
// Get index of top set bit. If x is 0 return nbits
|
||||
#define top_set_bit(x) ((x) ? sizeof(x)*8-leading_zeros(x)-1 : sizeof(x)*8)
|
||||
|
||||
#define roundup_bits2bytes(bits) (((bits)+7)/8)
|
||||
#define roundup_bits2words32(bits) (((bits)+31)/32)
|
||||
#define roundup_bits2words64(bits) (((bits)+63)/64)
|
||||
|
||||
// Round a number up to the nearest number that is a power of two
|
||||
#define roundup2pow(x) (1UL << (64 - leading_zeros(x)))
|
||||
|
||||
#define rot32(x,r) (((x)<<(r)) | ((x)>>(32-(r))))
|
||||
#define rot64(x,r) (((x)<<(r)) | ((x)>>(64-(r))))
|
||||
|
||||
// need to check for length == 0, undefined behaviour if uint64_t >> 64 etc
|
||||
#define bitmask(nbits,type) ((nbits) ? ~(type)0 >> (sizeof(type)*8-(nbits)): (type)0)
|
||||
#define bitmask32(nbits) bitmask(nbits,uint32_t)
|
||||
#define bitmask64(nbits) bitmask(nbits,uint64_t)
|
||||
|
||||
// A possibly faster way to combine two words with a mask
|
||||
//#define bitmask_merge(a,b,abits) ((a & abits) | (b & ~abits))
|
||||
#define bitmask_merge(a,b,abits) (b ^ ((a ^ b) & abits))
|
||||
|
||||
// Swap lowest four bits. A nibble is 4 bits (i.e. half a byte)
|
||||
#define rev_nibble(x) ((((x)&1)<<3)|(((x)&2)<<1)|(((x)&4)>>1)|(((x)&8)>>3))
|
||||
|
||||
//
|
||||
// Bit array (bitset)
|
||||
//
|
||||
// bitsetX_wrd(): get word for a given position
|
||||
// bitsetX_idx(): get index within word for a given position
|
||||
#define _VOLPTR(x) ((volatile __typeof(x) *)(&(x)))
|
||||
#define _VOLVALUE(x) (*_VOLPTR(x))
|
||||
|
||||
#define _TYPESHIFT(arr,word,shift) \
|
||||
((__typeof(*(arr)))((__typeof(*(arr)))(word) << (shift)))
|
||||
|
||||
#define bitsetX_wrd(wrdbits,pos) ((pos) / (wrdbits))
|
||||
#define bitsetX_idx(wrdbits,pos) ((pos) % (wrdbits))
|
||||
|
||||
#define bitset32_wrd(pos) ((pos) >> 5)
|
||||
#define bitset32_idx(pos) ((pos) & 31)
|
||||
|
||||
#define bitset64_wrd(pos) ((pos) >> 6)
|
||||
#define bitset64_idx(pos) ((pos) & 63)
|
||||
|
||||
//
|
||||
// Bit functions on arrays
|
||||
//
|
||||
#define bitset2_get(arr,wrd,idx) (((arr)[wrd] >> (idx)) & 0x1)
|
||||
#define bitset2_set(arr,wrd,idx) ((arr)[wrd] |= _TYPESHIFT(arr,1,idx))
|
||||
#define bitset2_del(arr,wrd,idx) ((arr)[wrd] &=~ _TYPESHIFT(arr,1,idx))
|
||||
#define bitset2_tgl(arr,wrd,idx) ((arr)[wrd] ^= _TYPESHIFT(arr,1,idx))
|
||||
#define bitset2_or(arr,wrd,idx,bit) ((arr)[wrd] |= _TYPESHIFT(arr,bit,idx))
|
||||
#define bitset2_xor(arr,wrd,idx,bit) ((arr)[wrd] = ~((arr)[wrd] ^ (~_TYPESHIFT(arr,bit,idx))))
|
||||
#define bitset2_and(arr,wrd,idx,bit) ((arr)[wrd] &= (_TYPESHIFT(arr,bit,idx) | ~_TYPESHIFT(arr,1,idx)))
|
||||
#define bitset2_cpy(arr,wrd,idx,bit) ((arr)[wrd] = ((arr)[wrd] &~ _TYPESHIFT(arr,1,idx)) | _TYPESHIFT(arr,bit,idx))
|
||||
|
||||
//
|
||||
// Thread safe versions
|
||||
//
|
||||
// They return the value of the bit (0 or 1) before it was updated
|
||||
#define bitset2_get_mt(arr,wrd,idx) bitset2_get(_VOLPTR(*(arr)),wrd,idx)
|
||||
#define bitset2_set_mt(arr,wrd,idx) ((__sync_fetch_and_or (_VOLPTR((arr)[wrd]), _TYPESHIFT(arr,1,idx)) >> (idx))&1)
|
||||
#define bitset2_del_mt(arr,wrd,idx) ((__sync_fetch_and_and(_VOLPTR((arr)[wrd]), ~_TYPESHIFT(arr,1,idx)) >> (idx))&1)
|
||||
#define bitset2_tgl_mt(arr,wrd,idx) ((__sync_fetch_and_xor(_VOLPTR((arr)[wrd]), _TYPESHIFT(arr,1,idx)) >> (idx))&1)
|
||||
#define bitset2_or_mt(arr,wrd,idx,bit) ((__sync_fetch_and_or (_VOLPTR((arr)[wrd]), _TYPESHIFT(arr,bit,idx)) >> (idx))&1)
|
||||
#define bitset2_xor_mt(arr,wrd,idx,bit) ((__sync_fetch_and_xor(_VOLPTR((arr)[wrd]), _TYPESHIFT(arr,bit,idx)) >> (idx))&1)
|
||||
#define bitset2_and_mt(arr,wrd,idx,bit) ((__sync_fetch_and_and(_VOLPTR((arr)[wrd]), (_TYPESHIFT(arr,bit,idx) | ~_TYPESHIFT(arr,1,idx))) >> (idx))&1)
|
||||
#define bitset2_cpy_mt(arr,wrd,idx,bit) ((bit) ? bitset2_set_mt(arr,wrd,idx) : bitset2_del_mt(arr,wrd,idx))
|
||||
|
||||
//
|
||||
// Auto detect size of type from pointer
|
||||
//
|
||||
#define bitset_wrd(arr,pos) bitsetX_wrd(sizeof(*(arr))*8,pos)
|
||||
#define bitset_idx(arr,pos) bitsetX_idx(sizeof(*(arr))*8,pos)
|
||||
#define bitset_op(func,arr,pos) func(arr, bitset_wrd(arr,pos), bitset_idx(arr,pos))
|
||||
#define bitset_op2(func,arr,pos,bit) func(arr, bitset_wrd(arr,pos), bitset_idx(arr,pos), bit)
|
||||
|
||||
// Auto-detect type size: bit functions
|
||||
#define bitset_get(arr,pos) bitset_op(bitset2_get, arr, pos)
|
||||
#define bitset_set(arr,pos) bitset_op(bitset2_set, arr, pos)
|
||||
#define bitset_del(arr,pos) bitset_op(bitset2_del, arr, pos)
|
||||
#define bitset_tgl(arr,pos) bitset_op(bitset2_tgl, arr, pos)
|
||||
#define bitset_or(arr,pos,bit) bitset_op2(bitset2_or, arr, pos, bit)
|
||||
#define bitset_xor(arr,pos,bit) bitset_op2(bitset2_xor, arr, pos, bit)
|
||||
#define bitset_and(arr,pos,bit) bitset_op2(bitset2_and, arr, pos, bit)
|
||||
#define bitset_cpy(arr,pos,bit) bitset_op2(bitset2_cpy, arr, pos, bit)
|
||||
|
||||
// Auto-detect type size: thread safe bit functions
|
||||
// They return the value of the bit (0 or 1) before it was updated
|
||||
#define bitset_get_mt(arr,pos) bitset_op(bitset2_get_mt, arr, pos)
|
||||
#define bitset_set_mt(arr,pos) bitset_op(bitset2_set_mt, arr, pos)
|
||||
#define bitset_del_mt(arr,pos) bitset_op(bitset2_del_mt, arr, pos)
|
||||
#define bitset_tgl_mt(arr,pos) bitset_op(bitset2_tgl_mt, arr, pos)
|
||||
#define bitset_or_mt(arr,pos,bit) bitset_op2(bitset2_or_mt, arr, pos, bit)
|
||||
#define bitset_xor_mt(arr,pos,bit) bitset_op2(bitset2_xor_mt, arr, pos, bit)
|
||||
#define bitset_and_mt(arr,pos,bit) bitset_op2(bitset2_and_mt, arr, pos, bit)
|
||||
#define bitset_cpy_mt(arr,pos,bit) bitset_op2(bitset2_cpy_mt, arr, pos, bit)
|
||||
|
||||
// Clearing a word does not return a meaningful value
|
||||
#define bitset_clear_word(arr,pos) ((arr)[bitset_wrd(arr,pos)] = 0)
|
||||
#define bitset_clear_word_mt(arr,pos) (_VOLVALUE((arr)[bitset_wrd(arr,pos)]) = 0)
|
||||
|
||||
//
|
||||
// Compact bit array of spin locks
|
||||
// These are most effecient when arr is of type: volatile char*
|
||||
//
|
||||
// Acquire a lock
|
||||
#define bitlock_acquire_block(arr,pos,wait,abandon) do { \
|
||||
size_t _w = bitset_wrd(arr,pos); \
|
||||
__typeof(*(arr)) _o, _n, _b = _TYPESHIFT(arr, 1, bitset_idx(arr,pos)); \
|
||||
do { \
|
||||
while((_o = _VOLVALUE((arr)[_w])) & _b) { wait } \
|
||||
abandon \
|
||||
_n = _o | _b; \
|
||||
} while(!__sync_bool_compare_and_swap(_VOLPTR((arr)[_w]), _o, _n)); \
|
||||
__sync_synchronize(); /* Must not move commands to before acquiring lock */ \
|
||||
} while(0)
|
||||
|
||||
// Undefined behaviour if you do not already hold the lock
|
||||
#define bitlock_release(arr,pos) do { \
|
||||
size_t _w = bitset_wrd(arr,pos); \
|
||||
__typeof(*(arr)) _mask = ~_TYPESHIFT(arr, 1, bitset_idx(arr,pos)); \
|
||||
__sync_synchronize(); /* Must get the lock before releasing it */ \
|
||||
__sync_and_and_fetch(_VOLPTR((arr)[_w]), _mask); \
|
||||
} while(0)
|
||||
|
||||
#define bitlock_acquire(arr,pos) bitlock_acquire_block(arr,pos,{},{})
|
||||
|
||||
// calls yield if cannot acquire the lock
|
||||
#define bitlock_yield_acquire(arr,pos) bitlock_acquire_block(arr,pos,sched_yield();,{})
|
||||
|
||||
// Block until we get the lock or someone else does
|
||||
// sets the memory pointed to by retptr to 1 if we got the lock, 0 otherwise
|
||||
#define bitlock_try_acquire(arr,pos,retptr) do { \
|
||||
*(retptr) = 1; /* default to success, set to zero if locked */ \
|
||||
bitlock_acquire_block(arr,pos,{*(retptr)=0;break;},if(!*(retptr)){break;}); \
|
||||
} while(0)
|
||||
|
||||
/*
|
||||
* Byteswapping
|
||||
*/
|
||||
|
||||
/* clang uses these to check for features */
|
||||
#ifndef __has_feature
|
||||
#define __has_feature(x) 0
|
||||
#endif
|
||||
|
||||
#ifndef __has_builtin
|
||||
#define __has_builtin(x) 0
|
||||
#endif
|
||||
|
||||
/* GCC versions < 4.3 do not have __builtin_bswapX() */
|
||||
#if ( defined(__clang__) && !__has_builtin(__builtin_bswap64) ) || \
|
||||
( !defined(__clang__) && defined(__GNUC__) && defined(__GNUC_MINOR__) && \
|
||||
( (__GNUC__ < 4) || (__GNUC__ == 4 && __GNUC_MINOR__ < 3)) )
|
||||
#define byteswap64(x) ( (((uint64_t)(x) << 56)) | \
|
||||
(((uint64_t)(x) << 40) & 0xff000000000000ULL) | \
|
||||
(((uint64_t)(x) << 24) & 0xff0000000000ULL) | \
|
||||
(((uint64_t)(x) << 8) & 0xff00000000ULL) | \
|
||||
(((uint64_t)(x) >> 8) & 0xff000000ULL) | \
|
||||
(((uint64_t)(x) >> 24) & 0xff0000ULL) | \
|
||||
(((uint64_t)(x) >> 40) & 0xff00ULL) | \
|
||||
(((uint64_t)(x) >> 56)) )
|
||||
|
||||
#define byteswap32(x) ( (((uint32_t)(x) << 24)) | \
|
||||
(((uint32_t)(x) << 8) & 0xff0000U) | \
|
||||
(((uint32_t)(x) >> 8) & 0xff00U) | \
|
||||
(((uint32_t)(x) >> 24)) )
|
||||
|
||||
/* uint16_t type might be bigger than 2 bytes, so need to mask */
|
||||
#define byteswap16(x) ( (((uint16_t)(x) & 0xff) << 8) | \
|
||||
(((uint16_t)(x) >> 8) & 0xff) )
|
||||
#else
|
||||
#define byteswap64(x) __builtin_bswap64(x)
|
||||
#define byteswap32(x) __builtin_bswap64(x)
|
||||
#define byteswap16(x) __builtin_bswap64(x)
|
||||
#endif
|
||||
|
||||
#endif /* BITLOCK_H_ */
|
||||
Reference in New Issue
Block a user