mirror of
https://codeberg.org/JasterV/Color-Plotter.git
synced 2026-04-26 18:10:05 +00:00
85 lines
2.1 KiB
Python
85 lines
2.1 KiB
Python
import requests
|
|
import json
|
|
from vector_utils import Vector as vec
|
|
from mpl_toolkits import mplot3d
|
|
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
import math
|
|
|
|
def get_content(url):
|
|
response = requests.get(url)
|
|
content = response.content.decode()
|
|
return content
|
|
|
|
|
|
def get_json(url):
|
|
content = get_content(url)
|
|
data = json.loads(content)
|
|
return data
|
|
|
|
|
|
def hex_to_rgb(s):
|
|
s = s.lstrip('#')
|
|
return [int(s[:2], 16), int(s[2:4], 16), int(s[4:6], 16)]
|
|
|
|
|
|
def parse_colors(data_set):
|
|
return dict((item['color'], hex_to_rgb(item['hex'])) for item in data_set['colors'])
|
|
|
|
|
|
def calculate_size(dist):
|
|
pass
|
|
|
|
|
|
def closest(colors, to, n):
|
|
closest_colors = ((c, vec.distance(to, c)) for c in colors.values())
|
|
return sorted(closest_colors, key=lambda c: c[1])[:n]
|
|
|
|
|
|
def plot_data(ax, values, colors, sizes):
|
|
xdata, ydata, zdata = zip(*values)
|
|
ax.scatter3D(xdata, ydata, zdata, c=colors /
|
|
255.0, s=sizes, depthshade=False)
|
|
|
|
|
|
def plot_closest(ax, colors, to):
|
|
distances = closest(colors, to, n=len(colors)//2)
|
|
max_distance = max(distances, key=lambda x: x[1])[1]
|
|
sizes = np.array(
|
|
list(map(lambda x: (math.log2(max_distance + 1) - math.log2(x[1] + 1))*100, distances)))
|
|
rgb_values = list(map(lambda x: x[0], distances))
|
|
plot_data(ax, rgb_values, np.array(rgb_values), sizes)
|
|
|
|
|
|
def init_projection(size=(13, 10)):
|
|
plt.figure(figsize=size, dpi=90)
|
|
ax = plt.axes(projection='3d')
|
|
ax.set_axis_off()
|
|
return ax
|
|
|
|
|
|
def print_colors(colors):
|
|
print("\n-----COLOR LIST-----")
|
|
for color in colors.keys():
|
|
print(color)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
data_set_url = "https://raw.githubusercontent.com/dariusk/corpora/master/data/colors/xkcd.json"
|
|
data_set = get_json(data_set_url)
|
|
colors = parse_colors(data_set)
|
|
|
|
print_colors(colors)
|
|
|
|
color = input(
|
|
"\nEnter a color from the color list or just enter an hexadecimal value: ")
|
|
|
|
if color in colors:
|
|
color = colors[color]
|
|
else:
|
|
color = hex_to_rgb(color)
|
|
|
|
ax = init_projection()
|
|
plot_closest(ax, colors, color)
|
|
plt.show()
|
|
|