iOS實現簡單長截圖

2022-07-20 22:00:16

本文範例為大家分享了iOS實現簡易的長截圖的具體程式碼，供大家參考，具體內容如下

長截圖的實現原理：

實際上是將view的內容繪製成圖片，再將各個view繪製出來的圖片拼接出來。

具體程式碼：

將view繪製成圖片

func getImage(in view:UIView?) -> UIImage? {
    guard let view = view else {return nil}
    let size = view.bounds.size
    UIGraphicsBeginImageContextWithOptions(size, false, UIScreen.main.scale)
    view.layer.render(in: UIGraphicsGetCurrentContext()!)
    view.layer.contents = nil
    let image = UIGraphicsGetImageFromCurrentImageContext()
    UIGraphicsEndImageContext()
    return image
}

1.繪製狀態列

// 先獲取狀態列view
func getStatusBar() -> UIView? {
    if Float(UIDevice.current.systemVersion)! >= 13 {
        guard let statusBarManager = UIApplication.shared.keyWindow?.windowScene?.statusBarManager,
              let localStatusBar = statusBarManager.value(forKey: "createLocalStatusBar") as? NSObject,
              let statusBar = localStatusBar.value(forKey: "statusBar") as? UIView else {return nil}
        return statusBar
    } else {
        guard let statusBarWindow = UIApplication.shared.value(forKey: "statusBarWindow") as? UIWindow  else {return nil}
        let statusBar = statusBarWindow.value(forKey: "statusBar") as? UIView
        return statusBar
    }
}

// 再繪製成圖片
let statusBarImage = getImage(in: getStatusBar())

2.繪製導航欄(如果有的話)

if let nav = vc.navigationController { // 表示有導航欄
    let navBar = nav.navigationBar
    let navBarImage = getImage(in: navBar)
}

在某些情況下，如自定義的導航欄，則需要另外自行獲取view再來繪製

3.繪製ScrollView

繪製scrollview長圖的時候，如果直接繪製的話，那麼得到的圖片就是scrollview的frame.size大小的圖片，而沒有捲動到的地方則不會繪製進去，所以要先將scrollview的size變成和contentSize一樣

func getScrollViewImage(scrollView:UIScrollView?) -> UIImage? {
    if let scroll = scrollView {
        let saveOffset = scroll.contentOffset // 儲存偏移量，用於繪製圖片完成後還原
        let saveFrame = scroll.frame // 儲存frame
        scroll.contentOffset = CGPoint.zero
        scroll.frame = CGRect(origin: saveFrame.origin, size: scroll.contentSize) // 設定size和contentSize一致
        
        UIGraphicsBeginImageContext(scroll.frame.size)
        UIGraphicsBeginImageContextWithOptions(CGSize(width: scroll.frame.size.width, height: scroll.frame.size.height), false, UIScreen.main.scale)
        scroll.layer.render(in: UIGraphicsGetCurrentContext()!)
        let scrollImage = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
        
        scroll.contentOffset = saveOffset // 還原偏移量，否則繪製圖片之後，scrollview偏移量不正確
        scroll.frame = saveFrame // 還原frame
        return scrollImage
    }
    return nil
}

到這裡，其實最難的一步已經完成了，但是這個方法也不是完美的。當這個scrollview新增到父檢視的時候是新增約束的方式來確定大小和位置的時候，使用這個方法繪製出來的圖片大小沒有問題，但是內容卻可能只有螢幕上顯示出來的部分多一點，其他部分是空白。具體原因我也不太清楚，只要在繪製成圖片之前將約束移除，繪製之後再新增回來

func getScrollViewImage(scrollView:UIScrollView?) -> UIImage? {
    if let scroll = scrollView {
        let saveOffset = scroll.contentOffset // 儲存偏移量，用於繪製圖片完成後還原
        let saveFrame = scroll.frame // 儲存frame
        scroll.contentOffset = CGPoint.zero
        scroll.frame = CGRect(origin: saveFrame.origin, size: scroll.contentSize) // 設定size和contentSize一致
        let layouts = scroll.superview!.constraints // 獲取的是scrollview的父容器的約束，這才是約束scrollview大小和位置的正確約束
        scroll.superview?.removeConstraints(layouts)
        
        UIGraphicsBeginImageContext(scroll.frame.size)
        UIGraphicsBeginImageContextWithOptions(CGSize(width: scroll.frame.size.width, height: scroll.frame.size.height), false, UIScreen.main.scale)
        scroll.layer.render(in: UIGraphicsGetCurrentContext()!)
        let scrollImage = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
        
        scroll.contentOffset = saveOffset // 還原偏移量，否則繪製圖片之後，scrollview偏移量不正確
        scroll.frame = saveFrame // 還原frame
        scroll.superview?.addConstraints(layouts) // 還原約束
        return scrollImage
    }
    return nil
}

如果專案中整合了SnapKit的話可以用其給scrollview重新設定約束，繪製圖片結束後在還原也可以

func getScrollViewImage(scrollView:UIScrollView?) -> UIImage? {
    if let scroll = scrollView {
        
        // ......

        let layouts = scroll.superview!.constraints // 獲取的是scrollview的父容器的約束，這才是約束scrollview大小和位置的正確約束
        scroll.snp.remakeConstraints { (make) in
            make.top.left.right.equalTo(0)
            make.height.equalTo(scroll.contentSize.height)
        }
        
        // ...... 
        
        scroll.snp.removeConstraints()
        scroll.superview?.addConstraints(layouts) // 還原約束
        
        return scrollImage
    }
    return nil
}

4.拼接圖片

已經得到了所需要的各個部分的元素，這裡按照上下位置將其拼接起來

func combineImages(with upImage:UIImage?, and downImage:UIImage?) -> UIImage? {
    if upImage == nil {
        return downImage
    }
    if downImage == nil {
        return upImage
    }
    guard let up = upImage,
          let down = downImage else {return nil}
    let size = CGSize(width: up.size.width, height: up.size.height + down.size.height)
    UIGraphicsBeginImageContextWithOptions(size, false, UIScreen.main.scale)
    
    let upRect = CGRect(origin: CGPoint(x: 0, y: 0), size: up.size)
    up.draw(in: upRect)
    let downRect = CGRect(x: (up.size.width - down.size.width) / 2, y: upRect.origin.y + upRect.size.height, width: down.size.width, height: down.size.height)
    down.draw(in: downRect)
    
    let result = UIGraphicsGetImageFromCurrentImageContext()
    UIGraphicsEndImageContext()
    return result
}

這裡是圖片的上下拼接，左右拼接與這個類似，只需要計算好圖片的左右位置即可。

效果圖：

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支援it145.com。

iOS實現簡單長截圖

熱門文章