//自定義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開始算的啊!!!  // 存儲(chǔ)每一列的總高度.因?yàn)樘砑訄D片的列高度會(huì)變,所以需要定義一個(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)需要的布局屬性對(duì)象,看后邊的參數(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ù)組 中最小列對(duì)應(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;      }    }  }// 思考下只使用上邊的代碼會(huì)出現(xiàn)什么問題// 并不能影響心情,請(qǐng)不要恐慌...// 提示:這個(gè)方法會(huì)被多次調(diào)用,數(shù)組}

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

// 3.返回CollectionView的滾動(dòng)范圍- (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;}