UIImage+Additions.h

UIImage+Additions.h

typedef void (^UIImageDrawBlock)( CGContextRef ctx, CGRect rect );

@interface UIImage (Additions)

+ (UIImage*)imageWithSize:(CGSize)size opaque:(BOOL)opaque scale:(CGFloat)scale draw:(UIImageDrawBlock)draw;

@end

UIImage+Additions.m

@implementation UIImage (Additions)

+ (UIImage*)imageWithSize:(CGSize)size opaque:(BOOL)opaque scale:(CGFloat)scale draw:(UIImageDrawBlock)draw
{
    UIGraphicsBeginImageContextWithOptions( size, opaque, scale );
    
    CGContextRef ctx = UIGraphicsGetCurrentContext();
    CGRect rect = CGRectMake( 0, 0, size.width, size.height );
    
    CGContextClearRect( ctx, rect );
    
    if( draw ) {
        draw( ctx, rect );
    }
    
    UIImage* image = UIGraphicsGetImageFromCurrentImageContext();
    
    UIGraphicsEndImageContext();
    
    return image;
}

@end