Ability to not load the thumbnails strip when the controller is created

Alua-Kinzhebayeva · Feb 14, 2017 · 7bf8f64 · 7bf8f64
1 parent 7cf9a70
commit 7bf8f64
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/PDFReader/Demo/StartViewController.swift b/PDFReader/Demo/StartViewController.swift
@@ -55,6 +55,8 @@ internal final class StartViewController: UIViewController {
     /// Presents a document
     ///
     /// - parameter document: document to present
+    ///
+    /// Add `thumbnailsEnabled:false` to `createNew` to not load the thumbnails in the controller.
     fileprivate func showDocument(_ document: PDFDocument) {
         let image = UIImage(named: "")
         let controller = PDFViewController.createNew(with: document, title: "", actionButtonImage: image, actionStyle: .activitySheet)

diff --git a/Sources/Classes/PDFViewController.swift b/Sources/Classes/PDFViewController.swift
@@ -17,7 +17,7 @@ extension PDFViewController {
     /// - parameter backButton:        button to override the default controller back button
     ///
     /// - returns: a `PDFViewController`
-    public class func createNew(with document: PDFDocument, title: String? = nil, actionButtonImage: UIImage? = nil, actionStyle: ActionStyle = .print, backButton: UIBarButtonItem? = nil) -> PDFViewController {
+    public class func createNew(with document: PDFDocument, title: String? = nil, actionButtonImage: UIImage? = nil, actionStyle: ActionStyle = .print, backButton: UIBarButtonItem? = nil, thumbnailsEnabled: Bool? = true) -> PDFViewController {
         let storyboard = UIStoryboard(name: "PDFReader", bundle: Bundle(for: PDFViewController.self))
         let controller = storyboard.instantiateInitialViewController() as! PDFViewController
         controller.document = document
@@ -30,6 +30,7 @@ extension PDFViewController {
         }
 
         controller.backButton = backButton
+        controller.thumbnailsEnabled = thumbnailsEnabled!
 
         if let actionButtonImage = actionButtonImage {
             controller.actionButton = UIBarButtonItem(image: actionButtonImage, style: .plain, target: controller, action: #selector(actionButtonPressed))