swift - Introspección e iteración en un Enum

enums swift2

Question

¿Es posible averiguar programáticamente cuántos "casos" tiene un Enum en Swift 2 e iterar sobre ellos?

Este código no se compila, pero le da una idea de lo que estoy tratando de lograr:

enum HeaderStyles{

    case h1
    case h2
    case h3
}

for item in HeaderStyles{
    print(item)
}

Answer 1

Puede escribir una estructura genérica que proporcione esa capacidad de iteración. En el siguiente ejemplo, los valores enum raw deben comenzar con 0 (lo hace de forma predeterminada) y no tienen espacios entre los valores sin procesar (no puede haber valores en bruto como 0,1,2,3,5 - 4 falta)

public struct EnumGenerator<T> : GeneratorType, SequenceType {
    private let enumInit: Int -> T?
    private var i: Int = 0
    public mutating func next() -> T? { return enumInit(i++) }
    public init(_ initFunc: Int -> T?){ self.enumInit = initFunc}
}

enum HeaderStyles:Int{
    case h1
    case h2
    case h3
}

for item in EnumGenerator(HeaderStyles.init){
    print(item)
}

Answer 2

La forma más fácil de iterar en todos los casos es hacer una propiedad calculada que devuelve una Array de ellos:

enum SomeEnum {
    case Zero, One, Two

    static var allCases: [SomeEnum] {
        return [.Zero, .One, .Two]
    }
}

Si desea una matriz automática, puede usar Int como rawValue para no tener que cambiar ningún código si agrega un case adicional:

Swift 3/4: ++ y -- se eliminaron y anyGenerator se renombró a AnyIterator

enum SomeEnum: Int {
    case Zero, One, Two

    static var allCases: [SomeEnum] {
        var i = 0
        return Array(AnyIterator{
            let newEnum = SomeEnum(rawValue: i)
            i += 1
            return newEnum
        })
    }
}

Swift 2

enum SomeEnum: Int {
    case Zero, One, Two

    static var allCases: [SomeEnum] {
        var i = 0
        return Array(anyGenerator{ SomeEnum(rawValue: i++) })
    }
}

En ambos casos, lo usarías así:

for num in SomeEnum.allCases {
    // ...
}