Ensure a supplementary view is of kind _ASCollectionReusableView befo…

…re setting its layoutAttributes
TextureGroup · Apr 19, 2017 · f39da70 · f39da70
1 parent 7c86faf
commit f39da70
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Source/ASCollectionView.mm b/Source/ASCollectionView.mm
@@ -1118,9 +1118,10 @@ - (void)collectionView:(UICollectionView *)collectionView didEndDisplayingCell:(
 
 - (void)collectionView:(UICollectionView *)collectionView willDisplaySupplementaryView:(_ASCollectionReusableView *)view forElementKind:(NSString *)elementKind atIndexPath:(NSIndexPath *)indexPath
 {
+  // Since _ASCollectionReusableView is not available for subclassing, comparing the classes is faster than isKindOfClass:
   // This is a safeguard similar to the behavior for cells in -[ASCollectionView collectionView:willDisplayCell:forItemAtIndexPath:]
   // It ensures _ASCollectionReusableView receives layoutAttributes and calls applyLayoutAttributes.
-  if (view.layoutAttributes == nil) {
+  if (([view class] == [_ASCollectionReusableView class]) && (view.layoutAttributes == nil)) {
     view.layoutAttributes = [collectionView layoutAttributesForSupplementaryElementOfKind:elementKind atIndexPath:indexPath];
   }