collectionViewのカスタムレイアウトについての質問 [swift]

Question

[今回聞きたいこと]
・numberOfItemsInSection と、numberOfSections の　return の値をどうすればいいでしょうか？

　
インスタグラムの類似アプリを作っています。

こちらのサイトのコードを、ほぼそのまま使って、CollectionViewのカスタムレイアウトを実装しています。
https://swiswiswift.com/2017/11/26/custum-collectionview/

今回実装しているcollectionViewのレイアウトはこの画像と同じものです。

CustomCollectionViewFlowLayoutクラスを、コレクションビューにアタッチしています。
コードはこちらのページのCustomCollectionViewFlowLayoutクラスのコードと全く同じです。

firebaseから引っ張ってきたデータ(今回の場合だとpostオブジェクトの中に入っている画像url)をコレクションビューに表示させたいです。
データーを一つ取ってくるごとに、collectionViewがreloadData()されるようになっています。

numberOfItemsInSection と、numberOfSections の　return の値をどのように変えれば、outofindexなどのエラーを出さずに表示させることができるでしょうか？

import UIKit
import SDWebImage
import FirebaseDatabase

class DiscoverViewController: UIViewController, UICollectionViewDelegate, UICollectionViewDataSource {
    
    @IBOutlet weak var myCollectionView: UICollectionView!
    var REF_POSTS = Database.database().reference().child("posts")
    var posts: [Post] = []
    
    override func viewDidLoad() {
        super.viewDidLoad()
        myCollectionView.delegate = self
        myCollectionView.dataSource = self
        
        loadTopPosts()
    }
    
    func loadTopPosts() {

            REF_POSTS.queryOrdered(byChild: "timestamp").observeSingleEvent(of: .value, with: { snapshot in
            let arraySnapshot = (snapshot.children.allObjects as! [DataSnapshot]).reversed()
            arraySnapshot.forEach({ (child) in
                if let dict = child.value as? [String: Any] {
                    let post = Post.transformPostPhoto(dict: dict, key: child.key)
                    self.posts.append(post)
                    self.myCollectionView.reloadData()
                }
            })
        }
    }
    
    //Cellが選択された際に呼び出される
    func collectionView(_ collectionView: UICollectionView, didSelectItemAt indexPath: IndexPath) {
        print("Section: (indexPath.section)")
        print("Num: (indexPath.row)")
        print("Number: (indexPath.section * 6 + indexPath.row)")
    }
    
    func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {

        return 6　//←ここの値を変える

    }
    //セクションの総数を返す
    func numberOfSections(in collectionView: UICollectionView) -> Int {
　　　　return 8　//←ここの値を変える
    }

    //Cellに値を設定する
    func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
        
        let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "MyCell", for: indexPath) as! CustomCollectionViewCell
        
       
        
        let photoUrlString = self.posts[indexPath.section * 6 + indexPath.row].imageUrls[0] //←エラー
        let photoUrl = URL(string: photoUrlString)
        cell.imageView1.sd_setImage(with: photoUrl)
        
       
        
        return cell
    }
}

Answer 1

表示される総セル数は6*8=48ですが、posts.countが48個未満なのではないでしょうか。

numberOfSectionは (posts.count-1)/6+1
numberOfItemsInSectionは section<posts.count/6 ? 6 : posts.count%6
をそれぞれ返せばいいのではないでしょうか？
動作は確認してないので、ミスがあるかもしれませんが

Comments

[kou009]

ありがとうございます。教えていただいたコードで動きました。