【Swift】collectionview cellの高さを可変にしたいです

Question

現状、正方形になっているセルのサイズは正方形になっていますが、幅はそのまま画面サイズの半分のままで高さを中身の画像のサイズにしたいです。一番ベストな方法教えて頂けると助かります。

import UIKit

class ChatViewController: UIViewController ,UICollectionViewDataSource, UICollectionViewDelegate, UICollectionViewDelegateFlowLayout {
    
    // 画像の名前を配列とする
    let photos = ["jump1", "jump2", "jump3", "jump4", "jump5", "jump6", "jump7", "jump8", "jump9", "jump10"]
    
    // インスタンス化された直後に発動
    override func viewDidLoad() {
        super.viewDidLoad()
    }
    
    // collectionViewのセルを返す
    func collectionView(collectionView: UICollectionView, cellForItemAtIndexPath indexPath: NSIndexPath) -> UICollectionViewCell{
        
        // dequeueReusableCellWithReuseIdentifier の働きは
        // 再利用できるセルがあればそれを使う
        // 再利用できるセルがなければ生成する
        // Cell はストーリーボードで設定したセルのID
        let testCell:UICollectionViewCell = collectionView.dequeueReusableCellWithReuseIdentifier("Cell", forIndexPath: indexPath)
        
        // Tag番号を使ってImageViewのインスタンス生成
        let imageView = testCell.contentView.viewWithTag(1) as! UIImageView
        // 画像配列の番号で指定された要素の名前の画像をUIImageとする
        let cellImage = UIImage(named: photos[indexPath.row])
        // UIImageをUIImageViewのimageとして設定
        imageView.image = cellImage
        
        return testCell
    }
    
    // Screenサイズに応じたセルサイズを返す
    // UICollectionViewDelegateFlowLayoutの設定が必要
    func collectionView(collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAtIndexPath indexPath: NSIndexPath) -> CGSize {
        let cellSize:CGFloat = self.view.frame.size.width/2
        // 正方形で返すためにwidth,heightを同じにする
        return CGSizeMake(cellSize, cellSize)
    }
    
    //section 数の設定、今回は１つにセット
    func numberOfSectionsInCollectionView(collectionView: UICollectionView) -> Int {
        return 1
    }
    
    // collectionViewの要素の数を返す
    func collectionView(collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        return 10;
    }
    
    override func didReceiveMemoryWarning() {
        super.didReceiveMemoryWarning()
    }
    
    //横の間隔 これで横並びを実装する
    func collectionView(collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, minimumInteritemSpacingForSectionAtIndex section: Int) -> CGFloat {
        return 0
    }
    
    // Cell が選択された場合
    func collectionView(collectionView: UICollectionView, didSelectItemAtIndexPath indexPath: NSIndexPath) {
        
        // [indexPath.row] から画像名を探し、UImage を設定
//        selectedImage = UIImage(named: photos[indexPath.row])
//        if selectedImage != nil {
            // SubViewController へ遷移するために Segue を呼び出す
            performSegueWithIdentifier("toChatListViewController",sender: nil)
//        }
    }
    
    // Segue 準備 画面遷移時に値を遷移先に渡すならここに書く
    override func prepareForSegue(segue: UIStoryboardSegue, sender: AnyObject!) {
        if (segue.identifier == "toChatListViewController") {
//            let subVC: SubViewController = (segue.destinationViewController as? SubViewController)!
            // SubViewController のselectedImgに選択された画像を設定する
//            subVC.selectedImg = selectedImage
        }
    }
    
}

Answer 1

func collectionView(collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAtIndexPath indexPath: NSIndexPath) -> CGSize {

    let image: UIImage = UIImage(named: photos[indexPath.row])
    let width: CGFloat = self.view.frame.size.width / 2
    let height: CGFloat = (image?.size.height)!

    return CGSizeMake(width, height)

}

cellの中のImageViewはAutoLayout設定済みの程だと、
セルサイズを画像の高さに変えてあげればいいはず。