//本地圖片資源
let url = 'tree.webp'
document.getElementById('originalImage').src = url
let themeColor = new ThemeColor(url,(shrinkUrl,color)=>{
    //縮略圖
    let img = document.getElementById('showImage')
    if(img){
        img.setAttribute('src',shrinkUrl)
    }
    //主色
    document.getElementById('showDiv').style.backgroundColor = color
})

class ThemeColor{
    // 原圖資源
    imgUrl = ''
    // 像素集合
    originalPiexls = null
    // 縮略圖
    shrinkUrl = ''
    // 主色
    themeColor = 'white'
    // 回調(diào)
    themeColorCallBack = null
    // 提取像素出現(xiàn)最大次數(shù)操作對(duì)象
    colorCountedSet = new ColorCountedSet()
    
    constructor(imgUrl,callBack){
        this.imgUrl = imgUrl
        this.themeColorCallBack = callBack
        this.startScreeningThemeColor()
    }
    
    // 開(kāi)始解析主色
    async startScreeningThemeColor(){
        try {
            await this.shrinkImage()
        } catch (error) {
            console.log('error:' + error)
        }
        this.screeningThemeColor()
    }

    // 圖片縮小
    async shrinkImage(){
        var image = new Image();
        image.src = this.imgUrl;
        await new Promise((resolve)=>{
            image.onload = resolve
        })
        let width = image.width
        let height = image.height
        let shrinkFactor = 10
        let shrinkWidth = width / shrinkFactor
        let shrinkHeight = height / shrinkFactor
        let canvas = document.createElement('canvas')
        canvas.setAttribute('width',`${shrinkWidth}px`)
        canvas.setAttribute('height',`${shrinkHeight}px`)
        var ctx = canvas.getContext("2d")
        ctx.drawImage(image,0,0,shrinkWidth,shrinkHeight)
        this.shrinkUrl = canvas.toDataURL('image/jpeg',1)
        try {
            //保存像素
            this.originalPiexls = ctx.getImageData(0,0,width,height)
        } catch (error) {
            console.log(error)
        }
    }

    // 開(kāi)始篩選主題色
    screeningThemeColor(){
        if(!this.originalPiexls || !this.originalPiexls.data || this.originalPiexls.data.length == 0){
            throw('像素為空')
        }
        for(let i = 0;i < this.originalPiexls.data.length;i+=4){
            let r = this.originalPiexls.data[i]
            let g = this.originalPiexls.data[i + 1]
            let b = this.originalPiexls.data[i + 2]
            let a = this.originalPiexls.data[i + 3] / 255.0
            //添加一個(gè)色值范圍，讓它能忽略一定無(wú)效的像素值
            if(a > 0 && (r < 200 && g < 200 && b < 200) && (r > 50 && g > 50 && b > 50)){
                this.colorCountedSet.push(r,g,b,a)
            }
        }

        let maxCount = 0
        // 尋找出現(xiàn)次數(shù)最多的像素定為主色調(diào)
        this.colorCountedSet.forEach((value,key)=>{
            if(maxCount <= value){
                maxCount = value
                this.themeColor = 'rgba(' + key + ')'
            }
        })
        //執(zhí)行回調(diào)
        if(this.themeColorCallBack){
            this.themeColorCallBack(this.shrinkUrl,this.themeColor)
        }
    }
 }

// 統(tǒng)計(jì)不同像素的出現(xiàn)次數(shù)
class ColorCountedSet{
    //像素集合
    map = new Map()
    
    //添加像素到集合
    push(r,g,b,a){
        //根據(jù)像素值生成一個(gè)map 元素 key值
        let identification = r + ',' + g + ',' + b + ',' + a
        if(!this.map.get(identification)){
            this.map.set(identification,1)
        } else {
            // 存在進(jìn)行次數(shù)自增
            let times = parseInt(this.map.get(identification)) + 1
            this.map.set(identification,times)
        }
    }

// 給 ColorCountedSet 操作類(lèi)添加一個(gè) forEach 方法 
forEach(cb){
        this.map.forEach(function(value,key){
                cb(value,key)
        });
    }
}