Swift 열거형

야곰님의 강의 자료

열거형

Swift 열거형은 다른 언어의 열거형과 많이 다릅니다. 강력한 기능을 지니고 있어 잘 알아두고 사용하면 좋습니다.
유사한 종류의 여러 값을 한 곳에 모아서 정의한 것입니다. ex) 요일, 월, 계절 등
enum 자체가 하나의 데이터 타입으로, 대문자 카멜케이스를 사용하여 이름을 정의합니다. 각 case는 소문자 카멜케이스로 정의합니다.
C언어의 열거형에서는 정수값이 자동으로 할당이 되는 반면, 각 case 그 자체가 고유의 값으로 취급됩니다.(각 case에 자동으로 정수값이 할당되지 않음)
각 case는 한 줄에 개별로도, 한 줄에 여러개도 정의할 수 있습니다.

enum 이름 { 
	case 이름1
	case 이름2
	case 이름3, 이름4, 이름5
	// ...
}

// ex)
enum BoostCamp {
        case iosCamp
        case androidCamp
        case webCamp
}

열거형 사용

타입이 명확할 경우, 열거형의 이름을 생략 할 수 있습니다.
switch 구문에서 사용하면 좋습니다.

enum Weekday {
    case mon
    case tue
    case wed
    case thu, fri, sat, sun
}

// weekday가 day의 타입으로 지정(선언시 타입 명시)
// 열거형 타입과 케이스를 모두 사용하여도 됩니다
var day: Weekday = Weekday.mon 

// 타입 명시후에는 축약가능
// 타입이 명확하다면 .케이스 처럼 표현해도 무방합니다
day = .tue 

print(day) // tue

// case는 한정되어 있음을 컴파일러 인지
// switch의 비교값에 열거형 타입이 위치할 때
// 모든 열거형 케이스를 포함한다면
// default를 작성할 필요가 없습니다
switch day {
case .mon, .tue, .wed, .thu:
    print("평일입니다")
case Weekday.fri:
    print("불금 파티!!")
case .sat, .sun:
    print("신나는 주말!!")
}

case는 한정되어 있음을 컴파일러 인지하여, switch의 비교값에 열거형 타입이 위치할 때 모든 열거형 케이스를 포함한다면 default를 작성할 필요가 없습니다. 단, 열거형 케이스를 하나라도 미입력시 default를 꼭 작성해줘야 합니다.

rawValue (원시값)

C언어의 enum 처럼 정수값을 가질 수 있습니다.
rawValue는 case별로 각각 다른 값을 가져야합니다.
자동으로 1이 증가된 값이 할당됩니다.
rawValue를 반드시 지닐 필요가 없다면 굳이 만들지 않아도 됩니다.
정수 타입 뿐만 아니라, Hashable 프로토콜을 따르는 모든 타입을 원시값의 타입으로 지정 할 수있습니다.

enum Fruit: Int { // 타입명시
// C언어처럼 자동으로 값이 1씩 증가
    case apple = 0
    case grape = 1
    case peach // 자동으로 2
}

print("Fruit.peach.rawValue == \(Fruit.peach.rawValue)")
// Fruit.peach.rawValue == 2

enum School: String {
    case elementary = "초등"
    case middle = "중등"
    case high = "고등"
    case university
}

// 정수 타입 뿐만 아니라, Hashable 프로토콜을 따르는 모든 타입을 원시값의 타입으로 지정 할 수있습니다.
print("School.middle.rawValue == \(School.middle.rawValue)")
// School.middle.rawValue == 중등

// 열거형의 원시값 타입이 String일 때, 원시값이 지정되지 않았다면
// case의 이름을 원시값으로 사용합니다
print("School.university.rawValue == \(School.university.rawValue)")
// School.middle.rawValue == university

원시값을 통한 초기화

rawValue를 통해 초기화 할 수 있습니다.
rawValue가 case에 해당하지 않을 수 있으므로, rawValue를 통해 초기화 한 인스턴스는 옵셔널 타입입니다.
case가 없으면 생성이 되지 않아 nill이 나올 수 있기 때문입니다.

// rawValue를 통해 초기화 한 열거형 값은 옵셔널 타입이므로 Fruit 타입이 아닙니다
let apple: Fruit = Fruit(rawValue: 0) // 오류(옵셔널 타입 명시) 
let apple: Fruit? = Fruit(rawValue: 0) // 옵셔널 타입 반환

// 옵셔널 바인딩
// if let 구문을 사용하면 rawValue에 해당하는 케이스를 곧바로 사용할 수 있습니다
if let orange: Fruit = Fruit(rawValue: 5) {
    print("rawValue 5에 해당하는 케이스는 \(orange)입니다")
} else {
    // rawValue 5에 해당하는 케이스가 없습니다
    print("rawValue 5에 해당하는 케이스가 없습니다")
} 

메서드

Swift의 열거형에는 메서드도 추가할 수 있습니다.

enum Month {
    case dec, jan, feb
    case mar, apr, may
    case jun, jul, aug
    case sep, oct, nov
    
    func printMessage() {
        switch self {
        case .mar, .apr, .may:
            print("따스한 봄~")
        case .jun, .jul, .aug:
            print("여름 더워요~")
        case .sep, .oct, .nov:
            print("가을은 독서의 계절!")
        case .dec, .jan, .feb:
            print("추운 겨울입니다")
        }
    }
}

Month.mar.printMessage() // 따스한 봄~