<reponame>zendai/ImageSoup
from collections import Counter

import webcolors


CSS3_COLORS = webcolors.CSS3_NAMES_TO_HEX.keys()


def color_analysis(image):
    if image.mode != 'RGB':
        image = image.convert('RGB')

    # initialize a Counter starting from zero
    color_counter = Counter({color: 0 for color in Counter(CSS3_COLORS)})
    for pixel_count, RGB in image.getcolors(image.width * image.height):
        color_name = get_color_name(RGB)
        color_counter[color_name] += pixel_count

    # Calculate percent for each color
    for color in color_counter:
        pixel_count = image.width * image.height
        color_counter[color] = color_counter[color] / pixel_count

    return color_counter


def closest_color(requested_color):
    min_colors = {}
    for key, name in webcolors.css3_hex_to_names.items():
        r_c, g_c, b_c = webcolors.hex_to_rgb(key)
        rd = (r_c - requested_color[0]) ** 2
        gd = (g_c - requested_color[1]) ** 2
        bd = (b_c - requested_color[2]) ** 2
        min_colors[(rd + gd + bd)] = name
    return min_colors[min(min_colors.keys())]


def get_color_name(requested_color):
    try:
        color_name = webcolors.rgb_to_name(requested_color)
    except ValueError:
        color_name = closest_color(requested_color)
    return color_name

from collections import Counter

import webcolors


CSS3_COLORS = webcolors.CSS3_NAMES_TO_HEX.keys()


def color_analysis(image):
    if image.mode != 'RGB':
        image = image.convert('RGB')

    # initialize a Counter starting from zero
    color_counter = Counter({color: 0 for color in Counter(CSS3_COLORS)})
    for pixel_count, RGB in image.getcolors(image.width * image.height):
        color_name = get_color_name(RGB)
        color_counter[color_name] += pixel_count

    # Calculate percent for each color
    for color in color_counter:
        pixel_count = image.width * image.height
        color_counter[color] = color_counter[color] / pixel_count

    return color_counter


def closest_color(requested_color):
    min_colors = {}
    for key, name in webcolors.css3_hex_to_names.items():
        r_c, g_c, b_c = webcolors.hex_to_rgb(key)
        rd = (r_c - requested_color[0]) ** 2
        gd = (g_c - requested_color[1]) ** 2
        bd = (b_c - requested_color[2]) ** 2
        min_colors[(rd + gd + bd)] = name
    return min_colors[min(min_colors.keys())]


def get_color_name(requested_color):
    try:
        color_name = webcolors.rgb_to_name(requested_color)
    except ValueError:
        color_name = closest_color(requested_color)
    return color_name
# initialize a Counter starting from zero # Calculate percent for each color