-
Notifications
You must be signed in to change notification settings - Fork 129
/
Sequence.swift
230 lines (194 loc) · 8.67 KB
/
Sequence.swift
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
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2022-2023 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
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
@available(macOS 9999, iOS 9999, tvOS 9999, watchOS 9999, *)
extension PredicateExpressions {
public struct SequenceContains<
LHS : PredicateExpression,
RHS : PredicateExpression
> : PredicateExpression
where
LHS.Output : Sequence,
LHS.Output.Element : Equatable,
RHS.Output == LHS.Output.Element
{
public typealias Output = Bool
public let sequence: LHS
public let element: RHS
public init(sequence: LHS, element: RHS) {
self.sequence = sequence
self.element = element
}
public func evaluate(_ bindings: PredicateBindings) throws -> Bool {
let a = try sequence.evaluate(bindings)
let b = try element.evaluate(bindings)
return a.contains(b)
}
}
public struct SequenceContainsWhere<
LHS : PredicateExpression,
RHS : PredicateExpression
> : PredicateExpression
where
LHS.Output : Sequence,
RHS.Output == Bool
{
public typealias Element = LHS.Output.Element
public typealias Output = Bool
public let sequence: LHS
public let test: RHS
public let variable: Variable<Element>
public init(_ sequence: LHS, builder: (Variable<Element>) -> RHS) {
self.variable = Variable()
self.sequence = sequence
self.test = builder(variable)
}
public func evaluate(_ bindings: PredicateBindings) throws -> Output {
var mutableBindings = bindings
return try sequence.evaluate(bindings).contains {
mutableBindings[variable] = $0
return try test.evaluate(mutableBindings)
}
}
}
public struct SequenceAllSatisfy<
LHS : PredicateExpression,
RHS : PredicateExpression
> : PredicateExpression
where
LHS.Output : Sequence,
RHS.Output == Bool
{
public typealias Element = LHS.Output.Element
public typealias Output = Bool
public let sequence: LHS
public let test: RHS
public let variable: Variable<Element>
public init(_ sequence: LHS, builder: (Variable<Element>) -> RHS) {
self.variable = Variable()
self.sequence = sequence
self.test = builder(variable)
}
public func evaluate(_ bindings: PredicateBindings) throws -> Output {
var mutableBindings = bindings
return try sequence.evaluate(bindings).allSatisfy {
mutableBindings[variable] = $0
return try test.evaluate(mutableBindings)
}
}
}
public struct SequenceStartsWith<
Base : PredicateExpression,
Prefix : PredicateExpression
> : PredicateExpression
where
Base.Output : Sequence,
Prefix.Output : Sequence,
Base.Output.Element == Prefix.Output.Element,
Base.Output.Element : Equatable
{
public typealias Output = Bool
public let base: Base
public let prefix: Prefix
public init(base: Base, prefix: Prefix) {
self.base = base
self.prefix = prefix
}
public func evaluate(_ bindings: PredicateBindings) throws -> Bool {
try base.evaluate(bindings).starts(with: try prefix.evaluate(bindings))
}
}
@_disfavoredOverload
public static func build_contains<LHS, RHS>(_ lhs: LHS, _ rhs: RHS) -> SequenceContains<LHS, RHS> {
SequenceContains(sequence: lhs, element: rhs)
}
public static func build_contains<LHS, RHS>(_ lhs: LHS, where builder: (Variable<LHS.Output.Element>) -> RHS) -> SequenceContainsWhere<LHS, RHS> {
SequenceContainsWhere(lhs, builder: builder)
}
public static func build_allSatisfy<LHS, RHS>(_ lhs: LHS, _ builder: (Variable<LHS.Output.Element>) -> RHS) -> SequenceAllSatisfy<LHS, RHS> {
SequenceAllSatisfy(lhs, builder: builder)
}
public static func build_starts<Base, Prefix>(_ base: Base, with prefix: Prefix) -> SequenceStartsWith<Base, Prefix> {
SequenceStartsWith(base: base, prefix: prefix)
}
}
@available(macOS 9999, iOS 9999, tvOS 9999, watchOS 9999, *)
extension PredicateExpressions.SequenceContains : StandardPredicateExpression where LHS : StandardPredicateExpression, RHS : StandardPredicateExpression {}
@available(macOS 9999, iOS 9999, tvOS 9999, watchOS 9999, *)
extension PredicateExpressions.SequenceContainsWhere : StandardPredicateExpression where LHS : StandardPredicateExpression, RHS : StandardPredicateExpression {}
@available(macOS 9999, iOS 9999, tvOS 9999, watchOS 9999, *)
extension PredicateExpressions.SequenceAllSatisfy : StandardPredicateExpression where LHS : StandardPredicateExpression, RHS : StandardPredicateExpression {}
@available(macOS 9999, iOS 9999, tvOS 9999, watchOS 9999, *)
extension PredicateExpressions.SequenceStartsWith : StandardPredicateExpression where Base : StandardPredicateExpression, Prefix : StandardPredicateExpression {}
@available(macOS 9999, iOS 9999, tvOS 9999, watchOS 9999, *)
extension PredicateExpressions.SequenceContains : Codable where LHS : Codable, RHS : Codable {
public func encode(to encoder: Encoder) throws {
var container = encoder.unkeyedContainer()
try container.encode(sequence)
try container.encode(element)
}
public init(from decoder: Decoder) throws {
var container = try decoder.unkeyedContainer()
sequence = try container.decode(LHS.self)
element = try container.decode(RHS.self)
}
}
@available(macOS 9999, iOS 9999, tvOS 9999, watchOS 9999, *)
extension PredicateExpressions.SequenceContainsWhere : Codable where LHS : Codable, RHS : Codable {
public func encode(to encoder: Encoder) throws {
var container = encoder.unkeyedContainer()
try container.encode(sequence)
try container.encode(test)
try container.encode(variable)
}
public init(from decoder: Decoder) throws {
var container = try decoder.unkeyedContainer()
sequence = try container.decode(LHS.self)
test = try container.decode(RHS.self)
variable = try container.decode(PredicateExpressions.Variable<Element>.self)
}
}
@available(macOS 9999, iOS 9999, tvOS 9999, watchOS 9999, *)
extension PredicateExpressions.SequenceAllSatisfy : Codable where LHS : Codable, RHS : Codable {
public func encode(to encoder: Encoder) throws {
var container = encoder.unkeyedContainer()
try container.encode(sequence)
try container.encode(test)
try container.encode(variable)
}
public init(from decoder: Decoder) throws {
var container = try decoder.unkeyedContainer()
sequence = try container.decode(LHS.self)
test = try container.decode(RHS.self)
variable = try container.decode(PredicateExpressions.Variable<Element>.self)
}
}
@available(macOS 9999, iOS 9999, tvOS 9999, watchOS 9999, *)
extension PredicateExpressions.SequenceStartsWith : Codable where Base : Codable, Prefix : Codable {
public func encode(to encoder: Encoder) throws {
var container = encoder.unkeyedContainer()
try container.encode(base)
try container.encode(prefix)
}
public init(from decoder: Decoder) throws {
var container = try decoder.unkeyedContainer()
self.base = try container.decode(Base.self)
self.prefix = try container.decode(Prefix.self)
}
}
@available(macOS 9999, iOS 9999, tvOS 9999, watchOS 9999, *)
extension PredicateExpressions.SequenceContains : Sendable where LHS : Sendable, RHS : Sendable {}
@available(macOS 9999, iOS 9999, tvOS 9999, watchOS 9999, *)
extension PredicateExpressions.SequenceContainsWhere : Sendable where LHS : Sendable, RHS : Sendable {}
@available(macOS 9999, iOS 9999, tvOS 9999, watchOS 9999, *)
extension PredicateExpressions.SequenceAllSatisfy : Sendable where LHS : Sendable, RHS : Sendable {}
@available(macOS 9999, iOS 9999, tvOS 9999, watchOS 9999, *)
extension PredicateExpressions.SequenceStartsWith : Sendable where Base : Sendable, Prefix : Sendable {}