原文鏈接(iOS—添加文字水印)
最近有個寫水印需求,搜到了上面的文章,修復了一些小問題,改成swift實現(xiàn),如下:
import Foundation
import UIKit
fileprivate let HORIZONTAL_SPACEING: CGFloat = 30 //水平間距
fileprivate let VERTICAL_SPACEING: CGFloat = 50 //豎直間距
fileprivate let CG_TRANSFORM_ROTATING = CGFloat(-Double.pi / 6) //旋轉角度(正旋45度 || 反旋45度)
class WaterMarkView: UIView {
public private(set) lazy var waterMarkImgV: UIImageView = {
let imgV = UIImageView(frame: self.bounds)
imgV.image = UIImage(named: "")
return imgV
}()
private static let shared = WaterMarkView(frame: UIScreen.main.bounds)
/// 水印文字
public private(set) var markText: NSString = "默認水印文字"
override init(frame: CGRect) {
super.init(frame: frame)
self.addSubview(waterMarkImgV)
waterMarkImgV.frame = frame
}
required init?(coder: NSCoder) {
fatalError("init(coder:) has not been implemented")
}
/// 顯示水印
public static func show(view: UIView, text: String = "默認水印文字") {
let markText = text as NSString
let image = waterImage(size: view.bounds.size, text: markText)
let shared = WaterMarkView.shared
shared.markText = markText
shared.waterMarkImgV.image = image
view.addSubview(shared)
}
/// 隱藏水印
public static func hide() {
let shared = WaterMarkView.shared
shared.removeFromSuperview()
}
/// 水印圖片
private static func waterImage(size: CGSize, text: NSString, font: UIFont = UIFont.systemFont(ofSize: 17), textColor: UIColor = .black) -> UIImage? {
let imgW: CGFloat = size.width
let imgH: CGFloat = size.height
// 1.開啟上下文
UIGraphicsBeginImageContextWithOptions(size, false, 0)
// 2.繪制圖片
//屬性字符串
let attr: [NSAttributedString.Key: Any] = [.font: font, .foregroundColor: textColor]
let attrStr = NSMutableAttributedString(string: text as String, attributes: attr)
//文字寬高
let strW: CGFloat = attrStr.size().width
let strH: CGFloat = attrStr.size().height
//獲取當前上下文
guard let context = UIGraphicsGetCurrentContext() else { return nil }
//調(diào)整矩陣錨點到中心
context.concatenate(CGAffineTransform(translationX: imgW * 0.5, y: imgH * 0.5))
//旋轉矩陣
context.concatenate(CGAffineTransform(rotationAngle: CG_TRANSFORM_ROTATING))
//恢復錨點位置到左上角
context.concatenate(CGAffineTransform(translationX: -imgW * 0.5, y: -imgH * 0.5))
//對角線長度(實際繪制范圍是對角線長度的正方形)
let sqrtLength: CGFloat = sqrt(pow(imgW, 2) + pow(imgH, 2))
//繪制的行數(shù)和列數(shù)
let countHor = Int(sqrtLength / (strW + HORIZONTAL_SPACEING)) + 1
let countVer = Int(sqrtLength / (strH + VERTICAL_SPACEING)) + 1
//畫布起點
let orginX: CGFloat = -(sqrtLength - imgW) * 0.5
let orginY: CGFloat = -(sqrtLength - imgH) * 0.5
//雙循環(huán)實現(xiàn)
var i = 0
var j = 0
while i < countVer {
j = 0
let overlayOriginY = orginY + CGFloat(i) * (strH + VERTICAL_SPACEING)
while j < countHor {
let overlayOrginX = orginX + CGFloat(j) * (strW + HORIZONTAL_SPACEING)
//繪制單個文字水印
let drawRect = CGRect(x: overlayOrginX, y: overlayOriginY, width: strW, height: strH)
text.draw(in: drawRect, withAttributes: attr)
j += 1
}
i += 1
}
//3.上下文獲取新圖片
let newImage = UIGraphicsGetImageFromCurrentImageContext()
//4.關閉上下文
UIGraphicsEndImageContext()
context.restoreGState()
return newImage
}
override func hitTest(_ point: CGPoint, with event: UIEvent?) -> UIView? {
return nil
}
}