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
|
//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift Collections open source project
//
// Copyright (c) 2023 - 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
//
//===----------------------------------------------------------------------===//
#if !COLLECTIONS_SINGLE_MODULE
import InternalCollectionsUtilities
#endif
extension Rope._UnmanagedLeaf: CustomStringConvertible {
@usableFromInline
internal var description: String {
_addressString(for: _ref.toOpaque())
}
}
extension Rope {
@inlinable
public var _nodeCount: Int {
_root?.nodeCount ?? 0
}
}
extension Rope._Node {
@inlinable
internal var nodeCount: Int {
guard !isLeaf else { return 1 }
return readInner { $0.children.reduce(into: 1) { $0 += $1.nodeCount } }
}
}
extension Rope {
public func _dump(
heightLimit: Int = .max,
firstPrefix: String = "",
restPrefix: String = ""
) {
guard _root != nil else {
print("<Empty>")
return
}
root.dump(heightLimit: heightLimit, firstPrefix: firstPrefix, restPrefix: restPrefix)
}
}
extension Rope._Node: CustomStringConvertible {
@usableFromInline
internal var description: String {
"""
\(height > 0 ? "Inner@\(height)" : "Leaf")(\
at: \(_addressString(for: object)), \
summary: \(summary), \
childCount: \(childCount)/\(Summary.maxNodeSize))
"""
}
}
extension Rope._Node {
@usableFromInline
internal func dump(
heightLimit: Int = .max,
firstPrefix: String = "",
restPrefix: String = ""
) {
print("\(firstPrefix)\(description)")
guard heightLimit > 0 else { return }
if height > 0 {
readInner {
let c = $0.children
for slot in 0 ..< c.count {
c[slot].dump(
heightLimit: heightLimit - 1,
firstPrefix: "\(restPrefix)\(slot): ",
restPrefix: "\(restPrefix) ")
}
}
} else {
readLeaf {
let c = $0.children
for slot in 0 ..< c.count {
print("\(restPrefix)\(slot): \(c[slot])")
}
}
}
}
}
|