• swift中如何识别二维码图片中的内容?
  • 发布于 2个月前
  • 75 热度
    0 评论
在如今的很多应用里都实现了识别应用内某个图片中二维码的功能,比如最常用的微信,在聊天中就可以长按图片来识别图中的二维码,今天来讲讲如何在 swift 中利用代码识别一张图片中的二维码内容。

CoreImage 识别二维码
其实要识别图片中的二维码内容非常简单,主要用到的是 CoreImage 中的 CIDetector,为了使用方便,我给 UIImage 写了个分类,来专门检测图片上的二维码内容:
extension UIImage {
    /// 检测图片中二维码内容
    /// - Returns: 二维码中的内容
    func detectQRCode() -> String? {
        guard let ciImage = ciImage ?? CIImage(image: self) else {
            return nil
        }
        // 堆代码 duidaima.com
        let options = [CIDetectorAccuracy: CIDetectorAccuracyHigh]
        let context = CIContext()
        
        guard let detector = CIDetector(ofType: CIDetectorTypeQRCode, context: context, options: options) else {
            return nil
        }
        
        let features = detector.features(in: ciImage)
        
        if let feature = features.first as? CIQRCodeFeature {
            return feature.messageString
        }
        
        return nil
    }
}
简单解释下上边的代码,我们首先将 UIImage 对象转换为 CIImage 对象。然后,我们创建一个 CIDetector 对象并设置识别精度为高。接下来,我们使用该 CIDetector 对象在 CIImage 中查找二维码特征。如果找到了二维码,我们将返回其字符串内容。如果未找到二维码,我们将返回 nil。使用非常简单,直接拿 UIImage 对象调用 detectQRCode 方法就行了。

识别 UIView 中的二维码
在做类似微信长按识别二维码需求时,可能会针对某个 UIView 下的二维码进行识别,我也为 UIView 写了个分类,来处理这种情况:
extension UIView {
    func detectQRCode(completion: @escaping (String?) -> Void) {
        guard let image = self.asImage() else {
            completion(nil)
            return
        }
        DispatchQueue.global(qos: .userInitiated).async {
            let qrCode = image.detectQRCode()
            DispatchQueue.main.async {
                completion(qrCode)
            }
        }
    }
    
    private func asImage() -> UIImage? {
        UIGraphicsBeginImageContextWithOptions(self.bounds.size, false, UIScreen.main.scale)
        guard let context = UIGraphicsGetCurrentContext() else {
            return nil
        }
        self.layer.render(in: context)
        guard let image = UIGraphicsGetImageFromCurrentImageContext() else {
            return nil
        }
        UIGraphicsEndImageContext()
        return image
    }
}
这个方案的思路是,先将 view 转为图片,然后再使用上一步中写的方法检测图片中的二维码内容。另外我还考虑到检测二维码的过程可能会比较耗时,所以我把方法改成了异步,检测过程放在了子线程中执行。

使用起来非常方便:
// 假设有一个名为 myView 的 UIView 对象
myView.detectQRCode { qrCode in
    if let qrCode = qrCode {
        print("检测到的二维码内容为: \(qrCode)")
    } else {
        print("没有检测到二维码内容")
    }
}
多个二维码的情况
细心的同学可能已经考虑到了,一个图片或者 view 里可能包含多张二维码,那么这种情况也是需要兼容的,其实我们上边的检测方法 detector.features 中返回的是一个数组,里边就包含了所有检测到的对象,我们只需要稍加改动代码,就可以实现这种情况:
extension UIImage {
    /// 检测图片中二维码内容
    /// - Returns: 二维码中的内容
    func detectQRCodes() -> [String] {
        guard let ciImage = ciImage ?? CIImage(image: self) else {
            return []
        }
        
        let options = [CIDetectorAccuracy: CIDetectorAccuracyHigh]
        let context = CIContext()
        
        guard let detector = CIDetector(ofType: CIDetectorTypeQRCode, context: context, options: options) else {
            return []
        }
        
        let features = detector.features(in: ciImage)
        
        var qrCodes: [String] = []
        
        for feature in features {
            if let qrCodeFeature = feature as? CIQRCodeFeature {
                if let qrCode = qrCodeFeature.messageString {
                    qrCodes.append(qrCode)
                }
            }
        }
        
        return qrCodes
    }
}
我把这个方法的返回值改成了一个字符串数组,如果图片中包含多个二维码,CIDetector 的 features(in:) 方法就可以获取所有的二维码特征。然后遍历这些特征,并提取每个二维码的内容。
用户评论