열거형 [ enum ]

정의

- 같은 주제로 연관된 데이터들을 멤버로 구성하여 나타내는 자료형입니다.

- Swift에서 열거형은 각각의 case가 고유의 값으로 취급 됩니다.

- enum은 타입이므로 대문자 카멜케이스를 사용해 이름을 정의합니다.

---

정의형식

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

---

열거형 예제

열거형 생성

- 한 줄에 case 하나를 작성할 수 있습니다.

- 한 줄에 연속된 여러가지 case를 작성할 수도 있습니다.

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

 

열거형 사용

- 열거형의 케이스를 나타내는 문법으로 열거형 타입으로 적어주고 . 뒤에 케이스 이름을 작성하면 됩니다.

- 다음 예제에서는 Weekday가 day의 타입으로 지정되어있습니다.

- day의 타입이 Weekday이기 때문에 축약하여, .tue 라고 케이스명만 작성하여 출력할 수도 있습니다.

var day: Weekday = Weekday.mon
day = .tue

print(day)

 

switch문

- 또한 열거형은 switch문에서도 자주 사용됩니다.

- switch의 비교값으로 day 열거형 값이 작성되었습니다.

- 컴파일러는 case의 올 수 있는 값이 한정된다라고 이해합니다.

- 그렇기때문에 오류를 최소화하기 위해 모든 케이스를 작성하거나 default를 사용할 수 있습니다.  

switch day {
case .mon, .tue, .wed, .thu:
    print("평일입니다")
case Weekday.fri:
    print("불금 파티!!")
case .sat, .sun:
    print("신나는 주말!!")
}

---

원시값 사용 : Int

- C언어와 같이 enum에 정수값을 갖게하는 방법입니다.

- 열거형의 이름 뒤에 콜론(:)을 찍고, 원시값을 가질 타입을 명시하면됩니다.

- 다음 코드를 보면 각 case별로 1씩 증가하는것을 확인 할 수 있습니다.

- rawValue값을 사용할 수 있습니다.

enum Fruit: Int {
    case apple = 0
    case grape = 1
    case peach
    //    case mango = 0
}

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

실행결과
// 2

 

원시값 사용 : String

- 정수 타입 뿐만 아니라 Hashable 프로토콜을 따르는 모든 타입이 원시값의 타입으로 지정될 수 있습니다.

- 마지막 university와 같이 case의 값에 어떤 값이 들어올지 모르는 경우엔 case의 이름을 출력합니다.

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

print("School.middle.rawValue == \(School.middle.rawValue)")
print("School.university.rawValue == \(School.university.rawValue)")

실행결과
// School.middle.rawValue == 중등
// School.university.rawValue == university

 

원시값을 통한 초기화

- rawValue를 통해 초기화 할 수 있습니다.

- rawValue가 case에 해당하지 않을 수 있으므로 rawValue를 통해 초기화한 인스턴스는 옵셔널 타입입니다.

enum Fruit: Int {
    case apple = 0
    case grape = 1
    case peach
}

let apple: Fruit? = Fruit(rawValue: 0)

if let orange: Fruit = Fruit(rawValue: 5) {
    print("rawValue 5에 해당하는 케이스는 \(orange)입니다")
} else {
    print("rawValue 5에 해당하는 케이스가 없습니다")
} 

실행결과
// 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()

실행결과
// 따스한 봄~

---

내용출처

https://www.youtube.com/watch?v=uKJiaYYNJN4&list=PLz8NH7YHUj_ZmlgcSETF51Z9GSSU6Uioy&index=15