don't attempt to render pages at size (0, 0)

brow · Sep 17, 2013 · b4f6f23 · b4f6f23
1 parent 326930b
commit b4f6f23
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/Leaves/LeavesCache.m b/Leaves/LeavesCache.m
@@ -31,6 +31,9 @@ - (void) dealloc
 
 
 - (CGImageRef) imageForPageIndex:(NSUInteger)pageIndex {
+    if (CGSizeEqualToSize(pageSize, CGSizeZero))
+        return NULL;
+
 	CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();
 	CGContextRef context = CGBitmapContextCreate(NULL, 
 												 pageSize.width, 
@@ -61,10 +64,12 @@ - (CGImageRef) cachedImageForPageIndex:(NSUInteger)pageIndex {
 	}
 	if (!pageImage) {
 		CGImageRef pageCGImage = [self imageForPageIndex:pageIndex];
-		pageImage = [UIImage imageWithCGImage:pageCGImage];
-		@synchronized (pageCache) {
-			[pageCache setObject:pageImage forKey:pageIndexNumber];
-		}
+        if (pageCGImage) {
+            pageImage = [UIImage imageWithCGImage:pageCGImage];
+            @synchronized (pageCache) {
+                [pageCache setObject:pageImage forKey:pageIndexNumber];
+            }
+        }
 	}
 	return pageImage.CGImage;
 }