mirror of
https://github.com/NicklasVraa/Color-manager.git
synced 2026-01-18 17:27:20 +01:00
531 lines
18 KiB
Python
531 lines
18 KiB
Python
# Desc: A program for recoloring icon packs, themes and wallpapers. For NovaOS.
|
|
# Auth: Nicklas Vraa
|
|
|
|
from typing import List, Set, Tuple, Dict, Optional
|
|
from tqdm import tqdm
|
|
from colormath.color_objects import sRGBColor, LabColor
|
|
from colormath.color_conversions import convert_color
|
|
from colormath.color_diff import delta_e_cie2000
|
|
from PIL import Image, ImageDraw
|
|
import os, re, shutil, json, subprocess
|
|
|
|
# Color conversion -------------------------------------------------------------
|
|
|
|
def hex_to_rgb(hex:str) -> Tuple[int,int,int]:
|
|
return int(hex[1:3], 16), int(hex[3:5], 16), int(hex[5:7], 16)
|
|
|
|
def hex_to_hsl(hex:str) -> Tuple[float,float,float]:
|
|
return rgb_to_hsl(hex_to_rgb(hex))
|
|
|
|
def hex_to_gray(hex:str) -> str:
|
|
r, g, b = hex_to_rgb(hex)
|
|
return '#' + format(int(0.21*r + 0.72*g + 0.07*b), '02x')*3
|
|
|
|
def rgb_to_hex(rgb:Tuple[int,int,int]) -> str:
|
|
r, g, b = rgb
|
|
return "#{:02x}{:02x}{:02x}".format(r, g, b)
|
|
|
|
def rgba_to_hex(rgba:Tuple[int,int,int,Optional[float]]) -> str:
|
|
r, g, b = rgba[:3]
|
|
hex = "#{:02X}{:02X}{:02X}".format(r, g, b)
|
|
|
|
if len(rgba) > 3:
|
|
if rgba[3] != 1.0:
|
|
hex += format(int(rgba[3] * 255), '02X')
|
|
|
|
return hex
|
|
|
|
def rgb_to_hsl(rgb:Tuple[int,int,int]) -> Tuple[float,float,float]:
|
|
r, g, b = rgb
|
|
r /= 255.0; g /= 255.0; b /= 255.0
|
|
max_val = max(r, g, b); min_val = min(r, g, b)
|
|
h = s = l = (max_val + min_val) / 2.0
|
|
|
|
if max_val == min_val:
|
|
h = s = 0
|
|
else:
|
|
d = max_val - min_val
|
|
s = d / (2.0 - max_val - min_val)
|
|
|
|
if max_val == r: h = (g - b) / d + (6.0 if g < b else 0.0)
|
|
elif max_val == g: h = (b - r) / d + 2.0
|
|
else: h = (r - g) / d + 4.0
|
|
h /= 6.0
|
|
|
|
return h, s, l
|
|
|
|
def rgb_to_gray(rgb:Tuple[int,int,int]) -> Tuple[int,int,int]:
|
|
r, g, b = rgb
|
|
weighed_avg = int(0.21*r + 0.72*g + 0.07*b)
|
|
return weighed_avg, weighed_avg, weighed_avg
|
|
|
|
def hsl_to_rgb(hsl:Tuple[float,float,float]) -> Tuple[int,int,int]:
|
|
h, s, l = hsl
|
|
|
|
if s == 0:
|
|
r = g = b = l
|
|
else:
|
|
if l < 0.5: q = l * (1 + s)
|
|
else: q = l + s - l * s
|
|
p = 2 * l - q
|
|
r = hue_to_rgb(p, q, h + 1 / 3)
|
|
g = hue_to_rgb(p, q, h)
|
|
b = hue_to_rgb(p, q, h - 1 / 3)
|
|
|
|
return int(round(r * 255)), int(round(g * 255)), int(round(b * 255))
|
|
|
|
def hue_to_rgb(p:float, q:float, t:float) -> float:
|
|
if t < 0: t += 1
|
|
if t > 1: t -= 1
|
|
if t < 1 / 6: return p + (q - p) * 6 * t
|
|
if t < 1 / 2: return q
|
|
if t < 2 / 3: return p + (q - p) * (2 / 3 - t) * 6
|
|
return p
|
|
|
|
def norm_hsl(h:int, s:int, l:int) -> Tuple[float,float,float]:
|
|
return h/360, s/100, l/100
|
|
|
|
# File conversion --------------------------------------------------------------
|
|
|
|
def svg_to_png(src_path:str, dest_path:str, width:int = 300) -> None:
|
|
"""Generate pngs from svgs with a given width."""
|
|
|
|
src_path = os.path.abspath(os.path.expanduser(src_path))
|
|
dest_path = os.path.abspath(os.path.expanduser(dest_path))
|
|
|
|
try:
|
|
subprocess.run(['inkscape', '--version'], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
except FileNotFoundError:
|
|
raise RuntimeError("Inkscape is not installed.")
|
|
|
|
os.makedirs(dest_path, exist_ok=True)
|
|
svgs = [file for file in os.listdir(src_path) if file.endswith('.svg')]
|
|
|
|
for svg in svgs:
|
|
svg_path = os.path.join(src_path, svg)
|
|
png = os.path.splitext(svg)[0] + '.png'
|
|
png_path = os.path.join(dest_path, png)
|
|
command = ['inkscape', svg_path, '-o', png_path, '-w', str(width)]
|
|
subprocess.run(command)
|
|
|
|
# CSS handling -----------------------------------------------------------------
|
|
|
|
def expand_css_rgba(match) -> None:
|
|
return rgba_to_hex((
|
|
int(match.group(1)), int(match.group(2)),
|
|
int(match.group(3)), float(match.group(4))
|
|
))
|
|
|
|
def css_to_hex(text:str):
|
|
text = re.sub(r"rgba\((\d+)\s*,\s*(\d+)\s*,\s*(\d+)\s*,\s*([\d.]+)\)",
|
|
lambda match: expand_css_rgba(match), text)
|
|
|
|
for key in name_to_hex_dict:
|
|
text = re.sub(key + "([,;])", name_to_hex_dict[key] + "\\1", text)
|
|
|
|
return text
|
|
|
|
def expand_hex(hex:str) -> str:
|
|
"""Returns 6-digit version of any hexadecimal color code."""
|
|
hex = hex.lstrip('#')
|
|
|
|
if len(hex) == 3:
|
|
hex = ''.join([c * 2 for c in hex])
|
|
|
|
return '#' + hex.ljust(6, '0')
|
|
|
|
# Color comparision ------------------------------------------------------------
|
|
|
|
def generate_palette_dict(colors:List[str]) -> Dict[str,LabColor]:
|
|
palette_dict = {}
|
|
|
|
for color in colors:
|
|
r, g, b = hex_to_rgb(color)
|
|
palette_dict[color] = convert_color(sRGBColor(r,g,b), LabColor)
|
|
|
|
return palette_dict
|
|
|
|
def load_json_file(path:str) -> Dict:
|
|
with open(path, 'r') as file:
|
|
palette = json.load(file)
|
|
return palette
|
|
|
|
def get_input_colors(resource) -> Tuple[List[str],bool,bool]:
|
|
"""Returns an HSL tuple, or a list of colors, depending on the input, as well as a boolean indicating which one, as well as if the palettes specifies smoothing pngs/jpgs."""
|
|
|
|
if isinstance(resource, tuple) and len(resource) == 3:
|
|
return resource, False, True
|
|
|
|
elif type(resource) is str:
|
|
palette_file = load_json_file(resource)
|
|
return generate_palette_dict(palette_file["colors"]), palette_file["smooth"], False
|
|
|
|
else:
|
|
return generate_palette_dict(resource["colors"]), palette_file["smooth"], False
|
|
|
|
def get_file_colors(file:str) -> Set[str]:
|
|
"""Return a set of all unique colors within a given string
|
|
representing an svg-file."""
|
|
|
|
colors = set()
|
|
matches = re.findall(r"#[A-Fa-f0-9]{6}", file)
|
|
|
|
for match in matches:
|
|
colors.add(match)
|
|
|
|
return colors
|
|
|
|
def closest_match(color:str, palette:Dict[str,LabColor]) -> str:
|
|
"""Compare the similarity of colors in the CIELAB colorspace. Return the
|
|
closest match, i.e. the palette entry with the smallest euclidian distance
|
|
to the given color."""
|
|
|
|
closest_color = None
|
|
min_distance = float('inf')
|
|
|
|
for entry in palette:
|
|
|
|
# Prior dictionary lookup and update.
|
|
lab_color = hex_to_lab_dict.get(color)
|
|
|
|
if lab_color is None:
|
|
r, g, b = hex_to_rgb(color)
|
|
lab_color = convert_color(sRGBColor(r,g,b), LabColor)
|
|
hex_to_lab_dict[color] = lab_color
|
|
|
|
distance = delta_e_cie2000(lab_color, palette[entry])
|
|
|
|
if distance < min_distance:
|
|
min_distance = distance
|
|
closest_color = entry
|
|
|
|
return closest_color
|
|
|
|
# Pack management --------------------------------------------------------------
|
|
|
|
def expand_path(path:str) -> str:
|
|
return os.path.abspath(os.path.expanduser(path))
|
|
|
|
def get_paths(folder: str, exts: List[str]) -> List[str]:
|
|
"""Return paths of every file with the given extensions within a folder
|
|
and its subfolders, excluding symbolic links."""
|
|
|
|
paths = []
|
|
|
|
for item in os.listdir(folder):
|
|
item_path = os.path.join(folder, item)
|
|
|
|
if os.path.islink(item_path): # Link.
|
|
continue
|
|
|
|
if os.path.isfile(item_path): # File.
|
|
for ext in exts:
|
|
if item.lower().endswith(ext):
|
|
paths.append(item_path)
|
|
|
|
elif os.path.isdir(item_path): # Folder.
|
|
subfolder_paths = get_paths(item_path, exts)
|
|
paths.extend(subfolder_paths)
|
|
|
|
return paths
|
|
|
|
def copy_file_structure(src_path:str, dest_path:str) -> None:
|
|
"""Copies a directory tree, but changes symbolic links to point
|
|
to files within the destination folder instead of the source.
|
|
Assumes that no link points to files outside the source folder."""
|
|
|
|
shutil.rmtree(dest_path, ignore_errors=True)
|
|
shutil.copytree(src_path, dest_path, symlinks=True)
|
|
|
|
for root, _, files in os.walk(dest_path):
|
|
for file in files:
|
|
file_path = os.path.join(root, file)
|
|
|
|
if os.path.islink(file_path):
|
|
link_target = os.readlink(file_path)
|
|
|
|
if not os.path.isabs(link_target):
|
|
continue
|
|
|
|
# Make link relative and update.
|
|
link_base = os.path.dirname(file_path)
|
|
relative_target = os.path.relpath(link_target, link_base)
|
|
os.remove(file_path)
|
|
|
|
# Replace root source folder with root destination folder.
|
|
relative_target = relative_target.replace(src_path, dest_path, 1)
|
|
os.symlink(relative_target, file_path)
|
|
|
|
def rename_pack(src_path, dest_path:str, name:str) -> None:
|
|
"""If an index.theme file exists within the given folder, apply
|
|
appropiate naming."""
|
|
|
|
index_path = os.path.join(dest_path, "index.theme")
|
|
|
|
print(index_path)
|
|
if os.path.exists(index_path):
|
|
with open(index_path, 'r') as file:
|
|
text = file.read()
|
|
|
|
text = re.sub(r"(Name=).*", "\\1" + name, text, count=1)
|
|
text = re.sub(r"(GtkTheme=).*", "\\1" + name, text, count=1)
|
|
text = re.sub(r"(MetacityTheme=).*", "\\1" + name, text, count=1)
|
|
text = re.sub(r"(IconTheme=).*", "\\1" + name, text, count=1)
|
|
|
|
text = re.sub(r"(Comment=).*", "\\1" + "A variant of " + os.path.basename(src_path) + " created by nicklasvraa/color-manager", text, count=1)
|
|
|
|
with open(index_path, 'w') as file:
|
|
file.write(text)
|
|
|
|
def copy_pack(src_path:str, dest_path:str, name:str) -> str:
|
|
"""Copy pack and return the resulting copy's directory path."""
|
|
|
|
src_path = expand_path(src_path)
|
|
dest_path = os.path.join(expand_path(dest_path), name)
|
|
|
|
copy_file_structure(src_path, dest_path)
|
|
rename_pack(src_path, dest_path, name)
|
|
|
|
return dest_path
|
|
|
|
# SVG/CSS recoloring -----------------------------------------------------------
|
|
|
|
def monochrome_vec(svg:str, colors:Set[str], hsl:Tuple[float,float,float]) -> str:
|
|
"""Replace every instance of color within the given list with their
|
|
monochrome equivalent in the given string representing an svg-file,
|
|
determined by the given hue, saturation and lightness offset."""
|
|
|
|
h, s, l_offset = hsl
|
|
|
|
if s == 0:
|
|
for color in colors:
|
|
graytone = hex_to_gray(color)
|
|
svg = re.sub(color, graytone, svg)
|
|
else:
|
|
l_offset = (l_offset - 0.5) * 2 # Remapping.
|
|
|
|
for color in colors:
|
|
graytone = hex_to_gray(color)
|
|
r, g, b = hex_to_rgb(graytone)
|
|
l = (0.21*r + 0.72*g + 0.07*b)/255
|
|
l = max(-1, min(l+l_offset, 1))
|
|
monochrome = rgb_to_hex(hsl_to_rgb((h, s, l)))
|
|
svg = re.sub(color, monochrome, svg)
|
|
|
|
return svg
|
|
|
|
def multichrome_vec(svg:str, colors:Set[str], new_colors:Dict[str,LabColor]) -> str:
|
|
"""Replace colors in a given svg/css with the closest match within a given
|
|
color palette."""
|
|
|
|
for color in colors:
|
|
new_color = closest_match(color, new_colors)
|
|
svg = re.sub(color, new_color, svg)
|
|
|
|
return svg
|
|
|
|
# PNG/JPG recoloring -----------------------------------------------------------
|
|
|
|
def monochrome_img(img:Image, hsl:Tuple[float,float,float]) -> Image:
|
|
"""Replace every instance of color within the given list with their
|
|
monochrome equivalent in the given image, determined by the given hue, saturation and lightness offset."""
|
|
|
|
mode = img.mode
|
|
h, s, l_offset = hsl
|
|
|
|
if s == 0:
|
|
if mode == "RGBA": img = img.convert("LA")
|
|
else: img = img.convert("L")
|
|
else:
|
|
width, height = img.size
|
|
l_offset = (l_offset - 0.5) * 2 # Remapping.
|
|
|
|
for x in range(width):
|
|
for y in range(height):
|
|
if mode == "RGBA":
|
|
r, g, b, a = img.getpixel((x, y))
|
|
else:
|
|
r, g, b = img.getpixel((x, y))
|
|
|
|
l = (0.21*r + 0.72*g + 0.07*b)/255
|
|
l = max(-1, min(l+l_offset, 1))
|
|
new_color = hsl_to_rgb((h, s, l))
|
|
|
|
if mode == "RGBA":
|
|
img.putpixel((x,y), new_color + (a,))
|
|
else:
|
|
img.putpixel((x,y), new_color)
|
|
|
|
return img
|
|
|
|
def multichrome_img(img:Image, new_colors:Dict[str,LabColor], smooth:bool) -> Image:
|
|
"""Replace colors in a given image with the closest match within a given
|
|
color palette."""
|
|
|
|
if smooth: img = img.convert("P", palette=Image.ADAPTIVE, colors=256)
|
|
else: img = img.convert("P")
|
|
|
|
palette = img.getpalette()
|
|
|
|
rgb_palette = [(palette[i], palette[i+1], palette[i+2]) for i in range(0, len(palette), 3)]
|
|
|
|
hex_palette = ["#%02x%02x%02x" % rgb for rgb in rgb_palette]
|
|
new_palette = []
|
|
|
|
for color in hex_palette:
|
|
new_color = hex_to_rgb(closest_match(color, new_colors))
|
|
new_palette.extend(new_color)
|
|
|
|
img.putpalette(new_palette)
|
|
return img
|
|
|
|
# User interface functions -----------------------------------------------------
|
|
|
|
def recolor(src_path:str, dest_path:str, name:str, replacement) -> None:
|
|
"""Recursively copies and converts a source folder into a destination, given a either a color or a palette."""
|
|
|
|
new_colors, smooth, is_mono = get_input_colors(replacement)
|
|
dest_path = copy_pack(src_path, dest_path, name)
|
|
svg_paths = get_paths(dest_path, [".svg"])
|
|
png_paths = get_paths(dest_path, [".png"])
|
|
jpg_paths = get_paths(dest_path, [".jpg", ".jpeg"])
|
|
css_paths = get_paths(dest_path, [".css", "rc"])
|
|
|
|
for path in tqdm(svg_paths, desc="svg", unit="file"):
|
|
with open(path, 'r') as file: x = file.read()
|
|
|
|
colors = get_file_colors(x)
|
|
|
|
if is_mono: x = monochrome_vec(x, colors, new_colors)
|
|
else: x = multichrome_vec(x, colors, new_colors)
|
|
|
|
with open(path, 'w') as file: file.write(x)
|
|
|
|
for path in tqdm(png_paths, desc="png", unit="file"):
|
|
x = Image.open(path)
|
|
x = x.convert("RGBA")
|
|
a = x.split()[3] # Save original alpha channel.
|
|
|
|
if is_mono: x = monochrome_img(x, new_colors)
|
|
else: x = multichrome_img(x, new_colors, smooth)
|
|
|
|
x = x.convert("RGBA")
|
|
r,g,b,_ = x.split()
|
|
x = Image.merge("RGBA",(r,g,b,a)) # Restore original alpha channel.
|
|
x.save(path)
|
|
|
|
for path in tqdm(jpg_paths, desc="jpg", unit="file"):
|
|
x = Image.open(path)
|
|
x = x.convert("RGB")
|
|
|
|
if is_mono: x = monochrome_img(x, new_colors)
|
|
else: x = multichrome_img(x, new_colors, smooth)
|
|
|
|
x = x.convert("RGB")
|
|
x.save(path)
|
|
|
|
for path in tqdm(css_paths, desc="css", unit="file"):
|
|
with open(path, 'r') as file: x = file.read()
|
|
|
|
x = css_to_hex(x)
|
|
colors = get_file_colors(x)
|
|
|
|
if is_mono: x = monochrome_vec(x, colors, new_colors)
|
|
else: x = multichrome_vec(x, colors, new_colors)
|
|
|
|
with open(path, 'w') as file: file.write(x)
|
|
|
|
def extract_colors(img_path:str, num_colors:int=8, save_path:str=None, pixels:int=50, cols:int=10) -> List[str]:
|
|
"""Returns and optionally saves the color palette of the given image, finding the specified number of colors."""
|
|
|
|
_, ext = os.path.splitext(img_path)
|
|
|
|
if ext == ".svg":
|
|
with open(img_path, 'r') as file:
|
|
svg = file.read()
|
|
|
|
colors = list(get_file_colors(svg))
|
|
num_colors = len(colors)
|
|
|
|
else:
|
|
img = Image.open(img_path)
|
|
|
|
colors = img.convert('P', palette=Image.ADAPTIVE, colors=num_colors)
|
|
colors = colors.getpalette()[0:num_colors*3]
|
|
|
|
colors = ['#{:02X}{:02X}{:02X}'.format(colors[i], colors[i+1], colors[i+2]) for i in range(0, len(colors), 3)]
|
|
|
|
if save_path != None:
|
|
if num_colors < cols: cols = num_colors
|
|
|
|
rows = -(-len(colors) // cols)
|
|
width = cols * pixels; height = rows * pixels
|
|
|
|
img = Image.new("RGBA", (width, height))
|
|
draw = ImageDraw.Draw(img)
|
|
|
|
for i, hex_color in enumerate(colors):
|
|
row = i // cols; col = i % cols
|
|
x0 = col * pixels; y0 = row * pixels
|
|
x1 = x0 + pixels; y1 = y0 + pixels
|
|
draw.rectangle([x0, y0, x1, y1], fill=hex_color)
|
|
|
|
img.save(save_path, format="png")
|
|
|
|
return colors
|
|
|
|
def clean_svg(input_path:str, output_path:str=None) -> str:
|
|
"""Removes needless metadata from svgs and optionally saves as copy, if output path is specified."""
|
|
|
|
with open(input_path, 'r') as f:
|
|
svg = f.read()
|
|
|
|
patterns = [
|
|
r".*xmlns:.*\n",
|
|
r"\s*<metadata[\s\S]*?<\/metadata.*",
|
|
r"\s*<sodipodi[\s\S]*?<\/sodipodi.*",
|
|
]
|
|
|
|
for pattern in patterns:
|
|
svg = re.sub(pattern, '', svg)
|
|
|
|
if output_path is None:
|
|
output_path = input_path
|
|
|
|
with open(output_path, 'w') as f:
|
|
f.write(svg)
|
|
|
|
def add_backdrop(src_path:str, dest_path:str, name:str, color:str="#000000", padding=0, rounding=0):
|
|
|
|
dest_path = copy_pack(src_path, dest_path, name)
|
|
svg_paths = get_paths(dest_path, [".svg"])
|
|
|
|
for path in tqdm(svg_paths, desc="Changing svgs ", unit="file"):
|
|
with open(path, 'r') as file:
|
|
svg = file.read()
|
|
|
|
width = int(re.search(r'<svg.*width=\"(\d*)\"', svg).group(1))
|
|
height = int(re.search(r'<svg.*height=\"(\d*)\"', svg).group(1))
|
|
pos = re.search(r'<svg.*>\n', svg).end()
|
|
|
|
backdrop = '<rect fill="' + color + '" x="' + str(padding) + '" y="' + str(padding) + '" width="' + str(width-2*padding) + '" height="' + str(height-2*padding) + '" rx="' + str(rounding * (width / 2)) + '" ry="' + str(rounding * (height / 2)) + '"/>'
|
|
|
|
credit = "\n<!-- Inserted by Color Manager -->\n"
|
|
svg = svg[:pos] + credit + backdrop + credit + svg[pos:]
|
|
|
|
with open(path, 'w') as file:
|
|
file.write(svg)
|
|
|
|
# Global constants -------------------------------------------------------------
|
|
|
|
this_path = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
# A dynamic dictionary to avoid multiple color conversions.
|
|
hex_to_lab_dict = {
|
|
"#ffffff": LabColor(9341.568974319263, -0.037058350415009045, -0.6906417562959177), # White.
|
|
"#000000": LabColor(0,0,0) # Black.
|
|
}
|
|
|
|
# A static dictionary of named colors from the css standard.
|
|
name_to_hex_dict = load_json_file(os.path.join(this_path, "named_colors.json"))
|