如果觉得代码有更好的实现方式或者优化方案，请留言，大家共同交流一哈
Swift
swift全部代码，可直接拖入项目使用
透明度一直设置不成功，所以注释掉了，如果各位大神实现，请不吝赐教。
import UIKit
import CoreImage
import CoreGraphics
/// 像素，每个像素包含红，蓝，绿，透明度
public struct CWPixel {
// 完整像素值
var value: UInt32
// 红色值
var red: UInt8 {
get {
return UInt8(value & 0xFF)
}
set {
value = UInt32(newValue) | (value & 0xFFFFFF00)
}
}
// 绿色
var green: UInt8 {
get {
return UInt8((value >> 8) & 0xFF)
}
set {
value = (UInt32(newValue) << 8) | (value & 0xFFFF00FF)
}
}
// 蓝色
var blue: UInt8 {
get {
return UInt8((value >> 16) & 0xFF)
}
set {
value = (UInt32(newValue) << 16) | (value & 0xFF00FFFF)
}
}
// 透明
var alpha: UInt8 {
get {
return UInt8((value >> 24) & 0xFF)
}
set {
value = (UInt32(newValue) << 24) | (value & 0x00FFFFFF)
}
}
// 是否为白色
var isWhite: Bool {
get {
return value == 0xFFFFFFFF
}
}
}
// MARK: - 生成二维码
extension UIImage {
class func cwQRCodeImage(content: String,
logo: UIImage? = nil,
logoFrame: CGRect = CGRect.zero,
size: CGFloat = -1,
highCorrection: Bool = false,
tintColor: UIColor? = nil) -> UIImage? {
var retImage: UIImage? = nil
var openHighCorrection = highCorrection
// 如果要添加logo或者绘制颜色，自动开启高容错
if logo != nil || tintColor != nil {
openHighCorrection = true
}
let correctionLevel = openHighCorrection ? "H" : "M"
// 生成二维码并设置纠错级别
let outputImage = qrCodeOutputImage(content: content,
correctionLevel: correctionLevel)
if let output = outputImage {
// 调整尺寸之后的二维码
retImage = adjustHDQRCode(outputImage: output,
size:size)
}
// 二维码着色
if let mTintColor = tintColor {
retImage = retImage?.cwModifyTintColor(tintColor: mTintColor)
}
// 二维码添加logo
if let mLogo = logo {
retImage = retImage?.cwAddLogoImage(image: mLogo,
frame: logoFrame)
}
return retImage
}
/// 生成原始二维码CIImage
///
/// - Parameters:
/// - content: 二维码包含的内容
/// - correctionLevel: 二维码纠错级别
/// ，调高纠错级别可以往上面加一个头像或者变色，但是识别速度也会降低
/// - correctionLevel:
/// - 二维码纠错级别,通过调高二维码的纠错级别可以往上面加一个头像或者变色，但是识别速度也会降低
/// - L（低） 7％的码字可以被恢复。
/// - M（中） 15％的码字可以被恢复。
/// - Q（四分）25％的码字可以被恢复。
/// - H（高） 30％的码字可以被恢复。
private class func qrCodeOutputImage(content: String,
correctionLevel: String) -> CIImage? {
// 初始化filter
let data = content.data(using: .utf8)
let qrFilter = CIFilter(name: "CIQRCodeGenerator")
// 设置内容
qrFilter?.setValue(data, forKey: "inputMessage")
qrFilter?.setValue(correctionLevel, forKey: "inputCorrectionLevel")
return qrFilter?.outputImage
}
/// 调整二维码的尺寸并返回高清图片
/// - Code:
/// - let transform = CGAffineTransform(scaleX: 5, y: 5)
/// - outputImage?.transformed(by: transform)
///
/// - Parameters:
/// - outputImage: CIFilter生成的二维码
/// - size: 放大之后的尺寸
/// - Returns: 返回调整之后的图片
private class func adjustHDQRCode(outputImage: CIImage,
size: CGFloat) -> UIImage? {
let integralRect: CGRect = outputImage.extent.integral
// 设置默认放大5倍
var scale: CGFloat = 5.0
if size != -1 {
// 需要计算出最佳缩放比例
scale = min(size/integralRect.width, size/integralRect.height)
}
let width = integralRect.width * scale
let height = integralRect.height * scale
let colorSpaceRef = CGColorSpaceCreateDeviceGray()
let bitmapRef = CGContext(data: nil,
width: Int(width),
height: Int(height),
bitsPerComponent: 8,
bytesPerRow: 0,
space: colorSpaceRef,
bitmapInfo: CGImageAlphaInfo.none.rawValue)
let context = CIContext(options: nil)
//放大重新绘制
if let bitmapImage = context.createCGImage(outputImage,
from: integralRect) {
bitmapRef?.interpolationQuality = .none
bitmapRef?.scaleBy(x: scale, y: scale)
bitmapRef?.draw(bitmapImage, in: integralRect)
}
let scaledImage: CGImage! = bitmapRef?.makeImage()
return UIImage(cgImage: scaledImage)
}
/// 改变二维码图片的像素的颜色
private func cwModifyTintColor(tintColor: UIColor) -> UIImage {
// 必须可以转换为CGImage
guard let mCgImage = self.cgImage else {
return self
}
// 将Float类型保存为Int进行地址操作
let width = Int(self.size.width)
let height = Int(self.size.height)
// 创建一个大小为 4 * width * height 的缓存区
let bytesPerRow = width * 4
let byteSize = width * height
// 初始化一个指针类型并分配空间
let rgbImageBuf = UnsafeMutablePointer<CWPixel>.allocate(capacity: byteSize)
// 初始化绘制，颜色空间跟随设备
let colorSpaceRef = CGColorSpaceCreateDeviceRGB()
// 用BGRA创建一个位图
var bitmapInfo: UInt32 = CGBitmapInfo.byteOrder32Big.rawValue
bitmapInfo = bitmapInfo | CGImageAlphaInfo.premultipliedLast.rawValue & CGBitmapInfo.alphaInfoMask.rawValue
// 创建位图
guard let context = CGContext(data: rgbImageBuf,
width: width,
height: height,
bitsPerComponent: 8,
bytesPerRow: bytesPerRow,
space: colorSpaceRef,
bitmapInfo: bitmapInfo) else {
return self
}
// 使用rgbImageBuf 填充 mCgImage
context.draw(mCgImage, in: CGRect(origin: .zero, size: self.size))
// 连续存储的内存集合
var bufferPointer = UnsafeMutableBufferPointer<CWPixel>(start: rgbImageBuf,
count: byteSize)
// 释放分配的空间
// rgbImageBuf.deallocate()
//获取红绿蓝色值
var red: CGFloat = 0
var green: CGFloat = 0
var blue: CGFloat = 0
var alpha: CGFloat = 0
tintColor.getRed(&red, green: &green, blue: &blue, alpha: &alpha)
// 遍历像素，改变像素点颜色
for currentPixel in 0..<byteSize {
var pixel = bufferPointer[currentPixel]
if !pixel.isWhite {
//当不是白色的时候才使用
pixel.red = UInt8(red * 255)
pixel.green = UInt8(green * 255)
pixel.blue = UInt8(blue * 255)
// pixel.alpha = UInt8(alpha * 255)
let retPixel = pixel
bufferPointer[currentPixel] = retPixel
}
}
let inputColorSpace = CGColorSpaceCreateDeviceRGB()
var inputBitmapInfo: UInt32 = CGBitmapInfo.byteOrder32Big.rawValue
let inputBytesPerRow = width * 4
inputBitmapInfo |= CGImageAlphaInfo.premultipliedLast.rawValue & CGBitmapInfo.alphaInfoMask.rawValue
guard let imageContext = CGContext(data: bufferPointer.baseAddress,
width: width,
height: height,
bitsPerComponent: 8,
bytesPerRow: inputBytesPerRow,
space: inputColorSpace,
bitmapInfo: inputBitmapInfo,
releaseCallback: nil,
releaseInfo: nil) else {
return self
}
guard let cgImage = imageContext.makeImage() else {
return self
}
let image = UIImage(cgImage: cgImage)
return image
}
/// 给二维码图片添加logo
private func cwAddLogoImage(image: UIImage,
frame: CGRect) -> UIImage? {
var resultImage: UIImage? = nil
if frame != CGRect.zero {
// 开始绘制
UIGraphicsBeginImageContext(self.size)
//绘制二维码图片
self.draw(in: CGRect.init(origin: CGPoint.zero,
size:self.size))
//绘制Logo
image.draw(in: frame)
//获取绘制之后的image
resultImage = UIGraphicsGetImageFromCurrentImageContext();
//结束绘制
UIGraphicsEndImageContext();
}
return resultImage
}
}
OC
二维码生成
/**
*
* @brief 生成二维码图片
*
* @param content 内容
* @param width 二维码宽度，-1为自动
* @param highLevel 是否是高的纠错级别，设为YES,可往上添加logo
*
* @return 生成的二维码图片
*/
+ (UIImage *)qrCodeImageWithContent:(NSString *)content width:(CGFloat)width highCorrectionLevel:(BOOL)highLevel
{
UIImage *resultImage = nil;
NSData *data = [content dataUsingEncoding:NSUTF8StringEncoding];
//创建filter
CIFilter *QRFilter = [CIFilter filterWithName:@"CIQRCodeGenerator"];
// 设置内容
[QRFilter setValue:data forKey:@"inputMessage"];
//二维码纠错级别,通过调高二维码的纠错级别可以往上面加一个头像或者变色，但是识别速度也会降低
/**
L（低） 7％的码字可以被恢复。
M（中） 15％的码字可以被恢复。
Q（四分）25％的码字可以被恢复。
H（高） 30％的码字可以被恢复。
*
*/
[QRFilter setValue:highLevel?@"H":@"M" forKey:@"inputCorrectionLevel"];
CIImage *qrOutPutImage = QRFilter.outputImage;
CGRect integralRect = CGRectIntegral(qrOutPutImage.extent);
CGFloat scale = 5;
if (width != -1) {
//计算出缩放比例
scale = MIN(width/CGRectGetWidth(integralRect), width/CGRectGetHeight(integralRect));
}
//计算出宽高
size_t width = CGRectGetWidth(integralRect)*scale;
size_t height = CGRectGetHeight(integralRect)*scale;
//二维码生成的位图不清晰，所以需要放大绘制
CGColorSpaceRef colorSpaceRef = CGColorSpaceCreateDeviceGray();
CGContextRef bitmapRef = CGBitmapContextCreate(nil, width, height, 8, 0, colorSpaceRef, (CGBitmapInfo)kCGImageAlphaNone);
CIContext *context = [CIContext contextWithOptions:nil];
CGImageRef bitmapImage = [context createCGImage:qrOutPutImage fromRect:integralRect];
CGContextSetInterpolationQuality(bitmapRef, kCGInterpolationNone);
CGContextScaleCTM(bitmapRef, scale, scale);
CGContextDrawImage(bitmapRef, integralRect, bitmapImage);
CGImageRef scaledImage = CGBitmapContextCreateImage(bitmapRef);
CGContextRelease(bitmapRef);
CGImageRelease(bitmapImage);
CGColorSpaceRelease(colorSpaceRef);
resultImage = [UIImage imageWithCGImage:scaledImage];
CGImageRelease(scaledImage);
return resultImage;
}
添加logo，修改配色
//生成二维码图片，可插入logo，可调整配色
+ (UIImage *)qrCodeImageWithContent:(NSString *)content codeImageWidth:(CGFloat)width logo:(UIImage *)logo logoFrame:(CGRect)logoFrame color:(UIColor *)color
{
//生成固定尺寸的二维码
UIImage *image = [self qrCodeImageWithContent:content width:width highCorrectionLevel:YES];
UIImage *resultImage = nil;
if (image) {
//修改配色
int imageWidth = image.size.width;
int imageHeight = image.size.height;
size_t bytesPerRow = imageWidth * 4;
uint32_t *rgbImageBuf = (uint32_t *)malloc(bytesPerRow * imageHeight);
CGColorSpaceRef colorSpaceRef = CGColorSpaceCreateDeviceRGB();
CGContextRef context = CGBitmapContextCreate(rgbImageBuf, imageWidth, imageHeight, 8, bytesPerRow, colorSpaceRef, kCGBitmapByteOrder32Little | kCGImageAlphaNoneSkipLast);
//获取红绿蓝色值
CGFloat red = 0.0f;
CGFloat green = 0.0f;
CGFloat blue = 0.0f;
CGFloat alpha = 0.0f;
[color getRed:&red green:&green blue:&blue alpha:&alpha];
//遍历像素，改变像素点颜色
int pixelNum = imageWidth * imageHeight;
uint32_t *pCurPtr = rgbImageBuf;
for (int i = 0; i < pixelNum; i++, pCurPtr++) {
if ((*pCurPtr & 0xFFFFFF00) < 0x99999900) {
uint8_t *ptr = (uint8_t *)pCurPtr;
ptr[3] = red*255;
ptr[2] = green*255;
ptr[1] = blue*255;
}
else
{
uint8_t *ptr = (uint8_t *)pCurPtr;
ptr[0] = 0;
}
}
//取出图片
CGDataProviderRef dataProvider = CGDataProviderCreateWithData(NULL, rgbImageBuf, bytesPerRow * imageHeight, ProviderReleaseData);
CGImageRef imageRef = CGImageCreate(imageWidth, imageHeight, 8, 32, bytesPerRow, colorSpaceRef,
kCGImageAlphaLast | kCGBitmapByteOrder32Little, dataProvider,
NULL, true, kCGRenderingIntentDefault);
CGDataProviderRelease(dataProvider);
resultImage = [UIImage imageWithCGImage:imageRef];
CGImageRelease(imageRef);
CGContextRelease(context);
CGColorSpaceRelease(colorSpaceRef);
}
// 绘制logo
if (resultImage) {
//存在logo,且logoFrame不为空
if (logo && !CGRectEqualToRect(logoFrame, CGRectZero)) {
//开始绘制
UIGraphicsBeginImageContext(resultImage.size);
//绘制二维码图片
[resultImage drawInRect:CGRectMake(0, 0, resultImage.size.width, resultImage.size.height)];
//绘制Logo
[logo drawInRect:logoFrame];
//获取绘制之后的image
resultImage = UIGraphicsGetImageFromCurrentImageContext();
//结束绘制
UIGraphicsEndImageContext();
}
}
return resultImage;
}
//数据释放
void ProviderReleaseData (void *info, const void *data, size_t size){
free((void*)data);
}
