带枚举的Swift协议扩展

时间:2015-11-06 16:19:55

标签: swift enums protocols

所以我有几个代表各种单位系统的枚举:

enum MassUnit:Double{
     case Pound = 453.59237, Ounce = 28.349523125, Gram = 1.0, Kilogram = 1000.0;
}
enum VolumeUnit:Double{
    case Teaspoon = 1, Tablespoon = 3, Cup = 48, Pint = 96, Quart = 192, Gallon = 768, Liter = 202.884136211, Milliliter = 0.202884136211
}
enum TimeUnit:Double{
    case Second = 1, Minute = 60, Hour = 3600, Day = 86400, Week = 604800, Year = 31536000
}

我想做的是能够从一个单位转换到另一个单位,例如从一年到几秒。为此,我确保我的枚举的原始值对应于转换乘数,例如1 Min = 60秒。因此,给定某个单位的x量,转换就是

x * rawValue1 / rawValue2 // rawValue2 = rawValue of desired unit.

虽然转换很简单,但我希望能够提高效率并使用协议:

protocol Convertable{
    func convert(inputAmount inputAmount:Double, outputUnit:Self)->Double;
}

然后,我可以扩展枚举:

extension TimeUnit:Convertable{
    func convert(inputAmount inputAmount: Double, outputUnit: TimeUnit) -> Double {
        return inputAmount * self.rawValue / outputUnit.rawValue;
    }
}

然后我可以像这样转换:

TimeUnit.Year.convert(inputAmount: 2.54, outputUnit: .Second)
// returns 80101440

这很好,但是,根据我想要转换的单位数量,会有很多重复的相同代码。

所以,我想做的是以某种方式使用协议扩展。

extension Convertable{
    func convert(inputAmount inputAmount: Double, outputUnit: Self) -> Double {
        return inputAmount * self.rawValue / outputUnit.rawValue;// Compile error...
    }
}

这是我遇到麻烦的地方,输出单元被声明为self,它对rawValue一无所知。

有什么想法吗?

1 个答案:

答案 0 :(得分:11)

当我问这个问题的时候,答案来了我:Convertible协议还需要一个rawValue变量:

protocol Convertable{
    var rawValue:Double{get}
    func convert(inputAmount inputAmount:Double, outputUnit:Self)->Double;
}

这样,引用self.rawValue是已知的,因为枚举已经有了rawValue,所以不需要额外的工作:

enum MassUnit:Double, Convertable{
     case Pound = 453.59237, Ounce = 28.349523125, Gram = 1.0, Kilogram = 1000.0;
}

enum VolumeUnit:Double, Convertable{
    case Teaspoon = 1, Tablespoon = 3, Cup = 48, Pint = 96, Quart = 192, Gallon = 768, Liter = 202.884136211, Milliliter = 0.202884136211
}

enum TimeUnit:Double, Convertable{
    case Second = 1, Minute = 60, Hour = 3600, Day = 86400, Week = 604800, Year = 31536000
}

然后使用转换器:

TimeUnit.Year.convert(inputAmount: 2.54, outputUnit: .Second) // 80101440
MassUnit.Gram.convert(inputAmount: 20.0, outputUnit: .Ounce) //0.70547
VolumeUnit.Pint.convert(inputAmount: 0.2, outputUnit: .Tablespoon)// 6.4000

即便是这样,在奇怪的情况下,转换不能像这样工作,我可以用我自己的实现覆盖convert函数,例如在Temperature:

enum TemperatureUnit:Double, Convertable{
    case Celsius, Fahrenheit

    func convert(inputAmount inputAmount: Double, outputUnit: TemperatureUnit) -> Double {
        if self == outputUnit {
            return inputAmount;
        } else if self == .Celsius {
            return inputAmount * 9.0/5.0 + 32.0
        } else {
            return (inputAmount - 32.0) * 5.0 / 9.0;
        }
    }
}

TemperatureUnit.Celsius.convert(inputAmount: 3, outputUnit: .Fahrenheit) // 37.4
TemperatureUnit.Fahrenheit.convert(inputAmount: 0, outputUnit: .Celsius) // -17.7778

美丽!

相关问题