aboutsummaryrefslogtreecommitdiff
path: root/pywal/backends/colorz.py
blob: 776678e727d6dc234595c4b247dcd82159f72401 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
"""
Generate a colorscheme using Colorz.
"""
import shutil
import subprocess
import sys

from .. import util


def gen_colors(img):
    """Generate a colorscheme using Colorz."""
    cmd = ["colorz", "-n", "6", "--bold", "0", "--no-preview"]
    return subprocess.check_output([*cmd, img]).splitlines()


def adjust(colors, light):
    """Create palette."""
    # Create list with placeholder values.
    raw_colors = ["#000000", *colors, "#FFFFFF",
                  "#333333", *colors, "#FFFFFF"]

    # Update placeholder values.
    if light:
        for color in raw_colors:
            color = util.saturate_color(color, 0.50)
            color = util.darken_color(color, 0.4)

        raw_colors[0] = util.lighten_color(colors[0], 0.9)
        raw_colors[7] = util.darken_color(colors[0], 0.75)
        raw_colors[8] = util.darken_color(colors[0], 0.25)
        raw_colors[15] = raw_colors[7]

    else:
        raw_colors[0] = util.darken_color(colors[0], 0.75)
        raw_colors[7] = util.lighten_color(colors[0], 0.75)
        raw_colors[8] = util.darken_color(colors[0], 0.25)
        raw_colors[15] = raw_colors[7]

    return raw_colors


def get(img, light=False):
    """Get colorscheme."""
    if not shutil.which("colorz"):
        print("error: Colorz wasn't found on your system.",
              "Try another backend. (wal --backend)")
        sys.exit(1)

    colors = gen_colors(img)
    colors = [color.decode('UTF-8').split()[0] for color in colors]
    return adjust(colors, light)