2015-08-25 45 views
3

是否可以通过编程的方式找出Enum在Swift 2中有多少个“案例”并迭代它们?Enum的反省和迭代

此代码不能编译,但它给哟我想要实现一个想法:

enum HeaderStyles{ 

    case h1 
    case h2 
    case h3 
} 

for item in HeaderStyles{ 
    print(item) 
} 
+1

枚举必须在编译时声明,因此您 - 开发人员 - 确切知道枚举具有多少个案例。 ;-) – vadian

回答

3

您可以编写一个提供该迭代​​功能的通用结构。在下面的例子中,枚举原始值必须从0开始(默认情况下它是这样做的)并且在原始值之间没有间隔(不能有原始值,例如0,1,2,3,5-4缺失)

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) 
} 
5

遍历所有的情况下,最简单的方法是使返回的一个计算的属性他们的Array

enum SomeEnum { 
    case Zero, One, Two 

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

如果你想自动阵列可以使用Int作为rawValue所以你没有,如果你添加一个额外的case更改任何代码:

斯威夫特3/4:++--被拆除和anyGenerator更名为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 
     }) 
    } 
} 

斯威夫特2

enum SomeEnum: Int { 
    case Zero, One, Two 

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

在这两种情况下,你会使用这样的:

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