想要實現一個功能，需要知道圖片的主題色，網上搜了很多都是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

defer {free(bitmapData)}

//建立一個位圖

let context =CGContext(data: bitmapData,

                                width: imgWidth,

                                height: imgHeight,

                                bitsPerComponent:8,

                                bytesPerRow: imgWidth *

                                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);

//第二步取每個點的畫素值

unsignedchar* data =CGBitmapContextGetData (context);

if (data == NULL)

returnnil;

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)];

}