         callback?(row)
     }
 
+    func collectionView(collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAtIndexPath indexPath: NSIndexPath) -> CGSize {
+        return cellSize
+    }
+    
     func isHighlight(asset: PHAsset) -> Bool {
         for highlightRow in highlightRows {
             if highlightRow == asset {

        self.registerNib(nib, forCellWithReuseIdentifier: "ImageCell")
        self.delegate = self
        self.dataSource = self
    }



import UIKit
import Photos

class ImagePickerCollectionView: UICollectionView, UICollectionViewDataSource, UICollectionViewDelegate, UICollectionViewDelegateFlowLayout {

    let manager = PHImageManager.defaultManager()
    var rows = [PHAsset]()
    var highlightRows = [PHAsset]()
    var callback: (PHAsset -> Void)?
    var cellSize = CGSize(width: 80, height: 80)

    override func awakeFromNib() {
        super.awakeFromNib()
