pixel_art/pixel_art.py at master · Fel1-x/pixel_art · GitHub

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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
def define_bee():
    import numpy as np

    # make an empty numpy array for storing the image
    image_mat = np.full((16, 16, 3), 1.0)

    # define some colours
    black = [0, 0, 0]
    yellow = [1.0, 0.85, 0]
    pink = [1, 0.753, 0.796]
    grey = [0.65] * 3

    # specify which pixels are which colour
    image_mat[7:11, 2] = pink
    image_mat[6:12, 3:5] = pink
    image_mat[6:12, 5:7] = yellow
    image_mat[6:12, 7:9] = black
    image_mat[6:12, 9:11] = yellow
    image_mat[6:12, 11:13] = pink
    image_mat[7:11, 13] = pink
    image_mat[4:6, 5:11] = grey
    image_mat[3, 6:10] = grey

    return image_mat


def define_butterfly():
    import numpy as np

    # make an empty numpy array for storing the image
    image_mat = np.full((16, 16, 3), 1.0)

    # define some colours
    black = [0, 0, 0]
    other = [0.2, 0.4, 0.8]
    other = [0.5, 0.4, 0.2]

    # specify which pixels are which colour
    c = 8
    # centre of butterfly
    image_mat[2:13, c] = black

    # generate mirror image at the same time
    for i in [-1, 1]:
        # outline
        image_mat[1, i*1+c] = black
        image_mat[0, i*2+c] = black
        image_mat[4, i*1+c] = black
        image_mat[3, [i*j+c for j in [2, 3]]] = black
        image_mat[2, [i*j+c for j in [4, 5]]] = black
        image_mat[3, i*6+c] = black
        image_mat[4:7, i*7+c] = black
        image_mat[7, [i*j+c for j in [6, 5]]] = black
        image_mat[8, [i*j+c for j in [4, 3, 2]]] = black
        image_mat[7, i*1+c] = black
        image_mat[9, i*5+c] = black
        image_mat[10:12, i*6+c] = black
        image_mat[12, i*5+c] = black
        image_mat[13, [i*j+c for j in [4, 3]]] = black
        image_mat[12, i*2+c] = black
        image_mat[11, i*1+c] = black

        # fill in the centre line by line
        image_mat[3, [i*j+c for j in [4, 5]]] = other
        image_mat[4, [i*j+c for j in range(2, 7)]] = other
        image_mat[5:7, [i*j+c for j in range(1, 7)]] = other
        image_mat[7, [i*j+c for j in range(2, 5)]] = other
        image_mat[8, i*1+c] = other
        image_mat[9, [i*j+c for j in range(1, 5)]] = other
        image_mat[10, [i*j+c for j in range(1, 6)]] = other
        image_mat[11, [i*j+c for j in range(2, 6)]] = other
        image_mat[12, [i*j+c for j in range(3, 5)]] = other

    return image_mat


def plot_image(image):
    import matplotlib.pyplot as plt

    plt.axis('off')
    plt.imshow(image)
    plt.show()


bee = define_bee()
plot_image(bee)
butterfly = define_butterfly()
plot_image(butterfly)