[ios] swift 기본 문법 정리 7: Enum

사실 열거형 문법이야 언어마다 조금씩은 다르지만 큰 틀은 동일해서, 안다루고 넘어가려다 문법적인 표현 정도는 다루는 게 맞다고 생각을 해서 간단하게 다루고 넘어가려고 한다.

 

열거형 문법은 enum 키워드로 선언을 하고 case 키워드를 사용하여 구분 지어준다.

enum CompassPoint {
    case north
    case south
    case east
    case west
}

아래와 같이 case를 콤마로 구분하여 한 줄에 적을 수도 있다.

enum Planet {
    case mercury, venus, earth, mars, jupiter, saturn, uranus, neptune
}

 

열거형 타입을 최초로 변수에 정의할 때는 클래스의 타입을 명시해야 하지만, 이후에는 아래와 같이 생략하여도 무방하다.

var directionToHead = CompassPoint.east

// 타입 추론으로 다음 값 할당은 Enum Class 호출이 필요 없음
directionToHead = .west

directionToHead = .south

 

각 열거형의 값을 당연하게도 switch 문에서 매칭할 수 있다. 

directionToHead = .south
switch directionToHead {
case .north:
    print("Lots of planets have a north")
case .south:
    print("Watch out for penguins")
case .east:
    print("Where the sun rises")
case .west:
    print("Where the skies are blue")
}
// Prints "Watch out for penguins"

let somePlanet = Planet.earth
switch somePlanet {
case .earth:
    print("Mostly harmless")
default:
    print("Not a safe place for humans")
}

 

열거형의 각 case에 custom type의 추가적인 정보를 저장할 수 있다.

enum Barcode {
    case upc(Int, Int, Int, Int)
    case qrCode(String)
}

var productBarcode = Barcode.upc(8, 85909, 51226, 3)
productBarcode = .qrCode("ABCDEFGHIJKLMNOP")

switch productBarcode {
case .upc(let numberSystem, let manufacturer, let product, let check):
    print("UPC: \(numberSystem), \(manufacturer), \(product), \(check).")
case .qrCode(let productCode):
    print("QR code: \(productCode).")
}
// Prints "QR code: ABCDEFGHIJKLMNOP."

 

각 case에 아래와 같이 raw 값을 지정할 수도 있다. 단, 이 값들은 절대 중복되어서는 안된다.

enum ASCIIControlCharacter: Character {
    case tab = "\t"
    case lineFeed = "\n"
    case carriageReturn = "\r"
}

 

또한 명시적으로 raw값을 할당하지 않아도, swift가 암시적으로 할당해준다.

enum Planet: Int {
    case mercury = 1, venus, earth, mars, jupiter, saturn, uranus, neptune
}

 

raw 값은 rawValue 프로퍼티를 사용하여 접근할 수 있다.

let earthsOrder = Planet.earth.rawValue
// earthsOrder is 3

let sunsetDirection = CompassPoint.west.rawValue
// sunsetDirection is "west"

let possiblePlanet = Planet(rawValue: 7)
// possiblePlanet is of type Planet? and equals Planet.uranus

 

만약 열거형에 지정된  raw값이 없는 값으로 초기자를 지정하면 그 값은 nil이 된다.

enum Planet: Int {
    case mercury = 1, venus, earth, mars, jupiter, saturn, uranus, neptune
}

let positionToFind = 11

if let somePlenet = Planet(rawValue: positionToFind) {
    switch somePlenet {
    case .earth:
        print("Mostly harmless")
    default:
        print("Not a safe place for humans")
    }
} else {
    print("There isn't a planet at position \(positionToFind)")
}

 

물론 다른 열거 타입을 case로 갖는 재귀 형태도 가능하다. 재귀열거자는 case 앞에 indirect 키워드를 붙여 표시한다.

enum ArithmeticExpression {
    case number(Int)
    indirect case addition(ArithmeticExpression, ArithmeticExpression)
    indirect case multiplication(ArithmeticExpression, ArithmeticExpression)
}

모든 case가 재귀형태이면 아래와 같이 묶어서 표시할 수도 있다.

indirect enum ArithmeticExpression {
    case number(Int)
    case addition(ArithmeticExpression, ArithmeticExpression)
    case multiplication(ArithmeticExpression, ArithmeticExpression)
}

let five = ArithmeticExpression.number(5)
let four = ArithmeticExpression.number(4)
let sum = ArithmeticExpression.addition(five, four)
let product = ArithmeticExpression.multiplication(sum, ArithmeticExpression.number(2))

func evaluate(_ expression: ArithmeticExpression) -> Int {
    switch expression {
    case let .number(value):
        return value
    case let .addition(left, right):
        return evaluate(left) + evaluate(right)
    case let .multiplication(left, right):
        return evaluate(left) * evaluate(right)
    }
}

print(evaluate(product))
// Prints "18"