CustomLayout
#import
@interface CustomLayout : UICollectionViewFlowLayout
@end
#import "CustomLayout.h"
#define kScreenWidth ([UIScreen mainScreen].bounds.size.width)
static CGFloat const lineWidth = 1.0f;
@interface CustomLayout ()
@property (nonatomic,strong) NSMutableArray *attributesArray;
@property (nonatomic,assign) CGFloat y;
@end
@implementation CustomLayout
- (void)prepareLayout{
[super prepareLayout];
self.attributesArray = [NSMutableArray array];
for (int i = 0; i < [self.collectionView numberOfSections]; i++) {
for (int j = 0; j < [self.collectionView numberOfItemsInSection:i]; j++) {
[self.attributesArray addObject:[self layoutAttributesForItemAtIndexPath:[NSIndexPath indexPathForItem:j inSection:i]]];
}
//footer的UICollectionViewLayoutAttributes
UICollectionViewLayoutAttributes *attribute = [self layoutAttributesForSupplementaryViewOfKind:UICollectionElementKindSectionFooter atIndexPath:[NSIndexPath indexPathForItem:0 inSection:i]];
[self.attributesArray addObject:attribute];
}
}
- (NSArray *)layoutAttributesForElementsInRect:(CGRect)rect{
return self.attributesArray;
}
- (UICollectionViewLayoutAttributes *)layoutAttributesForItemAtIndexPath:(NSIndexPath *)indexPath{
UICollectionViewLayoutAttributes *attribute = [UICollectionViewLayoutAttributes layoutAttributesForCellWithIndexPath:indexPath];
CGFloat width = kScreenWidth/3.0;
CGFloat width1 = (kScreenWidth-lineWidth*2)/3.0;
CGFloat width2 = (kScreenWidth-lineWidth*3)/4.0;
if (indexPath.section == 0) {
attribute.frame = CGRectMake(indexPath.item*width, self.y, width, width);
NSInteger count = [self.collectionView numberOfItemsInSection:indexPath.section];
if (indexPath.item == count-1) {
self.y = self.y + width + 10;
}
}else if (indexPath.section == 1){
if (indexPath.item == 0) { // 4
attribute.frame = CGRectMake(0, _y, width1, width*2+lineWidth);
} else if (indexPath.item == 1) {
attribute.frame = CGRectMake(width1+lineWidth, _y, width1*2, width);
} else if (indexPath.item == 2) {
attribute.frame = CGRectMake(width1+lineWidth, width+lineWidth+_y, width1, width);
} else if (indexPath.item == 3) {
attribute.frame = CGRectMake((width1+lineWidth)*2, width+lineWidth+_y, width1, width);
}
NSInteger count = [self.collectionView numberOfItemsInSection:indexPath.section];
if (indexPath.item == count-1) {
self.y = self.y + width*2+lineWidth + 10;
}
}else if (indexPath.section == 2){
if (indexPath.item < 4) {
attribute.frame = CGRectMake((width2+lineWidth)*indexPath.item, _y, width2, width);
} else if (indexPath.item >= 4 && indexPath.item <7) {
attribute.frame = CGRectMake((width1+lineWidth)*(indexPath.item-4), width+lineWidth+_y, width1, width);
}
NSInteger count = [self.collectionView numberOfItemsInSection:indexPath.section];
if (indexPath.item == count-1) {
self.y = self.y + width*2+lineWidth + 10;
}
}
else if (indexPath.section == 3){
NSInteger row = indexPath.item/3;
NSInteger col = indexPath.item%3;
attribute.frame = CGRectMake((width1+lineWidth)*col, (width+lineWidth)*row+_y, width1, width);
NSInteger count = [self.collectionView numberOfItemsInSection:indexPath.section];
if (indexPath.item == count-1) {
NSInteger rows = 0;
if (count%3 == 0) {
rows = count/3;
}else{
rows = count/3 + 1;
}
self.y = self.y + rows * width + (rows-1)*1 + 10;
}
}
return attribute;
}
- (UICollectionViewLayoutAttributes *)layoutAttributesForSupplementaryViewOfKind:(NSString *)elementKind atIndexPath:(NSIndexPath *)indexPath
{
UICollectionViewLayoutAttributes *attribute = [UICollectionViewLayoutAttributes layoutAttributesForSupplementaryViewOfKind:elementKind withIndexPath:indexPath];
attribute.frame = CGRectMake(0, _y, kScreenWidth, 60);
_y += 60+10;
return attribute;
}
- (CGSize)collectionViewContentSize
{
CGSize size = self.collectionView.bounds.size;
size.height = self.y;
return size;
}
@end
用法
#import "ViewController.h"
#import "CustomLayout.h"
#define random(r, g, b, a) [UIColor colorWithRed:(r)/255.0 green:(g)/255.0 blue:(b)/255.0 alpha:(a)/255.0]
#define randomColor random(arc4random_uniform(256), arc4random_uniform(256), arc4random_uniform(256), 255)
@interface ViewController ()
@property (nonatomic,strong) UICollectionView *collectionView;
@property (nonatomic,strong) NSArray *dataArray;
@end
@implementation ViewController
- (void)viewDidLoad {
[super viewDidLoad];
self.dataArray = @[
@[@"1",@"2",@"3"],
@[@"4",@"5",@"6",@"7"],
@[@"8",@"9",@"10",@"11",@"12",@"13",@"14"],
@[@"15",@"16",@"17",@"18",@"19",@"20",@"21",@"22",@"22",@"22",@"22"]
];
CustomLayout *layout = [[CustomLayout alloc]init];
self.collectionView = [[UICollectionView alloc]initWithFrame:CGRectMake(0, 0, self.view.bounds.size.width, self.view.bounds.size.height) collectionViewLayout:layout];
self.collectionView.backgroundColor = [UIColor lightGrayColor];
self.collectionView.delegate = self;
self.collectionView.dataSource = self;
[self.view addSubview:self.collectionView];
[self.collectionView registerClass:[UICollectionViewCell class] forCellWithReuseIdentifier:@"id"];
[self.collectionView registerClass:[UICollectionReusableView class] forSupplementaryViewOfKind:UICollectionElementKindSectionFooter withReuseIdentifier:@"footer"];
}
- (NSInteger)numberOfSectionsInCollectionView:(UICollectionView *)collectionView{
return self.dataArray.count;
}
- (NSInteger)collectionView:(UICollectionView *)collectionView numberOfItemsInSection:(NSInteger)section{
NSArray *subArray = self.dataArray[section];
return subArray.count;
}
- ( UICollectionViewCell *)collectionView:(UICollectionView *)collectionView cellForItemAtIndexPath:(NSIndexPath *)indexPath{
UICollectionViewCell *cell = [collectionView dequeueReusableCellWithReuseIdentifier:@"id" forIndexPath:indexPath];
cell.backgroundColor = randomColor;
return cell;
}
- (UICollectionReusableView *)collectionView:(UICollectionView *)collectionView viewForSupplementaryElementOfKind:(NSString *)kind atIndexPath:(NSIndexPath *)indexPath{
UICollectionReusableView *footerView = [collectionView dequeueReusableSupplementaryViewOfKind:kind withReuseIdentifier:@"footer" forIndexPath:indexPath];
footerView.backgroundColor = [UIColor cyanColor];
for (UIView *subView in footerView.subviews) {
[subView removeFromSuperview];
}
UILabel *titleLabel = [[UILabel alloc]initWithFrame:footerView.bounds];
titleLabel.text = [NSString stringWithFormat:@"第%ld个分区的区尾",(long)indexPath.section];
[footerView addSubview:titleLabel];
return footerView;
}
@end
CB07B05D-0D50-438C-B9AF-5A8C5470E409.png
地址https://github.com/yinbowang/UICollectionVIewDemo.git