一、思路
思路一：比较每一行所有列的cell的高度，从上到下（也就是从第一行开始),从最短的开始计算，(记录下b的高度和索引，从开始计算，依次类推)
思路二：设置上、下、左、右间距和行间距、列间距及列数。
思路三：实现的重要的方法。
二、代码先行。
1.自定义layout类。
//入口
#import <UIKit/UIKit.h> @protocol STRWaterLayoutDelegate; @interface STRWaterLayout : UICollectionViewLayout @property (nonatomic, weak)id<STRWaterLayoutDelegate>delegate; @end
@protocol STRWaterLayoutDelegate <NSObject> @required
- (CGFloat)waterLayout:(STRWaterLayout *)waterLayout layoutHeightAtindexPath:(NSIndexPath *)indexpath layoutItemWidth:(CGFloat)width;
@optional
- (UIEdgeInsets)edgeInsetsWithWaterLayout:(STRWaterLayout *)waterLayout;
- (CGFloat)columsNumberWithWaterLayout:(STRWaterLayout *)waterLayout;
- (CGFloat)rowsMarginWithWaterLayout:(STRWaterLayout *)waterLayout;
- (CGFloat)columsMarginWithWaterLaout:(STRWaterLayout *)waterLayout;
@end
//出口
#import "STRWaterLayout.h"
static const UIEdgeInsets layoutEdgeInsets = {, , , }; //上、下、左、右的间距
static const CGFloat columsMar = ; //列间距
static const CGFloat rowMar = ; //行间距
static const NSInteger columsNums = ; //列数 @interface STRWaterLayout() /**每个cell的高度等信息*/
@property(nonatomic, strong)NSMutableArray *columMinGapArray;
/**所有布局信息的数组*/
@property(nonatomic, strong)NSMutableArray *layoutAttributesArray;
/**布局的最终的高度*/
@property(nonatomic, assign)CGFloat contentHeight; - (UIEdgeInsets)edgeInsets; //上、下、左、右的间距
- (CGFloat)columsNumber; //列数
- (CGFloat)rowMargin; //行间距
- (CGFloat)columsMargin; //列间距
@end
@implementation STRWaterLayout #pragma mark --- delegate methods - (UIEdgeInsets)edgeInsets{ if (self.delegate && [self.delegate respondsToSelector:@selector(edgeInsetsWithWaterLayout:)]){ return [self.delegate edgeInsetsWithWaterLayout:self];
}else{
return layoutEdgeInsets;
}
}
- (CGFloat)columsNumber{ if(self.delegate && [self.delegate respondsToSelector:@selector(columsNumberWithWaterLayout:)]){
return [self.delegate columsNumberWithWaterLayout:self];
}else{
return columsNums;
}
}
- (CGFloat)columsMargin{
if(self.delegate && [self.delegate respondsToSelector:@selector(columsMarginWithWaterLaout:)]){
return [self.delegate columsMarginWithWaterLaout:self];
}else{
return columsMar;
}
}
- (CGFloat)rowMargin{
if(self.delegate && [self.delegate respondsToSelector:@selector(rowsMarginWithWaterLayout:)]){
return [self.delegate rowsMarginWithWaterLayout:self];
}else{
return rowMar;
}
}
#pragma mark --- private methods /**在重新布局时会依次调用这四个方法*/ /**每次重新布局时都会调用它*/
- (void)prepareLayout{
[super prepareLayout];
self.contentHeight = ;
[self.columMinGapArray removeAllObjects];
[self.layoutAttributesArray removeAllObjects];
for (NSInteger i = ; i< [self columsNumber]; i++) { //遍历所有的列数
[self.columMinGapArray addObject: @([self edgeInsets].top)];
}
//当前只有一组，所以这么处理
NSInteger numbers = [self.collectionView numberOfItemsInSection:];
//处理所有的布局数据
for (NSInteger y = ; y <numbers;y++) {
//calls layouts methods,调用布局的方法
UICollectionViewLayoutAttributes *layoutAttributes = [self layoutAttributesForItemAtIndexPath:[NSIndexPath indexPathForRow:y inSection:]];
//再把算出来的布局加到布局数组中。
[self.layoutAttributesArray addObject:layoutAttributes];
}
}
/**返回横向滚动或纵向滚动的contentSize*/
- (CGSize)collectionViewContentSize{
return CGSizeMake(self.collectionView.frame.size.width, self.contentHeight+[self edgeInsets].bottom);
}
/**返回所有UICollectionViewLayoutAttributes的属性的数组*/
- (nullable NSArray<__kindof UICollectionViewLayoutAttributes *> *)layoutAttributesForElementsInRect:(CGRect)rect{
return self.layoutAttributesArray;
}
/**返回每个indexPath的布局UICollectionViewLayoutAttributes*/
// 1.首先通过indexPath取出每个collectionViewLayout的布局（uicollectionlayoutAttributes）.
// 2.从存储每个cell的高度的数组中取出第一个cell的高度，然后遍历，存储最小高度的cell,并记下它的索引，然后在这个索引下的cell增加下一个cell,然后再遍历看看哪个cell的高度最低，再把它的高度和索引记下来，增加下下个cell,依次类推。
// 3.因为布局的宽度和x,y可以自己设定（也就是能控制），不能控制的是布局的高度，需要从外部传进来，因为需要等比例的，所以需要传当前indexPath的布局的宽度，然后外面传的时候用cell的实际宽度*实际高度/当前布局的宽度。
- (nullable UICollectionViewLayoutAttributes *)layoutAttributesForItemAtIndexPath:(NSIndexPath *)indexPath{ UICollectionViewLayoutAttributes *layoutAttributes = [UICollectionViewLayoutAttributes layoutAttributesForCellWithIndexPath:indexPath]; CGFloat minContentHeight = [self.columMinGapArray[] doubleValue];
NSInteger minIndex = ;
for (NSInteger i = ; i < self.columMinGapArray.count; i++) {
if (minContentHeight > [self.columMinGapArray[i] doubleValue]) {
minContentHeight = [self.columMinGapArray[i] doubleValue];
minIndex = i;
}
}
CGFloat w = (self.collectionView.frame.size.width - [self edgeInsets].left - [self edgeInsets].right - ([self columsNumber]-) * [self columsMargin])/[self columsNumber]; CGFloat y = (minContentHeight == [self edgeInsets].top ? minContentHeight : minContentHeight + [self rowMargin]); CGFloat x = [self edgeInsets].left + minIndex *(w+[self columsMargin]); CGFloat h = [self.delegate waterLayout:self layoutHeightAtindexPath:indexPath layoutItemWidth:w]; layoutAttributes.frame = CGRectMake(x, y, w, h);
self.columMinGapArray[minIndex] = @(CGRectGetMaxY(layoutAttributes.frame)); CGFloat contentHeight = [self.columMinGapArray[minIndex] doubleValue];
if (self.contentHeight < contentHeight) {
self.contentHeight = contentHeight;
}
return layoutAttributes;
} #pragma mark --- getters and setters
- (NSMutableArray *)columMinGapArray{
if (_columMinGapArray == nil) { _columMinGapArray = [NSMutableArray array];
}
return _columMinGapArray;
}
- (NSMutableArray *)layoutAttributesArray{
if (_layoutAttributesArray == nil) {
_layoutAttributesArray = [NSMutableArray array];
}
return _layoutAttributesArray;
}
@end
2.设置collectionView.
1).注意事项1，需要设置布局，上面创建的layout布局。
2).注意事项2，需要设置代理和数据源，然后把必须实现的方法实现一下。
3).注意事项3，注册cell有两种形式，一个是xib,一个是自定义的cell类，自定义或xib的cell类一定是设置collectionViewcell方法中的cell对象。
- (UICollectionView *)collectionView{
if (!_collectionView) {
STRWaterLayout *layout = [[STRWaterLayout alloc] init];
layout.delegate = self;
UICollectionView *collectionView = [[UICollectionView alloc] initWithFrame:CGRectMake(, , self.view.bounds.size.width,self.view.bounds.size.height) collectionViewLayout:layout];
collectionView.delegate = self;
collectionView.dataSource = self;
collectionView.mj_header = [MJRefreshNormalHeader headerWithRefreshingTarget:self refreshingAction:@selector(headRefresh)];
collectionView.mj_footer = [MJRefreshAutoNormalFooter footerWithRefreshingTarget:self refreshingAction:@selector(foorRefresh)];
[collectionView registerClass:[CollectionViewCell class] forCellWithReuseIdentifier:customCell];
[self.view addSubview:collectionView];
self.collectionView = collectionView;
}
return _collectionView;
}
3.实现collectionView的数据源方法或代理方法。
- (NSInteger)collectionView:(UICollectionView *)collectionView numberOfItemsInSection:(NSInteger)section{
return self.dataArray.count; //存储的是布局模型数据，有宽度、高度（最基本的）及其它对象数据。
}
// The cell that is returned must be retrieved from a call to -dequeueReusableCellWithReuseIdentifier:forIndexPath:
- ( UICollectionViewCell *)collectionView:(UICollectionView *)collectionView cellForItemAtIndexPath:(NSIndexPath *)indexPath{
CollectionViewCell *cell = [collectionView dequeueReusableCellWithReuseIdentifier:customCell forIndexPath:indexPath];
cell.layoutModel = self.dataArray[indexPath.row];
return cell;
}
4.获取数据。（我这里是用的plist，这个可以根据项目来设置，这里是举例。）
- (NSMutableArray *)dataArray{
if (!_dataArray) {
NSArray *data = [[NSArray alloc] initWithContentsOfFile:[[NSBundle mainBundle]pathForResource:@"" ofType:@"plist"]];
NSMutableArray *dataA = [NSMutableArray array];
for (NSDictionary *dic in data) {
STRWaterLayoutModel *layoutModel = [STRWaterLayoutModel getCollectionModel:dic];
[dataA addObject:layoutModel];
}
_dataArray = dataA;
}
return _dataArray;
}
5.实现布局layout的代理，传入当前cell的高度和间距。
- (CGFloat)waterLayout:(STRWaterLayout *)waterLayout layoutHeightAtindexPath:(NSIndexPath *)indexpath layoutItemWidth:(CGFloat)width{
STRWaterLayoutModel *waterLayoutModel = [self.dataArray objectAtIndex:indexpath.row];
// return width * [waterLayoutModel.h doubleValue] / [waterLayoutModel.w doubleValue];
return [waterLayoutModel.h doubleValue];
}
- (CGFloat)columsNumberWithWaterLayout:(STRWaterLayout *)waterLayout{
return ;
}
- (CGFloat)columsMarginWithWaterLaout:(STRWaterLayout *)waterLayout{
return ;
}
- (CGFloat)rowsMarginWithWaterLayout:(STRWaterLayout *)waterLayout{
return ;
}
- (UIEdgeInsets)edgeInsetsWithWaterLayout:(STRWaterLayout *)waterLayout{
return UIEdgeInsetsMake(,, , );
}
三、总结。
1.通过上面瀑布流的核心思想。
2.创建collectionView的一些注意事项。
3.给某个类传数据时，可以用代理，这才是正儿八经遵守MVC的思想，后期改动的话，好扩展，如果用属性的话，后期会很麻烦。
4.代码地址：待后期上传。