iOS_WaterfallFlowLayout瀑布流用重写UICollectionViewFlowLayout类实现

//  WaterfallFlowLayout.h
@interface WaterfallFlowLayout : UICollectionViewFlowLayout

// 总列数
@property (nonatomic, assign) NSInteger columnCount;

// 商品数据数组
@property (nonatomic, strong) NSArray *goodsArray;

@end

<span style="font-family: Arial, Helvetica, sans-serif;">//  WaterfallFlowLayout.m</span>

#import "WaterfallFlowLayout.h"
#import "Good.h"

@interface WaterfallFlowLayout ()
// 所有item的属性的数组
@property (nonatomic, strong) NSArray *layoutAttributesArray;
@end

@implementation WaterfallFlowLayout

/**
 *  布局准备方法 当collectionView的布局发生变化时 会被调用
 *  通常是做布局的准备工作 itemSize.....
 *  UICollectionView 的 contentSize 是根据 itemSize 动态计算出来的
 */
- (void)prepareLayout {
    // 根据列数 计算item的宽度 宽度是一样的
    CGFloat contentWidth = self.collectionView.bounds.size.width - self.sectionInset.left - self.sectionInset.right; //减去分区的边框
    CGFloat marginX = self.minimumInteritemSpacing;  //最小左右间距
    CGFloat itemWidth = (contentWidth - marginX * (self.columnCount - 1)) / self.columnCount;
    
    // 计算布局属性
    [self computeAttributesWithItemWidth:itemWidth];
}

#pragma mark 根据itemWidth计算布局属性
- (void)computeAttributesWithItemWidth:(CGFloat)itemWidth {
    
    // 定义一个列高数组 记录每一列的总高度
    CGFloat columnHeight[self.columnCount];
    // 初始化
    for (int i = 0; i < self.columnCount; i++) {
        columnHeight[i] = self.sectionInset.top;
    }
    
    // 遍历 goodsList 数组计算相关的属性
    NSMutableArray *attributesArray = [NSMutableArray arrayWithCapacity:self.goodsArray.count];
    
    //因为item数是跟当前的商品数一样的 每次改变都动态计算一遍
    for (NSInteger i = 0; i < self.goodsArray.count; i++) {
        Good *good = self.goodsArray[i];
        // 建立布局属性
        NSIndexPath *indexPath = [NSIndexPath indexPathForItem:i inSection:0];
        // 获得当前item的布局属性
        UICollectionViewLayoutAttributes *attributes = [UICollectionViewLayoutAttributes layoutAttributesForCellWithIndexPath:indexPath];
        // 找出最短列号
        NSInteger column = [self shortestColumn:columnHeight];
        // X值
        CGFloat itemX = (itemWidth + self.minimumInteritemSpacing) * column + self.sectionInset.left;
        // Y值 = 当前列的总高度
        CGFloat itemY = columnHeight[column];
        // 等比例缩放 计算item的高度
        CGFloat itemH = good.h * itemWidth / good.w;
        // 设置frame
        attributes.frame = CGRectMake(itemX, itemY, itemWidth, itemH);
        [attributesArray addObject:attributes];
        
        //!!!!!!!!
        self.itemSize = CGSizeMake(itemWidth, itemH);

        // 累加当前列高
        columnHeight[column] += itemH + self.minimumLineSpacing;
        
    }
    // 给属性数组设置数值
    self.layoutAttributesArray = attributesArray.copy;
}

#pragma mark 找出columnHeight数组中最短列号 追加数据的时候追加在最短列中
- (NSInteger)shortestColumn:(CGFloat *)columnHeight {
    
    CGFloat min = CGFLOAT_MAX;
    NSInteger column = 0;
    // 循环列高数组
    for (int i = 0; i < self.columnCount; i++) {
        if (columnHeight[i] < min) {
            min = columnHeight[i];
            column = i;
        }
    }
    return column;
}

/**
 *  跟踪效果：当到达要显示的区域时 会计算所有显示item的属性
 *           一旦计算完成 所有的属性会被缓存 不会再次计算
 *  @return 返回布局属性(UICollectionViewLayoutAttributes)数组
 */
- (NSArray *)layoutAttributesForElementsInRect:(CGRect)rect {
    // 直接返回计算好的布局属性数组
    return self.layoutAttributesArray;
}