루프 모드에서 컬렉션을 무한 반복하는 사용자 정의 반복자

Question

루프 모드에서 컬렉션을 무한히 반복하는 iterator를 찾고 있습니다. 따라서 컬렉션의 끝 인덱스에 도달하면 반복기는 시작 인덱스에서 요소를 반환해야합니다.

다음 해결책은 효과가있는 것 같지만 더 나은 방법으로 만들 수 있기를 바랍니다.

public struct LoopIterator<T: Collection>: IteratorProtocol { 

    private let collection: T 
    private var startIndexOffset: T.IndexDistance 

    public init(collection: T) { 
     self.collection = collection 
     startIndexOffset = 0 
    } 

    public mutating func next() -> T.Iterator.Element? { 
     guard !collection.isEmpty else { 
     return nil 
     } 
     let index = collection.index(collection.startIndex, offsetBy: startIndexOffset) 
     startIndexOffset += T.IndexDistance(1) 
     if startIndexOffset >= collection.count { 
     startIndexOffset = 0 
     } 
     return collection[index] 
    } 
} 

extension Array { 
    func makeLoopIterator() -> LoopIterator<Array> { 
     return LoopIterator(collection: self) 
    } 
} 

// Testing... 
// Will print: 1, 2, 3, 1, 2, 3 
var it = [1, 2, 3].makeLoopIterator() 
for _ in 0..<6 { 
    print(it.next()) 
} 

맞춤 반복기는 올바른 방법입니까? 무엇을 개선 할 수 있습니까?

감사합니다.

Answer 1

Swift 3 (사용중인)에서 색인은 컬렉션 자체에서 고급으로 작성됩니다.

public struct LoopIterator<Base: Collection>: IteratorProtocol { 

    private let collection: Base 
    private var index: Base.Index 

    public init(collection: Base) { 
     self.collection = collection 
     self.index = collection.startIndex 
    } 

    public mutating func next() -> Base.Iterator.Element? { 
     guard !collection.isEmpty else { 
      return nil 
     } 

     let result = collection[index] 
     collection.formIndex(after: &index) // (*) See discussion below 
     if index == collection.endIndex { 
      index = collection.startIndex 
     } 
     return result 
    } 
} 

이제 우리는 단순히 앞으로 인덱스를 이동하고 지금 말을 가리키는 경우, 처음으로 재설정 다음과 같이두고, 당신은이를 단순화 할 수 있습니다. count 또는 IndexDistance은 필요 없습니다.

여기서 Iterator는 모든 콜렉션 (및 모든 색인)에서 작동하므로 다소 애매한 경우 (특히 AnyIndex)의 성능을 향상시키기 위해 여기에 formIndex을 사용했습니다. 더 간단한 버전은 index = collection.index(after: index)이고 대부분의 경우 더 좋을 수 있습니다.

스위프트 3 지수에 대한 자세한 내용은 SE-0065을 참조하십시오.