iOS对SDWebImage下载的图片进行圆角化处理

工作三年了,都没写过博客,没想到第一篇居然在简书上写的。
最近工作不好找啊,写点博客聊以自慰吧。不吐槽了,开始上正题吧。
我们经常会遇到头像圆角问题,当然如果一个界面里就一个圆角的imageView,感觉也没必要优化。
简单粗暴代码如下:

imageView.layer.cornerRadius = CGRectGetWidth(imageView.bounds)/2;
imageView.layer.masksToBounds = YES;

但是当在一个tableView的每一个Cell中的imageView都要设置圆角的话,麻烦就来了。就会导致离屏渲染,让CPU处理它不擅长而GPU擅长的切图工作,让GPU等着渲染。一帧下来,CPU还没处理完,GPU只能不渲染了,这就引起了掉帧。滑动卡顿是必须的。
当然了,用GPU去切图就好了,所以最好的办法就是对image进行处理,切圆角。我用了一个image的category添加一个方法处理图片圆角

@interface UIImage (WebRadius)
- (UIImage*)imageByRoundCornerRadius;
@end

@implementation UIImage (WebRadius)
- (UIImage*)imageByRoundCornerRadius{
    UIGraphicsBeginImageContextWithOptions(self.size, NO, self.scale);
    CGContextRef context = UIGraphicsGetCurrentContext();
    CGContextScaleCTM(context, 1, -1);
    CGContextTranslateCTM(context, 0, -self.size.height);
    CGFloat minSize = MIN(self.size.width, self.size.height);
    CGRect rect = CGRectMake((self.size.width-minSize)/2, (self.size.height-minSize)/2, minSize, minSize);
    UIBezierPath * path = [UIBezierPath bezierPathWithRoundedRect:rect cornerRadius:minSize/2];
    [path closePath];
    CGContextSaveGState(context);
    [path addClip];
    CGContextDrawImage(context, rect, self.CGImage);
    CGContextRestoreGState(context);
    UIImage * image = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    return image;
}
@end

图片只需要调用这个方法就可以进行image圆角处理。处理是在GPU中进行。(灵感来源于YYKit)
那么对于网络加载的图片该怎么处理?
我们很多人会使用SDWebImage进行网络请求图片,难道每次获取图片都要进行图片圆角化处理,然后显示吗?如果tableview一直滑动,那不是一直在圆角化处理吗?于是我想在SDWebImage从网络下载中对图片圆角化处理一下,然后保存在memory和disk中,再次从缓存中获取图片就不用圆角化处理了。SDWebImage在从网络下载图片成功后一般都会走SDImageCache的

storeImage:imageData:forKey:toDisk:completion:

方法进行保存。于是我使用了SDImageCache的category替换了storeImage:imageData:forKey:toDisk:completion:方法
代码如下:

@interface SDImageCache (WebRadius)
@property (nonatomic,retain)NSNumber * isCornerRadius;
@end

