當traitCollection發生變化時更改單元格佈局

Question

當水平尺寸類更改時，我在調整單元格佈局時遇到問題。

我的單元有一個stackView，我希望該軸對於緊湊尺寸類是水平的，對於常規是垂直的。

這是我的嘗試：

override func traitCollectionDidChange(previousTraitCollection: UITraitCollection?) { 
    if previousTraitCollection?.horizontalSizeClass != traitCollection.horizontalSizeClass { 
     self.collectionView?.reloadData() 
    } 
} 

override func collectionView(collectionView: UICollectionView, cellForItemAtIndexPath indexPath: NSIndexPath) -> UICollectionViewCell { 
    let cell = collectionView.dequeueReusableCellWithReuseIdentifier(reuseIdentifier, forIndexPath: indexPath) as! MyCollectionViewCell 

    switch traitCollection.horizontalSizeClass { 
    case .Compact: 
     cell.stackView.axis = .Horizontal 
    default: 
     cell.stackView.axis = .Vertical 
    } 

    return cell 
} 

但結果是，並非所有的細胞更新它們的佈局，SE以下的gif。

編輯： 我已經證實，軸是由cellForItem和在細胞類本身正確打印改變。所以，這個問題似乎是細胞沒有重繪..

任何建議，應該怎樣解決這個問題？

Github Repo

Answer 1

調用layoutIfNeeded()添加到cellForItemAtIndexPath方法，以獲取堆棧以重新佈局其內容：

override func collectionView(collectionView: UICollectionView, cellForItemAtIndexPath indexPath: NSIndexPath) -> UICollectionViewCell { 
    let cell = collectionView.dequeueReusableCellWithReuseIdentifier(reuseIdentifier, forIndexPath: indexPath) as! MyCollectionViewCell 

    switch traitCollection.horizontalSizeClass { 
    case .Compact: 
     cell.stackView.axis = .Horizontal 
    default: 
     cell.stackView.axis = .Vertical 
    } 

    cell.stackView.layoutIfNeeded() 

    return cell 
}