Output allowed

PhotographyKit.podspec

@@ -1,7 +1,7 @@
 Pod::Spec.new do |spec|
 
   spec.name         = "PhotographyKit"
-  spec.version      = "0.24"
+  spec.version      = "0.25"
   spec.license      = "MIT"
   spec.summary      = "A swift library for quickly integrating a built in camera session into your app."
   spec.homepage     = "https://github.com/appoly/PhotographyKit"

Sources/PhotographyKit/PhotographyKit.swift

@@ -23,6 +23,7 @@ public enum PhotographyKitError: Error {
     case failedToConnectToDeviceTorch
     case failedToConnectToDeviceMicrophone
     case failedToCreateVideoFile
+    case failedToAddOutput
 }
 
 
@@ -44,6 +45,8 @@ extension PhotographyKitError: LocalizedError {
                 return "Failed to connect to device microphone"
             case .failedToCreateVideoFile:
                 return "Failed to create video file for recording"
+            case .failedToAddOutput:
+                return "Failed to add output"
         }
     }
 }
@@ -67,7 +70,7 @@ public class PhotographyKit: NSObject {
 
     //User defined photo settings
     private let zoomSensitivity: CGFloat = 0.03
-    private let delegate: PhotographyKitDelegate
+    private let delegate: PhotographyKitDelegate?
     private let allowsVideo: Bool
 
     //Capture session
@@ -87,6 +90,11 @@ public class PhotographyKit: NSObject {
 
     // MARK: - Actions
 
+    /// Stop Capture session
+    public func stop() {
+        captureSession?.stopRunning()
+    }
+
     /// Tries to resets the capture session, starting the preview again.
     public func resetCamera() throws {
         photo = nil
@@ -116,6 +124,16 @@ public class PhotographyKit: NSObject {
     }
 
 
+    ///Adds output to capture session
+    public func addOutput(output: AVCaptureOutput) throws {
+        if(captureSession?.canAddOutput(output) ?? false) {
+            captureSession?.addOutput(output)
+        } else {
+            throw PhotographyKitError.failedToAddOutput
+        }
+    }
+
+
     public func endVideoRecording() {
         guard allowsVideo else { return }
         movieFileOutput.stopRecording()
@@ -226,7 +244,7 @@ public class PhotographyKit: NSObject {
 
     // MARK: - Initializers
 
-    public init?(view: UIView, delegate: PhotographyKitDelegate, allowsVideo: Bool) throws {
+    public init?(view: UIView, delegate: PhotographyKitDelegate?, allowsVideo: Bool) throws {
         self.allowsVideo = allowsVideo
         self.delegate = delegate
         super.init()
@@ -399,7 +417,7 @@ extension PhotographyKit: AVCapturePhotoCaptureDelegate {
         guard let image = getImageFrom(photo) else { return }
 
         let photo = PhotographyKitPhoto(image: image)
-        delegate.didCaptureImage(image: photo)
+        delegate?.didCaptureImage(image: photo)
     }
 
 }
@@ -409,16 +427,16 @@ extension PhotographyKit: AVCapturePhotoCaptureDelegate {
 extension PhotographyKit: AVCaptureFileOutputRecordingDelegate {
 
     public func fileOutput(_ output: AVCaptureFileOutput, didStartRecordingTo fileURL: URL, from connections: [AVCaptureConnection]) {
-        delegate.didStartRecordingVideo()
+        delegate?.didStartRecordingVideo()
     }
 
 
     public func fileOutput(_ output: AVCaptureFileOutput, didFinishRecordingTo outputFileURL: URL, from connections: [AVCaptureConnection], error: Error?) {
         guard error == nil else {
-            delegate.didFailRecordingVideo(error: error!)
+            delegate?.didFailRecordingVideo(error: error!)
             return
         }
-        delegate.didFinishRecordingVideo(url: outputFileURL)
+        delegate?.didFinishRecordingVideo(url: outputFileURL)
     }
 
 }