diff --git a/AsyncDisplayKit/ASCollectionView.h b/AsyncDisplayKit/ASCollectionView.h index 4b2048f6ec..a5b0aae9c7 100644 --- a/AsyncDisplayKit/ASCollectionView.h +++ b/AsyncDisplayKit/ASCollectionView.h @@ -119,6 +119,8 @@ */ - (void)reloadData; +- (void)registerSupplementaryViewOfKind:(NSString *)elementKind; + /** * Inserts one or more sections. * diff --git a/AsyncDisplayKit/ASCollectionView.mm b/AsyncDisplayKit/ASCollectionView.mm index d2288f39e6..8148e55d7b 100644 --- a/AsyncDisplayKit/ASCollectionView.mm +++ b/AsyncDisplayKit/ASCollectionView.mm @@ -379,6 +379,12 @@ static BOOL _isInterceptedSelector(SEL sel) [self performBatchAnimated:YES updates:updates completion:completion]; } +- (void)registerSupplementaryViewOfKind:(NSString *)elementKind +{ + NSString *identifier = [NSString stringWithFormat:@"_ASCollectionSupplementaryView_%@", elementKind]; + [self registerClass:[UIView class] forSupplementaryViewOfKind:elementKind withReuseIdentifier:identifier]; +} + - (void)insertSections:(NSIndexSet *)sections { ASDisplayNodeAssertMainThread(); diff --git a/examples/ASCollectionView/Sample/ViewController.m b/examples/ASCollectionView/Sample/ViewController.m index a6ceb20f4d..779f4e6666 100644 --- a/examples/ASCollectionView/Sample/ViewController.m +++ b/examples/ASCollectionView/Sample/ViewController.m @@ -39,6 +39,9 @@ _collectionView.asyncDelegate = self; _collectionView.backgroundColor = [UIColor whiteColor]; + [_collectionView registerSupplementaryViewOfKind:UICollectionElementKindSectionHeader]; + [_collectionView registerSupplementaryViewOfKind:UICollectionElementKindSectionFooter]; + return self; }