[iOS - swift 공식 문서] 20. Extensions (확장)

Extensions

• 기존 class, struct, enum, protocol에 새로운 기능 추가
• 추가 가능한 것
  • computed property
  • instance method
  • 새로운 init
  • sucscript
  • 새로운 nested type

extension - Computed Property

• 주의: stored property는 불가
• literal값에도 extension이 가능

extension Double {
    var km: Double { return self * 1_000.0 }
    var m: Double { return self }
    var cm: Double { return self / 100.0 }
    var mm: Double { return self / 1_000.0 }
    var ft: Double { return self / 3.28084 }
}
let oneInch = 25.4.mm
print("One inch is \(oneInch) meters")
// Prints "One inch is 0.0254 meters"

let threeFeet = 3.ft
print("Three feet is \(threeFeet) meters")
// Prints "Three feet is 0.914399970739201 meters"

let aMarathon = 42.km + 195.m
print("A marathon is \(aMarathon) meters long")
// Prints "A marathon is 42195.0 meters long"

extension - Initilizer (이니셜라이저)

• extension으로 convenience init은 추가가 가능하지만 클래스에 designated init이나 deinit은 추가 불가
• designated init과 deinit은 반드시 original class에서 구현 가능
• 단, strcut에서 init을 사용자가 지정하지 않은 경우(memberwise init인 경우) extension으로 init추가 가능

// struct에서 init을 지정해주지 않은 경우 > extension으로 init추가 가능

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

// init 사용
let centerRect = Rect(center: Point(x: 4.0, y: 4.0),
                      size: Size(width: 3.0, height: 3.0))
// centerRect's origin is (2.5, 2.5) and its size is (3.0, 3.0)

extension - Method

• 기존 type에 instance method 및 type method 추가 가능

// instance method인 repetitions 추가

extension Int {
    func repetitions(task: () -> Void) {
        for _ in 0..<self {
            task()
        }
    }
}

• extension으로 instance자체를 수정 가능 - mutating 키워드 사용

extension Int {
    mutating func square() {
        self = self * self
    }
}
var someInt = 3
someInt.square()
// someInt is now 9

Subscript

• subscript도 extension 사용 가능

// [n]: 오른쪽으로부터 n번째 숫자 반환

extension Int {
    subscript(digitIndex: Int) -> Int {
        var decimalBase = 1
        for _ in 0..<digitIndex {
            decimalBase *= 10
        }
        return (self / decimalBase) % 10
    }
}
746381295[0]
// returns 5
746381295[1]
// returns 9
746381295[2]
// returns 2
746381295[8]
// returns 7

extension - Nested Type

• 기존 type에 nested type 추가 가능

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