如何将正方形图像添加到QRCode | Swift

import UIKit
import CoreImage

func addSquareImageToQRCode(qrCodeImage: UIImage, squareImage: UIImage) -> UIImage? {
    guard let qrCodeCIImage = CIImage(image: qrCodeImage),
          let squareCIImage = CIImage(image: squareImage) else {
        return nil
    }
    
    let context = CIContext(options: nil)
    
    // 缩放squareImage以适应QRCode的大小
    let squareSize = qrCodeCIImage.extent.size.width / 4
    let resizedSquareCIImage = squareCIImage.transformed(by: CGAffineTransform(scaleX: squareSize / squareCIImage.extent.size.width, y: squareSize / squareCIImage.extent.size.height))
    
    // 将squareImage叠加到QRCode的中心
    let compositeCIImage = CIFilter(name: "CISourceOverCompositing")!
    compositeCIImage.setValue(resizedSquareCIImage, forKey: kCIInputImageKey)
    compositeCIImage.setValue(qrCodeCIImage, forKey: kCIInputBackgroundImageKey)
    
    guard let outputCIImage = compositeCIImage.outputImage else {
        return nil
    }
    
    // 渲染最终的图像
    guard let cgImage = context.createCGImage(outputCIImage, from: outputCIImage.extent) else {
        return nil
    }
    
    return UIImage(cgImage: cgImage)
}

// 示例使用
if let qrCodeImage = UIImage(named: "qrcode.png"),
   let squareImage = UIImage(named: "square.png"),
   let resultImage = addSquareImageToQRCode(qrCodeImage: qrCodeImage, squareImage: squareImage) {
    // 使用resultImage
}