Swift之下標

本文首發於我的博客

前言

Swift中對枚舉、結構體、類使用下標(subscript），就能夠像使用數組同樣來使用了

使用規則

• 使用subscript能夠給任意類型（枚舉、結構體、類）增長下標功能，有些地方也翻譯爲：下標腳本
• subscript的語法相似於實例方法、計算屬性，本質就是方法（函數）

例以下面的代碼中，類Point中，的屬性 x 和 y,能夠用下標訪問

class Point {
    var x = 0.0, y = 0.0
    subscript(index: Int) -> Double {
        set {
            if index == 0 {
                x = newValue
            } else if index == 1 {
                y = newValue
            }
        }
        get {
            if index == 0 {
                return x
            } else if index == 1 {
                return y
            }
            return 0
        }
    }
}
複製代碼

訪問的時候

var p = Point()
p[0] = 11.1
p[1] = 22.2
print(p.x) // 11.1
print(p.y) // 22.2
print(p[0]) // 11.1
print(p[1]) // 22.2

複製代碼

注意點

• subscript中定義的返回值類型決定了
  • get方法的返回值類型
  • set方法中newValue的類型
• subscript能夠接受多個參數，而且類型任意

subscript能夠沒有set方法

例以下面的代碼中，只提供了get，沒有set

class Point {
    var x = 0.0, y = 0.0
    subscript(index: Int) -> Double {
        get {
            if index == 0 {
                return x
            } else if index == 1 {
                return y
            }
            return 0
        }
    }
}
複製代碼

• 若是隻有get方法，能夠省略get

上面的代碼能夠寫成

class Point {
    var x = 0.0, y = 0.0
    subscript(index: Int) -> Double {
        if index == 0 {
            return x
        } else if index == 1 {
            return y
        }
        return 0
    }
}

複製代碼

能夠設置參數標籤

例以下面的代碼

class Point {
    var x = 0.0, y = 0.0
    subscript(index i: Int) -> Double {
        if i == 0 {
            return x
        } else if i == 1 {
            return y
        }
        return 0
    }
}

複製代碼

調用的時候

var p = Point()
p.y = 22.2
print(p[index: 1]) // 22.2
複製代碼

下標能夠是類型方法

以下

class Sum {
    static subscript(v1: Int, v2: Int) -> Int {
        return v1 + v2
    }
}

print(Sum[10, 20]) // 30
複製代碼

結構體、類做爲返回值對比

結構體

eg,以下代碼,結構體Point，用了下標subscript只有get方法

struct Point {
    var x = 0, y = 0
}

class PointManager {
    var point = Point()
    subscript(index: Int) -> Point {
        get { point }
    }
}
複製代碼

使用的時候報錯

var pm = PointManager()
pm[0].x = 11 //Cannot assign to property: subscript is get-only
// 等價於 pm[0] = Point(x: 11, y: pm[0].y)
pm[0].y = 22//Cannot assign to property: subscript is get-only
複製代碼

解決辦法一

• 加上set方法

struct Point {
    var x = 0, y = 0
}

class PointManager {
    var point = Point()
    subscript(index: Int) -> Point {
       set { point = newValue }
        get { point }
    }
}

//下面使用正常
var pm = PointManager()
pm[0].x = 11
// 等價於 pm[0] = Point(x: 11, y: pm[0].y)
pm[0].y = 22

複製代碼

解決辦法二

結構體改爲類

class Point {
    var x = 0, y = 0
}

class PointManager {
    var point = Point()
    subscript(index: Int) -> Point {
        get { point }
    }
}

//下面使用正常
var pm = PointManager()
pm[0].x = 11
// 等價於 pm[0] = Point(x: 11, y: pm[0].y)
pm[0].y = 22

複製代碼

緣由分析

類是引用類型的，傳遞的是地址 結構體是值類型，傳遞的是具體的值

接受多個參數的下標

eg,以下代碼

class Grid {
    var data = [
        [0, 1, 2],
        [3, 4, 5],
        [6, 7, 8]
    ]
    subscript(row: Int, column: Int) -> Int {
        set {
            guard row >= 0 && row < 3 && column >= 0 && column < 3 else {
                return
            }
            data[row][column] = newValue
        }
        get {
            guard row >= 0 && row < 3 && column >= 0 && column < 3 else {
                return 0
            }
            return data[row][column]
        }
    }
}

複製代碼

以下使用

var grid = Grid()
grid[0, 1] = 77
grid[1, 2] = 88
grid[2, 0] = 99
print(grid.data)
複製代碼

輸出爲

[[0, 77, 2], [3, 4, 88], [99, 7, 8]]

參考資料:

Swift官方源碼

從入門到精通Swift編程