//自定义UICollectionViewLayout必须要重写的三个协议方法
//1.计算每个item的大小和位置
- (void)prepareLayout;
//2.返回每个item的布局属性
- (nullable NSArray<__kindof UICollectionViewLayoutAttributes *> *)layoutAttributesForElementsInRect:(CGRect)rect;
//3.返回collectionView的总高度
- (CGSize)collectionViewContentSize;

- (instancetype)init {
  self = [super init];
  if (self) {
    _attributesArray = [NSMutableArray array];
    // 默认值设置为2列
    _numberOfColumns = 2;
    _contentHeight = 0.0f;
    _cellMargin = 5.0f;/**< 用来表示间距的属性 */
  }
  return self;
}

- (CGFloat)itemWidth {
  //所有边距的和.两列时有三个边距, 三列时有四个边距,逻辑强大就是好...
  CGFloat allMargin = (_numberOfColumns + 1) * _cellMargin;
  //除去边界之后的总宽度
  CGFloat noMarginWidth = CGRectGetWidth(self.collectionView.bounds) - allMargin;
  //出去边距的总宽度除以列数得到每一列的宽度(也就是itemWidth)
  return noMarginWidth / _numberOfColumns;
}

- (void)prepareLayout {
  // 定义变量记录高度最小的列,初始为第0列高度最小.
#pragma mark - 注意这个是从0开始算的啊!!!
  NSInteger shortestColumn = 0;
#pragma mark - 注意这个是从0开始算的啊!!!
  // 存储每一列的总高度.因为添加图片的列高度会变,所以需要定义一个数组来记录列的总高度.
  NSMutableArray *columnHeightArray = [NSMutableArray array];
  // 设置列的初始高度为边距的高度,没毛病!!!
  for (int i = 0; i < _numberOfColumns; i++) {
    // 所有列初始高度均设置为cell的间距
    [columnHeightArray addObject:@(_cellMargin)];
  }
  // 遍历collectionView中第 0 区中的所有item
  for (int i = 0; i < [self.collectionView numberOfItemsInSection:0]; i++) {
    //需要用到这个玩意,提前拿到.
    NSIndexPath *indexPath = [NSIndexPath indexPathForItem:i inSection:0];
    // 创建系统需要的布局属性对象,看后边的参数就知道这就是每个item的布局属性了
    UICollectionViewLayoutAttributes *layoutAttributes = [UICollectionViewLayoutAttributes layoutAttributesForCellWithIndexPath: indexPath];
    // 将布局属性放入数组中,这个数组当然是一开始定义的布局属性数组了
    [_attributesArray addObject:layoutAttributes];
    // 设置每个item的位置(x, y, width, height)
    // 横坐标的起始位置
#pragma mark - 比如一共两列,现在要放一张图片上去,需要放到高度最小的那一列.
#pragma mark - 假设第0列最短,那么item的x坐标就是从一个边距宽度那里开始.
#pragma mark - (itemWidth + cellMargin)为一个整体
    CGFloat x = (self.itemWidth + _cellMargin) * shortestColumn + _cellMargin;
    // 纵坐标就是 总高度数组 中最小列对应的高度
#pragma mark - 图片始终是添加在高度最小的那一列
    CGFloat y = [columnHeightArray[column] floatValue];/**<注意类型转换 */
    // 宽度没什么好说的
    CGFloat width = self.itemWidth;
#pragma mark - 这里给自定义的类声明了一个协议,通过协议得到图片的高度,调用时机就是需要item高度的时候
#pragma mark - 将Item的宽度传给代理人(ViewController),VC计算好高度后将高度返回给自定义类
#pragma mark - 也就是↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓
    CGFloat height = [self.delegate collectionView:self.collectionView
                        layout:self
                         width:self.itemWidth
               heightForItemAtIndexPath:indexPath];
    // 大功告成,设置item的位置信息,没什么好说
    layoutAttributes.frame = CGRectMake(x, y, width, height);
    // 上边废了半天劲放了一个item上去了,总高度数组是不是该更新一下数据了
    columnHeightArray[shortestColumn] = @([columnHeightArray[shortestColumn] floatValue] + height + _cellMargin);
    // 整个内容的高度,通过比较得到较大值作为整个内容的高度
    self.contentHeight = MAX(self.contentHeight, [columnHeightArray[shortestColumn] floatValue]);
    // 刚才放了一个item上去,那么此时此刻哪一列的高度比较低呢
    for (int i = 0; i < _numberOfColumns; i++) {
      //当前列的高度(刚才添加item的那一列)
      CGFloat currentHeight = [columnHeightArray[shortestColumn] floatValue];
      // 取出第i列中存放列高度
      CGFloat height = [columnHeightArray[i] floatValue];
      if (currentHeight > height) {
        //第i列高度(height)最低时,高度最低的列(shortestColumn)当然就是第i列了
        shortestColumn = i;
      }
    }
  }
// 思考下只使用上边的代码会出现什么问题
// 并不能影响心情,请不要恐慌...
// 提示:这个方法会被多次调用,数组
}

// 2.返回的是, 每个item对应的布局属性
- (NSArray<UICollectionViewLayoutAttributes *> *)layoutAttributesForElementsInRect:(CGRect)rect {
  return _attributesArray;
}

// 3.返回CollectionView的滚动范围
- (CGSize)collectionViewContentSize {
  return CGSizeMake(0, _contentHeight);
}

CustomCollectionViewLayout *layout = [[CustomCollectionViewLayout alloc] init];
layout.numberOfColumns = 2;/**< 在ViewController里设置瀑布流的列数,2列或3列为最佳 */
layout.delegate = self;/**< 指定VC为计算高度协议方法的代理人 */

- (CGFloat)collectionView:(UICollectionView *)collectionView
          layout:(UICollectionViewLayout *)collectionViewLayout
          width:(CGFloat)width
 heightForItemAtIndexPath:(nonnull NSIndexPath *)indexPath {
  UIImage *image = _imagesArray[indexPath.row];
  // 根据传过来的宽度来设置一个合适的矩形, 高度设为CGFLOAT_MAX表示以宽度来计算高度
  CGRect boundingRect = CGRectMake(0, 0, width, CGFLOAT_MAX);
  // 通过系统函数来得到最终的矩形,需要引入头文件
  // #import <AVFoundation/AVFoundation.h>
  CGRect imageCurrentRect = AVMakeRectWithAspectRatioInsideRect(image.size, boundingRect);
  return imageCurrentRect.size.height;
}