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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
|
#!/bin/python3
import sys
from colorsys import hls_to_rgb, rgb_to_hls
from math import sqrt
light_colours = [
"dc8a78",
"dd7878",
"ea76cb",
"8839ef",
"d20f39",
"e64553",
"fe640b",
"df8e1d",
"40a02b",
"179299",
"04a5e5",
"209fb5",
"1e66f5",
"7287fd",
]
dark_colours = [
"f5e0dc",
"f2cdcd",
"f5c2e7",
"cba6f7",
"f38ba8",
"eba0ac",
"fab387",
"f9e2af",
"a6e3a1",
"94e2d5",
"89dceb",
"74c7ec",
"89b4fa",
"b4befe",
]
colour_names = [
"rosewater",
"flamingo",
"pink",
"mauve",
"red",
"maroon",
"peach",
"yellow",
"green",
"teal",
"sky",
"sapphire",
"blue",
"lavender",
"success",
"error",
]
def hex_to_rgb(hex: str) -> tuple[float, float, float]:
"""Convert a hex string to an RGB tuple in the range [0, 1]."""
return tuple(int(hex[i : i + 2], 16) / 255 for i in (0, 2, 4))
def rgb_to_hex(rgb: tuple[float, float, float]) -> str:
"""Convert an RGB tuple in the range [0, 1] to a hex string."""
return "".join(f"{round(i * 255):02X}" for i in rgb)
def hex_to_hls(hex: str) -> tuple[float, float, float]:
return rgb_to_hls(*hex_to_rgb(hex))
def hls_to_hex(h: str, l: str, s: str) -> str:
return rgb_to_hex(hls_to_rgb(h, l, s))
def adjust(hex: str, light: float = 0, sat: float = 0) -> str:
h, l, s = hex_to_hls(hex)
return hls_to_hex(h, max(0, min(1, l + light)), max(0, min(1, s + sat)))
def grayscale(hex: str, light: bool) -> str:
h, l, s = hex_to_hls(hex)
return hls_to_hex(h, min(0.5, l) if light else max(0.5, l), 0)
def distance(colour: str, base: str) -> float:
h1, l1, s1 = hex_to_hls(colour)
h2, l2, s2 = hex_to_hls(base)
return abs(h1 - h2) * 0.4 + abs(l1 - l2) * 0.3 + abs(s1 - s2) * 0.3
def smart_sort(colours: list[str], base: list[str]) -> list[str]:
sorted_colours = [None] * len(colours)
distances = {}
for colour in colours:
dist = [(i, distance(colour, b)) for i, b in enumerate(base)]
dist.sort(key=lambda x: x[1])
distances[colour] = dist
for colour in colours:
while len(distances[colour]) > 0:
i, dist = distances[colour][0]
if sorted_colours[i] is None:
sorted_colours[i] = colour, dist
break
elif sorted_colours[i][1] > dist:
old = sorted_colours[i][0]
sorted_colours[i] = colour, dist
colour = old
distances[colour].pop(0)
return [i[0] for i in sorted_colours]
def mix(a: str, b: str, w: float) -> str:
r1, g1, b1 = hex_to_rgb(a)
r2, g2, b2 = hex_to_rgb(b)
return rgb_to_hex(
(r1 * (1 - w) + r2 * w, g1 * (1 - w) + g2 * w, b1 * (1 - w) + b2 * w)
)
def mix_colours(colours: list[str], base: list[str], amount: float) -> list[str]:
for i, b in enumerate(base):
colours[i] = mix(colours[i], b, amount)
if __name__ == "__main__":
light = sys.argv[1] == "light"
scheme = sys.argv[2]
colours_in = sys.argv[3] if scheme == "monochrome" else sys.argv[4]
added_sat = 0.5 if light else 0.1
base = light_colours if light else dark_colours
colours = []
for colour in colours_in.split(" "):
colours.append(adjust(colour, sat=added_sat)) # TODO: optional adjust
colours = smart_sort(colours, base) # TODO: optional smart sort
mix_colours(colours, base, 0) # TODO: customize mixing from config
# Success and error colours # TODO: customize mixing
colours.append(mix(colours[8], base[8], 0.9)) # Success (green)
colours.append(mix(colours[4], base[4], 0.9)) # Error (red)
if scheme == "monochrome":
for i, colour in enumerate(colours):
colours[i] = grayscale(colour, light)
for i, colour in enumerate(colours):
print(f"{colour_names[i]} {colour}")
|
