Exit PDF Drawing Mode Automatically in Swift for iOS

1
//
2
//  Copyright © 2017-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 ExitDrawingModeAutomaticallyExample: Example, PDFViewControllerDelegate {
12

13
    override init() {
14
        super.init()
15

16
        title = "Exit drawing mode automatically"
17
        contentDescription = "Exit drawing mode automatically after a line has been drawn."
18
        category = .annotations
19
        priority = 201
20
    }
21

22
    private weak var pdfController: PDFViewController?
23

24
    override func invoke(with delegate: ExampleRunnerDelegate) -> UIViewController? {
25
        let document = AssetLoader.document(for: .welcome)
26
        let pdfController = PDFViewController(document: document)
27
        self.pdfController = pdfController
28

29
        // Example doesn’t have a cleanup callback. Avoid registering twice when opening and closing this example two times.
30
        NotificationCenter.default.removeObserver(self, name: .PSPDFAnnotationsAdded, object: nil)
31

32
        NotificationCenter.default.addObserver(self, selector: #selector(annotationsAdded), name: .PSPDFAnnotationsAdded, object: nil)
33

34
        return pdfController
35
    }
36

37
    @objc private func annotationsAdded(_ notification: Notification) {
38
        // Example doesn’t have a cleanup callback. The weak reference being nil shows if another example is generating this notification.
39
        guard let pdfController else {
40
            NotificationCenter.default.removeObserver(self, name: .PSPDFAnnotationsAdded, object: nil)
41
            return
42
        }
43

44
        // bail out if this is for another controller (e.g. split screen)
45
        // or if this inserts another annotation type.
46
        guard let annotation = (notification.object as? NSArray)?.firstObject as? InkAnnotation, annotation.document == pdfController.document else {
47
            return
48
        }
49

50
        // if we are in drawing mode and a new ink annotation is added, finish it.
51
        if pdfController.annotationStateManager.state == .ink {
52
            pdfController.annotationStateManager.state = nil
53
        }
54
    }
55
}