associated valueを持つenumの配列から配列のindexを返す方法

Question

いつも勉強させてもらっています。

タイトルの通り、associated valueを持つenumを作成し、
それを配列にして利用しています。

配列の変更が起こり得るので、その時点でのenumの条件のindexを取得し、
それをもとにenumにわけ、処理をしたいのですが、
どのようにしたらうまくindexを取得できるでしょうか？

stackoverflowにぴったり一致するものがありましたので、
参考に適応してみましたが、変数にして行うと、型が一致しているのに、しませんってエラーが出て困っております。

皆様の智恵をいただければ幸いです。

簡略化したサンプルを載せてみます。

// CDCellのEnum
enum CDPresentCell {   
    case titleCell(String?)
    case artistCell(String?)
    case releaseCell(Date?)
}

// CollectionView
import UIKit

class CDViewController: UIViewController, UICollectionViewDataSource, UICollectionViewDelegate {
    @IBOutlet weak var collectionView: UICollectionView!
    @IBOutlet weak var datePicker: UIDatePicker!
    var cells:[CDPresentCell]

    init() {
        self.cells = [.titleCell(nil),
                      .artistCell(nil),
                      .releaseCell(nil)
        ]
    }

    // 試した1個目 <- CDPresentCell と CDPresentCellが一致しないというエラーが出ました。
    private fun returnRow1(cell: CDPresentCell) -> Int? {
        var m: Int? = nil
        for n in 0 ..< self.cells.count {
            if cell == self.cells[n] {
                m = n
            } 
        }
        return m
    }

    // コレ試した2個目 <- CDPresentCell と CDPresentCellが一致しないというエラーが出ました。
    private func returnRow2(cell: CDPresentCell) -> Int? {
        if let index = self.cells.firstIndex(where: { if case cell.self = $0 { return true }; return false }) {
            return index
        }
        return nil
    }

    override func viewDidLoad() {
    }

    @IBAction func doneButtonPressed(_ sender: Any) {
        if let row = self.collectionView.indexPathsForSelectedItems?.first?.row {
            let cellType = self.cells[row]
            switch cellType {
            case .releaseCell(let data):
                self.datePicker.alpha = 1.0
            default: break
            }
        }
    }

    func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        return  self.cells.count
    }

    func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
        let cellType = self.cells[indexPath.row]
        switch cellType {
        case .titleCell(let data):
            let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "title", for: indexPath) as! TitleCell
            cell.bind(data: data)
            return cell
        case .artistCell(let data):
            let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "artist", for: indexPath) as! ArtistCell
            cell.bind(data: data)
            return cell
        case .releaseCell(let data):
            let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "release", for: indexPath) as! ReleaseCell
            cell.bind(data: data)
            return cell
    }

    func collectionView(_ collectionView: UICollectionView, didSelectItemAt indexPath: IndexPath) {
        let cellType = self.cells[row]
        switch cellType {
        case .releaseCell(let data):
            self.datePicker.alpha = 1.0
        default: break
        }
    }

    // collectionCellのtextFieldなどにdelegateを設定し、
    // returnする場合にrowを数値で指定せずに、cellPresenterのself.cellsの
    // enumからindexを取得し、更新に利用したいです。
    func artistTextFieldDidReturn(text: String) {
        let row = returnRow2(cell: .artist)  // コレが働くようにしたい
        let cellType = self.cells[row]
        switch cellType {
        case .titleCell(let data):
            self.cells[row] = .titleCell(text)
        default: break
        }
        self.collectionView.reloadItems(at: [[0, row]])
    }
}

Comments

[MasakiHori]

case が一致しているかどうかを調べる方法を質問していますか？
それともassociated valueも一致しているかどうかを調べる方法を質問していますか？

[hameji]

わかりずづらくてすみません。
case が一致するindexを返したいです。
associated valueの内容は判定しなくていいです。

Answer 1

switch文を使うのが分かりやすいでしょう。

let a = CDPresentCell.titleCell(nil)
let b = CDPresentCell.titleCell("b")

switch (a, b) {
case (.titleCell, .titleCell): print("same")
case (.artistCell, .artistCell): print("same")
case (.releaseCell, .releaseCell): print("same")
default: print("not same")
}

これをenum自身に判別させるのが楽でしょう

extension CDPresentCell {
    func isSameCase(other: CDPresentCell) -> Bool {
        switch (self, other) {
        case (.titleCell, .titleCell): return true
        case (.artistCell, .artistCell): return true
        case (.releaseCell, .releaseCell): return true
        default: return false
        }
    }
}

こうしておけば

private func returnRow2(cell: CDPresentCell) -> Int? {
    return cells.firstIndex { $0.isSameCase(other: cell) }
}

となります。

Comments

[hameji]

ありがとうございます。
そんなふうに、2つの要素を用いて、switchで比較できるんですね。
勉強になりました。スッキリです。