Thumbnail click logic bug fix for last update

Author: davinc

Demo/Demo/RootViewController.m

@@ -149,19 +149,11 @@ - (NSInteger)numberOfThumbnailsInRowForThumbnailGridView:(VCThumbnailGridView *)
 	return 3;
 }
 
-- (UIImage*)thumbnailGridView:(VCThumbnailGridView*)thumbnailGridView imageAtIndex:(NSInteger)index 
+- (VCThumbnailView *)thumbnailGridView:(VCThumbnailGridView *)thumbnailGridView thumbnailViewAtIndex:(NSInteger)index
 {
-	return [UIImage imageNamed:@"Icon.png"];
-}
-
-- (NSString*)thumbnailGridView:(VCThumbnailGridView*)thumbnailGridView imageUrlAtIndex:(NSInteger)index
-{
-	static bool something = false;
-	something = !something;
-	if (something) {
-		return [NSString stringWithString:@"http://images.apple.com/support/home/images/featured_itunes.png"];
-	}
-	return nil;
+	VCThumbnailView *thumbnailView = [[VCThumbnailView alloc] initWithFrame:CGRectZero];
+	[thumbnailView setImage:[UIImage imageNamed:@"Icon.png"]];
+	return [thumbnailView autorelease];
 }
 
 #pragma mark - VCThumbnailGridViewDelegate

VCThumbnailGridView/VCThumbnailGridView.h

@@ -33,11 +33,12 @@
 
 #import <UIKit/UIKit.h>
 
+#import "VCThumbnailViewCell.h"
+
 @protocol VCThumbnailGridViewDataSource;
 @protocol VCThumbnailGridViewDelegate;
 
 @class VCThumbnailView;
-@class VCThumbnailViewCell;
 
 @interface VCThumbnailGridView : UIView <UITableViewDataSource, UITableViewDelegate> {
 @private
@@ -68,7 +69,7 @@
 @optional
 - (NSInteger)numberOfThumbnailsInRowForThumbnailGridView:(VCThumbnailGridView *)thumbnailGridView;
 - (VCThumbnailView *)thumbnailGridView:(VCThumbnailGridView *)thumbnailGridView thumbnailViewAtIndex:(NSInteger)index;
-- (UIImage *)thumbnailGridView:(VCThumbnailGridView *)thumbnailGridView imageAtIndex:(NSInteger)index;
+//- (UIImage *)thumbnailGridView:(VCThumbnailGridView *)thumbnailGridView imageAtIndex:(NSInteger)index;
 
 @end
 

VCThumbnailGridView/VCThumbnailGridView.m

@@ -140,7 +140,6 @@ - (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(N
 		int indexOfImage = indexPath.row * _numberOfThumbnailsInRow;
 
 		VCThumbnailView *thumbnail = nil;
-		BOOL respondsToSelectorImage = [self.dataSource respondsToSelector:@selector(thumbnailGridView:imageAtIndex:)];
 		BOOL respondsToSelectorView = [self.dataSource respondsToSelector:@selector(thumbnailGridView:thumbnailViewAtIndex:)];
 		for (int i = 0; i < _numberOfThumbnailsInRow; i++) {
 			if (indexOfImage < _numberOfThumbnails) {
@@ -153,25 +152,16 @@ - (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(N
 				}
 
 				// set proerties
-				thumbnail.backgroundColor = [UIColor whiteColor];
+				thumbnail.tag = indexOfImage;
+				[thumbnail addTarget:self withSelector:@selector(didTapImageThumbnail:)];
 				[cell addSubview:thumbnail];
 				[cell.thumbnails addObject:thumbnail];
-				
-				thumbnail.hidden = NO;
-				if (respondsToSelectorImage) {
-					[thumbnail setImage:[self.dataSource thumbnailGridView:self imageAtIndex:indexOfImage]];
-					[thumbnail addTarget:self withSelector:@selector(didTapImageThumbnail:)];
-				}
 				thumbnail.tag = indexOfImage++;
-			}else {
-				thumbnail.hidden = YES;
 			}
-			
 			thumbnail = nil;
 		}
     }
 
-	
     return cell;
 }
 

VCThumbnailGridView/VCThumbnailView.m

@@ -44,7 +44,7 @@ - (id)initWithFrame:(CGRect)frame {
 		shouldAutoRotateToFit = NO;
 		self.autoresizesSubviews = YES;
 		self.backgroundColor = [UIColor clearColor];
-		self.userInteractionEnabled = NO;
+		self.userInteractionEnabled = YES;
 		self.contentMode = UIViewContentModeScaleAspectFit;
 		self.layer.borderColor = [[UIColor lightGrayColor] CGColor];
 		self.layer.borderWidth = 1.0f;