美文网首页iOS 识别技术
iOS代码获取图片主色调

iOS代码获取图片主色调

作者: Skylpy | 来源:发表于2017-09-25 09:43 被阅读70次

    最近的开发过程中,有一个功能是根据银行的图标来确定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.

    相关文章

      网友评论

        本文标题:iOS代码获取图片主色调

        本文链接:https://www.haomeiwen.com/subject/mxerextx.html