add AppendBit()

NethermindEth · Jan 19, 2025 · 46de9ca · 46de9ca
1 parent 7c4aeb3
commit 46de9ca
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/core/trie/bitarray.go b/core/trie/bitarray.go
@@ -13,8 +13,9 @@ import (
 )
 
 const (
-	maxUint64 = uint64(math.MaxUint64) // 0xFFFFFFFFFFFFFFFF
-	maxUint8  = uint8(math.MaxUint8)
+	maxUint64       = uint64(math.MaxUint64) // 0xFFFFFFFFFFFFFFFF
+	maxUint8        = uint8(math.MaxUint8)
+	MaxBitArraySize = 33 // (1 + 4 * 8) bytes
 )
 
 var emptyBitArray = new(BitArray)
@@ -318,6 +319,11 @@ func (b *BitArray) Append(x, y *BitArray) *BitArray {
 	return b.Lsh(x, y.len).Or(b, y)
 }
 
+// Sets the bit array to the concatenation of x and a single bit.
+func (b *BitArray) AppendBit(x *BitArray, bit uint8) *BitArray {
+	return b.Append(b, new(BitArray).SetBit(bit))
+}
+
 // Sets the bit array to x | y and returns the bit array.
 func (b *BitArray) Or(x, y *BitArray) *BitArray {
 	b.words[0] = x.words[0] | y.words[0]