Add delegate method for minimumColumnSpacing

Fix chiahsien#207
ApolloZhu · Dec 10, 2021 · de37d0b · de37d0b
1 parent 88b81c2
commit de37d0b
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/Source/CHTCollectionViewWaterfallLayout.swift b/Source/CHTCollectionViewWaterfallLayout.swift
@@ -53,6 +53,10 @@ private func > <T: Comparable>(lhs: T?, rhs: T?) -> Bool {
     @objc optional func collectionView(_ collectionView: UICollectionView,
                                        layout collectionViewLayout: UICollectionViewLayout,
                                        columnCountFor section: Int) -> Int
+
+    @objc optional func collectionView(_ collectionView: UICollectionView,
+                                       layout collectionViewLayout: UICollectionViewLayout,
+                                       minimumColumnSpacingFor section: Int) -> CGFloat
 
     @available(*, unavailable, renamed: "collectionView(_:layout:sizeForItemAt:)")
     @objc optional func collectionView (_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout,
@@ -205,10 +209,13 @@ public class CHTCollectionViewWaterfallLayout: UICollectionViewLayout {
     }
 
     public func itemWidth(inSection section: Int) -> CGFloat {
-        let columnCount = self.columnCount(forSection: section)
-        let spaceColumCount = CGFloat(columnCount - 1)
         let width = collectionViewContentWidth(ofSection: section)
-        return floor((width - (spaceColumCount * minimumColumnSpacing)) / CGFloat(columnCount))
+
+        let columnCount = self.columnCount(forSection: section)
+        let spaceColumnCount = CGFloat(columnCount - 1)
+        let columnSpacing = delegate?.collectionView?(collectionView!, layout: self, minimumColumnSpacingFor: section) ?? minimumColumnSpacing
+
+        return floor((width - (spaceColumnCount * columnSpacing)) / CGFloat(columnCount))
     }
 
     override public func prepare() {