refactored for requested functionality

Author: Senbazuru1075

LUX/Classes/Base/CollectionViews/LUXPinterestStyleLayout.swift

@@ -20,8 +20,8 @@ public class LUXPinterestStyleLayout: UICollectionViewLayout {
     private var _baseItemWidth: CGFloat? { didSet { invalidateLayout() }}
     public func setBaseItemWidth(_ width: CGFloat) { _baseItemWidth = width }
 
-    public var setWidthForColumn: Int?
-    private var widthForColumn: ((Int) -> CGFloat?)? = { _ in return nil }
+    public var setWidthCalculatorForColumn: CGFloat?
+    public var calculateMultiWidthColumnForColumn: ((_ multiWidth: CGFloat...) -> [CGFloat?])?
 
     private var contentHeight: CGFloat = 0.0
     private var contentWidth: CGFloat {
@@ -72,18 +72,28 @@ public class LUXPinterestStyleLayout: UICollectionViewLayout {
         var offSet: CGFloat = 0
 
         if let columnCount = _columnCount {
-            let columnWidth = contentWidth / CGFloat(_columnCount!)
+            let columnWidth = contentWidth / CGFloat(columnCount)
             offSet = columnWidth
+            
             for column in 0..<columnCount {
-                let width = (widthForColumn ?? { _ in return 0 })(setWidthForColumn ?? 0) ?? offSet
-                xOffSet.append(width * CGFloat(column))
+                
+                if let multiWidth = calculateMultiWidthColumnForColumn {
+                    if let setWidth = setWidthCalculatorForColumn {
+                        for ( _, size) in multiWidth(setWidth).enumerated() {
+                            xOffSet.append(size! * CGFloat(column))
+                        }
+                    }
+                }
+                xOffSet.append(offSet * CGFloat(column))
             }
         } else if let baseWidth = _baseItemWidth {
             if _columnCount == nil {
                 columnCount = Int(contentWidth / baseWidth)
                 offSet = baseWidth / CGFloat(columnCount)
 
                 for column in 0..<columnCount {
+            
+                    
                     xOffSet.append(offSet * CGFloat(column))
                 }
             } else {
@@ -109,7 +119,7 @@ public class LUXPinterestStyleLayout: UICollectionViewLayout {
             var column = 0
             var yOffset: [CGFloat] = .init(repeating: 0, count: _columnCount!)
 
-            for item in 0..<itemCount {
+                for item in 0..<itemCount {
             let indexPath = IndexPath(item: item, section: 0)
 
             let photoHeight = heightForCellAtIndexPath?(collectionView, indexPath) ?? 180
@@ -125,11 +135,9 @@ public class LUXPinterestStyleLayout: UICollectionViewLayout {
         let attributes = UICollectionViewLayoutAttributes(forCellWith: indexPath)
         attributes.frame = insetFrame
         cache.append(attributes)
-
         contentHeight = max(contentHeight, frame.maxY)
         yOffset[column] = yOffset[column] + height
-
-            column = column < ( _columnCount! - 1) ? (column + 1) : 0
+        column = column < ( _columnCount! - 1) ? (column + 1) : 0
         }
     }