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
58
59
|
/*
* Copyright (C) 2014 ~ 2018 Deepin Technology Co., Ltd.
*
* Author: jouyouyun <jouyouwen717@gmail.com>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package graphic
import (
"fmt"
"image"
)
// GetDominantColorOfImage return the dominant hsv color of an image.
func GetDominantColorOfImage(imgfile string) (h, s, v float64, err error) {
img, err := LoadImage(imgfile)
if err != nil {
return
}
return doGetDominantColorOfImage(img)
}
func doGetDominantColorOfImage(img image.Image) (h, s, v float64, err error) {
// loop all points in image
var sumR, sumG, sumB, count uint64
mx := img.Bounds().Max.X
my := img.Bounds().Max.Y
count = uint64(mx * my)
if count == 0 {
err = fmt.Errorf("image is empty")
return
}
for x := 0; x < mx; x++ {
for y := 0; y < my; y++ {
c := img.At(x, y)
rr, gg, bb, _ := c.RGBA()
r, g, b := rr>>8, gg>>8, bb>>8
sumR += uint64(r)
sumG += uint64(g)
sumB += uint64(b)
}
}
h, s, v = Rgb2Hsv(uint8(sumR/count), uint8(sumG/count), uint8(sumB/count))
return
}
|
