iOS 中獲取某個視圖的截圖

最近在作SDK的截圖，想觸發相似系統的截屏功能，找了一圈，總結一下靠譜的幾種方式。 我寫了個UIView 的category，將這幾種方式封裝和簡化了一下。

第一種情形截圖

這種是最最普通的截圖，針對通常的視圖上添加視圖的狀況，基本均可以使用。 源碼：

/**
 普通的截圖
 該API僅能夠在未使用layer和OpenGL渲染的視圖上使用
 
 @return 截取的圖片
 */
- (UIImage *)nomalSnapshotImage
{
    UIGraphicsBeginImageContextWithOptions(self.frame.size, NO, [UIScreen mainScreen].scale);
    [self.layer renderInContext:UIGraphicsGetCurrentContext()];
    UIImage *snapshotImage = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    
    return snapshotImage;
}
複製代碼

第二種情形截圖

若是一些視圖是用OpenGL渲染出來的，那麼使用上面的方式就沒法截圖到OpenGL渲染的部分，這時候就要用到改進後的截圖方案：

/**
 針對有用過OpenGL渲染過的視圖截圖
 
 @return 截取的圖片
 */
- (UIImage *)openglSnapshotImage
{
    CGSize size = self.bounds.size;
    UIGraphicsBeginImageContextWithOptions(size, NO, [UIScreen mainScreen].scale);
    CGRect rect = self.frame;
    [self drawViewHierarchyInRect:rect afterScreenUpdates:YES];
    UIImage *snapshotImage = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    
    return snapshotImage;
}
複製代碼

第三種情形截圖

有一些特殊的Layer（好比：AVCaptureVideoPreviewLayer 和 AVSampleBufferDisplayLayer） 添加到某個View 上後，使用上面的幾種方式都沒法截取到Layer上的內容，這個時候能夠使用系統的一個API，可是該API只能返回一個UIView，返回的UIView 能夠修改frame 等參數。

/**
 截圖
 以UIView 的形式返回(_UIReplicantView)
 
 @return 截取出來的圖片轉換的視圖
 */
- (UIView *)snapshotView
{
    UIView *snapView = [self snapshotViewAfterScreenUpdates:YES];
    return snapView;
}
複製代碼

遺留問題： 經過方式三截取的UIView，沒法轉換爲UIImage，我試過將返回的截圖View寫入位圖再轉換成UIImage，可是返回的UIImage 要麼爲空，要麼沒有內容。若是有人知道解決方案請告知我。

UIWebView的截圖

去年在作藍牙打印的時候，嘗試過將UIWebView 的內容轉換爲UIImage，寫過一個UIWebView的category，也算是對UIWebView 的截圖，順便也貼出來吧

- (UIImage *)imageForWebView
{
    // 1.獲取WebView的寬高
    CGSize boundsSize = self.bounds.size;
    CGFloat boundsWidth = boundsSize.width;
    CGFloat boundsHeight = boundsSize.height;

    // 2.獲取contentSize
    CGSize contentSize = self.scrollView.contentSize;
    CGFloat contentHeight = contentSize.height;
    // 3.保存原始偏移量，便於截圖後復位
    CGPoint offset = self.scrollView.contentOffset;
    // 4.設置最初的偏移量爲(0,0);
    [self.scrollView setContentOffset:CGPointMake(0, 0)];

    NSMutableArray *images = [NSMutableArray array];
    while (contentHeight > 0) {
        // 5.獲取CGContext 5.獲取CGContext
        UIGraphicsBeginImageContextWithOptions(boundsSize, NO, 0.0);
        CGContextRef ctx = UIGraphicsGetCurrentContext();
        // 6.渲染要截取的區域
        [self.layer renderInContext:ctx];
        UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
        UIGraphicsEndImageContext();
        // 7.截取的圖片保存起來
        [images addObject:image];

        CGFloat offsetY = self.scrollView.contentOffset.y;
        [self.scrollView setContentOffset:CGPointMake(0, offsetY + boundsHeight)];
        contentHeight -= boundsHeight;
    }
    // 8 webView 恢復到以前的顯示區域
    [self.scrollView setContentOffset:offset];
    CGFloat scale = [UIScreen mainScreen].scale;
    CGSize imageSize = CGSizeMake(contentSize.width * scale,
                                  contentSize.height * scale);
   // 9.根據設備的分辨率從新繪製、拼接成完整清晰圖片
    UIGraphicsBeginImageContext(imageSize);
    [images enumerateObjectsUsingBlock:^(UIImage *image, NSUInteger idx, BOOL *stop) {
        [image drawInRect:CGRectMake(0,scale * boundsHeight * idx,scale * boundsWidth,scale * boundsHeight)];
    }];
    UIImage *fullImage = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    
    return fullImage;
}
複製代碼