iOS：基于实时摄像头的实时 OCR（类似于 iTunes 兑换礼品卡）

iOS：基于实时摄像头的实时 OCR（类似于 iTunes 兑换礼品卡）

swift
import Vision
import AVFoundation
class ViewController: UIViewController, AVCaptureVideoDataOutputSampleBufferDelegate {
    var captureSession: AVCaptureSession?
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        setupCaptureSession()
    }
    
    func setupCaptureSession() {
        captureSession = AVCaptureSession()
        
        guard let captureDevice = AVCaptureDevice.default(for: .video) else { return }
        
        guard let input = try? AVCaptureDeviceInput(device: captureDevice) else { return }
        
        captureSession?.addInput(input)
        
        let output = AVCaptureVideoDataOutput()
        output.setSampleBufferDelegate(self, queue: DispatchQueue(label: "videoQueue"))
        captureSession?.addOutput(output)
        
        captureSession?.startRunning()
    }
    
    func captureOutput(_ output: AVCaptureOutput, didOutput sampleBuffer: CMSampleBuffer, from connection: AVCaptureConnection) {
        guard let pixelBuffer: CVPixelBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) else { return }
        
        let request = VNRecognizeTextRequest { (request, error) in
            if let error = error {
                print("OCR Error:", error)
                return
            }
            
            guard let observations = request.results as? [VNRecognizedTextObservation] else { return }
            
            let recognizedStrings = observations.compactMap { observation in
                return observation.topCandidates(1).first?.string
            }
            
            DispatchQueue.main.async {
                // 处理识别到的文本
                self.handleRecognizedStrings(recognizedStrings)
            }
        }
        
        request.recognitionLevel = .accurate
        request.recognitionLanguages = ["en", "zh-Hans"]
        
        try? VNImageRequestHandler(cvPixelBuffer: pixelBuffer, options: [:]).perform([request])
    }
    
    func handleRecognizedStrings(_ strings: [String]) {
        // 处理识别到的文本
    }
}

swift
func handleRecognizedStrings(_ strings: [String]) {
    guard let recognizedString = strings.first else { return }
    
    if recognizedString.contains("iTunes") && recognizedString.contains("兑换码") {
        // 提取兑换码
        let redeemCode = recognizedString.replacingOccurrences(of: "iTunes兑换码：", with: "")
        
        // 将兑换码添加到用户的iTunes账户中
        iTunesAPI.redeem(redeemCode) { success in
            if success {
                print("兑换成功")
            } else {
                print("兑换失败")
            }
        }
    }
}