iOS截圖的示例分析

小編給大家分享一下iOS截圖的示例分析，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

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 從上下文中獲取的圖片即為完整截圖。

核心代碼如下（代碼為演示用途，完整代碼請從這里 （本地下載） 查看)：

- (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];
 }
}

以上是“iOS截圖的示例分析”這篇文章的所有內容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內容對大家有所幫助，如果還想學習更多知識，歡迎關注億速云行業資訊頻道！