124 lines
4.7 KiB
Objective-C
124 lines
4.7 KiB
Objective-C
/*
|
|
* This file is part of the SDWebImage package.
|
|
* (c) Olivier Poitrey <rs@dailymotion.com>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
#import "UIButton+WebCache.h"
|
|
#import "objc/runtime.h"
|
|
|
|
static char operationKey;
|
|
|
|
@implementation UIButton (WebCache)
|
|
|
|
- (void)setImageWithURL:(NSURL *)url forState:(UIControlState)state
|
|
{
|
|
[self setImageWithURL:url forState:state placeholderImage:nil options:0 completed:nil];
|
|
}
|
|
|
|
- (void)setImageWithURL:(NSURL *)url forState:(UIControlState)state placeholderImage:(UIImage *)placeholder
|
|
{
|
|
[self setImageWithURL:url forState:state placeholderImage:placeholder options:0 completed:nil];
|
|
}
|
|
|
|
- (void)setImageWithURL:(NSURL *)url forState:(UIControlState)state placeholderImage:(UIImage *)placeholder options:(SDWebImageOptions)options
|
|
{
|
|
[self setImageWithURL:url forState:state placeholderImage:placeholder options:options completed:nil];
|
|
}
|
|
|
|
- (void)setImageWithURL:(NSURL *)url forState:(UIControlState)state completed:(SDWebImageCompletedBlock)completedBlock
|
|
{
|
|
[self setImageWithURL:url forState:state placeholderImage:nil options:0 completed:completedBlock];
|
|
}
|
|
- (void)setImageWithURL:(NSURL *)url forState:(UIControlState)state placeholderImage:(UIImage *)placeholder completed:(SDWebImageCompletedBlock)completedBlock
|
|
{
|
|
[self setImageWithURL:url forState:state placeholderImage:placeholder options:0 completed:completedBlock];
|
|
}
|
|
|
|
- (void)setImageWithURL:(NSURL *)url forState:(UIControlState)state placeholderImage:(UIImage *)placeholder options:(SDWebImageOptions)options completed:(SDWebImageCompletedBlock)completedBlock
|
|
{
|
|
[self cancelCurrentImageLoad];
|
|
|
|
[self setImage:placeholder forState:state];
|
|
|
|
if (url)
|
|
{
|
|
id<SDWebImageOperation> operation = [SDWebImageManager.sharedManager downloadWithURL:url options:options progress:nil completed:^(UIImage *image, NSError *error, BOOL fromCache, BOOL finished)
|
|
{
|
|
if (image)
|
|
{
|
|
[self setImage:image forState:state];
|
|
}
|
|
if (completedBlock && finished)
|
|
{
|
|
completedBlock(image, error, fromCache);
|
|
}
|
|
}];
|
|
objc_setAssociatedObject(self, &operationKey, operation, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
|
|
}
|
|
}
|
|
|
|
- (void)setBackgroundImageWithURL:(NSURL *)url forState:(UIControlState)state
|
|
{
|
|
[self setBackgroundImageWithURL:url forState:state placeholderImage:nil options:0 completed:nil];
|
|
}
|
|
|
|
- (void)setBackgroundImageWithURL:(NSURL *)url forState:(UIControlState)state placeholderImage:(UIImage *)placeholder
|
|
{
|
|
[self setBackgroundImageWithURL:url forState:state placeholderImage:placeholder options:0 completed:nil];
|
|
}
|
|
|
|
- (void)setBackgroundImageWithURL:(NSURL *)url forState:(UIControlState)state placeholderImage:(UIImage *)placeholder options:(SDWebImageOptions)options
|
|
{
|
|
[self setBackgroundImageWithURL:url forState:state placeholderImage:placeholder options:options completed:nil];
|
|
}
|
|
|
|
- (void)setBackgroundImageWithURL:(NSURL *)url forState:(UIControlState)state completed:(SDWebImageCompletedBlock)completedBlock
|
|
{
|
|
[self setBackgroundImageWithURL:url forState:state placeholderImage:nil options:0 completed:completedBlock];
|
|
}
|
|
|
|
- (void)setBackgroundImageWithURL:(NSURL *)url forState:(UIControlState)state placeholderImage:(UIImage *)placeholder completed:(SDWebImageCompletedBlock)completedBlock
|
|
{
|
|
[self setBackgroundImageWithURL:url forState:state placeholderImage:placeholder options:0 completed:completedBlock];
|
|
}
|
|
|
|
- (void)setBackgroundImageWithURL:(NSURL *)url forState:(UIControlState)state placeholderImage:(UIImage *)placeholder options:(SDWebImageOptions)options completed:(SDWebImageCompletedBlock)completedBlock
|
|
{
|
|
[self cancelCurrentImageLoad];
|
|
|
|
[self setBackgroundImage:placeholder forState:state];
|
|
|
|
if (url)
|
|
{
|
|
id<SDWebImageOperation> operation = [SDWebImageManager.sharedManager downloadWithURL:url options:options progress:nil completed:^(UIImage *image, NSError *error, BOOL fromCache, BOOL finished)
|
|
{
|
|
if (image)
|
|
{
|
|
[self setBackgroundImage:image forState:state];
|
|
}
|
|
if (completedBlock && finished)
|
|
{
|
|
completedBlock(image, error, fromCache);
|
|
}
|
|
}];
|
|
objc_setAssociatedObject(self, &operationKey, operation, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
|
|
}
|
|
}
|
|
|
|
|
|
- (void)cancelCurrentImageLoad
|
|
{
|
|
// Cancel in progress downloader from queue
|
|
id<SDWebImageOperation> operation = objc_getAssociatedObject(self, &operationKey);
|
|
if (operation)
|
|
{
|
|
[operation cancel];
|
|
objc_setAssociatedObject(self, &operationKey, nil, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
|
|
}
|
|
}
|
|
|
|
@end
|