#import "UIImage+ASConvenience.h"

#import <UIKit/UIKit.h>

#import "ASInternalHelpers.h"

#import "ASAssert.h"

@implementation UIImage (ASDKAdditions)

+ (UIImage *)as_resizableRoundedImageWithCornerRadius:(CGFloat)cornerRadius

cornerColor:(UIColor *)cornerColor

fillColor:(UIColor *)fillColor

{

return [self as_resizableRoundedImageWithCornerRadius:cornerRadius

cornerColor:cornerColor

fillColor:fillColor

borderColor:nil

borderWidth:1.0

roundedCorners:UIRectCornerAllCorners

scale:0.0];

}

+ (UIImage *)as_resizableRoundedImageWithCornerRadius:(CGFloat)cornerRadius

cornerColor:(UIColor *)cornerColor

fillColor:(UIColor *)fillColor

borderColor:(UIColor *)borderColor

borderWidth:(CGFloat)borderWidth

{

return [self as_resizableRoundedImageWithCornerRadius:cornerRadius

cornerColor:cornerColor

fillColor:fillColor

borderColor:borderColor

borderWidth:borderWidth

roundedCorners:UIRectCornerAllCorners

scale:0.0];

}

+ (UIImage *)as_resizableRoundedImageWithCornerRadius:(CGFloat)cornerRadius

cornerColor:(UIColor *)cornerColor

fillColor:(UIColor *)fillColor

borderColor:(UIColor *)borderColor

borderWidth:(CGFloat)borderWidth

roundedCorners:(UIRectCorner)roundedCorners

scale:(CGFloat)scale

{

static NSCache *__pathCache = nil;

static dispatch_once_t onceToken;

dispatch_once(&onceToken, ^{

__pathCache = [[NSCache alloc] init];

// UIBezierPath objects are fairly small and these are equally sized. 20 should be plenty for many different parameters.

__pathCache.countLimit = 20;

});

// Treat clear background color as no background color

if ([cornerColor isEqual:[UIColor clearColor]]) {

cornerColor = nil;

}

CGFloat dimension = (cornerRadius * 2) + 1;

CGRect bounds = CGRectMake(0, 0, dimension, dimension);

typedef struct {

UIRectCorner corners;

CGFloat radius;

} PathKey;

PathKey key = { roundedCorners, cornerRadius };

NSValue *pathKeyObject = [[NSValue alloc] initWithBytes:&key objCType:@encode(PathKey)];

UIBezierPath *path = [__pathCache objectForKey:pathKeyObject];

if (path == nil) {

CGSize cornerRadii = CGSizeMake(cornerRadius, cornerRadius);

path = [UIBezierPath bezierPathWithRoundedRect:bounds byRoundingCorners:roundedCorners cornerRadii:cornerRadii];

[__pathCache setObject:path forKey:pathKeyObject];

}

// We should probably check if the background color has any alpha component but that

// might be expensive due to needing to check mulitple color spaces.

UIGraphicsBeginImageContextWithOptions(bounds.size, cornerColor != nil, scale);

BOOL contextIsClean = YES;

if (cornerColor) {

contextIsClean = NO;

[cornerColor setFill];

// Copy "blend" mode is extra fast because it disregards any value currently in the buffer and overwrites directly.

UIRectFillUsingBlendMode(bounds, kCGBlendModeCopy);

}

BOOL canUseCopy = contextIsClean || (CGColorGetAlpha(fillColor.CGColor) == 1);

[fillColor setFill];

[path fillWithBlendMode:(canUseCopy ? kCGBlendModeCopy : kCGBlendModeNormal) alpha:1];

if (borderColor) {

[borderColor setStroke];

// Inset border fully inside filled path (not halfway on each side of path)

CGRect strokeRect = CGRectInset(bounds, borderWidth / 2.0, borderWidth / 2.0);

// It is rarer to have a stroke path, and our cache key only handles rounded rects for the exact-stretchable

// size calculated by cornerRadius, so we won't bother caching this path. Profiling validates this decision.

UIBezierPath *strokePath = [UIBezierPath bezierPathWithRoundedRect:strokeRect cornerRadius:cornerRadius];

[strokePath setLineWidth:borderWidth];

BOOL canUseCopy = (CGColorGetAlpha(borderColor.CGColor) == 1);

[strokePath strokeWithBlendMode:(canUseCopy ? kCGBlendModeCopy : kCGBlendModeNormal) alpha:1];

}

UIImage *result = UIGraphicsGetImageFromCurrentImageContext();

UIGraphicsEndImageContext();

UIEdgeInsets capInsets = UIEdgeInsetsMake(cornerRadius, cornerRadius, cornerRadius, cornerRadius);

result = [result resizableImageWithCapInsets:capInsets resizingMode:UIImageResizingModeStretch];

return result;

}