(本文主要讲一下现在比较流行的一种布局方式----瀑布流布局 如有写的不好的地方 还请多多指正 感谢)

1、功能分析

如图所示: 我们可以看到该布局中的每个元素有一个共同的特点就是等宽不等高. 而且当一行排列完成之后在下一行进行排列时都是从最短的那一列开始排,否则的话就会让每一列的差距越来越大从而显得非常不美观.

2、实现思路

• 根据需求 该页面需要有滚动效果 而且可以展示很多数据 所以决定用UICollectionView来完成 那么UICollectionView中具体的每一个cell如何排列就是我们需要解决的问题了.也就是说我们需要计算出每一个cell的frame.
• 接下来就对cell的x值,y值,宽度,高度进行逐一计算
• 宽度w
  我们可以很直观的从图中看出宽度w=(collectionView的宽度 - cell左边的边距 - cell右边的边距 - (总共的列数 - 1) * 每一列之间的间距) / 总共的列数
• 高度h
  高度h是根据具体项目中的模型本身的高度来决定
• x,y值
  根据上图可以发现每一列中所有cell的x值是一样的 所以要算x值只需要求出列号就行了. y值就是最短的那一列的cell最大y值再加上间距
  综上所述,现在需要做的首要任务就是找出最短的那一列.所以我们需要通过遍历每一列的高度来找出最短的那一列.

3、code

以上进行简单分析之后就要开始动手了 既然是自定义布局 我们就需要创建一个继承自UICollectionViewLayout的类来实现布局 在这个类中我们需要实现以下几个方法

• - (void)prepareLayout这个方法是用来进行初始化的 实现代码如下

-(void)prepareLayout
{
    [super prepareLayout];
    //清除之前计算的所有高度
    [self.colunmHeights removeAllObjects];
    for (NSInteger i = 0; i < ZDDefaultColumnCount; i++) {
        [self.colunmHeights addObject:@(ZDDefaultEdgeInsets.top)];
    }
    //清除之前所有的布局
    [self.attrsArray removeAllObjects];
    //创建每一个cell对应的布局属性
    NSInteger count = [self.collectionView numberOfItemsInSection:0];
    for (NSInteger i = 0; i < count; i++) {
        //创建位置
        NSIndexPath *indexPath = [NSIndexPath indexPathForItem:i inSection:0];
        //获取indexPath位置cell对应的布局属性
        UICollectionViewLayoutAttributes *attrs = [self layoutAttributesForItemAtIndexPath:indexPath];
        [self.attrsArray addObject:attrs];
    }
}

其中self.colunmHeights和self.attrsArray是自己定义的两个属性 分别用来保存所有列的当前高度以及所有cell的布局属性(两个都是可变数组类型)

• - (NSArray<UICollectionViewLayoutAttributes *> *)layoutAttributesForElementsInRect:(CGRect)rect这个方法是用来决定cell的排布 实现代码如下

-(NSArray<UICollectionViewLayoutAttributes *> *)layoutAttributesForElementsInRect:(CGRect)rect
{
    return self.attrsArray;
}

• - (UICollectionViewLayoutAttributes *)layoutAttributesForItemAtIndexPath:(NSIndexPath *)indexPath这个方法是用来返回indexPath的位置所对应的cell的布局属性的 实现代码如下

-(UICollectionViewLayoutAttributes *)layoutAttributesForItemAtIndexPath:(NSIndexPath *)indexPath
{
    //创建布局属性
    UICollectionViewLayoutAttributes *attrs = [UICollectionViewLayoutAttributes layoutAttributesForCellWithIndexPath:indexPath];
    //collectionView的宽度
    CGFloat collectionViewW = self.collectionView.frame.size.width;
    //设置布局属性的frame
    CGFloat w = (collectionViewW - ZDDefaultEdgeInsets.left - ZDDefaultEdgeInsets.right - (ZDDefaultColumnCount - 1) * ZDDefaultColumnMargin) / ZDDefaultColumnCount;
    CGFloat h = 50 + arc4random_uniform(100);
    
    //找出高度最短的那一列
    NSInteger destColumn = 0;
    CGFloat minColumnHeight = [self.colunmHeights[0] doubleValue];
    for (NSInteger i = 1; i < ZDDefaultColumnCount; i++) {
        CGFloat columnHeight = [self.colunmHeights[i] doubleValue];
        
        if (minColumnHeight > columnHeight) {
            minColumnHeight = columnHeight;
            destColumn = i;
        }
    }
    CGFloat x = ZDDefaultEdgeInsets.left + destColumn * (w + ZDDefaultColumnMargin);
    CGFloat y = minColumnHeight;
    if (y != ZDDefaultEdgeInsets.top) {
        y += ZDDefaultRowMargin;
    }
    attrs.frame = CGRectMake(x, y, w, h);
    
    //更新最短那列的高度
    self.colunmHeights[destColumn] = @(CGRectGetMaxY(attrs.frame));
    
    // 记录内容的高度
    CGFloat columnHeight = [self.columnHeights[destColumn] doubleValue];
    if (self.contentHeight < columnHeight) {
        self.contentHeight = columnHeight;
    }
    return attrs;
}

