Protecting against nil images

kgn · Jun 2, 2012 · b3f9938 · b3f9938
1 parent 21bd19f
commit b3f9938
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/UIImage+BBlock.m b/UIImage+BBlock.m
@@ -24,6 +24,10 @@ + (NSCache *)drawingCache{
 }
 
 + (UIImage *)imageForSize:(CGSize)size withDrawingBlock:(void(^)())drawingBlock{
+    if(size.width <= 0 || size.width <= 0){
+        return nil;
+    }
+
     UIGraphicsBeginImageContextWithOptions(size, NO, 0.0f);
     @autoreleasepool{
         drawingBlock();
@@ -41,7 +45,9 @@ + (UIImage *)imageWithIdentifier:(NSString *)identifier forSize:(CGSize)size and
     UIImage *image = [[[self class] drawingCache] objectForKey:identifier];
     if(image == nil){
         image = [[self class] imageForSize:size withDrawingBlock:drawingBlock];
-        [[[self class] drawingCache] setObject:image forKey:identifier];
+        if(image != nil){
+            [[[self class] drawingCache] setObject:image forKey:identifier];
+        }
     }
     return image;
 }