SDWebImage(5.0.0) 加載大圖內(nèi)存崩潰問題

網(wǎng)上的思路基本就是把下載的圖片進行壓縮,但是我這邊遇到點問題,一個是網(wǎng)上的基本是老版本的SDWebImage,我用的5.0.0的不太適用,再一個就是用的pod,直接在源碼更改不太理想。

我這邊的解決思路就是找到SDWebImage下載圖片的方法對下載完的圖片進行壓縮,當然實現(xiàn)方式使用runtime的swizzledSelector,這樣不用改動源碼。

SDWebImage下載圖片的方法

網(wǎng)上查了一些關于SDWebImage(5.0.0)源碼解析的資料,最后基本定位在這個兩個方法

- (void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)task didCompleteWithError:(NSError *)error

- (void)callCompletionBlocksWithImage:(nullable UIImage *)image
                            imageData:(nullable NSData *)imageData
                                error:(nullable NSError *)error
                             finished:(BOOL)finished

hook一個類中的實例方法

這里貼一下代碼

+ (void)swizzlingInClass:(Class)cls originalSelector:(SEL)originalSelector swizzledSelector:(SEL)swizzledSelector
{
    Class class = cls;
    Method originalMethod = class_getInstanceMethod(class, originalSelector);
    Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);
    if (!originalMethod || !swizzledMethod) {
        return;
    }

    IMP originalIMP = method_getImplementation(originalMethod);
    IMP swizzledIMP = method_getImplementation(swizzledMethod);
    const char *originalType = method_getTypeEncoding(originalMethod);
    const char *swizzledType = method_getTypeEncoding(swizzledMethod);

    class_replaceMethod(class,swizzledSelector,originalIMP,originalType);
    class_replaceMethod(class,originalSelector,swizzledIMP,swizzledType);
}

這里是具體實現(xiàn)

創(chuàng)建SDWebImageDownloaderOperation一個分類做hook,代碼供參考

#import "SDWebImageDownloaderOperation.h"
NS_ASSUME_NONNULL_BEGIN


@interface SDWebImageDownloaderOperation (HDImageCompressHook)

@end

NS_ASSUME_NONNULL_END
#import "SDWebImageDownloaderOperation+HDImageCompressHook.h"
#import "HDBJHookTool.h"

@implementation SDWebImageDownloaderOperation (HDImageCompressHook)
+ (void)load {
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        SEL originalSelector1 = @selector(URLSession: task: didCompleteWithError:);
        SEL swizzledSelector1 = @selector(hd_URLSession: task: didCompleteWithError:);
        [HDBJHookTool swizzlingInClass:[self class] originalSelector:originalSelector1 swizzledSelector:swizzledSelector1];
        [HDBJHookTool swizzlingInClass:[self class] originalSelector:originalSelector1 swizzledSelector:swizzledSelector1];
        
        SEL originalSelector2 = @selector(callCompletionBlocksWithImage:
                                          imageData:
                                          error:
                                          finished:);
        SEL swizzledSelector2 = @selector(hd_callCompletionBlocksWithImage:
                                          imageData:
                                          error:
                                          finished:);
        [HDBJHookTool swizzlingInClass:[self class] originalSelector:originalSelector2 swizzledSelector:swizzledSelector2];
    });
}
#pragma mark - Method Swizzling
- (void)hd_URLSession:(NSURLSession *)session task:(NSURLSessionTask *)task didCompleteWithError:(NSError *)error{

    Ivar ivar1 = class_getInstanceVariable([self class], "_imageData");
    if (object_getIvar(self, ivar1)) {
        NSMutableData * imageData = (NSMutableData *)object_getIvar(self, ivar1);
        NSData *imgData = UIImageJPEGRepresentation([UIImage imageWithData:[imageData copy]],1);
        NSUInteger sizeOriginKB = imageData.length / 1024;
         if (sizeOriginKB > 500){
            imgData = UIImageJPEGRepresentation([UIImage imageWithData:[imageData copy]],0.1);
            [[SDImageCache sharedImageCache] setValue:nil forKey:@"memCache"];
        }
//        imgData = [self resetSizeOfImageData:[UIImage imageWithData:[imageData copy]] maxSize:50];
        
        if (imgData.length/1024 > 200) {
            UIImage *image = [self compressImageWith:[UIImage imageWithData:imgData]];
            imgData = UIImagePNGRepresentation(image);
            MyLog(@"didCompleteWithError---------%ld",imgData.length/1024);
            if (imgData.length/1024 > 200) {
                imageData = [NSMutableData dataWithData:[self resetSizeOfImageData:image maxSize:90]];
                MyLog(@"didCompleteWithErrorresetSizeOfImageData---------%ld",imageData.length/1024);
            }
        }
        imageData = [NSMutableData dataWithData:imgData];
        object_setIvar(self, ivar1, imageData);
    }
    [self hd_URLSession:session task:task didCompleteWithError:error];
}

