iOS 识别技术

iOS代码获取图片主色调

2017-09-25  本文已影响70人  Skylpy

最近的开发过程中,有一个功能是根据银行的图标来确定cell的背景颜色,即获取图片的主色调。网上搜索了一些方法,发现有位博主分享了一个很好的代码方法,不过效果感觉不是特别明显,或者说变化不是很大,如图:

后来仔细想了一下,代码和方法是没有错误的,但是我们把透明颜色以及白色也计算进去了,所以效果才会那么不明显。所以在进行rgb色值获取的时候,排除掉透明色及白色即可,修改后的代码如下:

//根据图片获取图片的主色调

+(UIColor*)mostColor:(UIImage*)image{

#if __IPHONE_OS_VERSION_MAX_ALLOWED > __IPHONE_6_1

intbitmapInfo = kCGBitmapByteOrderDefault | kCGImageAlphaPremultipliedLast;

#else

intbitmapInfo = kCGImageAlphaPremultipliedLast;

#endif

//第一步 先把图片缩小 加快计算速度. 但越小结果误差可能越大

CGSize thumbSize=CGSizeMake(image.size.width/2, image.size.height/2);

CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();

CGContextRef context = CGBitmapContextCreate(NULL,

thumbSize.width,

thumbSize.height,

8,//bits per component

thumbSize.width*4,

colorSpace,

bitmapInfo);

CGRect drawRect = CGRectMake(0,0, thumbSize.width, thumbSize.height);

CGContextDrawImage(context, drawRect, image.CGImage);

CGColorSpaceRelease(colorSpace);

//第二步 取每个点的像素值

unsignedchar* data = CGBitmapContextGetData (context);

if(data ==NULL)returnnil;

NSCountedSet*cls=[NSCountedSetsetWithCapacity:thumbSize.width*thumbSize.height];

for(intx=0; x

for(inty=0; y

intoffset =4*(x*y);

intred = data[offset];

intgreen = data[offset+1];

intblue = data[offset+2];

intalpha =  data[offset+3];

if(alpha>0) {//去除透明

if(red==255&&green==255&&blue==255) {//去除白色

}else{

NSArray*clr=@[@(red),@(green),@(blue),@(alpha)];

[clsaddObject:clr];

}

}

}

}

CGContextRelease(context);

//第三步 找到出现次数最多的那个颜色

NSEnumerator*enumerator = [clsobjectEnumerator];

NSArray*curColor =nil;

NSArray*MaxColor=nil;

NSUInteger MaxCount=0;

while( (curColor = [enumeratornextObject]) !=nil)

{

NSUInteger tmpCount = [clscountForObject:curColor];

if( tmpCount < MaxCount )continue;

MaxCount=tmpCount;

MaxColor=curColor;

}

return[UIColorcolorWithRed:([MaxColor[0]intValue]/255.0f)green:([MaxColor[1]intValue]/255.0f)blue:([MaxColor[2]intValue]/255.0f)alpha:([MaxColor[3]intValue]/255.0f)];

}

最后的效果也是特别明显的,如图:

源文章的url地址:http://www.cocoachina.com/bbs/read.php?tid=181490.

上一篇 下一篇

猜你喜欢

热点阅读