Updated the source files for #19

Athlee · Jan 24, 2017 · 2911798 · 2911798
1 parent a9cec86
commit 2911798
Showing 1 changed file with 10 additions and 12 deletions.
diff --git a/Source/ATHImagePickerAssetsViewController.swift b/Source/ATHImagePickerAssetsViewController.swift
@@ -249,20 +249,18 @@ extension ATHImagePickerAssetsViewController: UICollectionViewDataSource, UIColl
       targetSize: UIScreen.main.bounds.size,
       contentMode: .aspectFill,
       options: fetchingOptions) { result, info in
-        if info!["PHImageFileURLKey"] != nil  {
-          if let previewController = self.holder.previewController, previewController.state == .folded {
-            let floatingView = self.holder.floatingView
-            previewController.restore(view: floatingView, to: .unfolded, animated: true)
-            previewController.animationCompletion = { _ in
-              self.collectionView.scrollToItem(at: indexPath, at: .top, animated: true)
-              previewController.animationCompletion = nil
-            }
-          }
-
-          DispatchQueue.main.async {
-            self.holder.previewController?.image = result
+        if let previewController = self.holder.previewController, previewController.state == .folded {
+          let floatingView = self.holder.floatingView
+          previewController.restore(view: floatingView, to: .unfolded, animated: true)
+          previewController.animationCompletion = { _ in
+            self.collectionView.scrollToItem(at: indexPath, at: .top, animated: true)
+            previewController.animationCompletion = nil
           }
         }
+
+        DispatchQueue.main.async {
+          self.holder.previewController?.image = result
+        }
     }
   }
 }