//自定義UICollectionViewLayout必須要重寫的三個(gè)協(xié)議方法
//1.計(jì)算每個(gè)item的大小和位置
- (void)prepareLayout;
//2.返回每個(gè)item的布局屬性
- (nullable NSArray<__kindof UICollectionViewLayoutAttributes *> *)layoutAttributesForElementsInRect:(CGRect)rect;
//3.返回collectionView的總高度
- (CGSize)collectionViewContentSize;

- (instancetype)init {
  self = [super init];
  if (self) {
    _attributesArray = [NSMutableArray array];
    // 默認(rèn)值設(shè)置為2列
    _numberOfColumns = 2;
    _contentHeight = 0.0f;
    _cellMargin = 5.0f;/**< 用來表示間距的屬性 */
  }
  return self;
}

- (CGFloat)itemWidth {
  //所有邊距的和.兩列時(shí)有三個(gè)邊距, 三列時(shí)有四個(gè)邊距,邏輯強(qiáng)大就是好...
  CGFloat allMargin = (_numberOfColumns + 1) * _cellMargin;
  //除去邊界之后的總寬度
  CGFloat noMarginWidth = CGRectGetWidth(self.collectionView.bounds) - allMargin;
  //出去邊距的總寬度除以列數(shù)得到每一列的寬度(也就是itemWidth)
  return noMarginWidth / _numberOfColumns;
}

- (void)prepareLayout {
  // 定義變量記錄高度最小的列,初始為第0列高度最小.
#pragma mark - 注意這個(gè)是從0開始算的啊!!!
  NSInteger shortestColumn = 0;
#pragma mark - 注意這個(gè)是從0開始算的啊!!!
  // 存儲每一列的總高度.因?yàn)樘砑訄D片的列高度會變,所以需要定義一個(gè)數(shù)組來記錄列的總高度.
  NSMutableArray *columnHeightArray = [NSMutableArray array];
  // 設(shè)置列的初始高度為邊距的高度,沒毛病!!!
  for (int i = 0; i < _numberOfColumns; i++) {
    // 所有列初始高度均設(shè)置為cell的間距
    [columnHeightArray addObject:@(_cellMargin)];
  }
  // 遍歷collectionView中第 0 區(qū)中的所有item
  for (int i = 0; i < [self.collectionView numberOfItemsInSection:0]; i++) {
    //需要用到這個(gè)玩意,提前拿到.
    NSIndexPath *indexPath = [NSIndexPath indexPathForItem:i inSection:0];
    // 創(chuàng)建系統(tǒng)需要的布局屬性對象,看后邊的參數(shù)就知道這就是每個(gè)item的布局屬性了
    UICollectionViewLayoutAttributes *layoutAttributes = [UICollectionViewLayoutAttributes layoutAttributesForCellWithIndexPath: indexPath];
    // 將布局屬性放入數(shù)組中,這個(gè)數(shù)組當(dāng)然是一開始定義的布局屬性數(shù)組了
    [_attributesArray addObject:layoutAttributes];
    // 設(shè)置每個(gè)item的位置(x, y, width, height)
    // 橫坐標(biāo)的起始位置
#pragma mark - 比如一共兩列,現(xiàn)在要放一張圖片上去,需要放到高度最小的那一列.
#pragma mark - 假設(shè)第0列最短,那么item的x坐標(biāo)就是從一個(gè)邊距寬度那里開始.
#pragma mark - (itemWidth + cellMargin)為一個(gè)整體
    CGFloat x = (self.itemWidth + _cellMargin) * shortestColumn + _cellMargin;
    // 縱坐標(biāo)就是 總高度數(shù)組 中最小列對應(yīng)的高度
#pragma mark - 圖片始終是添加在高度最小的那一列
    CGFloat y = [columnHeightArray[column] floatValue];/**<注意類型轉(zhuǎn)換 */
    // 寬度沒什么好說的
    CGFloat width = self.itemWidth;
#pragma mark - 這里給自定義的類聲明了一個(gè)協(xié)議,通過協(xié)議得到圖片的高度,調(diào)用時(shí)機(jī)就是需要item高度的時(shí)候
#pragma mark - 將Item的寬度傳給代理人(ViewController),VC計(jì)算好高度后將高度返回給自定義類
#pragma mark - 也就是↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓
    CGFloat height = [self.delegate collectionView:self.collectionView
                        layout:self
                         width:self.itemWidth
               heightForItemAtIndexPath:indexPath];
    // 大功告成,設(shè)置item的位置信息,沒什么好說
    layoutAttributes.frame = CGRectMake(x, y, width, height);
    // 上邊廢了半天勁放了一個(gè)item上去了,總高度數(shù)組是不是該更新一下數(shù)據(jù)了
    columnHeightArray[shortestColumn] = @([columnHeightArray[shortestColumn] floatValue] + height + _cellMargin);
    // 整個(gè)內(nèi)容的高度,通過比較得到較大值作為整個(gè)內(nèi)容的高度
    self.contentHeight = MAX(self.contentHeight, [columnHeightArray[shortestColumn] floatValue]);
    // 剛才放了一個(gè)item上去,那么此時(shí)此刻哪一列的高度比較低呢
    for (int i = 0; i < _numberOfColumns; i++) {
      //當(dāng)前列的高度(剛才添加item的那一列)
      CGFloat currentHeight = [columnHeightArray[shortestColumn] floatValue];
      // 取出第i列中存放列高度
      CGFloat height = [columnHeightArray[i] floatValue];
      if (currentHeight > height) {
        //第i列高度(height)最低時(shí),高度最低的列(shortestColumn)當(dāng)然就是第i列了
        shortestColumn = i;
      }
    }
  }
// 思考下只使用上邊的代碼會出現(xiàn)什么問題
// 并不能影響心情,請不要恐慌...
// 提示:這個(gè)方法會被多次調(diào)用,數(shù)組
}

// 2.返回的是, 每個(gè)item對應(yīng)的布局屬性
- (NSArray<UICollectionViewLayoutAttributes *> *)layoutAttributesForElementsInRect:(CGRect)rect {
  return _attributesArray;
}

// 3.返回CollectionView的滾動范圍
- (CGSize)collectionViewContentSize {
  return CGSizeMake(0, _contentHeight);
}

CustomCollectionViewLayout *layout = [[CustomCollectionViewLayout alloc] init];
layout.numberOfColumns = 2;/**< 在ViewController里設(shè)置瀑布流的列數(shù),2列或3列為最佳 */
layout.delegate = self;/**< 指定VC為計(jì)算高度協(xié)議方法的代理人 */

- (CGFloat)collectionView:(UICollectionView *)collectionView
          layout:(UICollectionViewLayout *)collectionViewLayout
          width:(CGFloat)width
 heightForItemAtIndexPath:(nonnull NSIndexPath *)indexPath {
  UIImage *image = _imagesArray[indexPath.row];
  // 根據(jù)傳過來的寬度來設(shè)置一個(gè)合適的矩形, 高度設(shè)為CGFLOAT_MAX表示以寬度來計(jì)算高度
  CGRect boundingRect = CGRectMake(0, 0, width, CGFLOAT_MAX);
  // 通過系統(tǒng)函數(shù)來得到最終的矩形,需要引入頭文件
  // #import <AVFoundation/AVFoundation.h>
  CGRect imageCurrentRect = AVMakeRectWithAspectRatioInsideRect(image.size, boundingRect);
  return imageCurrentRect.size.height;
}