adding completion block to UIImageView category

jakemarsh · Dec 14, 2012 · 03fcf3d · 03fcf3d
1 parent 73bfc6f
commit 03fcf3d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/UIImageView+JMImageCache.h b/UIImageView+JMImageCache.h
@@ -12,6 +12,8 @@
 
 - (void) setImageWithURL:(NSURL *)url;
 - (void) setImageWithURL:(NSURL *)url placeholder:(UIImage *)placeholderImage;
+- (void) setImageWithURL:(NSURL *)url placeholder:(UIImage *)placeholderImage completionBlock:(void (^)(UIImage *))completionBlock;
 - (void) setImageWithURL:(NSURL *)url key:(NSString*)key placeholder:(UIImage *)placeholderImage;
+- (void) setImageWithURL:(NSURL *)url key:(NSString*)key placeholder:(UIImage *)placeholderImage completionBlock:(void (^)(UIImage *image))completionBlock;
 
 @end
diff --git a/UIImageView+JMImageCache.m b/UIImageView+JMImageCache.m
@@ -43,7 +43,13 @@ - (void) setImageWithURL:(NSURL *)url {
 - (void) setImageWithURL:(NSURL *)url placeholder:(UIImage *)placeholderImage {
  [self setImageWithURL:url key:nil placeholder:placeholderImage];
 }
+- (void) setImageWithURL:(NSURL *)url placeholder:(UIImage *)placeholderImage completionBlock:(void (^)(UIImage *))completionBlock {
+ [self setImageWithURL:url key:nil placeholder:placeholderImage completionBlock:completionBlock];
+}
 - (void) setImageWithURL:(NSURL *)url key:(NSString*)key placeholder:(UIImage *)placeholderImage {
+ [self setImageWithURL:url key:key placeholder:placeholderImage completionBlock:nil];
+}
+- (void) setImageWithURL:(NSURL *)url key:(NSString*)key placeholder:(UIImage *)placeholderImage completionBlock:(void (^)(UIImage *image))completionBlock {
  self.jm_imageURL = url;
  self.image = placeholderImage;
 
@@ -91,6 +97,8 @@ - (void) setImageWithURL:(NSURL *)url key:(NSString*)key placeholder:(UIImage *)
 
  [safeSelf setNeedsLayout];
  [safeSelf setNeedsDisplay];
+
+ if (completionBlock) completionBlock(image);
  });
  }
  }];