Swift 一つのtableviewで複数のデータの表示切り替えを出来る様にしたい。

Question

やりたい事
以下のような一つのtableViewでUISegmentedControlを使って、「魚」を押したら"sakana"の配列に入っているデータのみを表示し、「野菜」を押したら"yasai"の配列に入ってるデータを表示し、また全てのデータを見たい時は「全て」を押すと元に戻るような切り替えをしたい。

現況のコード

import UIKit

class ViewController: UIViewController,UITableViewDelegate,UITableViewDataSource {
    
    func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        return alllist.count
    }
    
    func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        ///セルの構築
        let cell = tableView.dequeueReusableCell(withIdentifier: "Cell", for: indexPath)
        //セルのテキストラベルにindexPath.row番目の配列の要素を入れる
        cell.textLabel?.text = alllist[indexPath.row]
        //UITableViewCell型を返す
        return cell
    }   
    
    //表示する配列
    var alllist = [String]()
    var sakana = ["トビウオ","シャケ","マグロ"]
    var kudamono = ["リンゴ","バナナ","メロン"]
    var yasai = ["トマト","キュウリ","ブロッコリー"]
    
    @IBOutlet weak var table: UITableView!
    
    override func viewDidLoad() {
        super.viewDidLoad()
        // Do any additional setup after loading the view.
        
        table.delegate = self
        table.dataSource = self
        
        alllist = sakana + kudamono + yasai
    }       
}

わからない事・聞きたいこと
UITableviewの中身をUIButtonやUISegmentedControlで更新(切り替え)する方法が知りたいです。
特にどのようにUITableViewのnumberOfRowsInSection・cellForRowAtのcell.textLabel?.textの部分でデータをセットすれば良いのかが思い付きません。
アドバイスでも良いのでお待ちしてますm(_ _)m

Comments

[hoshi-takanori]

var currentList: [String] を用意して、表示したいものをそれで切り替えるようにすればいいのでは。

[samson66]

ありがとうございます！
無事に切り替えることが出来ました！

Answer 1

segmentedControlをIBOutletで接続はできるでしょうか？
hoshi-takanoriさんのご助言を利用させていただいて書いてみました。
こちらで切替できるかと思います。

func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        return currentList.count
    }

    func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        ///セルの構築
        let cell = tableView.dequeueReusableCell(withIdentifier: "Cell", for: indexPath)
        //セルのテキストラベルにindexPath.row番目の配列の要素を入れる
        cell.textLabel?.text = currentList[indexPath.row]
        //UITableViewCell型を返す
        return cell
    }   

    //表示する配列
    var alllist = [String]()
    var sakana = ["トビウオ","シャケ","マグロ"]
    var kudamono = ["リンゴ","バナナ","メロン"]
    var yasai = ["トマト","キュウリ","ブロッコリー"]

    @IBOutlet weak var table: UITableView!
@IBOutlet segmentedControl: UISegmentedControl!
var currentList: [String] {
    switch segmentedControl.selectedSegmentIndex {
    case 0:
        return allList
    case 1:
        return sakana
    case 2:
        return kudamono
    case 3:
        return yasai
    }
}
override func viewDidLoad() {
    super.viewDidLoad()
    // Do any additional setup after loading the view.

    table.delegate = self
    table.dataSource = self

    alllist = sakana + kudamono + yasai
    segmentedControl.addTarget(self, action: #selector(handleSegmentedControlSelected(_:)), for: .valueChanged)
}       

@objc private func handleSegmentedControlSelected(_ sender: UISegmentedControl) {
    table.reloadData()
}