Add checks in case viewer get out of sync with collection view

freyzou · May 7, 2017 · 65063bc · 65063bc
1 parent 5b83fdb
commit 65063bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Source/ViewerController.swift b/Source/ViewerController.swift
@@ -240,7 +240,7 @@ public class ViewerController: UIViewController {
             } else {
                 self.scrollView.configure()
             }
-            if !self.collectionView.indexPathsForVisibleItems.contains(self.currentIndexPath) {
+            if !self.collectionView.indexPathsForVisibleItems.contains(self.currentIndexPath) && self.collectionView.numberOfSections > self.currentIndexPath.section && self.collectionView.numberOfItems(inSection: self.currentIndexPath.section) > self.currentIndexPath.item {
                 self.collectionView.scrollToItem(at: self.currentIndexPath, at: .bottom, animated: true)
             }
         }