DDGScreenShot--iOS 各類截圖，webView wkWebView 生成長圖，加logo，打標籤

寫在前面

最近有這麼一個需求，分享頁面，分享的是web訂單截圖，既然是web 就會有超出屏幕的部分，
生成的圖片還要加上咱們的二維碼，這就涉及到圖片的合成了。
有了這樣的需求，就是各類google.也算解決了本身的需求，另外，也總結了一下，對需求作了下拓展，目前是swift4.0版本。
複製代碼

---

整合成一個三方庫，如下只是部分代碼，詳細代碼及demo請見，github地址https://github.com/dudongge/DDGScreenShot

DDGScreenShot 全部功能演示

---

View生成圖片

代碼也比較簡單 screenShotImage 就是截取後的圖片
        let context = UIGraphicsGetCurrentContext()
        context?.saveGState()
        context?.translateBy(x: -self.frame.origin.x, y: -self.frame.origin.y);
         self.layer.render(in: context!)
        let screenShotImage = UIGraphicsGetImageFromCurrentImageContext()
        context?.restoreGState();
        UIGraphicsEndImageContext()
複製代碼

---

ScrollView生成圖片

只要實現原理是計算偏移量，每一屏繪製一次，放在內存裏，最後將全部的圖片組合成一張圖片 screenShotImage就是最終圖片
public func DDGContentScrollScreenShot (_ completionHandler: @escaping (_ screenShotImage: UIImage?) -> Void) {
        
        self.isShoting = true
        
        let snapShotView = self.snapshotView(afterScreenUpdates: true)
        snapShotView?.frame = CGRect(x: self.frame.origin.x, y: self.frame.origin.y, width: (snapShotView?.frame.size.width)!, height: (snapShotView?.frame.size.height)!)
        self.superview?.addSubview(snapShotView!)

        let bakOffset    = self.contentOffset
        
        let page  = floorf(Float(self.contentSize.height / self.bounds.height))
        
        UIGraphicsBeginImageContextWithOptions(self.contentSize, false, UIScreen.main.scale)
        
        self.DDGContentScrollPageDraw(0, maxIndex: Int(page), drawCallback: { [weak self] () -> Void in
            let strongSelf = self
            
            let screenShotImage = UIGraphicsGetImageFromCurrentImageContext()
            UIGraphicsEndImageContext()
            
            strongSelf?.setContentOffset(bakOffset, animated: false)
            snapShotView?.removeFromSuperview()
            
            strongSelf?.isShoting = false
            
            completionHandler(screenShotImage)
        })
        
    }
    
    fileprivate func DDGContentScrollPageDraw (_ index: Int, maxIndex: Int, drawCallback: @escaping () -> Void) {
        
        self.setContentOffset(CGPoint(x: 0, y: CGFloat(index) * self.frame.size.height), animated: false)
        let splitFrame = CGRect(x: 0, y: CGFloat(index) * self.frame.size.height, width: bounds.size.width, height: bounds.size.height)
        
        DispatchQueue.main.asyncAfter(deadline: DispatchTime.now() + Double(Int64(0.3 * Double(NSEC_PER_SEC))) / Double(NSEC_PER_SEC)) { () -> Void in
            self.drawHierarchy(in: splitFrame, afterScreenUpdates: true)
            
            if index < maxIndex {
                self.DDGContentScrollPageDraw(index + 1, maxIndex: maxIndex, drawCallback: drawCallback)
            }else{
                drawCallback()
            }
        }
    }
複製代碼

---

UIwebView生成圖片 && wkwebView

繪製時大同小異，只是wkwebView 調用的繪製方法爲：drawHierarchy 其核心代碼以下
採用遞歸，直到拿到最後一個偏移量。
fileprivate func DDGRenderImageView(_ completionHandler: @escaping (_ screenShotImage: UIImage?) -> Void) {
        let ddgTempRenderView = UIView(frame: CGRect(x: 0, y: 0, width: self.contentSize.width, height: self.contentSize.height))
        self.removeFromSuperview()
        ddgTempRenderView.addSubview(self)
        
        self.contentOffset = CGPoint.zero
        self.frame         = ddgTempRenderView.bounds
        
        DispatchQueue.main.asyncAfter(deadline: DispatchTime.now() + Double(Int64(0.3 * Double(NSEC_PER_SEC))) / Double(NSEC_PER_SEC)) { () -> Void in
            let bounds = self.bounds
            UIGraphicsBeginImageContextWithOptions(bounds.size, false, UIScreen.main.scale)
            
            if (self.DDGContainsWKWebView()) {
                self.drawHierarchy(in: bounds, afterScreenUpdates: true)
            }else{
                self.layer.render(in: UIGraphicsGetCurrentContext()!)
            }
            let screenShotImage = UIGraphicsGetImageFromCurrentImageContext()
            UIGraphicsEndImageContext()
            
            completionHandler(screenShotImage)
        }
    }
