iOS 截圖的那些事兒

同時按下 Home 鍵和電源鍵，咔嚓一聲，就獲得了一張手機的截圖，這操做想必 iPhone 用戶再熟悉不過了。咱們做爲研發人員，面對的是一個個的 View，那麼該怎麼用代碼對 View 進行截圖呢？
這篇文章主要討論的是如何在包括 UIWebView 和 WKWebView 的網頁中進行長截圖。

UIWebView 截圖

對 UIWebView 截圖比較簡單，renderInContext 這個方法相信你們都不會陌生，這個方法是 CALayer 的一個實例方法，能夠用來對大部分 View 進行截圖。咱們知道，UIWebView 承載內容的實際上是做爲其子 View 的 UIScrollView，因此對 UIWebView 截圖應該對其 scrollView 進行截圖。具體的截圖方法以下：

- (void)snapshotForScrollView:(UIScrollView *)scrollView
{
    // 1. 記錄當前 scrollView 的偏移和位置
    CGPoint currentOffset = scrollView.contentOffset;
    CGRect currentFrame = scrollView.frame;

    scrollView.contentOffset = CGPointZero;
    // 2. 將 scrollView 展開爲其實際內容的大小
    scrollView.frame = CGRectMake(0, 0, scrollView.contentSize.width, scrollView.contentSize.height);

    // 3. 第三個參數設置爲 0 表示設置爲屏幕的默認縮放因子
    UIGraphicsBeginImageContextWithOptions(scrollView.contentSize, YES, 0);
    [scrollView.layer renderInContext:UIGraphicsGetCurrentContext()];
    UIImage *snapshotImage = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    
    // 4. 從新設置 scrollView 的偏移和位置，還原現場
    scrollView.contentOffset = currentOffset;
    scrollView.frame = currentFrame;
}

WKWebView 截圖

雖然 WKWebView 裏也有 scrollView，可是直接對這個 scrollView 截圖獲得的是一片空白的，具體緣由不明。一番 Google 以後能夠看到好些人提到 drawViewHierarchyInRect 方法， 能夠看到這個方法是 iOS 7.0 開始引入的。官方文檔中描述爲：

Renders a snapshot of the complete view hierarchy as visible onscreen into the current context.

注意其中的 visible onscreen，也就是將屏幕中可見部分渲染到上下文中，這也解釋了爲何對 WKWebView 中的 scrollView 展開爲實際內容大小，再調用 drawViewHierarchyInRect 方法老是獲得一張不完整的截圖（只有屏幕可見區域被正確截到，其餘區域爲空白）。
不過，這樣卻是給咱們提供了一個思路，能夠將 WKWebView 按屏幕高度裁成 n 頁，而後將 WKWebView 一頁一頁的往上推，每推一頁就調用一次 drawViewHierarchyInRect 將當前屏幕的截圖渲染到上下文中，最後調用 UIGraphicsGetImageFromCurrentImageContext 從上下文中獲取的圖片即爲完整截圖。

在這裏我仍是要推薦下我本身建的iOS開發學習羣：680565220，羣裏都是學ios開發的，若是你正在學習ios ，小編歡迎你加入，今天分享的這個案例已經上傳到羣文件，你們都是軟件開發黨，不按期分享乾貨（只有iOS軟件開發相關的），包括我本身整理的一份2018最新的iOS進階資料和高級開發教程

核心代碼以下（代碼爲演示用途，完整代碼請從這裏查看)：

- (void)snapshotForWKWebView:(WKWebView *)webView
{
    // 1
    UIView *snapshotView = [webView snapshotViewAfterScreenUpdates:YES];
    [webView.superview addSubview:snapshotView];

    // 2
    CGPoint currentOffset = webView.scrollView.contentOffset;
    ...

    // 3
    UIView *containerView = [[UIView alloc] initWithFrame:webView.bounds];
    [webView removeFromSuperview];
    [containerView addSubview:webView];

    
    // 4
    CGSize totalSize = webView.scrollView.contentSize;
    NSInteger page = ceil(totalSize.height / containerView.bounds.size.height);

    webView.scrollView.contentOffset = CGPointZero;
    webView.frame = CGRectMake(0, 0, containerView.bounds.size.width, webView.scrollView.contentSize.height);

    UIGraphicsBeginImageContextWithOptions(totalSize, YES, UIScreen.mainScreen.scale);
    [self drawContentPage:0 maxIndex:page completion:^{
        UIImage *snapshotImage = UIGraphicsGetImageFromCurrentImageContext();
        UIGraphicsEndImageContext();

        // 8
        [webView removeFromSuperview];
        ...
    }];
}

- (void)drawContentPage(NSInteger)index maxIndex:(NSInteger)maxIndex completion:(dispatch_block_t)completion
{
    // 5
    CGRect splitFrame = CGRectMake(0, index * CGRectGetHeight(containerView.bounds), containerView.bounds.size.width, containerView.frame.size.height);
    CGRect myFrame = webView.frame;
    myFrame.origin.y = -(index * containerView.frame.size.height);
    webView.frame = myFrame;

    // 6
    [targetView drawViewHierarchyInRect:splitFrame afterScreenUpdates:YES];

    // 7
    if (index < maxIndex) {
        [self drawContentPage:index + 1 maxIndex:maxIndex completion:completion];
    } else {
        completion();
    }
}

代碼注意項以下（對應代碼註釋中的序號）：

1. 爲了截圖時對 frame 進行操做不會出現閃屏等現象，咱們須要蓋一個「假」的 webView 到如今的位置上，並將真正的 webView 「摘下來」。調用 snapshotViewAfterScreenUpdates 便可獲得這樣一個「假」的 webView
2. 保存真正的 webView 的偏移、位置等信息，以便截圖完成以後「還原現場」
3. 用一個新的視圖承載「真正的」 webView，這個視圖也是繪圖所用到的上下文
4. 將 webView 按照實際內容高度和屏幕高度分紅 page 頁
5. 獲得每一頁的實際位置，並將 webView 往上推到該位置
6. 調用 drawViewHierarchyInRect 將當前位置的 webView 渲染到上下文中
7. 若是還未到達最後一頁，則遞歸調用 drawViewHierarchyInRect 方法進行渲染；若是已經渲染完了所有頁，則回調通知截圖完成
8. 調用 UIGraphicsGetImageFromCurrentImageContext 方法從當前上下文中獲取到完整截圖，將第 2 步中保存的信息從新賦予到 webView 上，「還原現場」

注意：咱們的截圖方法中有對 webView 的 frame 進行操做，若是其餘地方若是有對 frame 進行操做的話，是會影響咱們截圖的。因此在截圖時應該禁用掉其餘地方對 frame 的改變，就像這樣：

- (void)layoutWebView
{
    if (!_isCapturing) {
        self.wkWebView.frame = [self frameForWebView];
    }
}

結語

當前 WKWebView 的使用愈來愈普遍了，我隨意查看了內存佔用：打開一樣一個網頁，UIWebView 直接佔用了 160 MB 內存，而 WKWebView 只佔用了 40 MB 內存，差距是至關明顯的。若是咱們的業務中用到了 WKWebView 且有截圖需求的話，那麼仍是得老老實實完成的。

更多編程分享請關注微信公衆號：程序員大牛！