Skip to content

Instantly share code, notes, and snippets.

@khanlou
Created November 15, 2016 18:19
Show Gist options
  • Save khanlou/246e62115e72555ef90f64efb8ffb991 to your computer and use it in GitHub Desktop.
Save khanlou/246e62115e72555ef90f64efb8ffb991 to your computer and use it in GitHub Desktop.
count and allValues on Int-based enums
protocol CaseCountable { }
extension CaseCountable where Self : RawRepresentable, Self.RawValue == Int {
static var count: Int {
var count = 0
while let _ = Self(rawValue: count) { count+=1 }
return count
}
static var allValues: [Self] {
return (0..<count).flatMap({ Self(rawValue: $0) })
}
}
enum Someenum: Int {
case one, two, three, four
}
extension Someenum: CaseCountable { }
Someenum.count
Someenum.allValues
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment