[collections] Add support for supplementary views.

muhtasib · Apr 2, 2013 · 72a2a59 · 72a2a59
1 parent edb7291
commit 72a2a59
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 3 deletions.
diff --git a/src/collections/src/NICollectionViewCellFactory.m b/src/collections/src/NICollectionViewCellFactory.m
@@ -70,9 +70,9 @@ + (UICollectionViewCell *)cellWithClass:(Class)collectionViewCellClass
 
 ///////////////////////////////////////////////////////////////////////////////////////////////////
 + (UICollectionViewCell *)collectionViewModel:(NICollectionViewModel *)collectionViewModel
-                   cellForCollectionView:(UICollectionView *)collectionView
-                        atIndexPath:(NSIndexPath *)indexPath
-                         withObject:(id)object {
+                        cellForCollectionView:(UICollectionView *)collectionView
+                                  atIndexPath:(NSIndexPath *)indexPath
+                                   withObject:(id)object {
   UICollectionViewCell* cell = nil;
 
   // If this assertion fires then your app is about to crash. You need to either add an explicit

diff --git a/src/collections/src/NICollectionViewModel.h b/src/collections/src/NICollectionViewModel.h
@@ -79,6 +79,19 @@
                                   atIndexPath:(NSIndexPath *)indexPath
                                    withObject:(id)object;
 
+@optional
+
+/**
+ * Fetches a supplementary collection view element at a given indexPath.
+ *
+ * The value of the kind property and indexPath are implementation-dependent
+ * based on the type of UICollectionViewLayout being used.
+ */
+- (UICollectionReusableView *)collectionViewModel:(NICollectionViewModel *)collectionViewModel
+                                   collectionView:(UICollectionView *)collectionView
+                viewForSupplementaryElementOfKind:(NSString *)kind
+                                      atIndexPath:(NSIndexPath *)indexPath;
+
 @end
 
 /**

diff --git a/src/collections/src/NICollectionViewModel.m b/src/collections/src/NICollectionViewModel.m
@@ -194,6 +194,19 @@ - (UICollectionViewCell *)collectionView:(UICollectionView *)collectionView cell
 }
 
 
+///////////////////////////////////////////////////////////////////////////////////////////////////
+- (UICollectionReusableView *)collectionView:(UICollectionView *)collectionView viewForSupplementaryElementOfKind:(NSString *)kind atIndexPath:(NSIndexPath *)indexPath {
+  if ([self.delegate respondsToSelector:
+       @selector(collectionViewModel:collectionView:viewForSupplementaryElementOfKind:atIndexPath:)]) {
+    return [self.delegate collectionViewModel:self
+                               collectionView:collectionView
+            viewForSupplementaryElementOfKind:kind
+                                  atIndexPath:indexPath];
+  }
+  return nil;
+}
+
+
 ///////////////////////////////////////////////////////////////////////////////////////////////////
 ///////////////////////////////////////////////////////////////////////////////////////////////////
 #pragma mark -