UITableView和UICollectionView 头尾视图的创建

UITableView

头(尾)视图的View

TableHeaderView.h
@interface TableHeaderView : UITableViewHeaderFooterView
+ (instancetype)headerViewWithTableView:(UITableView *)tableView;
@end

TableHeaderView.m
@interface TableHeaderView ()

@end
@implementation TableHeaderView
+ (instancetype)headerViewWithTableView:(UITableView *)tableView
{
    static NSString *headerID = @"header";
    TableHeaderView *headerView = [tableView dequeueReusableHeaderFooterViewWithIdentifier:headerID];
    if (headerView == nil) {
        headerView = [[TableHeaderView alloc]initWithReuseIdentifier:headerID];
    }
    return headerView;
}

- (instancetype)initWithReuseIdentifier:(NSString *)reuseIdentifier
{
    if (self = [super initWithReuseIdentifier:reuseIdentifier]) {
         //添加头(尾)视图中的控件
   }
    return self;
}
@end

控制器

ViewController.m
@interface ViewController ()
@property (nonatomic, strong) UITableView *tableView;
@property (nonatomic, strong) TableHeaderView *headerView;
@end
@implementation ViewController
- (void)viewDidLoad {
    [super viewDidLoad];
    _tableView = [[UITableView alloc]initWithFrame:CGRectMake(0, 0,[UIScreen mainScreen].bounds.size.width,[UIScreen mainScreen].bounds.size.height)style:UITableViewStyleGrouped];
    _tableView.delegate = self;
    _tableView.dataSource = self;
    [self.view addSubview:_tableView];
}

......

- (UIView *)tableView:(UITableView *)tableView viewForHeaderInSection:(NSInteger)section     //头视图
- (UIView *)tableView:(UITableView *)tableView viewForHeaderInSection:(NSInteger)section     //尾视图
{
    _headerView = [TableHeaderView headerViewWithTableView:tableView];
    return _headerView;
}

- (CGFloat)tableView:(UITableView *)tableView heightForHeaderInSection:(NSInteger)section   //头视图
- (CGFloat)tableView:(UITableView *)tableView heightForFooterInSection:(NSInteger)section    //尾视图
{
    return  ;  //头(尾)视图的高度
}

@end

UICollectionView

头(尾)视图的View

CollectionHeaderView.h
@interface CollectionHeaderView : UICollectionReusableView
@end

CollectionHeaderView.m
@implementation CollectionHeaderView
- (instancetype)initWithFrame:(CGRect)frame {
    self = [super initWithFrame:frame];
    if (self) {
        //添加头(尾)视图中的控件      
    }
    return self;
}
@end

同理创建尾视图的View: CollectionFooterView

控制器

ViewController.m
@interface ViewController ()
@property (nonatomic, strong) UICollectionView *collectionView;
@property (nonatomic, strong) CollectionHeaderView *headerView;
@property (nonatomic, strong) CollectionFooterView *footerView;
@end

@implementation ViewController
- (void)viewDidLoad {
    [super viewDidLoad];
    _collectionView = [[UICollectionView alloc] initWithFrame:CGRectMake(0, 0, [UIScreen mainScreen].bounds.size.width,[UIScreen mainScreen].bounds.size.height) collectionViewLayout:[UICollectionViewFlowLayout new]];
    _collectionView.delegate = self;
    _collectionView.dataSource = self;
    [self.view addSubview:_collectionView];
    [_collectionView registerClass:[CollectionHeaderView class] forSupplementaryViewOfKind:UICollectionElementKindSectionHeader withReuseIdentifier:@"Header"];             //注册头视图
    [_collectionView registerClass:[CollectionFooterView class] forSupplementaryViewOfKind:UICollectionElementKindSectionFooter withReuseIdentifier:@"Footer"];              //注册尾视图
}

......

- (UICollectionReusableView *)collectionView:(UICollectionView *)collectionView viewForSupplementaryElementOfKind:(NSString *)kind atIndexPath:(NSIndexPath *)indexPath {
    UICollectionReusableView *reusableView = nil;
    if (kind == UICollectionElementKindSectionHeader) {       //头视图
        _headerView = [collectionView dequeueReusableSupplementaryViewOfKind:UICollectionElementKindSectionHeader withReuseIdentifier:@"Header" forIndexPath:indexPath];
        reusableView = _headerView;
    }else if (kind == UICollectionElementKindSectionFooter) {    //尾视图
        _footerView = [collectionView dequeueReusableSupplementaryViewOfKind:UICollectionElementKindSectionFooter withReuseIdentifier:@"Footer" forIndexPath:indexPath];
        _footerView.delegate = self;
        reusableView = _footerView;
    }
    return reusableView;
}

- (CGSize)collectionView:(UICollectionView *)collectionView layout:(UICollectionViewLayout *)collectionViewLayout referenceSizeForHeaderInSection:(NSInteger)section {
    return CGSizeMake(头视图的宽, 头视图的高);
}

- (CGSize)collectionView:(UICollectionView *)collectionView layout:(UICollectionViewLayout *)collectionViewLayout referenceSizeForFooterInSection:(NSInteger)section {
    return CGSizeMake(尾视图的宽, 尾视图的高);
}