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
|
//===----------------------------------------------------------------------===//
//
// 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 swift(>=5.8)
@available(macOS 13.3, iOS 16.4, watchOS 9.4, tvOS 16.4, *)
extension BigString {
struct Builder {
typealias _Chunk = BigString._Chunk
typealias _Ingester = BigString._Ingester
typealias _Rope = BigString._Rope
var base: _Rope.Builder
var suffixStartState: _CharacterRecognizer
var prefixEndState: _CharacterRecognizer
init(
base: _Rope.Builder,
prefixEndState: _CharacterRecognizer,
suffixStartState: _CharacterRecognizer
) {
self.base = base
self.suffixStartState = suffixStartState
self.prefixEndState = prefixEndState
}
init() {
self.base = _Rope.Builder()
self.suffixStartState = _CharacterRecognizer()
self.prefixEndState = _CharacterRecognizer()
}
}
}
@available(macOS 13.3, iOS 16.4, watchOS 9.4, tvOS 16.4, *)
extension Rope<BigString._Chunk>.Builder {
internal func _breakState() -> _CharacterRecognizer {
let chars = self.prefixSummary.characters
assert(self.isPrefixEmpty || chars > 0)
let metric = BigString._CharacterMetric()
var state = _CharacterRecognizer()
_ = self.forEachElementInPrefix(from: chars - 1, in: metric) { chunk, i in
if let i {
state = .init(partialCharacter: chunk.string[i...])
} else {
state.consumePartialCharacter(chunk.string[...])
}
return true
}
return state
}
}
@available(macOS 13.3, iOS 16.4, watchOS 9.4, tvOS 16.4, *)
extension BigString.Builder {
mutating func append(_ str: __owned some StringProtocol) {
append(Substring(str))
}
mutating func append(_ str: __owned String) {
append(str[...])
}
mutating func append(_ str: __owned Substring) {
guard !str.isEmpty else { return }
var ingester = _Ingester(str, startState: self.prefixEndState)
if var prefix = base._prefix._take() {
if let slice = ingester.nextSlice(maxUTF8Count: prefix.value.availableSpace) {
prefix.value._append(slice)
}
self.base._prefix = prefix
}
while let next = ingester.nextChunk() {
base.insertBeforeTip(next)
}
self.prefixEndState = ingester.state
}
mutating func append(_ newChunk: __owned _Chunk) {
var state = _CharacterRecognizer()
append(newChunk, state: &state)
}
mutating func append(_ newChunk: __owned _Chunk, state: inout _CharacterRecognizer) {
var newChunk = newChunk
newChunk.resyncBreaksFromStartToEnd(old: &state, new: &self.prefixEndState)
self.base.insertBeforeTip(newChunk)
}
mutating func append(_ other: __owned BigString) {
var state = _CharacterRecognizer()
append(other._rope, state: &state)
}
mutating func append(_ other: __owned BigString, in range: Range<BigString.Index>) {
let extract = BigString(other, in: range, state: &self.prefixEndState)
self.base.insertBeforeTip(extract._rope)
}
mutating func append(_ other: __owned _Rope, state: inout _CharacterRecognizer) {
guard !other.isEmpty else { return }
var other = BigString(_rope: other)
other._rope.resyncBreaksToEnd(old: &state, new: &self.prefixEndState)
self.base.insertBeforeTip(other._rope)
}
mutating func append(from ingester: inout _Ingester) {
//assert(ingester.state._isKnownEqual(to: self.prefixEndState))
if var prefix = base._prefix._take() {
if let first = ingester.nextSlice(maxUTF8Count: prefix.value.availableSpace) {
prefix.value._append(first)
}
base._prefix = prefix
}
let suffixCount = base._suffix?.value.utf8Count ?? 0
while let chunk = ingester.nextWellSizedChunk(suffix: suffixCount) {
base.insertBeforeTip(chunk)
}
precondition(ingester.isAtEnd)
self.prefixEndState = ingester.state
}
}
@available(macOS 13.3, iOS 16.4, watchOS 9.4, tvOS 16.4, *)
extension BigString.Builder {
mutating func finalize() -> BigString {
// Resync breaks in suffix.
_ = base.mutatingForEachSuffix { chunk in
chunk.resyncBreaksFromStart(old: &suffixStartState, new: &prefixEndState)
}
// Roll it all up.
let rope = self.base.finalize()
let string = BigString(_rope: rope)
string._invariantCheck()
return string
}
}
#endif
|
