Zacharias-Brohn
124 lines
3.6 KiB
Python
124 lines
3.6 KiB
Python
import json
|
|
import argparse
|
|
from pathlib import Path
|
|
from PIL import Image
|
|
from materialyoucolor.quantize import QuantizeCelebi
|
|
from materialyoucolor.score.score import Score
|
|
from materialyoucolor.dynamiccolor.material_dynamic_colors import MaterialDynamicColors
|
|
from materialyoucolor.hct.hct import Hct
|
|
|
|
|
|
parser = argparse.ArgumentParser(
|
|
description="Generate color scheme from wallpaper image"
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--path",
|
|
required=True,
|
|
help="Path to the wallpaper image"
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--output",
|
|
required=True,
|
|
help="Path to save the color scheme JSON file"
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--thumbnail",
|
|
required=True,
|
|
help="Path to save the thumbnail image"
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--scheme",
|
|
required=False,
|
|
type=str,
|
|
default="vibrant",
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.scheme == 'fruit-salad':
|
|
from materialyoucolor.scheme.scheme_fruit_salad import SchemeFruitSalad as Scheme
|
|
elif args.scheme == 'expressive':
|
|
from materialyoucolor.scheme.scheme_expressive import SchemeExpressive as Scheme
|
|
elif args.scheme == 'monochrome':
|
|
from materialyoucolor.scheme.scheme_monochrome import SchemeMonochrome as Scheme
|
|
elif args.scheme == 'rainbow':
|
|
from materialyoucolor.scheme.scheme_rainbow import SchemeRainbow as Scheme
|
|
elif args.scheme == 'tonal-spot':
|
|
from materialyoucolor.scheme.scheme_tonal_spot import SchemeTonalSpot as Scheme
|
|
elif args.scheme == 'neutral':
|
|
from materialyoucolor.scheme.scheme_neutral import SchemeNeutral as Scheme
|
|
elif args.scheme == 'fidelity':
|
|
from materialyoucolor.scheme.scheme_fidelity import SchemeFidelity as Scheme
|
|
elif args.scheme == 'content':
|
|
from materialyoucolor.scheme.scheme_content import SchemeContent as Scheme
|
|
elif args.scheme == 'vibrant':
|
|
from materialyoucolor.scheme.scheme_vibrant import SchemeVibrant as Scheme
|
|
else:
|
|
from materialyoucolor.scheme.scheme_tonal_spot import SchemeTonalSpot as Scheme
|
|
|
|
def generate_thumbnail(image_path, thumbnail_path, size=(128, 128)):
|
|
thumbnail_file = Path(thumbnail_path)
|
|
|
|
image = Image.open(image_path)
|
|
image = image.convert("RGB")
|
|
image.thumbnail(size, Image.NEAREST)
|
|
|
|
thumbnail_file.parent.mkdir(parents=True, exist_ok=True)
|
|
image.save(thumbnail_path, "JPEG")
|
|
|
|
|
|
def generate_color_scheme(thumbnail_path, output_path):
|
|
image = Image.open(thumbnail_path)
|
|
pixel_len = image.width * image.height
|
|
image_data = image.getdata()
|
|
|
|
quality = 1
|
|
pixel_array = [image_data[_] for _ in range(0, pixel_len, quality)]
|
|
|
|
result = QuantizeCelebi(pixel_array, 128)
|
|
score = Score.score(result)[0]
|
|
|
|
scheme = Scheme(
|
|
Hct.from_int(score),
|
|
True,
|
|
0.0
|
|
)
|
|
|
|
color_dict = {}
|
|
for color in vars(MaterialDynamicColors).keys():
|
|
color_name = getattr(MaterialDynamicColors, color)
|
|
if hasattr(color_name, "get_hct"):
|
|
color_int = color_name.get_hct(scheme).to_int()
|
|
color_dict[color] = int_to_hex(color_int)
|
|
|
|
output_dict = {
|
|
"name": "dynamic",
|
|
"flavour": "default",
|
|
"mode": "dark",
|
|
"variant": "tonalspot",
|
|
"colors": color_dict
|
|
}
|
|
|
|
output_file = Path(output_path)
|
|
output_file.parent.mkdir(parents=True, exist_ok=True)
|
|
|
|
with open(output_file, "w") as f:
|
|
json.dump(output_dict, f, indent=4)
|
|
|
|
|
|
def int_to_hex(argb_int):
|
|
return "#{:06X}".format(argb_int & 0xFFFFFF)
|
|
|
|
|
|
try:
|
|
generate_thumbnail(args.path, str(args.thumbnail))
|
|
generate_color_scheme(str(args.thumbnail), args.output)
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
with open(args.output + ".txt", "w") as f:
|
|
f.write(f"Error: {e}")
|