获取图片的主题色Swift和OC

想要实现一个功能，需要知道图片的主题色，网上搜了很多都是OC写的，Swift写的几乎没有，自己找了很多博客和文档，对UIImage做了extension，重写了一个mostColor的属性get方法。方便直接调用。有更好的办法希望大家多多分享，欢迎留言。为了方便大家，那我把OC写的也放在下边了

需要注明！！！！！

我写的这个swift版本计算主题色，为了追求精确度，计算了所有的，如果你觉得得到的结果很慢，一般主题色也不会要求很精确可以这样改：

在这个for循环的地方

改成

let offSet = x * y *4

就OK啦

Swift

extensionUIImage {

   var mostColor:UIColor {

       //获取图片信息

       let imgWidth:Int =Int(self.size.width) /2

       let imgHeight:Int =Int(self.size.height) /2

        

       //位图的大小＝图片宽＊图片高＊图片中每点包含的信息量

       let bitmapByteCount = imgWidth * imgHeight *4

        

       //使用系统的颜色空间

       let colorSpace =CGColorSpaceCreateDeviceRGB()

        

       //根据位图大小，申请内存空间

       let bitmapData =malloc(bitmapByteCount)

       defer {free(bitmapData)}

        

       //创建一个位图

       let context =CGContext(data: bitmapData,

                                width: imgWidth,

                                height: imgHeight,

                                bitsPerComponent:8,

                                bytesPerRow: imgWidth *4,

                                space: colorSpace,

                                bitmapInfo:CGImageAlphaInfo.premultipliedLast.rawValue)

        

       //图片的rect

       let rect =CGRect(x:0, y:0, width:CGFloat(imgWidth), height:CGFloat(imgHeight))

       //将图片画到位图中

        context?.draw(self.cgImage!, in: rect)

        

       //获取位图数据

       let bitData = context?.data

       let data =unsafeBitCast(bitData, to:UnsafePointer<CUnsignedChar>.self)

 

       let cls =NSCountedSet.init(capacity: imgWidth * imgHeight)

       for xin0..<imgWidth {

           for yin0..<imgHeight {

               let offSet = (y * imgWidth + x) *4

               let r = (data+ offSet    ).pointee

               let g = (data+ offSet+1).pointee

               let b = (data+ offSet+2).pointee

               let a = (data+ offSet+3).pointee

               if a >0 {

                   //去除透明

                   if!(r ==255&& g ==255&& b ==255) {

                       //去除白色

                        cls.add([CGFloat(r),CGFloat(g),CGFloat(b),CGFloat(a)])

                    }

                }

                

            }

        }

        

       //找到出现次数最多的颜色

       let enumerator = cls.objectEnumerator()

       var maxColor:Array<CGFloat>? =nil

       var maxCount =0

       whilelet curColor = enumerator.nextObject() {

           let tmpCount = cls.count(for: curColor)

           if tmpCount >= maxCount{

                maxCount = tmpCount

                maxColor = curColoras?Array<CGFloat>

            }

        }

       returnUIColor.init(red: (maxColor![0] /255), green: (maxColor![1] /255), blue: (maxColor![2] /255), alpha: (maxColor![3] /255))

    }

   

}

OC

- (UIColor*)mainColor{

    

    int bitmapInfo = kCGBitmapByteOrderDefault | kCGImageAlphaPremultipliedLast;

    

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

    CGSize thumbSize =CGSizeMake(self.size.width /2, self.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,self.CGImage);

    CGColorSpaceRelease(colorSpace);

    

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

    unsigned char* data =CGBitmapContextGetData (context);

    if (data == NULL)

        return nil;

    

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

    

    for (int x =0; x < thumbSize.width; x++) {

        for (int y =0; y < thumbSize.height; y++) {

//            int offset = 4 * (x + y * thumbSize.width);

            int offset =4 * x * y;

            int red = data[offset];

            int green = data[offset +1];

            int blue = data[offset +2];

            int alpha =  data[offset +3];

            if (alpha >0) {

                //去除透明

                if (!(red ==255 && green ==255 && blue ==255)) {

                    //去除白色

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

                    [cls addObject: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)];

}