117 lines
4.1 KiB
Python
117 lines
4.1 KiB
Python
import itertools
|
|
import json
|
|
import random
|
|
from collections import Counter
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
|
|
|
|
def make_wallpaper(
|
|
background: str,
|
|
speckle_colours: list[str],
|
|
filename: str | Path,
|
|
density: int = 0.6,
|
|
dimensions: tuple[float | int, float | int] = (1920, 1080),
|
|
) -> None:
|
|
x, y = dimensions
|
|
speckles_per_colour = x / 100 * y / 100 * density
|
|
|
|
fig, ax = plt.subplots(figsize=(x / 100, y / 100), facecolor=background)
|
|
ax.set_facecolor(background)
|
|
[spine.set_color(background) for spine in ax.spines.values()]
|
|
plt.xticks([])
|
|
plt.yticks([])
|
|
plt.margins(0, 0)
|
|
|
|
for color, size in itertools.product(
|
|
speckle_colours,
|
|
np.logspace(1, 6, 8, base=2),
|
|
):
|
|
plt.scatter(
|
|
[random.random() * x / 8 for _ in range(speckles_per_colour)],
|
|
[random.random() * y / 8 for _ in range(speckles_per_colour)],
|
|
c=color,
|
|
s=size,
|
|
)
|
|
|
|
plt.tight_layout()
|
|
# plt.xlim(0, x)
|
|
# plt.ylim(0, y)
|
|
# plt.axis("off")
|
|
plt.savefig(
|
|
filename,
|
|
dpi=128,
|
|
bbox_inches="tight",
|
|
pad_inches=0,
|
|
)
|
|
# plt.show()
|
|
plt.close()
|
|
|
|
|
|
# palette = random.choice(palettes)
|
|
def speckles(palette):
|
|
for i, background in enumerate(palette):
|
|
speckle_colours = palette[:i] + palette[i + 1 :]
|
|
_id = "_".join(speckle_colours).replace("#", "")
|
|
speckle_colours += ["#000000", "#000000", "#ffffff"]
|
|
make_wallpaper(
|
|
background,
|
|
speckle_colours,
|
|
f"speckles/speckles_{_id}.png",
|
|
)
|
|
|
|
|
|
def night_sky(palette):
|
|
speckle_colours = palette + ["#ffffff"]
|
|
_id = "_".join(palette).replace("#", "")
|
|
make_wallpaper(
|
|
"#000000",
|
|
speckle_colours,
|
|
f"speckles/night_sky_{_id}.png",
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
palettes = [
|
|
{"colours": ["#eee4ab", "#e5cb9f", "#99c4c8", "#68a7ad"], "votes": 100000},
|
|
{"colours": ["#e4d192", "#cba0ae", "#af7ab3", "#80558c"], "votes": 100000},
|
|
{"colours": ["#eeeeee", "#e1d4bb", "#cbb279", "#537188"], "votes": 100000},
|
|
{"colours": ["#c0dbea", "#ba90c6", "#e8a0bf", "#fdf4f5"], "votes": 100000},
|
|
{"colours": ["#f5ffc9", "#b3e5be", "#a86464", "#804674"], "votes": 100000},
|
|
{"colours": ["#ffde7d", "#f6416c", "#f8f3d4", "#00b8a9"], "votes": 100000},
|
|
{"colours": ["#53354a", "#903749", "#e84545", "#2b2e4a"], "votes": 100000},
|
|
{"colours": ["#967e76", "#d7c0ae", "#eee3cb", "#b7c4cf"], "votes": 100000},
|
|
{"colours": ["#fc5185", "#f5f5f5", "#3fc1c9", "#364f6b"], "votes": 100000},
|
|
{"colours": ["#eaeaea", "#ff2e63", "#252a34", "#08d9d6"], "votes": 100000},
|
|
{"colours": ["#eeeeee", "#00adb5", "#393e46", "#222831"], "votes": 100000},
|
|
{"colours": ["#2cd3e1", "#a459d1", "#f266ab", "#ffb84c"], "votes": 100000},
|
|
{"colours": ["#ffe194", "#e8f6ef", "#1b9c85", "#4c4c6d"], "votes": 100000},
|
|
{"colours": ["#146c94", "#19a7ce", "#b0daff", "#feff86"], "votes": 100000},
|
|
{"colours": ["#4c3d3d", "#c07f00", "#ffd95a", "#fff7d4"], "votes": 100000},
|
|
{"colours": ["#8bacaa", "#b04759", "#e76161", "#f99b7d"], "votes": 100000},
|
|
{"colours": ["#146c94", "#19a7ce", "#afd3e2", "#f6f1f1"], "votes": 100000},
|
|
{"colours": ["#9ba4b5", "#212a3e", "#394867", "#f1f6f9"], "votes": 100000},
|
|
{"colours": ["#00ffca", "#05bfdb", "#088395", "#0a4d68"], "votes": 100000},
|
|
{"colours": ["#7c9070", "#9ca777", "#fee8b0", "#f97b22"], "votes": 100000},
|
|
{"colours": ["#002a19", "#000000", "#ffffff", "#e0512f"], "votes": 100000},
|
|
]
|
|
|
|
palette_files = ["colorhunt.json"]
|
|
for palette_file in palette_files:
|
|
with open(palette_file, "rt") as f:
|
|
palettes += json.load(f)
|
|
|
|
print(len(palettes))
|
|
palettes = [
|
|
palette["colours"]
|
|
for palette in palettes
|
|
if palette["votes"] > 30000 or len(palette["colours"]) > 5
|
|
]
|
|
print(len(palettes))
|
|
for palette in palettes:
|
|
night_sky(palette)
|
|
speckles(palette)
|