Merge pull request hyperoslo#101 from aronse/feature/fix-reset-assets…

…-collectionview fix reloadAssets still showing pictures as selected in collection view
shift00 · Feb 15, 2016 · abfec59 · abfec59
2 parents 8fabb2f + 38aa96b
commit abfec59
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/Source/ImagePickerController.swift b/Source/ImagePickerController.swift
@@ -158,7 +158,7 @@ public class ImagePickerController: UIViewController {
       object: nil)
 
     NSNotificationCenter.defaultCenter().addObserver(self,
-      selector: "adjustButtonTitle:",
+      selector: "didReloadAssets:",
       name: ImageStack.Notifications.stackDidReload,
       object: nil)
 
@@ -168,6 +168,12 @@ public class ImagePickerController: UIViewController {
       object: nil)
   }
 
+  func didReloadAssets(notification: NSNotification) {
+    adjustButtonTitle(notification)
+    galleryView.collectionView.reloadData()
+    galleryView.collectionView.setContentOffset(CGPointZero, animated: false)
+  }
+
   func volumeChanged(notification: NSNotification) {
     guard let slider = volumeView.subviews.filter({ $0 is UISlider }).first as? UISlider,
       let userInfo = notification.userInfo,