Add Annotation Buttons to PDF Toolbar in Swift for iOS

1
//
2
//  Copyright © 2021-2025 PSPDFKit GmbH. All rights reserved.
3
//
4
//  The Nutrient sample applications are licensed with a modified BSD license.
5
//  Please see License for details. This notice may not be removed from this file.
6
//
7

8
import PSPDFKit
9
import PSPDFKitUI
10

11
class AnnotationButtonInNavigationBarExample: Example {
12

13
    override init() {
14
        super.init()
15
        title = "Annotation Buttons in Navigation Bar"
16
        contentDescription = "Shows how to add annotation buttons to the navigation bar and update their state."
17
        category = .barButtons
18
        priority = 80
19
    }
20

21
    override func invoke(with delegate: ExampleRunnerDelegate) -> UIViewController? {
22
        CustomPDFViewController(document: AssetLoader.document(for: .annualReport))
23
    }
24

25
}
26

27
private class CustomPDFViewController: PDFViewController, AnnotationStateManagerDelegate {
28

29
    override init(document: Document?, configuration: PDFConfiguration?) {
30
        super.init(document: document, configuration: configuration)
31
        annotationStateManager.add(self)
32
    }
33

34
    required init?(coder: NSCoder) {
35
        fatalError("init(coder:) has not been implemented")
36
    }
37

38
    private lazy var inkBarButtonItem = UIBarButtonItem(
39
        image: PSPDFKit.SDK.imageNamed("ink"),
40
        style: .plain,
41
        target: self,
42
        action: #selector(inkBarButtonItemPressed)
43
    )
44

45
    private lazy var eraserBarButtonItem = UIBarButtonItem(
46
        image: PSPDFKit.SDK.imageNamed("eraser"),
47
        style: .plain,
48
        target: self,
49
        action: #selector(eraserBarButtonItemPressed)
50
    )
51

52
    override func viewDidLoad() {
53
        super.viewDidLoad()
54
        navigationItem.setRightBarButtonItems([annotationButtonItem, eraserBarButtonItem, inkBarButtonItem], for: .document, animated: false)
55
    }
56

57
    @objc private func inkBarButtonItemPressed(_ sender: UIBarButtonItem) {
58
        annotationStateManager.toggleState(.ink, variant: .inkPen)
59
    }
60

61
    @objc private func eraserBarButtonItemPressed(_ sender: UIBarButtonItem) {
62
        annotationStateManager.toggleState(.eraser)
63
    }
64

65
    func annotationStateManager(_ manager: AnnotationStateManager, didChangeState oldState: Annotation.Tool?, to newState: Annotation.Tool?, variant oldVariant: Annotation.Variant?, to newVariant: Annotation.Variant?) {
66
        // Set the buttons to "x" once the corresponding tool is activated.
67
        inkBarButtonItem.image = (newState == .ink && newVariant == .inkPen) ? PSPDFKit.SDK.imageNamed("close") : PSPDFKit.SDK.imageNamed("ink")
68
        eraserBarButtonItem.image = (newState == .eraser) ? PSPDFKit.SDK.imageNamed("close") : PSPDFKit.SDK.imageNamed("eraser")
69
    }
70

71
}