iOS 原生实现生成二维码(CoreImage)工具类,不依赖第三方库,可自定义背景颜色,添加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

    }

}

完整使用示例

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

}

简洁使用

let qrModel = QRCreateModel()

qrModel.text = "https://www.yaozuopan.top"

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

    // imageView.image = qrImage

}

代码及示例下载(有实现二维码扫描)

前往

巴特西

iOS 原生实现生成二维码(CoreImage)工具类,不依赖第三方库,可自定义背景颜色,添加logo(Swift 4.0)

最新文章

热门文章