torch control (#117)

iOS/PolishBanknotesApp/PolishBanknotesApp.xcodeproj/project.xcworkspace/xcshareddata/IDEWorkspaceChecks.plist

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+	<key>IDEDidComputeMac32BitWarning</key>
+	<true/>
+</dict>
+</plist>

iOS/PolishBanknotesApp/PolishBanknotesApp.xcodeproj/xcshareddata/xcschemes/PolishBanknotesApp.xcscheme

@@ -0,0 +1,78 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<Scheme
+   LastUpgradeVersion = "1400"
+   version = "1.3">
+   <BuildAction
+      parallelizeBuildables = "YES"
+      buildImplicitDependencies = "YES">
+      <BuildActionEntries>
+         <BuildActionEntry
+            buildForTesting = "YES"
+            buildForRunning = "YES"
+            buildForProfiling = "YES"
+            buildForArchiving = "YES"
+            buildForAnalyzing = "YES">
+            <BuildableReference
+               BuildableIdentifier = "primary"
+               BlueprintIdentifier = "356516C0291C51F4006A460B"
+               BuildableName = "PolishBanknotesApp.app"
+               BlueprintName = "PolishBanknotesApp"
+               ReferencedContainer = "container:PolishBanknotesApp.xcodeproj">
+            </BuildableReference>
+         </BuildActionEntry>
+      </BuildActionEntries>
+   </BuildAction>
+   <TestAction
+      buildConfiguration = "Debug"
+      selectedDebuggerIdentifier = "Xcode.DebuggerFoundation.Debugger.LLDB"
+      selectedLauncherIdentifier = "Xcode.DebuggerFoundation.Launcher.LLDB"
+      shouldUseLaunchSchemeArgsEnv = "YES">
+      <Testables>
+      </Testables>
+   </TestAction>
+   <LaunchAction
+      buildConfiguration = "Debug"
+      selectedDebuggerIdentifier = "Xcode.DebuggerFoundation.Debugger.LLDB"
+      selectedLauncherIdentifier = "Xcode.DebuggerFoundation.Launcher.LLDB"
+      launchStyle = "0"
+      useCustomWorkingDirectory = "NO"
+      ignoresPersistentStateOnLaunch = "NO"
+      debugDocumentVersioning = "YES"
+      debugServiceExtension = "internal"
+      allowLocationSimulation = "YES">
+      <BuildableProductRunnable
+         runnableDebuggingMode = "0">
+         <BuildableReference
+            BuildableIdentifier = "primary"
+            BlueprintIdentifier = "356516C0291C51F4006A460B"
+            BuildableName = "PolishBanknotesApp.app"
+            BlueprintName = "PolishBanknotesApp"
+            ReferencedContainer = "container:PolishBanknotesApp.xcodeproj">
+         </BuildableReference>
+      </BuildableProductRunnable>
+   </LaunchAction>
+   <ProfileAction
+      buildConfiguration = "Release"
+      shouldUseLaunchSchemeArgsEnv = "YES"
+      savedToolIdentifier = ""
+      useCustomWorkingDirectory = "NO"
+      debugDocumentVersioning = "YES">
+      <BuildableProductRunnable
+         runnableDebuggingMode = "0">
+         <BuildableReference
+            BuildableIdentifier = "primary"
+            BlueprintIdentifier = "356516C0291C51F4006A460B"
+            BuildableName = "PolishBanknotesApp.app"
+            BlueprintName = "PolishBanknotesApp"
+            ReferencedContainer = "container:PolishBanknotesApp.xcodeproj">
+         </BuildableReference>
+      </BuildableProductRunnable>
+   </ProfileAction>
+   <AnalyzeAction
+      buildConfiguration = "Debug">
+   </AnalyzeAction>
+   <ArchiveAction
+      buildConfiguration = "Release"
+      revealArchiveInOrganizer = "YES">
+   </ArchiveAction>
+</Scheme>

iOS/PolishBanknotesApp/PolishBanknotesApp/Torch/TorchManager.swift

@@ -0,0 +1,44 @@
+//
+//  FlashlightManager.swift
+//  PolishBanknotesApp
+//
+//  Created by Miłosz Chojnacki on 20/06/2023.
+//
+
+import Foundation
+import AVFoundation
+import UIKit
+
+class TorchManager {
+    var isOn = false
+
+    func  activateTorch(interval: TimeInterval) {
+        isOn = true
+        DispatchQueue.global().async {
+            self.toggleTorch(on: true)
+            Thread.sleep(forTimeInterval: interval)
+            self.toggleTorch(on: false)
+        }
+        isOn = false
+    }
+
+    func toggleTorch(on: Bool) {
+        guard let device = AVCaptureDevice.default(for: AVMediaType.video) else { return }
+        guard device.hasTorch else { print("Torch isn't available"); return }
+
+        do {
+            try device.lockForConfiguration()
+            device.torchMode = on ? .on : .off
+            if on { try device.setTorchModeOn(level: AVCaptureDevice.maxAvailableTorchLevel) }
+            device.unlockForConfiguration()
+
+            if on {
+                UIAccessibility.post(notification: .announcement, argument: "Flash turned on!")
+            } else {
+                UIAccessibility.post(notification: .announcement, argument: "Flash turned off!")
+            }
+        } catch {
+            print("Torch can't be used")
+        }
+    }
+}

iOS/PolishBanknotesApp/PolishBanknotesApp/ViewModel/ContentViewModel.swift

@@ -9,6 +9,8 @@ import Foundation
 import CoreImage
 import Vision
 import Combine
+import AVFoundation
+import UIKit
 
 class ContentViewModel: ObservableObject {
     @Published var error: Error?
@@ -19,10 +21,13 @@ class ContentViewModel: ObservableObject {
 
     private let cameraManager = CameraManager.shared
     private let frameManager = FrameManager.shared
+    private let torchManager = TorchManager()
     private var frameCount = 0
     private let classifier = Classifier.shared
 
-    private var framePredictionSubscription: AnyCancellable?;
+    private var framePredictionSubscription: AnyCancellable?
+
+    // fps ~= 30, predictionRate = 2 => ~ 15fps (every 2nd frame)
     private let predictionRate = 5
     private let predictionSize = 5
 
@@ -41,7 +46,6 @@ class ContentViewModel: ObservableObject {
 
         framePredictionSubscription?.cancel()
 
-        // fps ~= 30, predictionRate = 2 => ~ 15fps (every 2nd frame)
         framePredictionSubscription = predictionSubscription()
     }
 
@@ -79,8 +83,20 @@ class ContentViewModel: ObservableObject {
             .assign(to: &$error)
 
         frameManager.$current
-            .receive(on: RunLoop.main)
             .compactMap { buffer in
+                if (buffer != nil) {
+                    let rawMetadata = CMCopyDictionaryOfAttachments(allocator: nil, target: buffer!, attachmentMode: CMAttachmentMode(kCMAttachmentMode_ShouldPropagate))
+                    let metadata = CFDictionaryCreateMutableCopy(nil, 0, rawMetadata) as NSMutableDictionary
+                    let directory = metadata.value(forKey: "MetadataDictionary") as? NSMutableDictionary
+                    let luxLevel = directory?.value(forKey: "LuxLevel") as? Int
+
+                    if luxLevel != 0 && luxLevel ?? 21 < 20 {
+                        if !self.torchManager.isOn {
+                            self.torchManager.activateTorch(interval: 30)
+                        }
+                    }
+                }
+
                 return CGImage.create(from: buffer)
             }
             .assign(to: &$frame)
@@ -93,4 +109,6 @@ class ContentViewModel: ObservableObject {
 
         self.framePredictionSubscription = self.predictionSubscription()
     }
+
+
 }