- (void)hd_callCompletionBlocksWithImage:(nullable UIImage *)image
                               imageData:(nullable NSData *)imageData
                                   error:(nullable NSError *)error
                                finished:(BOOL)finished{
    NSUInteger sizeOriginKB = imageData.length / 1024;
    if (sizeOriginKB > 500){
        imageData = UIImageJPEGRepresentation(image,0.1);
        image = [UIImage imageWithData:imageData];
    }
//    imageData = [self resetSizeOfImageData:image maxSize:90];
    image = [UIImage imageWithData:imageData];
    if (imageData.length/1024 > 200) {
        image = [self compressImageWith:image];
        imageData = UIImagePNGRepresentation(image);
        MyLog(@"hd_callCompletionBlocksWithImage---------%ld",imageData.length/1024);
        if (imageData.length/1024 > 200) {
            imageData = [self resetSizeOfImageData:image maxSize:90];
            image = [UIImage imageWithData:imageData];
            MyLog(@"hd_callCompletionBlocksWithImageresetSizeOfImageData---------%ld",imageData.length/1024);
        }
    }
    
    [self hd_callCompletionBlocksWithImage:image
                                 imageData:imageData
                                     error:error
                                  finished:finished];
}
-(UIImage *)compressImageWith:(UIImage *)image{
    
    float imageWidth = image.size.width;
    float imageHeight = image.size.height;
    float width = 320;
    float height = image.size.height/(image.size.width/width);
    float widthScale = imageWidth /width;
    float heightScale = imageHeight /height;
    // 創(chuàng)建一個bitmap的context
    // 并把它設置成為當前正在使用的context
    UIGraphicsBeginImageContext(CGSizeMake(width, height));
    if (widthScale > heightScale) {
        [image drawInRect:CGRectMake(0, 0, imageWidth /heightScale , height)];
    }else{
        [image drawInRect:CGRectMake(0, 0, width , imageHeight /widthScale)];
    }
    // 從當前context中創(chuàng)建一個改變大小后的圖片
    UIImage *newImage = UIGraphicsGetImageFromCurrentImageContext();
    // 使當前的context出堆棧
    UIGraphicsEndImageContext();
    return newImage;
}
#pragma mark - 圖片壓縮
- (NSData *)resetSizeOfImageData:(UIImage *)sourceImage maxSize:(NSInteger)maxSize {
    //先判斷當前質量是否滿足要求,不滿足再進行壓縮
    __block NSData *finallImageData = UIImageJPEGRepresentation(sourceImage,1.0);
    NSUInteger sizeOrigin   = finallImageData.length;
    NSUInteger sizeOriginKB = sizeOrigin / 1024;
    
    if (sizeOriginKB <= maxSize) {
        return finallImageData;
    }
    
    //獲取原圖片寬高比
    CGFloat sourceImageAspectRatio = sourceImage.size.width/sourceImage.size.height;
    //先調(diào)整分辨率
    CGSize defaultSize = CGSizeMake(1024, 1024/sourceImageAspectRatio);
    UIImage *newImage = [self newSizeImage:defaultSize image:sourceImage];
    
    finallImageData = UIImageJPEGRepresentation(newImage,1.0);
    
    //保存壓縮系數(shù)
    NSMutableArray *compressionQualityArr = [NSMutableArray array];
    CGFloat avg   = 1.0/250;
    CGFloat value = avg;
    for (int i = 250; i >= 1; i--) {
        value = i*avg;
        [compressionQualityArr addObject:@(value)];
    }
    
    /*
     調(diào)整大小
     說明:壓縮系數(shù)數(shù)組compressionQualityArr是從大到小存儲。
     */
    //思路:使用二分法搜索
    finallImageData = [self halfFuntion:compressionQualityArr image:newImage sourceData:finallImageData maxSize:maxSize];
    //如果還是未能壓縮到指定大小,則進行降分辨率
    while (finallImageData.length == 0) {
        //每次降100分辨率
        CGFloat reduceWidth = 100.0;
        CGFloat reduceHeight = 100.0/sourceImageAspectRatio;
        if (defaultSize.width-reduceWidth <= 0 || defaultSize.height-reduceHeight <= 0) {
            break;
        }
        defaultSize = CGSizeMake(defaultSize.width-reduceWidth, defaultSize.height-reduceHeight);
        UIImage *image = [self newSizeImage:defaultSize
                                      image:[UIImage imageWithData:UIImageJPEGRepresentation(newImage,[[compressionQualityArr lastObject] floatValue])]];
        finallImageData = [self halfFuntion:compressionQualityArr image:image sourceData:UIImageJPEGRepresentation(image,1.0) maxSize:maxSize];
    }
    return finallImageData;
}
#pragma mark 調(diào)整圖片分辨率/尺寸(等比例縮放)
- (UIImage *)newSizeImage:(CGSize)size image:(UIImage *)sourceImage {
    CGSize newSize = CGSizeMake(sourceImage.size.width, sourceImage.size.height);
    
    CGFloat tempHeight = newSize.height / size.height;
    CGFloat tempWidth = newSize.width / size.width;
    
    if (tempWidth > 1.0 && tempWidth > tempHeight) {
        newSize = CGSizeMake(sourceImage.size.width / tempWidth, sourceImage.size.height / tempWidth);
    } else if (tempHeight > 1.0 && tempWidth < tempHeight) {
        newSize = CGSizeMake(sourceImage.size.width / tempHeight, sourceImage.size.height / tempHeight);
    }
    
    UIGraphicsBeginImageContext(newSize);
    [sourceImage drawInRect:CGRectMake(0,0,newSize.width,newSize.height)];
    UIImage *newImage = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    return newImage;
}
#pragma mark 二分法
- (NSData *)halfFuntion:(NSArray *)arr image:(UIImage *)image sourceData:(NSData *)finallImageData maxSize:(NSInteger)maxSize {
    NSData *tempData = [NSData data];
    NSUInteger start = 0;
    NSUInteger end = arr.count - 1;
    NSUInteger index = 0;
    
    NSUInteger difference = NSIntegerMax;
    while(start <= end) {
        index = start + (end - start)/2;
        
        finallImageData = UIImageJPEGRepresentation(image,[arr[index] floatValue]);
        
        NSUInteger sizeOrigin = finallImageData.length;
        NSUInteger sizeOriginKB = sizeOrigin / 1024;
//        NSLog(@"當前降到的質量:%ld", (unsigned long)sizeOriginKB);
//        NSLog(@"\nstart:%zd\nend:%zd\nindex:%zd\n壓縮系數(shù):%lf", start, end, (unsigned long)index, [arr[index] floatValue]);
        
        if (sizeOriginKB > maxSize) {
            start = index + 1;
        } else if (sizeOriginKB < maxSize) {
            if (maxSize-sizeOriginKB < difference) {
                difference = maxSize-sizeOriginKB;
                tempData = finallImageData;
            }
            if (index<=0) {
                break;
            }
            end = index - 1;
        } else {
            break;
        }
    }
    return tempData;
}
@end

