Watermarks- PDFImageDelegate Implementation

Source/API/Image/PDFImage.swift

@@ -94,3 +94,16 @@ public class PDFImage: PDFDocumentObject {
                  cornerRadius: self.cornerRadius)
     }
 }
+
+
+extension PDFImage: Hashable {
+    public func hash(into hasher: inout Hasher) {
+        hasher.combine(image)
+        hasher.combine(caption)
+        hasher.combine(NSCoder.string(for: size))
+        hasher.combine(sizeFit)
+        hasher.combine(quality)
+        hasher.combine(options.rawValue)
+        hasher.combine(cornerRadius)
+    }
+}

Source/API/PDFGenerator+Generation.swift

@@ -341,6 +341,9 @@ extension PDFGenerator {
      */
     internal func render(object: PDFRenderObject, in container: PDFContainer, in context: CGContext) throws {
         try object.draw(generator: self, container: container, in: context)
+        if let imageObject = object as? PDFImageObject {
+            imageDelegate?.generator(willBeginDrawingImage: imageObject.image, with: context, in: object.frame)
+        }
     }
 
     // MARK: - INTERNAL STATIC FUNCS

Source/API/PDFGenerator.swift

@@ -62,6 +62,11 @@ public class PDFGenerator: PDFGeneratorProtocol, CustomStringConvertible {
      Relative value tracking progress
      */
     public let progress = Progress.discreteProgress(totalUnitCount: 3)
+
+    /**
+     Delegate for image processing. Called after resizing and scaling and before drawing.
+     */
+    public var imageDelegate: PDFGeneratorImageDelegate?
 
     /**
      Font of each container.

Source/API/PDFGeneratorImageDelegate.swift

@@ -0,0 +1,14 @@
+//
+//  PDFGeneratorImageDelegate.swift
+//  TPPDF
+//
+//  Created by Chris Gonzales on 10/10/20.
+//
+
+import Foundation
+
+public protocol PDFGeneratorImageDelegate: class {
+    func generator(willBeginDrawingImage image: PDFImage, with context: CGContext, in frame: CGRect)
+}
+
+typealias PDFGeneratorDelegate = PDFGeneratorImageDelegate // for extensibility

Source/API/Text/PDFAttributedText.swift

@@ -38,4 +38,18 @@ public class PDFAttributedText: PDFText {
     override internal var copy: PDFText {
         PDFAttributedText(text: text)
     }
+
+    public override func hash(into hasher: inout Hasher) {
+        hasher.combine(text)
+    }
+
+    /**
+     TODO: Documentation
+     */
+    public static func == (lhs: PDFAttributedText, rhs: PDFAttributedText) -> Bool {
+        guard lhs.text == rhs.text else {
+            return false
+        }
+        return true
+    }
 }

Source/API/Text/PDFSimpleText.swift

@@ -52,4 +52,19 @@ public class PDFSimpleText: PDFText {
     override internal var copy: PDFText {
         PDFSimpleText(text: self.text, spacing: self.spacing, style: self.style)
     }
+
+    override public func hash(into hasher: inout Hasher) {
+        hasher.combine(text)
+        hasher.combine(spacing)
+    }
+
+    public static func == (lhs: PDFSimpleText, rhs: PDFSimpleText) -> Bool {
+        guard lhs.text == rhs.text else {
+            return false
+        }
+        guard lhs.spacing == rhs.spacing else {
+            return false
+        }
+        return true
+    }
 }

Source/API/Text/PDFText.swift

@@ -8,7 +8,7 @@
 /**
  Protocol all text objects should implement
  */
-public class PDFText: PDFDocumentObject, CustomStringConvertible {
+public class PDFText: PDFDocumentObject, CustomStringConvertible, Hashable {
 
     /**
      TODO: Documentation
@@ -35,4 +35,9 @@ public class PDFText: PDFDocumentObject, CustomStringConvertible {
         }
         return false
     }
+
+    public func hash(into hasher: inout Hasher) {
+        fatalError()
+    }
 }
+