Animate UICollectionView sizeForItemAtIndexPath (Swift)

Me gustaría animar el cambio de tamaño de UICollectionViewCell. He escrito el código a continuación pero no puedo tener la línea de return dentro del bloque de animation. ¿Algunas ideas?

 func collectionView(collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAtIndexPath indexPath: NSIndexPath) -> CGSize { var newSize = CGSize(width: (self.view.frame.width), height: 0) UIView.animateWithDuration(2.0, animations: { () -> Void in return newSize }) } 

Llame al siguiente método si desea animar,

 self.collectionView.performBatchUpdates(updates: (() -> Void), completion:((Bool) -> Void)?) 

Más específicamente, también debe manejar el cambio de orientación, como a continuación,

 override func didRotateFromInterfaceOrientation(fromInterfaceOrientation: UIInterfaceOrientation) { self.collectionView.performBatchUpdates(nil, completion: nil) }