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
|
//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift Collections open source project
//
// Copyright (c) 2022 - 2024 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
//
//===----------------------------------------------------------------------===//
extension BitArray {
/// Set the count of this bit array to `count`, by either truncating it or
/// extending it by appending the specified Boolean value (`false` by default).
///
/// - Parameter count: The desired count of the bit array on return.
/// - Parameter padding: The value to append the array if the array is
/// currently too short.
public mutating func truncateOrExtend(
toCount count: Int,
with padding: Bool = false
) {
precondition(count >= 0, "Negative count")
if count < _count {
_removeLast(self.count - count)
} else if count > _count {
_extend(by: count - self.count, with: padding)
}
}
}
extension BitArray {
public mutating func insert(
repeating value: Bool,
count: Int,
at index: Int
) {
precondition(count >= 0, "Can't insert a negative number of values")
precondition(index >= 0 && index <= count, "Index out of bounds")
guard count > 0 else { return }
_extend(by: count, with: false)
_copy(from: index ..< self.count - count, to: index + count)
fill(in: index ..< index + count, with: value)
}
public mutating func append(repeating value: Bool, count: Int) {
precondition(count >= 0, "Can't append a negative number of values")
guard count > 0 else { return }
_extend(by: count, with: value)
}
}
|
