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
|
import collections
import numpy as np
import PIL.Image
import PIL.ImageDraw
import PIL.ImageFont
from .. import utils
def triangle(src, center, size, fill=None, outline=None):
"""Draw triangle on numpy array with Pillow.
Parameters
----------
src: numpy.ndarray
Input image.
center: (2,) array-like
center is (cy, cx).
size: float
Diameter to create the star.
fill: int or (3,) array-like, optional
RGB color to fill the mark. None for no fill. (default: None)
outline: int or (3,) array-like, optional
RGB color to draw the outline.
Returns
-------
dst: numpy.ndarray
Output image.
"""
dst = utils.numpy_to_pillow(src)
triangle_(img=dst, center=center, size=size, fill=fill, outline=outline)
return utils.pillow_to_numpy(dst)
def triangle_(img, center, size, fill=None, outline=None):
if isinstance(fill, collections.abc.Iterable):
fill = tuple(fill)
if isinstance(outline, collections.abc.Iterable):
outline = tuple(outline)
draw = PIL.ImageDraw.Draw(img)
radius = size / 2
cy, cx = center
x = cx + radius * np.cos(np.deg2rad(np.arange(0, 3) * 120 + 90))
y = cy - radius * np.sin(np.deg2rad(np.arange(0, 3) * 120 + 90))
xy = np.stack((x, y), axis=1)
xy = xy.flatten().tolist()
draw.polygon(xy, fill=fill, outline=outline)
|