主要的壓縮圖片代碼是這里

NSUInteger sizeOriginKB = imageData.length / 1024;
    if (sizeOriginKB > 500){
        imageData = UIImageJPEGRepresentation(image,0.1);
        image = [UIImage imageWithData:imageData];
    }
//    imageData = [self resetSizeOfImageData:image maxSize:90];
    image = [UIImage imageWithData:imageData];
    if (imageData.length/1024 > 200) {
        image = [self compressImageWith:image];
        imageData = UIImagePNGRepresentation(image);
        MyLog(@"hd_callCompletionBlocksWithImage---------%ld",imageData.length/1024);
        if (imageData.length/1024 > 200) {
            imageData = [self resetSizeOfImageData:image maxSize:90];
            image = [UIImage imageWithData:imageData];
            MyLog(@"hd_callCompletionBlocksWithImageresetSizeOfImageData---------%ld",imageData.length/1024);
        }
    }

這些壓縮圖片的方法是網(wǎng)上找的,在使用- (NSData *)resetSizeOfImageData:(UIImage *)sourceImage maxSize:(NSInteger)maxSize 這個方法直接壓縮比較大的圖片時比較慢,最后使用了現(xiàn)在這種多級判斷壓縮圖片來實現(xiàn)。

大概就是這么一個思路了誰有更好的方法也可提供一下。

?著作權歸作者所有,轉載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容