详解IOS中如何实现瀑布流效果

//自定义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;/**

- (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  height) {
        //第i列高度(height)最低时,高度最低的列(shortestColumn)当然就是第i列了
        shortestColumn = i;
      }
    }
  }
// 思考下只使用上边的代码会出现什么问题
// 并不能影响心情,请不要恐慌...
// 提示:这个方法会被多次调用,数组
}

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

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

CustomCollectionViewLayout *layout = [[CustomCollectionViewLayout alloc] init];
layout.numberOfColumns = 2;/**

- (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 
  CGRect imageCurrentRect = AVMakeRectWithAspectRatioInsideRect(image.size, boundingRect);
  return imageCurrentRect.size.height;
}