wallpaper/prim_wallpaper/make_wallpaper.py

import itertools

from PIL import Image


def make_wallpaper(a, b, c):
    out = Image.new("RGB", (1 << 12, 1 << 12))
    prim_a = (pow(a, 257, i) - 1 for i in range(1, 256 + 1))
    prim_b = (pow(b, 257, i) - 1 for i in range(1, 256 + 1))
    prim_c = (pow(c, 257, i) - 1 for i in range(1, 256 + 1))
    rgb_values = itertools.product(prim_a, prim_b, prim_c)
    for i, rgb in enumerate(rgb_values):
        y, x = divmod(i, 1 << 12)
        out.putpixel((x, y), rgb)
    return out