getSnapshotFromView

.m

+ (nullable UIImage *)getSnapshotFromView:(UIView *)view useScreenUpdate:(BOOL)useScreenUpdate
{
    UIGraphicsBeginImageContextWithOptions(view.bounds.size, view.opaque, [UIScreen mainScreen].scale);
    
    UIImage * snapshotImage = NULL;
    
    if ([view respondsToSelector:@selector(drawViewHierarchyInRect:afterScreenUpdates:)] && useScreenUpdate) {
        BOOL drawSuccess = [view drawViewHierarchyInRect:view.bounds afterScreenUpdates:NO];
        if (drawSuccess) {
            snapshotImage = UIGraphicsGetImageFromCurrentImageContext();
        }
    } else {
        [view.layer renderInContext:UIGraphicsGetCurrentContext()];
        snapshotImage = UIGraphicsGetImageFromCurrentImageContext();
    }
    
    UIGraphicsEndImageContext();
    
    return snapshotImage;
}