PaintCodeDrawableImages

PaintCodeDrawableImages.m

typedef void (^PaintCodeSourceBlock)();

@implementation PaintCodeDrawableImages

+ (UIImage *)drawImageWithSize:(CGSize)size paintCodeSource:(PaintCodeSourceBlock)paintCodeSourceBlock
{
    // Begin the image context.
    UIGraphicsBeginImageContextWithOptions(size, NO, 0.0f);

    // Draw the paint code image.
    paintCodeSourceBlock();

    // Create UIImage from context.
    UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    return image;
}