Added estimated number of rows for UICollectionViewFlowLayout with ve…

…rtical scroll direction. (#385)
Juanpe · Apr 30, 2021 · 62193db · 62193db
1 parent 19e7866
commit 62193db
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/Sources/Collections/CollectionViews/UICollectionView+CollectionSkeleton.swift b/Sources/Collections/CollectionViews/UICollectionView+CollectionSkeleton.swift
@@ -12,6 +12,14 @@ extension UICollectionView: CollectionSkeleton {
     var estimatedNumberOfRows: Int {
         guard let flowlayout = collectionViewLayout as? UICollectionViewFlowLayout else { return 0 }
         return Int(ceil(frame.height / flowlayout.itemSize.height))
+        switch flowlayout.scrollDirection {
+        case .vertical:
+            return Int(ceil(frame.height / flowlayout.itemSize.height))
+        case .horizontal:
+            return Int(ceil(frame.width / flowlayout.itemSize.width))
+        default:
+            return 0
+        }
     }
 
     var skeletonDataSource: SkeletonCollectionDataSource? {