|
|
@@ -40,7 +40,7 @@ enum KMWatermarkKey: String {
|
|
|
case lastAddData = "kAddLastWatermarkDataKey1"
|
|
|
case id = "kWatermarkIDKey"
|
|
|
case tag = "kWatermarkTagKey"
|
|
|
-
|
|
|
+ case watermarkName = "watermarkNameKey"
|
|
|
|
|
|
}
|
|
|
|
|
|
@@ -63,6 +63,9 @@ class KMWatermarkManager: NSObject {
|
|
|
let watermarkDict: NSDictionary = dataDict?.object(forKey: key) as! NSDictionary
|
|
|
|
|
|
let model = parseDictionary(dict: watermarkDict)
|
|
|
+ let image = self.drawImageAtpageRect(rect: CGRectMake(0, 0, 136, 181), data: model)
|
|
|
+ model.resultImage = image
|
|
|
+
|
|
|
self.watermarks.append(model)
|
|
|
}
|
|
|
self.watermarks.sort(){$0.tag > $1.tag}
|
|
|
@@ -296,6 +299,206 @@ class KMWatermarkManager: NSObject {
|
|
|
return result
|
|
|
}
|
|
|
|
|
|
+ func drawImageAtpageRect(rect: NSRect, data: KMPDFWatermarkData) -> NSImage? {
|
|
|
+ var size = NSZeroSize
|
|
|
+ let text: String = data.text ?? ""
|
|
|
+ if data.watermarkType == .image {
|
|
|
+ if let image = data.image() {
|
|
|
+ size = image.size
|
|
|
+ size.width *= data.scale
|
|
|
+ size.height *= data.scale
|
|
|
+ }
|
|
|
+ } else if data.watermarkType == .text {
|
|
|
+ var font = NSFont.systemFont(ofSize: 52)
|
|
|
+ if data.isTilePage {
|
|
|
+ font = NSFont.systemFont(ofSize: data.fontSize)
|
|
|
+ }
|
|
|
+ let style = NSMutableParagraphStyle()
|
|
|
+ style.alignment = .center
|
|
|
+ style.lineBreakMode = .byCharWrapping
|
|
|
+ let attributes: [NSAttributedString.Key: Any] = [ .paragraphStyle: style, .font: font ]
|
|
|
+ size = text.boundingRect(with: CGSize(width: CGFloat.greatestFiniteMagnitude, height: CGFloat.greatestFiniteMagnitude), options: [.usesLineFragmentOrigin, .usesFontLeading], attributes: attributes).size
|
|
|
+ }
|
|
|
+
|
|
|
+ let radian = data.rotation * (CGFloat.pi/180)
|
|
|
+ let t = CGAffineTransform(rotationAngle: radian)
|
|
|
+ var newRect = rect
|
|
|
+ if !data.isTilePage {
|
|
|
+ newRect = CGRect(x: 0, y: 0, width: size.width + 40, height: size.height + 40).applying(t)
|
|
|
+ }
|
|
|
+ let image = NSImage(size: newRect.size)
|
|
|
+ image.lockFocus()
|
|
|
+ NSGraphicsContext.current?.imageInterpolation = .high
|
|
|
+ NSGraphicsContext.saveGraphicsState()
|
|
|
+ NSColor.clear.set()
|
|
|
+ rect.fill()
|
|
|
+ NSGraphicsContext.restoreGraphicsState()
|
|
|
+ guard let context = NSGraphicsContext.current?.cgContext else { return nil }
|
|
|
+ let imageSize = newRect.size
|
|
|
+ NSGraphicsContext.current = NSGraphicsContext(cgContext: context, flipped: false)
|
|
|
+ NSGraphicsContext.saveGraphicsState()
|
|
|
+
|
|
|
+ if data.watermarkType == .image {
|
|
|
+ guard let image = data.image() else { return nil }
|
|
|
+ var size = image.size
|
|
|
+ size.width *= data.scale
|
|
|
+ size.height *= data.scale
|
|
|
+ let width = sqrt(image.size.height * image.size.height + image.size.width * image.size.width)
|
|
|
+ let newRect = CGRect(x: -(width - image.size.width)/2, y: -(width - image.size.height)/2, width: width, height: width)
|
|
|
+ let new_w = newRect.size.width
|
|
|
+ let new_h = newRect.size.height
|
|
|
+ let radian = data.rotation * (CGFloat.pi / 180)
|
|
|
+ let t = CGAffineTransform(rotationAngle: radian)
|
|
|
+ if data.isTilePage {
|
|
|
+ context.translateBy(x: image.size.width/2, y: image.size.height/2)
|
|
|
+ context.concatenate(t)
|
|
|
+ context.translateBy(x: -(image.size.width/2), y: -(image.size.height/2))
|
|
|
+
|
|
|
+ let verticalWidth = size.width + data.horizontalSpacing * data.scale
|
|
|
+ let horizontalHeight = size.height + data.verticalSpacing * data.scale
|
|
|
+ let line: Int = Int((new_h - data.verticalSpacing * data.scale)/horizontalHeight + 1)
|
|
|
+ let row: Int = Int((new_w - data.horizontalSpacing * data.scale) / verticalWidth + 1)
|
|
|
+ let point = CGPoint(x: image.size.width/2 - size.width/2 + data.tx*data.scale, y:image.size.height/2 - size.height/2 + data.ty * data.scale)
|
|
|
+ for i in 0..<line/2+1 {
|
|
|
+ for j in 0..<row {
|
|
|
+ let area = CGRect(x: point.x + CGFloat(j) * verticalWidth, y: point.y + CGFloat(i) * horizontalHeight, width: size.width, height: size.height)
|
|
|
+ image.draw(in: area, from: NSZeroRect, operation: .sourceOver, fraction: data.opacity)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ for i in 1..<line/2+1 {
|
|
|
+ for j in 0..<row {
|
|
|
+ let area = CGRect(x: point.x + CGFloat(j) * verticalWidth, y: point.y - CGFloat(i) * horizontalHeight, width: size.width, height: size.height)
|
|
|
+ image.draw(in: area, from: NSZeroRect, operation: .sourceOver, fraction: data.opacity)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ for i in 0..<line/2+1 {
|
|
|
+ for j in 1..<row {
|
|
|
+ let area = CGRect(x: point.x - CGFloat(j) * verticalWidth, y: point.y + CGFloat(i) * horizontalHeight, width: size.width, height: size.height)
|
|
|
+ image.draw(in: area, from: NSZeroRect, operation: .sourceOver, fraction: data.opacity)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ for i in 1..<(line/2 + 1) {
|
|
|
+ for j in 1..<row {
|
|
|
+ let area = CGRect(x: point.x - CGFloat(j)*verticalWidth, y: point.y - CGFloat(i)*horizontalHeight, width: size.width, height: size.height)
|
|
|
+ image.draw(in: area, from: NSZeroRect, operation: .sourceOver, fraction: data.opacity)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ var size = image.size
|
|
|
+ size.width *= data.scale
|
|
|
+ size.height *= data.scale
|
|
|
+ let radian = data.rotation * (CGFloat.pi / 180)
|
|
|
+ let t = CGAffineTransform(rotationAngle: radian)
|
|
|
+ var imageRect = CGRect(origin: .zero, size: size).applying(t)
|
|
|
+ if data.verticalPosition == .bottom {
|
|
|
+ imageRect.origin.y = imageSize.height - imageRect.size.height
|
|
|
+ } else if data.verticalPosition == .center {
|
|
|
+ imageRect.origin.y = (imageSize.height - imageRect.size.height) / 2.0
|
|
|
+ } else {
|
|
|
+ imageRect.origin.y = 0
|
|
|
+ }
|
|
|
+ if data.horizontalPosition == .left {
|
|
|
+ imageRect.origin.x = 0
|
|
|
+ } else if data.horizontalPosition == .center {
|
|
|
+ imageRect.origin.x = (imageSize.width - imageRect.size.width) / 2.0
|
|
|
+ } else {
|
|
|
+ imageRect.origin.x = imageSize.width - imageRect.size.width
|
|
|
+ }
|
|
|
+ let contextCenter = CGPoint(x: imageRect.midX, y: imageRect.midY)
|
|
|
+ context.translateBy(x: contextCenter.x, y: contextCenter.y)
|
|
|
+ context.rotate(by: radian)
|
|
|
+ context.translateBy(x: -contextCenter.x, y: -contextCenter.y)
|
|
|
+ image.draw(in: CGRect(x: imageRect.origin.x+(imageRect.size.width-size.width)/2.0, y: imageRect.origin.y+(imageRect.size.height-size.height)/2.0, width: size.width, height: size.height), from: NSZeroRect, operation: .sourceOver, fraction: data.opacity)
|
|
|
+ }
|
|
|
+ }else if text.count > 0 {
|
|
|
+ var font = NSFont.systemFont(ofSize: 52)
|
|
|
+ if data.isTilePage {
|
|
|
+ font = NSFont.systemFont(ofSize: data.fontSize)
|
|
|
+ }
|
|
|
+ var color = data.textColor
|
|
|
+ var red: CGFloat = 0, green: CGFloat = 0, blue: CGFloat = 0
|
|
|
+ color.usingColorSpaceName(.calibratedRGB)?.getRed(&red, green: &green, blue: &blue, alpha: nil)
|
|
|
+ color = NSColor.init(calibratedRed: red, green: green, blue: blue, alpha: data.opacity)
|
|
|
+
|
|
|
+ size = .zero
|
|
|
+
|
|
|
+ let alpha = data.opacity
|
|
|
+ let tileHorizontalSpace = data.horizontalSpacing
|
|
|
+ let tileVerticalSpace = data.verticalSpacing
|
|
|
+ let horizontalSpace = data.tx
|
|
|
+ let verticalSpace = data.ty
|
|
|
+ let style = NSMutableParagraphStyle()
|
|
|
+ style.alignment = .center
|
|
|
+ style.lineBreakMode = .byCharWrapping
|
|
|
+ let attributes: [NSAttributedString.Key: Any] = [ .paragraphStyle: style, .foregroundColor: NSColor(calibratedRed: red, green: green, blue: blue, alpha: alpha), .font: font ]
|
|
|
+ size = text.boundingRect(with: CGSize(width: CGFloat.greatestFiniteMagnitude, height: CGFloat.greatestFiniteMagnitude), options: [.usesLineFragmentOrigin, .usesFontLeading], attributes: attributes).size
|
|
|
+ let textRect = CGRect(x: 0, y: 0, width: size.width, height: size.height)
|
|
|
+ if data.isTilePage {
|
|
|
+ let width = sqrt(image.size.height * image.size.height + image.size.width * image.size.width)
|
|
|
+ let newRect = CGRect(x: -(width - image.size.width) / 2, y: -(width - image.size.height) / 2, width: width, height: width)
|
|
|
+ let new_w = newRect.size.width
|
|
|
+ let new_h = newRect.size.height
|
|
|
+ context.translateBy(x: image.size.width / 2, y: image.size.height / 2)
|
|
|
+ context.concatenate(CGAffineTransform(rotationAngle: radian))
|
|
|
+ context.translateBy(x: -(image.size.width / 2), y: -(image.size.height / 2))
|
|
|
+ let verticalWidth = size.width + tileHorizontalSpace / 3.0
|
|
|
+ let horizontalHeight = size.height + tileVerticalSpace / 3.0
|
|
|
+ let line: Int = Int((new_h - tileHorizontalSpace / 3.0) / horizontalHeight + 1)
|
|
|
+ let row: Int = Int((new_w - tileVerticalSpace / 3.0) / verticalWidth + 1)
|
|
|
+ let point = CGPoint(x: image.size.width / 2 - size.width / 2 + horizontalSpace / 3.0, y: image.size.height / 2 - size.height / 2 + verticalSpace / 3.0)
|
|
|
+ for i in 0..<line {
|
|
|
+ for j in 0..<row {
|
|
|
+ let drawRect = CGRect(x: point.x + CGFloat(j) * verticalWidth, y: point.y + CGFloat(i) * horizontalHeight, width: size.width, height: size.height)
|
|
|
+ text.draw(in: drawRect, withAttributes: attributes)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ for i in 1..<line {
|
|
|
+ for j in 0..<row {
|
|
|
+ let drawRect = CGRect(x: point.x + CGFloat(j) * verticalWidth, y: point.y - CGFloat(i) * horizontalHeight, width: size.width, height: size.height)
|
|
|
+ text.draw(in: drawRect, withAttributes: attributes)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ for i in 0..<line {
|
|
|
+ for j in 1..<row {
|
|
|
+ let drawRect = CGRect(x: point.x - CGFloat(j) * verticalWidth, y: point.y + CGFloat(i) * horizontalHeight, width: size.width, height: size.height)
|
|
|
+ text.draw(in: drawRect, withAttributes: attributes)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ for i in 1..<line {
|
|
|
+ for j in 1..<row {
|
|
|
+ let drawRect = CGRect(x: point.x - CGFloat(j) * verticalWidth, y: point.y - CGFloat(i) * horizontalHeight, width: size.width, height: size.height)
|
|
|
+ text.draw(in: drawRect, withAttributes: attributes)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ var textRect = textRect
|
|
|
+ if data.verticalPosition == .bottom {
|
|
|
+ textRect.origin.y = imageSize.height - textRect.size.height
|
|
|
+ } else if data.verticalPosition == .center {
|
|
|
+ textRect.origin.y = (imageSize.height - textRect.size.height) / 2.0
|
|
|
+ } else {
|
|
|
+ textRect.origin.y = 0
|
|
|
+ }
|
|
|
+ if data.horizontalPosition == .left {
|
|
|
+ textRect.origin.x = 0
|
|
|
+ } else if data.horizontalPosition == .center {
|
|
|
+ textRect.origin.x = (imageSize.width - textRect.size.width) / 2.0
|
|
|
+ } else {
|
|
|
+ textRect.origin.x = imageSize.width - textRect.size.width
|
|
|
+ }
|
|
|
+ let contextCenter = CGPoint(x: textRect.midX, y: textRect.midY)
|
|
|
+ context.translateBy(x: contextCenter.x, y: contextCenter.y)
|
|
|
+ context.rotate(by: radian)
|
|
|
+ context.translateBy(x: -contextCenter.x, y: -contextCenter.y)
|
|
|
+ (data.text ?? "").draw(in: CGRect(x: textRect.origin.x + (textRect.size.width - size.width) / 2.0, y: textRect.origin.y + (textRect.size.height - size.height) / 2.0, width: size.width, height: size.height), withAttributes: attributes)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ NSGraphicsContext.restoreGraphicsState()
|
|
|
+ NSGraphicsContext.current?.imageInterpolation = .default
|
|
|
+ image.unlockFocus()
|
|
|
+ return image
|
|
|
+ }
|
|
|
+
|
|
|
//MARK: - private
|
|
|
private func parseWaterMark(model: KMPDFWatermarkData) -> Dictionary<String, Any> {
|
|
|
|
|
|
@@ -334,6 +537,7 @@ class KMWatermarkManager: NSObject {
|
|
|
dict[KMWatermarkKey.pageRangeType.rawValue] = model.pageRangeType.rawValue
|
|
|
dict[KMWatermarkKey.pageRangeString.rawValue] = model.pageRangeString
|
|
|
dict[KMWatermarkKey.tag.rawValue] = model.tag
|
|
|
+ dict[KMWatermarkKey.watermarkName.rawValue] = model.watermarkName
|
|
|
|
|
|
return dict
|
|
|
}
|
|
|
@@ -354,8 +558,7 @@ class KMWatermarkManager: NSObject {
|
|
|
model.imagePath = path ?? ""
|
|
|
}
|
|
|
}
|
|
|
-
|
|
|
-
|
|
|
+
|
|
|
model.isScale = (dict.object(forKey: KMWatermarkKey.isScale.rawValue) as? Bool) ?? false
|
|
|
model.scale = (dict.object(forKey: KMWatermarkKey.scale.rawValue) as? CGFloat) ?? 1
|
|
|
model.rotation = dict.object(forKey: KMWatermarkKey.rotate.rawValue) as? CGFloat ?? 0
|
|
|
@@ -379,6 +582,7 @@ class KMWatermarkManager: NSObject {
|
|
|
model.horizontalSpacing = (dict.object(forKey: KMWatermarkKey.tileHorizontalSpace.rawValue) as? CGFloat) ?? 0
|
|
|
model.verticalSpacing = (dict.object(forKey: KMWatermarkKey.tileVerticalSpace.rawValue) as? CGFloat) ?? 0
|
|
|
model.tag = (dict.object(forKey: KMWatermarkKey.tag.rawValue) as? String) ?? ""
|
|
|
+ model.watermarkName = (dict.object(forKey: KMWatermarkKey.watermarkName.rawValue) as? String) ?? ""
|
|
|
|
|
|
return model
|
|
|
}
|
