Added moveToBeginningOfDocument / moveToEndOfDocument support

jwilling · Jun 1, 2016 · ce2d3c6 · ce2d3c6
1 parent 916315b
commit ce2d3c6
Showing 1 changed file with 40 additions and 1 deletion.
diff --git a/JNWCollectionView/JNWCollectionViewFramework.m b/JNWCollectionView/JNWCollectionViewFramework.m
@@ -429,6 +429,30 @@ - (NSArray *)allIndexPaths {
 	return indexPaths.copy;
 }
 
+- (NSIndexPath *)firstIndexPath {
+	NSIndexPath *indexPath = nil;
+	if (self.data.numberOfSections > 0) {
+		JNWCollectionViewSection section = self.data.sections[0];
+		if (section.numberOfItems > 0) {
+			indexPath = [NSIndexPath jnw_indexPathForItem:0 inSection:0];
+		}
+	}
+	return indexPath;
+}
+
+- (NSIndexPath *)lastIndexPath {
+	NSIndexPath *indexPath = nil;
+	NSInteger numberOfSections = self.data.numberOfSections;
+	if (numberOfSections > 0) {
+		JNWCollectionViewSection section = self.data.sections[numberOfSections - 1];
+		NSInteger numberOfItems = section.numberOfItems;
+		if (numberOfItems > 0) {
+			indexPath = [NSIndexPath jnw_indexPathForItem:numberOfItems - 1 inSection:numberOfSections - 1];
+		}
+	}
+	return indexPath;
+}
+
 - (NSArray *)indexPathsForItemsInRect:(CGRect)rect {
 	if (CGRectEqualToRect(rect, CGRectZero))
 		return [NSArray array];
@@ -1056,7 +1080,8 @@ - (void)rightClickInCollectionViewCell:(JNWCollectionViewCell *)cell withEvent:(
 
 - (void)moveUp:(id)sender {
 	NSIndexPath *toSelect = [self.collectionViewLayout indexPathForNextItemInDirection:JNWCollectionViewDirectionUp currentIndexPath:[self indexPathForSelectedItem]];
-	[self selectItemAtIndexPath:toSelect atScrollPosition:JNWCollectionViewScrollPositionNearest animated:YES];}
+	[self selectItemAtIndexPath:toSelect atScrollPosition:JNWCollectionViewScrollPositionNearest animated:YES];
+}
 
 - (void)moveUpAndModifySelection:(id)sender {
 	NSIndexPath *toSelect = [self.collectionViewLayout indexPathForNextItemInDirection:JNWCollectionViewDirectionUp currentIndexPath:[self indexPathForSelectedItem]];
@@ -1093,6 +1118,20 @@ - (void)moveLeftAndModifySelection:(id)sender {
 	[self selectItemAtIndexPath:toSelect atScrollPosition:JNWCollectionViewScrollPositionNearest animated:YES selectionType:JNWCollectionViewSelectionTypeExtending];
 }
 
+- (void)moveToBeginningOfDocument:(id)sender {
+	NSIndexPath *toSelect = [self firstIndexPath];
+	if (toSelect) {
+		[self selectItemAtIndexPath:toSelect atScrollPosition:JNWCollectionViewScrollPositionNearest animated:YES];
+	}
+}
+
+- (void)moveToEndOfDocument:(id)sender {
+	NSIndexPath *toSelect = [self lastIndexPath];
+	if (toSelect) {
+		[self selectItemAtIndexPath:toSelect atScrollPosition:JNWCollectionViewScrollPositionNearest animated:YES];
+	}
+}
+
 - (void)selectAll:(id)sender {
 	[self selectItemsAtIndexPaths:[self allIndexPaths] animated:YES];
 }