Added maximum number of scans property in DocScanner.swift

Signed-off-by: hjafri256 <[email protected]>

Author: HJafri256

DocumentScanner/Source/DocScanner.swift

@@ -37,6 +37,9 @@ import UIKit
 
     /// If true can scan and export multiple documents at onse, defaults to false
     public var exportMultiple: Bool = false
+    
+    ///Maximum number of scans. -1 for unlimited
+    public var maximumScans = -1
 
     /// Construct scanner object
     public init(presenter: UIViewController) {
@@ -132,8 +135,16 @@ import UIKit
 
             // perform perspective correction
             if let flattened = photo.flattened(rect: self.camera.observationRect) {
+                
                 self.scannedImages.append(flattened.noiseReducted.rotated)
-
+                
+                if self.maximumScans != -1
+                    && self.scannedImages.count >= self.maximumScans  {
+                    self.exportMultiple ? self.exportImages?(self.scannedImages) : self.exportImage?(self.scannedImages[0])
+                    self.scannedImages.removeAll()
+                    self.stopSession()
+                }
+                
                 return
             }
 
@@ -146,6 +157,7 @@ import UIKit
             // return from manual cropping
             cropView.onRetake = {
                 self.continueSession()
+                self.stopSession()
             }
 
             cropView.onRegionSave = {
@@ -157,6 +169,13 @@ import UIKit
                 if let flattened = photo.flattened(rect: region) {
                     self.scannedImages.append(flattened.noiseReducted.rotated)
 
+                    if self.maximumScans != -1
+                        && self.scannedImages.count >= self.maximumScans  {
+                        self.exportMultiple ? self.exportImages?(self.scannedImages) : self.exportImage?(self.scannedImages[0])
+                        self.scannedImages.removeAll()
+                        self.stopSession()
+                    }
+                    
                     return
                 }
 
@@ -186,6 +205,13 @@ import UIKit
 
         let croppedImage = photo.crop(toPreviewLayer: camera.cameraLayer, withRect: regionRect)
         scannedImages.append(croppedImage.noiseReducted.rotated)
+        
+        if self.maximumScans != -1
+            && self.scannedImages.count >= self.maximumScans  {
+            self.exportMultiple ? self.exportImages?(self.scannedImages) : self.exportImage?(self.scannedImages[0])
+            self.scannedImages.removeAll()
+            self.stopSession()
+        }
     }
 
     private func toggleIdle(disabled: Bool) {

Example/Podfile.lock

@@ -13,4 +13,4 @@ SPEC CHECKSUMS:
 
 PODFILE CHECKSUM: 7cfc6a5ca9cae37b2543fa20083bbc420031f30d
 
-COCOAPODS: 1.9.0
+COCOAPODS: 1.15.2

Sources/DocumentScanner/DocScanner.swift

@@ -37,6 +37,9 @@ import UIKit
 
     /// If true can scan and export multiple documents at onse, defaults to false
     public var exportMultiple: Bool = false
+    
+    ///Maximum number of scans. -1 for unlimited
+    public var maximumScans = -1
 
     /// Construct scanner object
     public init(presenter: UIViewController) {
@@ -132,8 +135,16 @@ import UIKit
 
             // perform perspective correction
             if let flattened = photo.flattened(rect: self.camera.observationRect) {
+                
                 self.scannedImages.append(flattened.noiseReducted.rotated)
-
+                
+                if self.maximumScans != -1
+                    && self.scannedImages.count >= self.maximumScans  {
+                    self.exportMultiple ? self.exportImages?(self.scannedImages) : self.exportImage?(self.scannedImages[0])
+                    self.scannedImages.removeAll()
+                    self.stopSession()
+                }
+                
                 return
             }
 
@@ -146,6 +157,7 @@ import UIKit
             // return from manual cropping
             cropView.onRetake = {
                 self.continueSession()
+                self.stopSession()
             }
 
             cropView.onRegionSave = {
@@ -157,6 +169,13 @@ import UIKit
                 if let flattened = photo.flattened(rect: region) {
                     self.scannedImages.append(flattened.noiseReducted.rotated)
 
+                    if self.maximumScans != -1
+                        && self.scannedImages.count >= self.maximumScans  {
+                        self.exportMultiple ? self.exportImages?(self.scannedImages) : self.exportImage?(self.scannedImages[0])
+                        self.scannedImages.removeAll()
+                        self.stopSession()
+                    }
+                    
                     return
                 }
 
@@ -186,6 +205,13 @@ import UIKit
 
         let croppedImage = photo.crop(toPreviewLayer: camera.cameraLayer, withRect: regionRect)
         scannedImages.append(croppedImage.noiseReducted.rotated)
+        
+        if self.maximumScans != -1
+            && self.scannedImages.count >= self.maximumScans  {
+            self.exportMultiple ? self.exportImages?(self.scannedImages) : self.exportImage?(self.scannedImages[0])
+            self.scannedImages.removeAll()
+            self.stopSession()
+        }
     }
 
     private func toggleIdle(disabled: Bool) {