Customize PDF annotation toolbar in Swift for iOS

1
//
2
//  Copyright © 2019-2026 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 CustomizeAnnotationToolbarExample: Example {
12

13
    override init() {
14
        super.init()
15
        title = "Customized Annotation Toolbar"
16
        contentDescription = "Customizes the buttons in the annotation toolbar."
17
        category = .barButtons
18
        priority = 200
19
    }
20

21
    override func invoke(with delegate: ExampleRunnerDelegate) -> UIViewController? {
22
        let document = AssetLoader.document(for: .annualReport)
23
        let controller = PDFViewController(document: document) {
24
            $0.overrideClass(AnnotationToolbar.self, with: CustomizedAnnotationToolbar.self)
25
        }
26
        return controller
27
    }
28
}
29

30
class CustomizedAnnotationToolbar: AnnotationToolbar {
31
    override init(annotationStateManager: AnnotationStateManager) {
32
        super.init(annotationStateManager: annotationStateManager)
33

34
        typealias Item = AnnotationToolConfiguration.ToolItem
35
        typealias Group = AnnotationToolConfiguration.ToolGroup
36
        let highlight = Item(type: .highlight)
37
        let underline = Item(type: .underline)
38
        let freeText = Item(type: .freeText)
39
        let note = Item(type: .note)
40

41
        let square = Item(type: .square)
42
        let circle = Item(type: .circle)
43
        let line = Item(type: .line)
44

45
        let compactGroups = [
46
            Group(items: [highlight, underline, freeText, note]),
47
            Group(items: [square, circle, line])
48
        ]
49
        let compactConfiguration = AnnotationToolConfiguration(annotationGroups: compactGroups)
50

51
        let regularGroups = [
52
            Group(items: [highlight, underline]),
53
            Group(items: [freeText]),
54
            Group(items: [note]),
55
            Group(items: [square, circle, line])
56
        ]
57
        let regularConfiguration = AnnotationToolConfiguration(annotationGroups: regularGroups)
58

59
        configurations = [compactConfiguration, regularConfiguration]
60
    }
61
}