思路

• 瀑布流核心思想是,每張圖片都是從高度最低的一列去拼接,同時圖片的尺寸取決于需求,而不是統一的尺寸,進而形成參差不齊的效果
• 通過UICollectionView的自定義UICollectionViewLayout(布局)去實現
• 相比用ScrollerView去實現瀑布流,UICollectionView更為簡單,因為它已經具備循環利用的功能,如果使用ScrollerView還需要自己去實現循環利用

步驟

• 新建一個UICollectionViewWaterLayout,繼承UICollectionViewLayout
• 設置代理UICollectionViewWaterLayoutDelegate，用于獲取每個item的高度、列數、列間距、行間距等,其中item的高度是必須的，外界通過內部提供的寬度,按比例計算出高度。之所以用代理的方式去獲取這些屬性，而不是直接用property屬性,是因為代理什么時候調用是內部決定的，當用property屬性去獲取這些屬性時，外界可以在任何時候去改變這些屬性，這就意味著內部要隨時針對這種改變去做響應處理(重寫set方法)比較麻煩。

protocol UICollectionViewWaterLayoutDelegate <NSObject>
  @required
  // 返回item的高度(根據width按比例去計算)
  - (CGFloat)waterLayout:(UICollectionViewWaterLayout *)waterLayout heightForItemAtIndexPath:(NSIndexPath *)indexPath andWidth:(CGFloat)width;
  @optional
  // 返回列數
  - (NSInteger)numberOfColumnsInWaterLayout:(UICollectionViewWaterLayout *)waterLayout;
  // 返回行間距
  - (CGFloat)rowMarginForWaterLayout:(UICollectionViewWaterLayout *)waterLayout;
  // 返回列間距
  - (CGFloat)columnMarginForWaterLayout:(UICollectionViewWaterLayout *)waterLayout;
  // 返回內邊距
  - (UIEdgeInsets)edgeInsetsForWaterLayout:(UICollectionViewWaterLayout *)waterLayout;
@end

• 添加成員變量(.h)

 /** 緩存每個item的UICollectionViewLayoutAttributes屬性,避免重復計算 */
property (nonatomic, strong) NSMutableArray *attributesArr;
/** 記錄每列的高度 */
property (nonatomic, strong) NSMutableArray *columnHeights;

• 重寫prepareLayout方法去初始化布局,注意要先調用父類同名方法

  - (void)prepareLayout { 
    // 要先調用父類的prepareLayout 
    [super prepareLayout]; 
    // 清空原來緩存的列的高度和cell的未知屬性,否則當重新調用prepareLayout,會累加 
    [self.attributesArr removeAllObjects]; 
    [self.columnHeights removeAllObjects]; 
    // 設置列高度初始值為頂部內邊距 
    for (int i = 0; i < layoutColumn; i++) {
       [self.columnHeights addObject:[NSNumber numberWithDouble:layoutInsets.top]];
     } 
    // 設置每個item的布局屬性,并緩存 
    NSInteger itemsTotal = [self.collectionView numberOfItemsInSection:0];
    for (int i = 0; i < itemsTotal; i++) { 
      NSIndexPath *indexPath = [NSIndexPath indexPathForItem:i inSection:0];
      UICollectionViewLayoutAttributes *attributes = [UICollectionViewLayoutAttributes layoutAttributesForCellWithIndexPath:indexPath]; 
      attributes.frame = [self getItemFrameWithIndexPath:indexPath]; [self.attributesArr addObject:attributes]; 
    }
}

• 瀑布流核心計算方法，之所以自己定義一個方法去計算,而不是直接在layoutAttributesForItemAtIndexPath:實現計算方法,并調用它來獲取每個item的位置屬性,是因為layoutAttributesForItemAtIndexPath：會被系統重復調用,導致列的高度計算錯誤

  /** 每個item都是放到當前高度最低的列下面 
    * 要計算x,y,就要找到高度最低的列 
    * 寬度 = (collectionView寬度 - 左內邊距 - 右內邊距 - (列數 - 1)*列間距)/列數 * 高度通過代理獲取 
    * x = 左內邊距 + 列號 * (寬度 + 列間距) 
    * y = 列的高度 + 行間距 
    */
   - (CGRect)getItemFrameWithIndexPath:(NSIndexPath *)indexPath { 
    // 高度最低的列 
    CGFloat minHeight = [self.columnHeights[0] doubleValue]; 
    NSInteger minHeightCol = 0;
    for (int i = 1; i < self.columnHeights.count; i++) { 
      CGFloat columnHeight = [self.columnHeights[i] doubleValue]; 
      if (minHeight > columnHeight) { 
        minHeight = columnHeight; minHeightCol = i; 
      } 
    }  
    // 計算item的位置 
    CGFloat collectionViewW = self.collectionView.frame.size.width;
    CGFloat itemW = (collectionViewW - self.insets.left - self.insets.right - (self.columns - 1)*self.columnMargin) / self.columns; 
    // 通過使用者實現的代理拿到item的高度
    CGFloat itemH = [self.delegate waterLayout:self heightForItemAtIndexPath:indexPath andWidth:itemW]; 
    CGFloat itemX = (itemW + self.columnMargin) * minHeightCol; 
    CGFloat itemY = minHeight + self.rowMargin; CGRect frame = CGRectMake(itemX, itemY, itemW, itemH);  
   // 更新列的高度 
    self.columnHeights[minHeightCol] = [NSNumber numberWithDouble:CGRectGetMaxY(frame)];  
    return frame;
}

• 重寫layoutAttributesForElementsInRect:,告訴collectionView每個item以什么形式排布,這個方法會被反復調用,所以不適合將計算過程寫在這個里面,而是直接用上面計算好的緩存數據

  - (NSArray<UICollectionViewLayoutAttributes *> *)layoutAttributesForElementsInRect:(CGRect)rect { 
    return self.attributesArr;
  }

• 重寫layoutAttributesForItemAtIndexPath:方法,如果不重寫,UICollectionView在調用切換布局方式的方法時會崩潰

  - (UICollectionViewLayoutAttributes *)layoutAttributesForItemAtIndexPath:(NSIndexPath *)indexPath { 
    return self.attributesArr[indexPath.item];
  }

• 重寫collectionViewContentSize,否則UICollectionView無法滾動,返回的就是滾動范圍

  - (CGSize)collectionViewContentSize { 
    // 找到高度最高的列 
    // 滾動高度 = 高度最高的列的高度 + 底部內邊距 
    // 記錄最低高度
    CGFloat maxHeight = [self.columnHeights[0] doubleValue];
    for(int i = 1; i < self.columnHeights.count; i++) { 
      CGFloat columnHeight = [self.columnHeights[i] doubleValue]; 
      if (maxHeight < columnHeight) { 
        maxHeight = columnHeight;
      }
    }
   return CGSizeMake(0, maxHeight + layoutInsets.bottom);
  }