創(chuàng)建 UIView 的Extension,見下圖
extension UIView {
//在這里面把下面代碼粘貼進(jìn)來,然后在你需要的地方調(diào)用即可
}
主要代碼
/**
* @method 截取View一部分,產(chǎn)生新的圖片
*
* @param shotSize(要截取的區(qū)域)
*
* @return image(截取后的圖片)
*/
func screenShot(shotSize: CGSize) -> UIImage? {
//1.開啟上下文
UIGraphicsBeginImageContextWithOptions(shotSize, false, 0)
//2.繪制圖形,把layer上的內(nèi)容繪制到上下文當(dāng)中
self.layer.render(in: UIGraphicsGetCurrentContext()!)
let newImage = UIGraphicsGetImageFromCurrentImageContext()
//3.關(guān)閉上下文
UIGraphicsEndImageContext()
return newImage
}
