arrays - 当你想使用 index.advanceBy 时，你应该使用什么类型？

Question

我想在数组中的另一个对象附近找到一些对象。我以为我可以写一个这样的扩展方法，但我得到了这个错误:

// Error: Cannot invoke 'advanceBy' with an argument list of type '(Int)'

Int 类型显然是错误的，但是indexOf 方法需要一个Self.Distance 参数，我不知道如何使用那作为参数类型。

extension CollectionType where Generator.Element : Equatable {
    func objectNear(object: Self.Generator.Element, indexModifier: Int) -> Self.Generator.Element? {
        if let index = self.indexOf(object) {
            let newIndex = index.advancedBy(indexModifier) // this doesn't work
            //let newIndex = index.advancedBy(1) // but this this works
            if self.indices.contains(newIndex) {
                return self[newIndex]
            }
        }
        return nil
    }
}

(如果有更 Swifty 的方法，我很乐意听到，但无论如何我都想了解以上内容。)

Answer 1

CollectionType 有方法

public func indexOf(element: Self.Generator.Element) -> Self.Index?

并符合

public protocol Indexable {
    typealias Index : ForwardIndexType
    // ...
}

最后，ForwardIndexType 有这个方法

public func advancedBy(n: Self.Distance) -> Self

因此正确的类型是Index.Distance:

func objectNear(object: Self.Generator.Element, indexModifier: Index.Distance) -> Self.Generator.Element? { ... }

但请注意，将索引推进到 endIndex 之外可能会崩溃，例如对于字符集:

let c = "abc".characters
print(c.objectNear("b", indexModifier: 1)) // Optional("c")
print(c.objectNear("b", indexModifier: 2)) // nil
print(c.objectNear("b", indexModifier: 3)) // fatal error: can not increment endIndex

一个安全的变体是:

func objectNear(object: Generator.Element, indexModifier: Index.Distance) -> Generator.Element? {
    if let index = indexOf(object) {
        if indexModifier > 0 && index.distanceTo(endIndex) <= indexModifier {
            return nil
        }
        if indexModifier < 0 && startIndex.distanceTo(index) < -indexModifier {
            return nil
        }
        return self[index.advancedBy(indexModifier)]
    }
    return nil
}

或者，如果您只需要用于集合的方法由Int(例如Array)索引，然后您可以定义

extension CollectionType where Generator.Element : Equatable, Index == Int {

    func objectNear(object: Generator.Element, indexModifier: Int) -> Generator.Element? {
        if let index = self.indexOf(object) {
            let newIndex = index + indexModifier
            if indices.contains(newIndex) {
                return self[newIndex]
            }
        }
        return nil
    }
}