为任意的、基于Int的枚举定义Swift协议

Question

我有这个表示颜色的枚举，我添加了几种方法来方便地基于对原始值的算术运算获得新实例：

enum Color : Int
{
    case Red = 0
    case Green
    case Blue
    case Cyan
    case Magenta
    case Yellow

    static func random() -> Color
    {
        return Color(rawValue: Int(arc4random_uniform(6)))!
    }
    func shifted(by offset:Int) -> Color
    {
        return Color(rawValue: (self.rawValue + offset) % 6)!
        // Cyclic: wraps around
    }
}

（这让人回想起旧的枚举只是int常量）

问题是，我还有其他几个基于int的枚举，我想在其中引入类似的功能，但不会重复代码。

我认为我应该在RawRepresentable上定义一个协议扩展，其中RawValue == Int:

extension RawRepresentable where RawValue == Int
{

但我对语法的理解到此为止。

理想情况下，我希望需要一个返回事例数的静态方法，并提供上面random()和shifted(_:)的默认实现（而不是这里的硬编码6）。

结论：我接受了Zoff Dino的回答。尽管Rob Napier给出的答案正是我想要的，但事实证明，我想要的并不是最优雅的设计，而另一个答案则提出了一个更好的方法。尽管如此，我还是对这两个答案投了赞成票；谢谢大家。

Answer 1

您应该扩展您的自定义协议，而不是RawRepresentable。试试这个：

protocol MyProtocol {
    static var maxRawValue : Int { get }
    static func random() ->  Self
    func shifted(by offset: Int) -> Self
}
enum Color : Int, MyProtocol
{
    case Red = 0
    case Green
    case Blue
    case Cyan
    case Magenta
    case Yellow
    // The maximum value of your Int enum
    static var maxRawValue: Int {
        return Yellow.rawValue
    }
}
extension MyProtocol where Self: RawRepresentable, Self.RawValue == Int {
    static func random() -> Self {
        let random = Int(arc4random_uniform(UInt32(Self.maxRawValue + 1)))
        return Self(rawValue: random)!
    }
    func shifted(by offset: Int) -> Self {
        return Self(rawValue: (self.rawValue + offset) % (Self.maxRawValue + 1))!
    }
}
let x = Color.random()
let y = x.shifted(by: 1)

Answer 2

您就快到了。你只需要Nate Cook的案件计数代码https://stackoverflow.com/a/27094913/97337.

extension RawRepresentable where RawValue == Int {
    // See http://natecook.com/blog/2014/10/loopy-random-enum-ideas/
    static var caseCount: Int {
        var max: Int = 0
        while let _ = self.init(rawValue: ++max) {}
        return max
    }
    static func random() -> Self {
        return Self(rawValue: Int(arc4random_uniform(UInt32(caseCount))))!
    }
    func shifted(by offset:Int) -> Self {
        return Self(rawValue: (self.rawValue + offset) % self.dynamicType.caseCount)!
        // Cyclic: wraps around
    }
}