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
|
// RUN: %target-swift-frontend %s -emit-ir
protocol SourceTargetTransformable {
associatedtype Source
associatedtype Target
// FIXME: should really be a typealias once we support that
associatedtype Transformer = (Source) -> Target
}
struct PiecewiseTransformedIteratorOf<
Source,
Target,
SourceIterator: IteratorProtocol,
TransformerIterator: IteratorProtocol,
Transformable: SourceTargetTransformable
>
: IteratorProtocol
where
Transformable.Source == Source,
Transformable.Target == Target,
SourceIterator.Element == Source,
TransformerIterator.Element == Transformable.Transformer {
typealias Element = Target
var sourceIterator: SourceIterator
var transformerIterator: TransformerIterator
mutating func next() -> Element? {
let source: Transformable.Source? = sourceIterator.next()
if let source: Transformable.Source = source {
let transformer: Transformable.Transformer? = transformerIterator.next()
if let transformer: Transformable.Transformer = transformer {
let tfunc: ((Source) -> Target)? = transformer as? ((Source) -> Target)
if let tfunc = tfunc {
return tfunc(source)
}
}
}
return nil
}
}
struct PiecewiseTransformedSequenceOf<
SourceSequence: Sequence,
TransformerSequence: Sequence,
Transformable: SourceTargetTransformable
>: Sequence
where
SourceSequence.Iterator.Element == Transformable.Source,
TransformerSequence.Iterator.Element == Transformable.Transformer {
typealias Source = SourceSequence.Iterator.Element
typealias Target = Transformable.Target
typealias Iterator = PiecewiseTransformedIteratorOf<Source, Target, SourceSequence.Iterator, TransformerSequence.Iterator, Transformable>
let inputs: SourceSequence
let transformers: TransformerSequence
init(inputs: SourceSequence, transformers: TransformerSequence) {
self.inputs = inputs
self.transformers = transformers
}
func makeIterator() -> Iterator {
return PiecewiseTransformedIteratorOf(sourceIterator: inputs.makeIterator(), transformerIterator: transformers.makeIterator())
}
}
|