如何使用扩展功能保存分层图像

时间:2019-10-13 23:22:41

标签: uiview uiimage extension-methods calayer uigraphicscontext

我正在使用扩展功能将uiview保存为uiimage。该代码可保存uiimage。但是,我想做的是在保存到照片库的图像上保存透明图像。因此,我尝试使用扩展功能保存分层图像。现在只有uiivew被保存,第二层没有被保存。

    class ViewController: UIViewController,UINavigationControllerDelegate {
    @IBAction func press(_ sender: Any) {
        let jake = drawingView.takeSnapshotOfView(view: drawingView)
              guard let selectedImage = jake else {
                              print("Image not found!")
                              return
                 }
                 UIImageWriteToSavedPhotosAlbum(selectedImage, self, #selector(image(_:didFinishSavingWithError:contextInfo:)), nil)
    }}

  func takeSnapshotOfView(view:UIView) -> UIImage? {
            UIGraphicsBeginImageContext(CGSize(width: view.frame.size.width, height: view.frame.size.height))
            view.drawHierarchy(in: CGRect(x: 0.0, y: 0.0, width: view.frame.size.width, height: view.frame.size.height), afterScreenUpdates: true)


            let image = UIGraphicsGetImageFromCurrentImageContext()
            UIGraphicsEndImageContext()

                let star:UIImage = UIImage(named: "e.png")!
                            let newSize = CGSize(width: star.size.width, height: star.size.height  )

                            UIGraphicsBeginImageContextWithOptions(newSize, false, star.scale)

                            star.draw(in: CGRect(x: newSize.width/12,
                                                 y: newSize.height/8,
                                                 width: newSize.width/1.2,
                                                 height: newSize.height/1.2),
                                      blendMode:CGBlendMode.normal, alpha:1)




                            UIGraphicsEndImageContext()

            return image
        }

1 个答案:

答案 0 :(得分:1)

这是一个包含CGRect和UIImage的UIView扩展,另外,您还可以为其提供另一个CGRect或CGSize,以使其对水印放置/大小更具动态感

extension UIView {        
    /// Takes a screenshot of a UIView, with an option to clip to view bounds and place a waterwark image 
    /// - Parameter rect: offset and size of the screenshot to take
    /// - Parameter clipToBounds: Bool to check where self.bounds and rect intersect and adjust size so there is no empty space
    /// - Parameter watermark: UIImage of the watermark to place on top
    func screenshot(for rect: CGRect, clipToBounds: Bool = true, with watermark: UIImage? = nil) -> UIImage {
        var imageRect = rect
        if clipToBounds {
            imageRect = bounds.intersection(rect)
        }
        return UIGraphicsImageRenderer(bounds: imageRect).image { _ in
        drawHierarchy(in: CGRect(origin: .zero, size: bounds.size), afterScreenUpdates: true)
        watermark?.draw(in: CGRect(origin: imageRect.origin, size: CGSize(width: 32, height: 32))) // update origin to place watermark where you want, with this update it will place it in top left or screenshot.
        }
    }
}

您可以这样称呼它:

let image = self.view.screenshot(for: CGRect(x: 0, y: 0, width: 200, height: 200), with: UIImage(named: "star"))

这将适用于调用屏幕快照(...)的视图的所有子视图

对于使用上述扩展名的任何人,我在此answer

中添加了其他信息