扩展就是为一个已有的类、结构体、枚举类型或者协议类型添加新功能。
1. 扩展语法
extension SomeType {
// code
}
可以通过扩展来扩展一个已有类型,使其采纳一个或多个协议:
extension SomeType: SomeProtocol, AnotherProtocol {
// protocol code
}
2. 计算型属性
extension Double {
var km: Double { return self * 1000 }
var m: Double { return self }
var cm: Double { return self / 100.0 }
var mm: Double { return self / 1000.0 }
var ft: Double { return self / 3.28084 }
}
let oneInch = 25.4.mm
print("One inch is \(oneInch) meters")
// print "One inch is 0.0254 meters"
注意:
扩展可以添加新的计算属性,但是不可以添加存储属性,也不可以为已有属性添加属性观察器。
3. 构造器
struct Size {
var width = 0.0, height = 0.0
}
struct Point {
var x = 0.0, y = 0.0
}
struct Rect {
var origin = Point()
var size = Size()
}
extension Rect {
init(center: Point, size: Size) {
let originX = center.x - (size.width / 2)
let originY = center.y - (size.height / 2)
self.init(origin: Point(x: originX, y: originY), size: size)
}
}
这个新的构造器首先根据提供的center
和size
的值计算一个合适的原点。然后调用该结构体的逐一成员构造器init(origin:size:)
,该构造器将新的原点和大小的值保存到了相应的属性中:
let centerRect = Rect(center: Point(x: 4.0, y: 4.0), size: Size(width: 3.0, height: 3.0))
注意:
如果你使用扩展提供了一个新的构造器,你依旧有责任确保构造过程能够让实例完全初始化。
4. 方法
extension Int {
func repetitions(task:() -> Void) {
for _ in 0..<self {
task()
}
}
}
3.repetitions {
print("Hello")
}
// print "Hello"
// print "Hello"
// print "Hello"
4.1 可变实例方法
extension Int {
mutating func square() {
self = self * self
}
}
var someInt = 3
someInt.square()
print(someInt)
// print "9"
5. 下标
extension Int {
subscript(digitIndex: Int) -> Int {
var decimalBase = 1
for _ in 0..<digitIndex {
decimalBase *= 10
}
return (self / decimalBase) % 10
}
}
print(9876543210[2])
// print "2"
print(9876543210[3])
// print "3"
print(9876543210[15])
// print "0"
6. 嵌套类型
扩展可以为已有的类、结构体和枚举添加新的嵌套类型。
extension Int {
enum Kind {
case Negative, Zero, Positive
}
var kind: Kind {
switch self {
case 0:
return .Zero
case let x where x > 0:
return .Positive
default:
return .Negative
}
}
}
func printIntegerKinds(_ numbers: [Int]) {
for number in numbers {
switch number.kind {
case .Negative:
print("-", terminator:"")
case .Zero:
print("0", terminator:"")
case .Positive:
print("+", terminator:"")
}
}
print("")
}
printIntegerKinds([3, 19, -27, 0, -6, 0, 7])
// print "++-0-0+"