其中self.contentHeight是自定义的一个属性 用来保存内容的高度

• - (CGSize)collectionViewContentSize这个方法是为了让collectionView可以滚动起来 实现代码如下

-(CGSize)collectionViewContentSize
{
    return CGSizeMake(0, self.contentHeight + ZDDefaultEdgeInsets.bottom);
}

4、优化

• 考虑到代码的复用性 方便直接将代码拖到下个项目中去
  (比如具体要展示多少列,每个cell之间的间距等等都需要根据具体的项目需求来决定) 所以对代码进行优化
• 优化思路是根据UITableViewDelegate tableView具体展示什么数据,展示多少数据都是由其数据源和代理方法来具体实现的,所以我也设计了一个代理属性 具体显示多少列 每个cell之间的间距都是有代理方法来实现的 具体实现代码如下:
• 在ZDWaterfallLayout.h文件中:

@class ZDWaterfallLayout;
@protocol ZDWaterfallLayoutDelegate <NSObject>
@required
-(CGFloat)waterfallLayout:(ZDWaterfallLayout *)waterfallLayout heightForItemAtIndex:(NSInteger)index itemWidth:(CGFloat)itemWidth;
@optional
//列数
-(CGFloat)columnCountInWaterfallLayout:(ZDWaterfallLayout *)waterfallLayout;
//每一列之间的间距
-(CGFloat)columnMarginInWaterfallLayout:(ZDWaterfallLayout *)waterfallLayout;
//每一行之间的间距
-(CGFloat)rowMarginInWaterfallLayout:(ZDWaterfallLayout *)waterfallLayout;
//cell的边距
-(UIEdgeInsets)edgeInsetsInWaterfallLayout:(ZDWaterfallLayout *)waterfallLayout;
@end
@interface ZDWaterfallLayout : UICollectionViewLayout
/**布局代理属性*/
@property (nonatomic,weak) id<ZDWaterfallLayoutDelegate>delegate ;
@end

• 在ZDWaterfallLayout.m文件中

-(CGFloat)rowMargin
{
    if ([self.delegate respondsToSelector:@selector(rowMarginInWaterfallLayout:)]) {
        return [self.delegate rowMarginInWaterfallLayout:self];
    }else{
        return ZDDefaultRowMargin;
    }
}
-(CGFloat)columnMargin
{
    if ([self.delegate respondsToSelector:@selector(columnMarginInWaterfallLayout:)]) {
        return [self.delegate columnMarginInWaterfallLayout:self];
    }else{
        return ZDDefaultColumnMargin;
    }
}
-(NSInteger)columnCount
{
    if ([self.delegate respondsToSelector:@selector(columnCountInWaterfallLayout:)]) {
        return [self.delegate columnCountInWaterfallLayout:self];
    }else{
        return ZDDefaultColumnCount;
    }
}
-(UIEdgeInsets)edgeInsets
{
    if ([self.delegate respondsToSelector:@selector(edgeInsetsInWaterfallLayout:)]) {
        return [self.delegate edgeInsetsInWaterfallLayout:self];
    }else{
        return ZDDefaultEdgeInsets;
    }
}

ZDDefaultRowMargin ZDDefaultColumnMargin ZDDefaultColumnCount ZDDefaultEdgeInsets这是我自定义的默认值
接下来想要改变布局效果就很简单了 只需要通过具体实现几个代理方法就可以搞定 比如我想排5列 只需要实现如下方法即可:

-(CGFloat)columnCountInWaterfallLayout:(ZDWaterfallLayout *)waterfallLayout
{
    return 5;
}

显示效果如下: