UIView+extension.swift

extension UIView {
	public func snapshotImage(bgColor: UIColor? = nil, rect: CGRect? = nil, state: Bool = false) -> UIImage? {
		var definedRect = CGRect.zero
		if let rect = rect {
			definedRect = rect
		} else {
			definedRect = self.bounds
		}
		UIGraphicsBeginImageContextWithOptions(definedRect.size, false, UIScreen.main.scale)
		guard let graphicContext = UIGraphicsGetCurrentContext() else {
			return nil
		}
		if bgColor != nil {
            bgColor!.set()
        }
		graphicContext.fill(definedRect)
		drawHierarchy(in: definedRect, afterScreenUpdates: state)
		self.layer.isOpaque = false
		self.layer.render(in: graphicContext)
		let snapshotImage = UIGraphicsGetImageFromCurrentImageContext()
		UIGraphicsEndImageContext()
		
		return snapshotImage
	}
}