iOS7以上使用CIFilter生成二维码

前言

前阵子做「优孕宝」这个项目,在医生端需要用到二维码生成,发现 iOS7 之后,可以使用原生的CIFilter创建二维码,但是生成的二维码只有黑白,而且大小不好控制,找了一下资料,发现解决的方法,使二维码透明背景,自定义颜色,还能加上阴影效果,方法很简单,直接调用即可。

二维码

生成二维码(生成CIImage)

首先是二维码的生成,使用CIFilter很简单,直接传入需要生成二维码的字符串即可。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
/**
生成二维码图片
@param qrString 二维码文本内容
@return 二维码图片(CIImage)
*/
- (CIImage *)createQRForString:(NSString *)qrString
{
NSData *stringData = [qrString dataUsingEncoding:NSUTF8StringEncoding];
// 创建一个二维码滤镜实例
CIFilter *qrFilter = [CIFilter filterWithName:@"CIQRCodeGenerator"];
// 恢复滤镜默认设置
[filter setDefaults];
// 设置滤镜内容
[qrFilter setValue:stringData forKey:@"inputMessage"];
// 设置滤镜纠错级别
[qrFilter setValue:@"M" forKey:@"inputCorrectionLevel"];
// 生成二维码 返回CIImage
return qrFilter.outputImage;
}

生成二维码 (转换UIImage)

因为生成的二维码是一个CIImage,我们直接转换成UIImage的话大小不好控制,所以使用下面方法返回需要大小的UIImage

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
/**
生成二维码图片
@param image CIImage
@param size 生成的二维码图片宽高
@return 二维码图片
*/
- (UIImage *)createQRImageFormCIImage:(CIImage *)image withSize:(CGFloat)size
{
CGRect extent = CGRectIntegral(image.extent);
CGFloat scale = MIN(size/CGRectGetWidth(extent), size/CGRectGetHeight(extent));
// 1.创建bitmap
size_t width = CGRectGetWidth(extent) * scale;
size_t height = CGRectGetHeight(extent) * scale;
CGColorSpaceRef cs = CGColorSpaceCreateDeviceGray();
CGContextRef bitmapRef = CGBitmapContextCreate(nil, width, height, 8, 0, cs, (CGBitmapInfo)kCGImageAlphaNone);
CIContext *context = [CIContext contextWithOptions:nil];
CGImageRef bitmapImage = [context createCGImage:image fromRect:extent];
CGContextSetInterpolationQuality(bitmapRef, kCGInterpolationNone);
CGContextScaleCTM(bitmapRef, scale, scale);
CGContextDrawImage(bitmapRef, extent, bitmapImage);
// 2.保存bitmap到图片
CGImageRef scaledImage = CGBitmapContextCreateImage(bitmapRef);
CGContextRelease(bitmapRef);
CGImageRelease(bitmapImage);
return [UIImage imageWithCGImage:scaledImage];
}

填充颜色

因为生成的二维码是黑白的,如果有颜色的需求,还要对二维码进行颜色填充,并转换为透明背景,使用遍历图片像素来更改图片颜色,因为使用的是CGContext,速度非常快。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
// Release Data
void ProviderReleaseData(void *info, const void *data, size_t size)
{
free((void *)data);
}
/**
设置图片颜色
@param image 图片
@param red 红色通道(0~255)
@param green 绿色通道(0~255)
@param blue 蓝色通道(0~255)
@return 背景透明、设置颜色的图片
*/
- (UIImage *)setupImageColorWithImage:(UIImage *)image red:(CGFloat)red green:(CGFloat)green blue:(CGFloat)blue
{
const int imageWidth = image.size.width;
const int imageHeight = image.size.height;
size_t bytesPerRow = imageWidth * 4;
uint32_t *rgbImageBuf = (uint32_t *)malloc(bytesPerRow * imageHeight);
CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();
CGContextRef context = CGBitmapContextCreate(rgbImageBuf, imageWidth, imageHeight, 8, bytesPerRow, colorSpace,
kCGBitmapByteOrder32Little | kCGImageAlphaNoneSkipLast);
CGContextDrawImage(context, CGRectMake(0, 0, imageWidth, imageHeight), image.CGImage);
// 遍历像素
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; // 0~255
ptr[2] = green;
ptr[1] = blue;
} 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, colorSpace,
kCGImageAlphaLast | kCGBitmapByteOrder32Little, dataProvider,
NULL, true, kCGRenderingIntentDefault);
CGDataProviderRelease(dataProvider);
UIImage *resultUIImage = [UIImage imageWithCGImage:imageRef];
// 清理空间
CGImageRelease(imageRef);
CGContextRelease(context);
CGColorSpaceRelease(colorSpace);
return resultUIImage;
}

添加阴影

经过这样的处理,基本上二维码就成型了,如果还想加上阴影,就在ImageViewLayer上添加阴影即可。

1
2
3
4
5
6
7
8
// 设置阴影的偏移量
imageView.layer.shadowOffset = CGSizeMake(0, 0.5);
// 设置阴影的半径
imageView.layer.shadowRadius = 1;
// 设置阴影颜色
imageView.layer.shadowColor = [UIColor blackColor].CGColor;
// 设置阴影的不透明度
imageView.layer.shadowOpacity = 0.3;

后话

一个完整的二维码就生成了,一般的需求应该已经满足。具体使用的时候可以再继续封装,比如二维码中间可以放一个头像,二维码下面还可以放一行文字,比如扫一扫向我付款或者加我为好友之类的。我已经封装了一个扫一扫的库,有时间再考虑封装一个生成二维码的库。

  • 本文作者: Bingo
  • 本文链接: https://blog.bingo.ren/46.html
  • 版权声明: 本博客所有文章除特别声明外,均采用 MIT 许可协议。转载请注明出处!