tester.py

import json
import math

scoreThreshold = 20


class Color:
    color = None
    RGB_values = []

    def __init__(self, color, rgb_values=None):
        self.color = color
        if rgb_values is None:
            self.load()
        else:
            self.RGB_values = rgb_values

    def save(self):
        colors = None
        with open("colors.json", "r") as colorFile:
            colors = json.load(colorFile)

        with open("colors.json", "w") as colorFile:
            rgb_push = []
            for value in self.RGB_values:
                rgb_push.append({"R": value[0], "G": value[1], "B": value[2]})
            colors[self.color] = rgb_push

            json.dump(colors, colorFile, indent=4)

    def load(self):
        with open("colors.json", "r") as colorFile:
            colors = json.load(colorFile)
            self.RGB_values = []
            for value in colors[self.color]:
                self.RGB_values.append((value["R"], value["G"], value["B"]))

def getColor(rgb):
    # The lower score the better
    redScore = 0
    blackScore = 0
    yellowScore = 0

    def getScore(color, rgb):
        scores = []
        for db_rgb in color.RGB_values:
            print(str(rgb) + " - " + str(db_rgb))
            vec_diff = (rgb[0]-db_rgb[0], rgb[1]-db_rgb[1], rgb[2]-db_rgb[2])
            print(str(vec_diff))
            vec_length = math.sqrt(vec_diff[0]**2+vec_diff[1]**2+vec_diff[2]**2)
            print(vec_length)
            scores.append(vec_length)
            print("----------------")

        return sum(scores)/len(scores)

    # Get color scores
    print("=================")
    print("red")
    redScore = getScore(red, rgb)
    print(redScore)
    print("black")
    blackScore = getScore(black, rgb)
    print(blackScore)
    print("yellow")
    yellowScore = getScore(yellow, rgb)
    print(yellowScore)
    print("=================")

    # Get the smallest score
    score_dict = {"red": redScore, "black": blackScore, "yellow": yellowScore}
    lowest_score_value = min([redScore, blackScore, yellowScore])
    chosenColor = None
    if score_dict['red'] == lowest_score_value:
        chosenColor = red
    elif score_dict['black'] == lowest_score_value:
        chosenColor = black
    elif score_dict['yellow'] == lowest_score_value:
        chosenColor = yellowScore

    print(str(lowest_score_value) + "/" + str(scoreThreshold))

    if chosenColor and lowest_score_value <= scoreThreshold:
        print("Got the color: " + chosenColor.color)
        return chosenColor
    return None


print("Load")

# Colors
black = Color("black")
red = Color("red")
yellow = Color("yellow")

print(black.__dict__, red.__dict__, yellow.__dict__)




# getColor((11, 14, 16))
# getColor((45, 9, 8))
# getColor((166, 73, 31))


# print("Edit")
#
#
# print("Save")
#
# black.save()
# red.save()
# yellow.save()
#
# print(black.__dict__, red.__dict__, yellow.__dict__)