Swift UIImage添加倾斜水印

首先上效果：

Simulator Screen Shot - iPhone 11 - 2020-07-31 at 14.02.09.png

代码

// 添加水印
extension UIImage {
    func addWaterText(viewSize: CGSize, text: String, color: UIColor = .lightGray, font: UIFont = .systemFont(ofSize: 13)) -> UIImage? {
        let scale = self.size.width/viewSize.width
        // 计算text渲染大小
        let sizeFont = UIFont.systemFont(ofSize: scale * font.pointSize)
        
        // 计算text size width:.greatestFiniteMagnitude height:.greatestFiniteMagnitude
        let textSize = text.hs_sizeWithConstrainedWidth(.greatestFiniteMagnitude, font: sizeFont)
        //原始image的宽高
        let viewWidth = self.size.width
        let viewHeight = self.size.height
        
        //进行尺寸重绘 为了防止图片失真，绘制区域宽高和原始图片宽高一样
        UIGraphicsBeginImageContext(self.size)
        self.draw(in: CGRect.init(x: 0, y: 0, width: viewWidth, height: viewHeight))
        //sqrtLength：原始image的对角线length。在水印旋转矩阵中只要矩阵的宽高是原始image的对角线长度，无论旋转多少度都不会有空白。
        let sqrtLength = CGFloat(sqrt(Double(viewWidth*viewWidth + viewHeight*viewHeight)))
        let attr = [NSAttributedString.Key.font: sizeFont, NSAttributedString.Key.foregroundColor: color]
        let attrStr = NSAttributedString.init(string: text, attributes: attr)
        //绘制文字的宽高
        let strWidth = textSize.width
        let strHeight = textSize.height
        
        //开始旋转上下文矩阵，绘制水印文字
        let context = UIGraphicsGetCurrentContext()
        context?.translateBy(x: CGFloat(viewWidth/2), y: CGFloat(viewHeight/2))
        context?.rotate(by: CGFloat(Double.pi / 16))
        context?.translateBy(x: CGFloat(-viewWidth/2), y: CGFloat(-viewHeight/2))
        //计算需要绘制的列数和行数
        let VSpace:CGFloat = 50 * scale
        let HSpace:CGFloat = 30 * scale
        let horCount = Int(sqrtLength / (strWidth + HSpace) + 1)
        let verCount = Int(sqrtLength / (strHeight + VSpace) + 1)
        
        //此处计算出需要绘制水印文字的起始点，由于水印区域要大于图片区域所以起点在原有基础上移
        let orignX = -(sqrtLength-viewWidth)/2
        let orignY = -(sqrtLength-viewHeight)/2
        //在每列绘制时X坐标叠加
        var tempOrignX = orignX
        //在每行绘制时Y坐标叠加
        var tempOrignY = orignY
        
        for i in 0...(horCount * verCount) {
            attrStr.draw(in: CGRect.init(x: tempOrignX, y: tempOrignY, width: strWidth, height: strHeight))
            if i % horCount == 0 && i != 0 {
                tempOrignX = orignX
                tempOrignY += strHeight + VSpace
            } else {
                tempOrignX += strWidth + HSpace
            }
        }
        let newImage = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
        return newImage
    }   
}

如何使用

        var image : UIImage = info[UIImagePickerController.InfoKey.editedImage] as! UIImage
        image = image.addWaterText(viewSize: self.imageView.size, text: "仅供XXXX试用")!
        self.imageView.image = image