iOS native implementation generates QR code (CoreImage) tool class, does not depend on third-party libraries, can customize background color, and add logo (Swift 4.0)

import Foundation
import CoreImage
import UIKit

extension UIColor {

    var coreImageColor: CIColor {
        return CIColor(color: self)
    }

    var components: (red: CGFloat, green: CGFloat, blue: CGFloat, alpha: CGFloat) {

        let coreImageColor = self.coreImageColor
        return (coreImageColor.red, coreImageColor.green, coreImageColor.blue, coreImageColor.alpha)
    }
}

class QRCreateModel {

    /// 文本
    var text: String!

    /// 二维码中间的logo
    var logo: String?

    /// 二维码缩放倍数{27*scale,27*scale}
    var scale: Float = 10

    /// 二维码背景颜色
    var backgroundColor: UIColor = UIColor.white

    /// 二维码颜色
    var contentColor: UIColor = UIColor.black
}

class QRCreator {

    static let shared = QRCreator()

    private let qrFilter: CIFilter
    private let colorFilter: CIFilter

    private init() {

        /// 创建二维码滤镜
        qrFilter = CIFilter(name: "CIQRCodeGenerator")!

        /// 创建颜色滤镜
        colorFilter = CIFilter(name: "CIFalseColor")!
    }

    private func createBase(text: String, scale: Float) -> CIImage? {

        qrFilter.setDefaults()
        guard let data = text.data(using: String.Encoding.utf8) else {

            return nil
        }

        /// 给二维码滤镜设置inputMessage
        qrFilter.setValue(data, forKey: "inputMessage")
        guard var outputImage = qrFilter.outputImage else {

            return nil
        }

        outputImage = outputImage.transformed(by: CGAffineTransform(scaleX: CGFloat(scale), y: CGFloat(scale)))
        return outputImage
    }

    private func colourFilter(ciImage: CIImage, model: QRCreateModel) -> CIImage? {

        /// 颜色滤镜恢复默认值
        colorFilter.setDefaults()

        /// 设置颜色滤镜的inputImage
        colorFilter.setValue(ciImage, forKey: "inputImage")

        /// 设置inputImage的backgroundColor(key: inputColor1)
        colorFilter.setValue(model.backgroundColor.coreImageColor, forKey: "inputColor1")

        /// 设置inputImage的contentColor(key: inputColor0)
        colorFilter.setValue(model.contentColor.coreImageColor, forKey: "inputColor0")

        return colorFilter.outputImage
    }

    private func addLogo(ciImage: CIImage, model: QRCreateModel) -> UIImage? {

        guard let _ = model.logo,
            let logoImage = UIImage(named: model.logo!) else {

                return nil
        }

        let image = UIImage(ciImage: ciImage)
        let originX = (image.size.width - logoImage.size.width)/2.0
        let originY = (image.size.height - logoImage.size.height)/2.0

        UIGraphicsBeginImageContext(image.size)
        image.draw(in: CGRect(x: 0, y: 0, width: image.size.width, height: image.size.height))
        logoImage.draw(in: CGRect(x: originX, y: originY, width: logoImage.size.width, height: logoImage.size.height))

        let outPutImage = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()

        return outPutImage
    }

    func create(_ model: QRCreateModel) -> UIImage? {

        guard var outputImage = createBase(text: model.text, scale: model.scale) else {

            return nil
        }

        if let colorOutputImage = colourFilter(ciImage: outputImage, model: model) {

            outputImage = colorOutputImage
        }

        guard let qrImageWithLogo = addLogo(ciImage: outputImage, model: model) else {

            return  UIImage(ciImage: outputImage)
        }

        return qrImageWithLogo
    }
}

Complete usage example

let qrModel = QRCreateModel()
qrModel.text = "https://www.yaozuopan.top"
qrModel.contentColor = UIColor.yellow
qrModel.backgroundColor = UIColor.gray

/// 二维码的容错率最大为30%(即二维码被遮挡的部分不能大于30%,否则二维码无法被识别)
/// e.g.
/// 假设二维码大小为{100,100}, logo的大小最大为{30,30}
/// 在设置loog请注意
// qrModel.logo = "mylogo"

if let qrImage = QRCreator.shared.create(qrModel) {

    print(qrImage.size.width)
    print(qrImage.size.height)
    // imageView.image = qrImage
}

Concise use

let qrModel = QRCreateModel()
qrModel.text = "https://www.yaozuopan.top"

if let qrImage = QRCreator.shared.create(qrModel) {

    // imageView.image = qrImage
}

Code and sample download (with QR code scanning)
go to

Guess you like

Origin http://43.154.161.224:23101/article/api/json?id=324523667&siteId=291194637