Ajouter “en” soutien pour itérer sur Swift classes personnalisées

Dire que vous avez une classe "Voiture" que vous voulez être en mesure de parcourir à l'aide d'une boucle for..in:

let cars = Cars()

for car in cars {
    println(car.name)
}

La façon la plus simple est d'utiliser AnyGenerator avec les classes comme ceci:

class Car {
    var name : String
    init(name : String) {
        self.name = name
    }
}

class Cars : SequenceType {

    var carList : [Car] = []

    func generate() -> AnyGenerator<Car> {
        //keep the index of the next car in the iteration
        var nextIndex = carList.count-1

        //Construct a AnyGenerator<Car> instance, passing a closure that returns the next car in the iteration
        return anyGenerator {
            if (nextIndex < 0) {
                return nil
            }
            return self.carList[nextIndex--]
        }
    }
}

Essayer un travail complet de l'échantillon ajouter les deux classes ci-dessus, puis essayez de les utiliser comme cela, l'ajout d'un couple d'items de test:

    let cars = Cars()

    cars.carList.append(Car(name: "Honda"))
    cars.carList.append(Car(name: "Toyota"))

    for car in cars {
        println(car.name)
    }

Ça y est, simple.

Plus d'infos: http://lillylabs.no/2014/09/30/make-iterable-swift-collection-type-sequencetype

@Matt Gibson est correct. Cependant, je voudrais ajouter plus d'informations pour référence future.

De Pointe Swift:

Ce code:

for x in someSequence {
    ...
}

Est converti en ceci:

var __g = someSequence.generate()
while let x = __g.next() {
    ...
}

Par conséquent, on doit adopter la Séquence, ce qui donne à la classe generate() et next(). Voici ces protocoles:

protocol Generator {
    typealias Element
    mutating func next() -> Element?
}
protocol Sequence {
    typealias GeneratorType : Generator
    func generate() -> GeneratorType
}

Accepté la réponse est correcte, et jusqu'à récemment, était la façon de traiter cette question. Toutefois, compte tenu de l'introduction des Extensions de Protocole dans Swift 2.0, plutôt que de la conformité à SequenceTypeet la mise en œuvre de func generate() -> GeneratorOf<Car> il y a maintenant une classe de base abstraite qui gère la mise en œuvre de cette fonctionnalité pour vous appelé AnyGenerator<T> (voir Apple docs) depuis GeneratorOf<T> n'existe plus.

Ce que cela signifie est que vous pouvez tout simplement sous-classe de cette classe de base abstraite, et en faisant faire, hériter de toutes les fonctionnalités du protocole précité de conformité:

class Cars: AnyGenerator<Car> {

     private var carList = [Car]()
     private var currentIndex:Int

     ...

}

Un alors besoin que de remplacer la next() méthode déclarée par le GeneratorType protocole (qui AnyGenerator<T> est également conforme à l') afin de définir le degré de l'itération de comportement:

class Cars: AnyGenerator<Car> {

     private var carList = [Car]()
     private var currentIndex:Int

     override func next() -> Car? {

         if (currentIndex < self.carList.count) {
            currentIndex++
            return self.carList[currentIndex-1]
         } else {
            currentIndex = 0;
            return nil
         }
     }

}