iOS Swift图像分类器APP相机闪光灯切换功能实现问询

嘿，我看了你这段代码，思路完全没问题——用开关控制闪光灯确实能帮用户在低光环境下更好地使用图像分类器，但当前的实现有几个小漏洞，会导致闪光灯切换失效甚至搞乱相机会话。让我帮你调整得更顺畅：

现有代码的核心问题 #

• 每次触发开关时你都新建了captureSession，但没正确替换原有会话，还可能导致旧会话残留浪费资源
• 你试图通过重启相机会话来切换闪光灯，其实完全没必要，动态操作已有设备就能实现，重启还会造成界面卡顿
• 全局的flash变量没被用来跟踪当前状态，逻辑上和开关动作脱节

修正后的实现方案 #

我们直接操作已有的captureDevice来动态切换闪光灯，不用重启会话，高效又流畅：

1. 重写开关的IBAction方法

把你的开关触发方法改成这样（注意拼写，你写的swithch如果和XIB关联了可以保留，建议后续改成switchFlash更清晰）：

@IBAction func swithch(_ sender: UISwitch) {
    guard let device = captureDevice else { return }
    
    // 先判断当前设备是否支持闪光灯
    guard device.hasTorch else {
        sender.isOn = false
        let alert = UIAlertController(title: "提示", message: "当前设备不支持闪光灯功能", preferredStyle: .alert)
        alert.addAction(UIAlertAction(title: "确定", style: .default))
        present(alert, animated: true)
        return
    }
    
    do {
        // 锁定设备配置，避免其他操作干扰
        try device.lockForConfiguration()
        
        // 根据开关状态切换闪光灯
        if sender.isOn {
            try device.setTorchModeOn(level: 1.0) // 1.0为最大亮度，可按需调整
            flash = 1
        } else {
            device.torchMode = .off
            flash = 0
        }
        
        // 解锁配置，允许其他操作
        device.unlockForConfiguration()
    } catch {
        print("闪光灯切换失败: \(error.localizedDescription)")
        sender.isOn = !sender.isOn // 切换失败后还原开关状态
    }
}

2. 调整初始化时的闪光灯逻辑

在你的setupCamera方法里，根据传入的flash参数设置初始闪光灯状态（如果设备支持的话）：

func setupCamera(flash: Int) {
    // 保留你原有的相机会话初始化、输入输出配置等代码
    // ...
    
    // 初始化闪光灯状态
    guard let device = captureDevice else { return }
    if device.hasTorch {
        do {
            try device.lockForConfiguration()
            if flash == 1 {
                try device.setTorchModeOn(level: 1.0)
                self.flash = 1
            } else {
                device.torchMode = .off
                self.flash = 0
            }
            device.unlockForConfiguration()
        } catch {
            print("初始化闪光灯失败: \(error.localizedDescription)")
        }
    }
}

3. 移除不必要的会话重建代码

把原IBAction里的stopActiveSession()和新建captureSession的代码删掉——我们不需要重启会话就能切换闪光灯，这些代码只会添乱。

整合后的关键代码片段 #

最终你的ImageDetectionViewController里闪光灯相关部分大概是这样：

class ImageDetectionViewController: UIViewController {
    var callBackImageDetection :(State)->Void = { state in }
    @IBOutlet weak var previewView: UIView!
    @IBOutlet weak var stackView: UIStackView!
    @IBOutlet weak var lowerView: UIView!
    
    var previewLayer: AVCaptureVideoPreviewLayer!
    let bubbleLayer = BubbleLayer(string: "")
    let queue = DispatchQueue(label: "videoQueue")
    var captureSession = AVCaptureSession()
    var captureDevice: AVCaptureDevice?
    let videoOutput = AVCaptureVideoDataOutput()
    var unknownCounter = 0
    let confidence: Float = 0.8
    let targetImageSize = CGSize(width: 227, height: 227)
    var flash = 0 // 0=关闭，1=开启，跟踪当前闪光灯状态
    
    @IBAction func swithch(_ sender: UISwitch) {
        guard let device = captureDevice else { return }
        
        guard device.hasTorch else {
            sender.isOn = false
            let alert = UIAlertController(title: "提示", message: "当前设备不支持闪光灯功能", preferredStyle: .alert)
            alert.addAction(UIAlertAction(title: "确定", style: .default))
            present(alert, animated: true)
            return
        }
        
        do {
            try device.lockForConfiguration()
            
            if sender.isOn {
                try device.setTorchModeOn(level: 1.0)
                flash = 1
            } else {
                device.torchMode = .off
                flash = 0
            }
            
            device.unlockForConfiguration()
        } catch {
            print("闪光灯切换失败: \(error.localizedDescription)")
            sender.isOn = !sender.isOn
        }
    }
    
    func setupCamera(flash: Int) {
        // 这里保留你原有的相机会话初始化代码
        // ...
        
        // 初始化闪光灯状态
        guard let device = captureDevice else { return }
        if device.hasTorch {
            do {
                try device.lockForConfiguration()
                if flash == 1 {
                    try device.setTorchModeOn(level: 1.0)
                    self.flash = 1
                } else {
                    device.torchMode = .off
                    self.flash = 0
                }
                device.unlockForConfiguration()
            } catch {
                print("初始化闪光灯失败: \(error.localizedDescription)")
            }
        }
    }
    
    // ... 其他原有方法 ...
}

额外小提示 #

• 记得在stopActiveSession()方法里也关闭闪光灯，避免APP退到后台后闪光灯还亮着
• 如果你的图像分类器在闪光灯开启时需要调整预处理逻辑（比如亮度补偿），可以根据flash变量的值做对应处理

内容的提问来源于stack exchange，提问作者Somdeb