fileprivate func shotScreenContentScrollPageDraw (_ index: Int, maxIndex: Int, drawCallback: @escaping () -> Void) {
        
        self.scrollView.setContentOffset(CGPoint(x: 0, y: CGFloat(index) * self.scrollView.frame.size.height), animated: false)
        let splitFrame = CGRect(x: 0, y: CGFloat(index) * self.scrollView.frame.size.height, width: bounds.size.width, height: bounds.size.height)
        
        DispatchQueue.main.asyncAfter(deadline: DispatchTime.now() + Double(Int64(0.3 * Double(NSEC_PER_SEC))) / Double(NSEC_PER_SEC)) { () -> Void in
            self.drawHierarchy(in: splitFrame, afterScreenUpdates: true)
            
            if index < maxIndex {
                self.shotScreenContentScrollPageDraw(index + 1, maxIndex: maxIndex, drawCallback: drawCallback)
            }else{
                drawCallback()
            }
        }
    }
複製代碼

---

兩張圖片合爲一張（一張底圖，一張logo）

在UIImage上作了拓展
let imageRef = self.cgImage
        let w: CGFloat = CGFloat((imageRef?.width)!)
        let h: CGFloat = CGFloat((imageRef?.height)!)
        //以1.png的圖大小爲畫布建立上下文
        UIGraphicsBeginImageContext(CGSize(width: w, height: h))
        self.draw(in: CGRect(x: 0, y: 0, width: w, height: h))
        //先把1.png 畫到上下文中
        logo.draw(in: CGRect(x: logoOrigin.x,
                             y: logoOrigin.y,
                             width: logoSize.width,
                             height:logoSize.height))
        //再把小圖放在上下文中
        let resultImg: UIImage? = UIGraphicsGetImageFromCurrentImageContext()
        //從當前上下文中得到最終圖片
        UIGraphicsEndImageContext()
        return resultImg!

複製代碼

在圖片上寫文字/加標籤

一樣對UIimage 作了拓展，增長了幾個參數
  public func drawTextInImage(text: String,
                         textColor: UIColor,
                         textFont: CGFloat,
                         textBgColor: UIColor,
                         textX: CGFloat,
                         textY: CGFloat )->UIImage {
        //開啓圖片上下文
        UIGraphicsBeginImageContext(self.size)
        //圖形重繪
        self.draw(in: CGRect.init(x: 0, y: 0, width: self.size.width, height: self.size.height))
        //水印文字屬性
        let att = [NSAttributedStringKey.foregroundColor: textColor,
                   NSAttributedStringKey.font: UIFont.systemFont(ofSize: textFont),
                   NSAttributedStringKey.backgroundColor: textBgColor]
        //水印文字大小
        let text = NSString(string: text)
        let size =  text.size(withAttributes: att)
        //繪製文字
        text.draw(in: CGRect.init(x: textX, y: textY, width: size.width, height: size.height), withAttributes: att)
        //從當前上下文獲取圖片
        let image = UIGraphicsGetImageFromCurrentImageContext()
        //關閉上下文
        UIGraphicsEndImageContext()
        return image!
    }
複製代碼

結束語

空間有限，所注代碼不全，我把上述功能整理成了最新的帶三方庫，
之後會加上圖片的相關處理，好比美顏，剪切，馬賽克，組圖等，
附上個人git地址：https://github.com/dudongge/DDGScreenShot
有什麼問題也能夠聯繫我QQ:532835032
若是對您有幫助，請您不吝star一下，增長我更新的動力
複製代碼

附上github上READER.ME文件部份內容

## view截屏：
    view.DDGScreenShot { (image) in
         拿到 image 
         各類複雜裝逼操做
         、、、、
    }
   ## ScrollView截屏：
    scrollView.DDGContentScrollScreenShot { (image) in
         拿到 image 
         各類複雜裝逼操做
         、、、、
    }
  ## webView截屏：
    webView.DDGContentscreenShot { (image) in
         拿到 image 
         各類複雜裝逼操做
         、、、、
    }
  ## wkwebView截屏： 方法和webView 同樣，內部作了校驗
      webView.DDGContentscreenShot { (image) in
           拿到 image 
           各類複雜裝逼操做
           、、、、
      }
  ## image 加 logo
     let image = image.composeImageWithLogo( logo: UIImage,
                               logoOrigin: CGPoint,
                               logoSize:CGSize) 
     傳入 logo圖片，logo位置 logo 大小 就能夠獲得一張生成好的圖片                         
         、、、、
  ## image 加 標籤，水印，文字
     let image = image.drawTextInImage(text: String,
                         textColor: UIColor,
                         textFont: CGFloat,
                         textBgColor: UIColor,
                         textX: CGFloat,
                         textY: CGFloat ) 
     傳入 文字、文字顏色、字體大小、背景顏色，字體起始位置 就能夠獲得一張生成好的帶標籤的圖片                         
         、、、、
         注，此方法在提交pod有問題，故將方法屏蔽，有須要的能夠拷貝代碼，到本地

## 使用pod
     iOS 9.0+, Swift 4.0+(Compatiable)
        使用pod 導入
        pod 'DDGScreenShot'
        ```

複製代碼