UICollectionView: 添加类似UITableView的tableHeaderView的页眉属性

在iOS开发中, 我们会经常使用到UICollectionView和UITableView, 但是在使用过程中总会出现各种各样的需求
由于UITableView拥有tableHeaderView, 可以在每个分区的header之外, 还可以设置页眉, 一般我们的首页轮播图就是放在这个位置
UICollectionView没有类似tableHeaderView的属性, 只能设置每个分区的header, 所以我们想要给UICollectionView添加页眉, 就需要自己设置

UICollectionView有一个属性: contentInset, 可以设置UICollectionView的内边距, 而不改变contentSize的大小
我们在设置页眉的时候都会先改变contentInset的top值, 让内容向下移动一段距离, 然后在这个空白的位置添加我们需要的页眉

下面是我自己写的一个分类, 添加了一个属性, 这个属性类似于UITableView的tableHeaderView
.h文件

#import <UIKit/UIKit.h>
@interface UICollectionView (LTExtension)
/** 页眉视图 */
@property (nonatomic) UIView *lt_collectionHeaderView;
/** 页脚视图 */
@property (nonatomic) UIView *lt_collectionFooterView;
/** 页眉向上偏移量 */
@property (nonatomic, assign) CGFloat lt_collectionHeaderViewOffset;
/** 页脚向下偏移量 */
@property (nonatomic, assign) CGFloat lt_collectionFooterViewOffset;
@end

.m文件

#import "UICollectionView+LTExtension.h"
#import <objc/runtime.h>

static const NSString *kCollectionHeaderViewKey = @"lt_collectionHeaderView";
static const NSString *kCollectionFooterViewKey = @"lt_collectionFooterView";

static const NSString *kCollectionHeaderViewOffsetKey = @"lt_collectionHeaderViewOffset";
static const NSString *kCollectionFooterViewOffsetKey = @"lt_collectionFooterViewOffset";

@implementation UICollectionView (LTExtension)

#pragma mark - < 页眉 >

/**
 设置页眉 - set方法
 */
- (void)setLt_collectionHeaderView:(UIView *)lt_collectionHeaderView
{
    UIView *headerView = self.lt_collectionHeaderView;
    if (headerView != lt_collectionHeaderView) {
        // 获取内边距
        UIEdgeInsets contentInset = self.contentInset;
        
        // 移除之前的lt_collectionHeaderView
        contentInset.top -= headerView.height;
        [headerView removeFromSuperview];
        
        // 设置内边距
        contentInset.top += lt_collectionHeaderView.height;
        self.contentInset = contentInset;
        
        // 添加lt_collectionHeaderView
        [self addSubview:lt_collectionHeaderView];
        
        objc_setAssociatedObject(self, &(kCollectionHeaderViewKey), lt_collectionHeaderView, OBJC_ASSOCIATION_ASSIGN);
    }
}

/**
 获取页眉 - get方法
 */
- (UIView *)lt_collectionHeaderView
{
    return objc_getAssociatedObject(self, &(kCollectionHeaderViewKey));
}

#pragma mark - < 页脚 >

/**
 设置页脚 - set方法
 */
- (void)setLt_collectionFooterView:(UIView *)lt_collectionFooterView
{
    UIView *footerView = self.lt_collectionFooterView;
    
    if (footerView != lt_collectionFooterView) {
        // 获取内边距
        UIEdgeInsets contentInset = self.contentInset;
        
        // 移除之前的
        contentInset.bottom -= footerView.height;
        [footerView removeFromSuperview];
        
        // 设置内边距
        contentInset.bottom += lt_collectionFooterView.height;
        self.contentInset= contentInset;
        
        // 添加页脚
        [self addSubview:lt_collectionFooterView];
        
        objc_setAssociatedObject(self, &(kCollectionFooterViewKey), lt_collectionFooterView, OBJC_ASSOCIATION_ASSIGN);
    }
}

/**
 获取页脚 - get方法
 */
- (UIView *)lt_collectionFooterView
{
    return objc_getAssociatedObject(self, &(kCollectionFooterViewKey));
}

#pragma mark - < 页眉偏移量 >

- (void)setLt_collectionHeaderViewOffset:(CGFloat)lt_collectionHeaderViewOffset
{
    objc_setAssociatedObject(self, &(kCollectionHeaderViewOffsetKey), @(lt_collectionHeaderViewOffset), OBJC_ASSOCIATION_ASSIGN);
    [self setNeedsLayout];
}

- (CGFloat)lt_collectionHeaderViewOffset
{
    NSNumber *number = objc_getAssociatedObject(self, &(kCollectionHeaderViewOffsetKey));
    return number.floatValue;
}

#pragma mark - < 页脚偏移量 >

- (void)setLt_collectionFooterViewOffset:(CGFloat)lt_collectionFooterViewOffset
{
    objc_setAssociatedObject(self, &(kCollectionFooterViewOffsetKey), @(lt_collectionFooterViewOffset), OBJC_ASSOCIATION_ASSIGN);
    [self setNeedsLayout];
}

- (CGFloat)lt_collectionFooterViewOffset
{
    NSNumber *number = objc_getAssociatedObject(self, &(kCollectionFooterViewOffsetKey));
    return number.floatValue;
}

#pragma mark - < 交换方法 >

/**
 加载过程中, 交换方法
 */
+ (void)load
{
    Method layoutSubviews = class_getInstanceMethod(self, @selector(layoutSubviews));
    Method lt_layoutSubviews = class_getInstanceMethod(self, @selector(lt_layoutSubviews));
    method_exchangeImplementations(layoutSubviews, lt_layoutSubviews);
}

/**
 加载过程中, 交换方法
 */
- (void)lt_layoutSubviews
{
    [self lt_layoutSubviews];
    UIView *lt_collectionHeaderView = self.lt_collectionHeaderView;
    lt_collectionHeaderView.y = -lt_collectionHeaderView.height - self.lt_collectionHeaderViewOffset;
    self.lt_collectionFooterView.y = self.contentSize.height + self.lt_collectionFooterViewOffset;
}
@end

这里对于frame的设置x, y, width, height是之前已经写好的一个分类, 用来设置获取view的frame, 我想这个大家都懂, 这里就直接使用了

使用方法

// 设置页眉
UIView *header = [[UIView alloc] initWithFrame:CGRectMake(0, 0, kScreen_width, 200)];
header.backgroundColor = [UIColor purpleColor];
self.collectionView.lt_collectionHeaderView = header;
// 设置页眉向上偏移量
self.collectionView.lt_collectionHeaderViewOffset = 0;

// 设置页脚
UIView *footer = [[UIView alloc] initWithFrame:CGRectMake(0, 0, kScreen_width, 200)];
footer.backgroundColor = [UIColor greenColor];
self.collectionView.lt_collectionFooterView = footer;
// 设置页脚向下偏移量
self.collectionView.lt_collectionHeaderViewOffset = 0;

效果图

最后, 如果有什么建议请提出来, 我会修改