[Swift] Subscript란?

 

안녕하세요 Foma 💻 입니다!

 

오늘은 해쉬 테이블(딕셔너리)을 직접 구현하는 중에 a["A"] 이런 식으로 값을 지정하고

 

값을 꺼내오는지 찾아보던 중에 subscript를 활용해야 한다는 것을 알고 글을 정리하게 되었습니다.

 

바로 시작할게요~

---

Subscript

 

Swift 공식 문서에는 아래와 같이 정의되어 있습니다!

클래스, 구조체 그리고 열거형에서 스크립트를 정의해 사용할 수 있습니다. 서브스크립트란 콜렉션, 리스트, 시퀀스 등 집합의 특정 멤버 엘리먼트에 간단하게 접근할 수 있는 문법입니다. 서브스크립트를 이용하면 추가적인 메소드 없이 특정 값을 할당(assign)하거나 가져올 수(retrieve) 있습니다. 예를들면, 배열(Array) 인스턴스의 특정 엘리먼트는 someArray[index] 문법으로, 사전(Dictionary) 인스턴스의 특정 엘리먼트는 someDictionary[key]로 접근할 수 있습니다. 하나의 타입에 여러 서브스크립트를 정의할 수 있고 오버로드(Overload)도 가능합니다. 뿐만아니라 단일 인자 값을 넘어, 필요 따라 복수 인자 값을 사용할 수 있습니다.

 

즉, 추가적인 메소드 없이 값을 할당하거나 가져올 수 있는 문법입니다.

---

Syntax

 

서브스크립트 선언 문법은 인스턴스 메소드와 계산된 프로퍼티를 선언하는 것과 비슷합니다.

 

인스턴스 메소드와 다른 점은, 서브스크립트는 읽고-쓰기(read-write) 혹은 읽기 전용(read only)만 가능하다는 것입니다.

 

정의는 계산된 프로퍼티 방식과 같이 setter, getter 방식을 따릅니다.

 

(만약 get,set을 지정하지 않으면 자동으로 읽기 전용으로 선언됩니다.)

 

아래와 같이 파라미터로 특정 값을 받고 get을 이용해서 특정 값을 반환할 수 있고 set을 통해 특정 값으로 설정할 수 있습니다.

 

subscript(index: Int) -> Int {
    get {
        // 적절한 반환 값
    }
    set(newValue) {
        // 적절한 set 액션
    }
}

---

Usage

 

Read-Only

 

아래와 같이 곱셈을 해주는 getter를 만들어서 특정 값을 넣었을 때 원하는 값을 곱해주어 반환해줍니다.

 

struct TimesTable {
    let multiplier: Int
    subscript(index: Int) -> Int {
        return multiplier * index
    }
}
let threeTimesTable = TimesTable(multiplier: 3)
print("six times three is \(threeTimesTable[6])")
// "six times three is 18" 출력

 

Read & Write

 

struct Matrix {
    let rows: Int, columns: Int
    var grid: [Double]
    init(rows: Int, columns: Int) {
        self.rows = rows
        self.columns = columns
        grid = Array(repeating: 0.0, count: rows * columns)
    }
    func indexIsValid(row: Int, column: Int) -> Bool {
        return row >= 0 && row < rows && column >= 0 && column < columns
    }
    subscript(row: Int, column: Int) -> Double {
        get {
            assert(indexIsValid(row: row, column: column), "Index out of range")
            return grid[(row * columns) + column]
        }
        set {
            assert(indexIsValid(row: row, column: column), "Index out of range")
            grid[(row * columns) + column] = newValue
        }
    }
}