しめ鯖日記

swift, iPhoneアプリ開発, ruby on rails等のTipsや入門記事書いてます

UIViewにUIImageへの変換メソッドを追加する(Swift)

下のようにUIViewを拡張してUIImage生成メソッドを追加すればOK。

extension UIView {
    func toImage() -> UIImage? {
        UIGraphicsBeginImageContext(self.frame.size)
        guard let context = UIGraphicsGetCurrentContext() else {
            return nil
        }
        self.layer.render(in: context)
        let image = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
        
        return image
    }
}

下のようにお手軽にUIViewをUIImageに変換できます。

let view = UIView(frame: CGRectMake(0.0, 0.0, 10.0, 10.0))
let view.imageValue()

ちなみにObjective-cだと下の通り。

@interface UIView(Hoge)

- (UIImage*)toImage;

@end

@implementation UIView(Hoge)

- (UIImage*)toImage {
    UIGraphicsBeginImageContextWithOptions(self.size, NO, 0);
    CGContextRef context = UIGraphicsGetCurrentContext();
    CGContextTranslateCTM(context, 0, 0);
    [self.layer renderInContext:context];
    UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    
    return image;
}

@end