@implementation SDImageCache (WebRadius)
+ (void)load{
    Method originalMethod = class_getInstanceMethod(self, @selector(storeImage:imageData:forKey:toDisk:completion:));
    Method myMethod = class_getInstanceMethod(self, @selector(lh_storeImage:imageData:forKey:toDisk:completion:));
    method_exchangeImplementations(originalMethod, myMethod);
}
static const char cornerRadiusKey = '\0';
- (void)setIsCornerRadius:(NSNumber *)isCornerRadius{
    [self willChangeValueForKey:@"isCornerRadius"];
    objc_setAssociatedObject(self, &cornerRadiusKey, isCornerRadius, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
    [self didChangeValueForKey:@"isCornerRadius"];
}
- (NSNumber*)isCornerRadius{
    return objc_getAssociatedObject(self, &cornerRadiusKey);
}
- (void)lh_storeImage:(nullable UIImage *)image
            imageData:(nullable NSData *)imageData
               forKey:(nullable NSString *)key
               toDisk:(BOOL)toDisk
           completion:(nullable SDWebImageNoParamsBlock)completionBlock
 {
    if (!image || !key) {
        if (completionBlock) {
            completionBlock();
        }
    return;
    }
    if ([self.isCornerRadius boolValue]) {
        image = [image imageByRoundCornerRadius];
        NSString * imageType = [self typeForImageData:imageData];
        if ([imageType isEqualToString:@"image/jpeg"]) {
            imageData = UIImageJPEGRepresentation(image, 1.0);
        }
        else if ([imageType isEqualToString:@"image/png"]){
            imageData = UIImagePNGRepresentation(image);
        }
    }
    Ivar memCachevar = class_getInstanceVariable([self class], [@"memCache" UTF8String]);
    if (memCachevar == nil) {
        memCachevar = class_getInstanceVariable([self class], [[NSString  stringWithFormat:@"_%@",@"memCache"] UTF8String]);
    }
    NSCache * memCache = object_getIvar(self, memCachevar);
    Ivar ioQueuevar = class_getInstanceVariable([self class], [@"ioQueue" UTF8String]);
    if (ioQueuevar == nil) {
        ioQueuevar = class_getInstanceVariable([self class], [[NSString stringWithFormat:@"_%@",@"ioQueue"] UTF8String]);
    }
    dispatch_queue_t  ioQueue = object_getIvar(self, ioQueuevar);
    if (self.config.shouldCacheImagesInMemory) {
        NSUInteger cost = image.size.height * image.size.width * image.scale * image.scale;
        [memCache setObject:image forKey:key cost:cost];
}
    if (toDisk) {
        dispatch_async(ioQueue, ^{
            @autoreleasepool {
                NSData *data = imageData;
                if (!data && image) {
                    SDImageFormat imageFormatFromData = [NSData sd_imageFormatForImageData:data];
                    data = [image sd_imageDataAsFormat:imageFormatFromData];
                }
                [self storeImageDataToDisk:data forKey:key];
            }
            if (completionBlock) {
                dispatch_async(dispatch_get_main_queue(), ^{
                    completionBlock();
                });
            }
        });
    }
    else {
        if (completionBlock) {
            completionBlock();
        }
    }
}
- (NSString *)typeForImageData:(NSData *)data {
    uint8_t c;
    [data getBytes:&c length:1];
    switch (c) {
        case 0xFF:
            return @"image/jpeg";
        case 0x89:
            return @"image/png";
        case 0x47:
            return @"image/gif";
        case 0x49:
        case 0x4D:
            return @"image/tiff";
     }
    return nil;
}

其中isCornerRadius是是否圆角的判断。这样就可以在不修改SDWebImage源码的情况下进行图片圆角操作了。不过主程序就得这样写

[SDImageCache sharedImageCache].isCornerRadius = [NSNumber numberWithBool:YES];
UIImageView * imageView = [[UIImageView alloc] init];
imageView.backgroundColor = [UIColor clearColor];
NSString * urlString = @"http://www.cdhdky.com/images/ttt.jpg";
UIImage * image = [[SDImageCache sharedImageCache] imageFromCacheForKey:urlString];
if (image) {
      imageView.frame = CGRectMake(0, 0, image.size.width, image.size.height);
      imageView.center = self.view.center;
      imageView.image = image;
}
else{
    __weak __typeof(self) weakSelf = self;
    [[SDWebImageManager sharedManager] loadImageWithURL:[NSURL URLWithString:urlString] options:SDWebImageRetryFailed progress:nil completed:^(UIImage * _Nullable image, NSData * _Nullable data, NSError * _Nullable error, SDImageCacheType cacheType, BOOL finished, NSURL * _Nullable imageURL) {
        imageView.frame = CGRectMake(0, 0, image.size.width, image.size.height);
        imageView.center = weakSelf.view.center;
        imageView.image = [[SDImageCache sharedImageCache]imageFromCacheForKey:imageURL.absoluteString];
    }];
}
[self.view addSubview:imageView];

本人有点懒,如果有什么不懂的请留言。这个category有点缺陷需要继续完善。从今天起,开始分享我的知识,虽然我很笨。
完整demo在github中

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 205,033评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,725评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,473评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,846评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,848评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,691评论 1 282
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,053评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,700评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 42,856评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,676评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,787评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,430评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,034评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,990评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,218评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,174评论 2 352
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,526评论 2 343

推荐阅读更多精彩内容