def _fill_annotations(a: Annotation, label: str, color=0xFF, out=None):
if out is None:
out = Image.new(mode='L', size=a.image.size)
brush = Brush(color)
d = Draw(out)
d.setantialias(False)
for o in a.iter_objects(label):
xy = a.points(o)
d.polygon(xy.flatten(), brush)
return d.flush()
def _stroke_annotations(a: Annotation,
label: str,
color=0xFF,
width=5,
out=None):
if out is None:
out = Image.new(mode='L', size=a.image.size)
pen = Pen(color, width)
d = Draw(out)
d.setantialias(False) # Should not antialias masks
for o in a.iter_objects(label):
xy = a.points(o)
d.polygon(xy.flatten(), pen)
return d.flush()
def test_path():
from aggdraw import Draw, Path
Path()
Path([0, 0])
Path([0, 0, 0, 0])
p = Path()
p.moveto(0, 0)
p.lineto(1, 1)
assert p.coords() == [0.0, 0.0, 1.0, 1.0]
p.curveto(0, 0, 0, 0, 0, 0)
p.close()
p.coords()
assert p.coords() == [0.0, 0.0, 1.0, 1.0, 0.125, 0.125, 0.0, 0.0]
draw = Draw("RGB", (800, 600))
draw.line(p)
draw.polygon(p)
draw.symbol((0, 0), p)
def cursor(color=None):
dc = Draw("RGBA", [32, 32], (0, 0, 0, 0))
if color is not None:
cursor_color = color[0:3]
else:
cursor_color = []
for c in P.default_fill_color:
cursor_color.append(abs(c - 255))
cursor_color = cursor_color[0:3]
# coordinate tuples are easier to read/modify but aggdraw needs a stupid x,y,x,y,x,y list
cursor_coords = [(6, 0), (6, 27), (12, 21), (18, 32), (20, 30), (15, 20),
(23, 20), (6, 0)]
cursor_xy_list = []
for point in cursor_coords:
cursor_xy_list.append(point[0])
cursor_xy_list.append(point[1])
brush = Brush(tuple(cursor_color), 255)
pen = Pen((255, 255, 255), 1, 255)
dc.polygon(cursor_xy_list, pen, brush)
cursor_surface = aggdraw_to_array(dc)
return cursor_surface
def test_path():
from aggdraw import Draw, Path
Path()
Path([0, 0])
Path([0, 0, 0, 0])
p = Path()
p.moveto(0, 0)
p.lineto(1, 1)
assert p.coords() == [0.0, 0.0, 1.0, 1.0]
p.curveto(2, 1, 2, 1, 2, 0)
p.close()
p.coords()
# Changed test w.r.t. aggdraw for agg 2.4
# Correctness of this may be seen in the file test-path-proof.ps
assert p.coords() == [0.0, 0.0, 1.0, 1.0, 1.625, 1.0, 2.0, 0.625, 2.0, 0.0]
draw = Draw("RGB", (800, 600))
draw.line(p)
draw.polygon(p)
draw.symbol((0, 0), p)
def test_graphics():
from aggdraw import Draw, Pen, Brush
draw = Draw("RGB", (500, 500))
pen = Pen("black")
brush = Brush("black")
draw.line((50, 50, 100, 100), pen)
draw.rectangle((50, 150, 100, 200), pen)
draw.rectangle((50, 220, 100, 270), brush)
draw.rectangle((50, 290, 100, 340), brush, pen)
draw.rectangle((50, 360, 100, 410), pen, brush)
draw.ellipse((120, 150, 170, 200), pen)
draw.ellipse((120, 220, 170, 270), brush)
draw.ellipse((120, 290, 170, 340), brush, pen)
draw.ellipse((120, 360, 170, 410), pen, brush)
draw.polygon((190+25, 150, 190, 200, 190+50, 200), pen)
draw.polygon((190+25, 220, 190, 270, 190+50, 270), brush)
draw.polygon((190+25, 290, 190, 340, 190+50, 340), brush, pen)
draw.polygon((190+25, 360, 190, 410, 190+50, 410), pen, brush)
from PIL import Image
from aggdraw import Draw, Brush
transBlack = (0, 0, 0, 0) # shows your example with visible edges
solidBlack = (0, 0, 0, 255) # shows shape on a black background
transWhite = (255, 255, 255, 0)
solidWhite = (255, 255, 255, 255)
im = Image.new("RGBA", (600, 600), solidBlack)
draw = Draw(im)
brush = Brush("yellow")
draw.polygon((
50,
50,
550,
60,
550,
550,
60,
550,
), None, brush)
draw.flush()
im.save("squar.png")
im.show()
class Canvas(CanvasBase):
# fonts appear smaller in aggdraw than with cairo
# fix that here:
fontScale = 1.2
def __init__(self,
img=None,
imageType=None, # determines file type
fileName=None, # if set determines output file name
size=None, ):
if img is None:
try:
import Image
except ImportError:
from PIL import Image
if size is None:
raise ValueError('please provide either an image or a size')
img = Image.new('RGBA', size, "white")
self.image = img
self.draw = Draw(img)
self.draw.setantialias(True)
if size is None:
self.size = self.draw.size
else:
self.size = size
if imageType and imageType not in ('png', 'jpg'):
raise ValueError('unsupported image type for agg canvas')
self.drawType = imageType
self.fileName = fileName
def _doLine(self, p1, p2, pen, **kwargs):
if kwargs.get('dash', (0, 0)) == (0, 0):
self.draw.line((p1[0], p1[1], p2[0], p2[1]), pen)
else:
dash = kwargs['dash']
pts = self._getLinePoints(p1, p2, dash)
currDash = 0
dashOn = True
while currDash < (len(pts) - 1):
if dashOn:
p1 = pts[currDash]
p2 = pts[currDash + 1]
self.draw.line((p1[0], p1[1], p2[0], p2[1]), pen)
currDash += 1
dashOn = not dashOn
def addCanvasLine(self, p1, p2, color=(0, 0, 0), color2=None, **kwargs):
if color2 and color2 != color:
mp = (p1[0] + p2[0]) / 2., (p1[1] + p2[1]) / 2.
color = convertColor(color)
self._doLine(p1, mp, Pen(color, kwargs.get('linewidth', 1)), **kwargs)
color2 = convertColor(color2)
self._doLine(mp, p2, Pen(color2, kwargs.get('linewidth', 1)), **kwargs)
else:
color = convertColor(color)
self._doLine(p1, p2, Pen(color, kwargs.get('linewidth', 1)), **kwargs)
def addCanvasText(self, text, pos, font, color=(0, 0, 0), **kwargs):
orientation = kwargs.get('orientation', 'E')
color = convertColor(color)
aggFont = Font(color, faceMap[font.face], size=font.size * self.fontScale)
blocks = list(re.finditer(r'\<(.+?)\>(.+?)\</\1\>', text))
w, h = 0, 0
supH = 0
subH = 0
if not len(blocks):
w, h = self.draw.textsize(text, aggFont)
tw, th = w, h
offset = w * pos[2]
dPos = pos[0] - w / 2. + offset, pos[1] - h / 2.
self.draw.text(dPos, text, aggFont)
else:
dblocks = []
idx = 0
for block in blocks:
blockStart, blockEnd = block.span(0)
if blockStart != idx:
# untagged text:
tblock = text[idx:blockStart]
tw, th = self.draw.textsize(tblock, aggFont)
w += tw
h = max(h, th)
dblocks.append((tblock, '', tw, th))
fmt = block.groups()[0]
tblock = block.groups()[1]
if fmt in ('sub', 'sup'):
lFont = Font(color, faceMap[font.face], size=0.8 * font.size * self.fontScale)
else:
lFont = aggFont
tw, th = self.draw.textsize(tblock, lFont)
w += tw
if fmt == 'sub':
subH = max(subH, th)
elif fmt == 'sup':
supH = max(supH, th)
else:
h = max(h, th)
dblocks.append((tblock, fmt, tw, th))
idx = blockEnd
if idx != len(text):
# untagged text:
tblock = text[idx:]
tw, th = self.draw.textsize(tblock, aggFont)
w += tw
h = max(h, th)
dblocks.append((tblock, '', tw, th))
supH *= 0.5
subH *= 0.5
h += supH + subH
offset = w * pos[2]
if orientation == 'W':
dPos = [pos[0] - w + offset, pos[1] - h / 2.]
elif orientation == 'E':
dPos = [pos[0] + offset, pos[1] - h / 2.]
else:
dPos = [pos[0] - w / 2. + offset, pos[1] - h / 2.]
if supH:
dPos[1] += supH
for txt, fmt, tw, th in dblocks:
tPos = dPos.copy()
if fmt == 'sub':
tPos[1] += subH
elif fmt == 'sup':
tPos[1] -= supH
if fmt in ('sub', 'sup'):
lFont = Font(color, faceMap[font.face], size=0.8 * font.size * self.fontScale)
else:
lFont = aggFont
self.draw.text(tPos, txt, lFont)
dPos[0] += tw
return (tw + th * .4, th + th * .4, offset)
def addCanvasPolygon(self, ps, color=(0, 0, 0), fill=True, stroke=False, **kwargs):
if not fill and not stroke:
return
dps = []
for p in ps:
dps.extend(p)
color = convertColor(color)
brush = None
pen = None
if fill:
brush = Brush(color)
if stroke:
pen = Pen(color)
self.draw.polygon(dps, pen, brush)
def addCanvasDashedWedge(self, p1, p2, p3, dash=(2, 2), color=(0, 0, 0), color2=None, **kwargs):
pen = Pen(color, kwargs.get('linewidth', 1))
dash = (3, 3)
pts1 = self._getLinePoints(p1, p2, dash)
pts2 = self._getLinePoints(p1, p3, dash)
if len(pts2) < len(pts1):
pts2, pts1 = pts1, pts2
for i in range(len(pts1)):
self.draw.line((pts1[i][0], pts1[i][1], pts2[i][0], pts2[i][1]), pen)
def flush(self):
self.draw.flush()
if self.fileName:
self.image.save(self.fileName)
class Canvas(CanvasBase):
# fonts appear smaller in aggdraw than with cairo
# fix that here:
fontScale=1.2
def __init__(self, img=None,
imageType=None, # determines file type
fileName=None, # if set determines output file name
size=None,
):
if img is None:
import Image
if size is None:
raise ValueError,'please provide either an image or a size'
img = Image.new('RGBA',size,"white")
self.image = img
self.draw = Draw(img)
self.draw.setantialias(True)
if size is None:
self.size = self.draw.size
else:
self.size = size
if imageType and imageType not in ('png','jpg'):
raise ValueError,'unsupported image type for agg canvas'
self.drawType=imageType
self.fileName=fileName
def _doLine(self, p1, p2, pen, **kwargs):
if kwargs.get('dash',(0,0)) == (0,0):
self.draw.line((p1[0],p1[1],p2[0],p2[1]),pen)
else:
dash = kwargs['dash']
pts = self._getLinePoints(p1,p2,dash)
currDash = 0
dashOn = True
while currDash<(len(pts)-1):
if dashOn:
p1 = pts[currDash]
p2 = pts[currDash+1]
self.draw.line((p1[0],p1[1],p2[0],p2[1]),pen)
currDash+=1
dashOn = not dashOn
def addCanvasLine(self, p1, p2, color=(0,0,0), color2=None, **kwargs):
if color2 and color2!=color:
mp = (p1[0]+p2[0])/2.,(p1[1]+p2[1])/2.
color = convertColor(color)
self._doLine(p1,mp,Pen(color,kwargs.get('linewidth',1)),**kwargs)
color2 = convertColor(color2)
self._doLine(mp,p2,Pen(color2,kwargs.get('linewidth',1)),**kwargs)
else:
color = convertColor(color)
self._doLine(p1,p2,Pen(color,kwargs.get('linewidth',1)),**kwargs)
def addCanvasText(self,text,pos,font,color=(0,0,0),**kwargs):
orientation=kwargs.get('orientation','E')
color = convertColor(color)
aggFont = Font(color,faceMap[font.face],size=font.size*self.fontScale)
blocks = list(re.finditer(r'\<(.+?)\>(.+?)\</\1\>',text))
w,h = 0,0
supH=0
subH=0
if not len(blocks):
w,h=self.draw.textsize(text,aggFont)
bw,bh=w*1.1,h*1.1
dPos = pos[0]-bw/2.,pos[1]-bh/2.
bgColor=kwargs.get('bgColor',(1,1,1))
bgColor = convertColor(bgColor)
self.draw.rectangle((dPos[0],dPos[1],dPos[0]+bw,dPos[1]+bh),
None,Brush(bgColor))
dPos = pos[0]-w/2.,pos[1]-h/2.
self.draw.text(dPos,text,aggFont)
else:
dblocks=[]
idx=0
for block in blocks:
blockStart,blockEnd=block.span(0)
if blockStart != idx:
# untagged text:
tblock = text[idx:blockStart]
tw,th=self.draw.textsize(tblock,aggFont)
w+=tw
h = max(h,th)
dblocks.append((tblock,'',tw,th))
fmt = block.groups()[0]
tblock = block.groups()[1]
if fmt in ('sub','sup'):
lFont = Font(color,faceMap[font.face],size=0.8*font.size*self.fontScale)
else:
lFont = aggFont
tw,th=self.draw.textsize(tblock,lFont)
w+=tw
if fmt == 'sub':
subH = max(subH,th)
elif fmt=='sup':
supH = max(supH,th)
else:
h = max(h,th)
dblocks.append((tblock,fmt,tw,th))
idx = blockEnd
if idx!=len(text):
# untagged text:
tblock = text[idx:]
tw,th=self.draw.textsize(tblock,aggFont)
w+=tw
h = max(h,th)
dblocks.append((tblock,'',tw,th))
supH *= 0.25
subH *= 0.25
h += supH + subH
bw,bh=w*1.1,h
#dPos = pos[0]-bw/2.,pos[1]-bh/2.
dPos = [pos[0]-w/2.,pos[1]-h/2.]
if orientation=='W':
dPos = [pos[0]-w,pos[1]-h/2.]
elif orientation=='E':
dPos = [pos[0],pos[1]-h/2.]
else:
dPos = [pos[0]-w/2,pos[1]-h/2.]
bgColor=kwargs.get('bgColor',(1,1,1))
bgColor = convertColor(bgColor)
self.draw.rectangle((dPos[0],dPos[1],dPos[0]+bw,dPos[1]+bh),
None,Brush(bgColor))
if supH: dPos[1]+=supH
for txt,fmt,tw,th in dblocks:
tPos = dPos[:]
if fmt=='sub':
tPos[1]+=subH
elif fmt=='sup':
tPos[1]-=supH
if fmt in ('sub','sup'):
lFont = Font(color,faceMap[font.face],size=0.8*font.size*self.fontScale)
else:
lFont = aggFont
self.draw.text(tPos,txt,lFont)
dPos[0]+=tw
def addCanvasPolygon(self,ps,color=(0,0,0),fill=True,stroke=False,**kwargs):
if not fill and not stroke: return
dps = []
for p in ps:
dps.extend(p)
color = convertColor(color)
brush=None
pen=None
if fill:
brush = Brush(color)
if stroke:
pen = Pen(color)
self.draw.polygon(dps,pen,brush)
def addCanvasDashedWedge(self,p1,p2,p3,dash=(2,2),color=(0,0,0),
color2=None,**kwargs):
pen = Pen(color,kwargs.get('linewidth',1))
dash = (3,3)
pts1 = self._getLinePoints(p1,p2,dash)
pts2 = self._getLinePoints(p1,p3,dash)
if len(pts2)<len(pts1): pts2,pts1=pts1,pts2
for i in range(len(pts1)):
self.draw.line((pts1[i][0],pts1[i][1],pts2[i][0],pts2[i][1]),pen)
def flush(self):
self.draw.flush()
if self.fileName:
self.image.save(self.fileName)
class Drawbject(object):
"""An abstract class that serves as the foundation for all KLDraw shapes. All Drawbjects
are drawn on an internal surface using the aggdraw drawing library, which can then be drawn
to the display buffer using blit() and displayed on the screen using flip(). For more
infomration on drawing in KLibs, please refer to the guide in the documentation.
Args:
width (int): The width of the shape in pixels.
height (int): The height of the shape in pixels.
stroke (List[width, Tuple[color], alignment]): The stroke of the shape, indicating
the width, color, and alignment (inner, center, or outer) of the stroke.
fill (Tuple[color]): The fill color for the shape expressed as an iterable of integer
values from 0 to 255 representing an RGB or RGBA color (e.g. (255,0,0,128)
for bright red with 50% transparency.)
rotation (int|float, optional): The degrees by which to rotate the Drawbject during
rendering. Defaults to 0.
Attributes:
stroke_color (None or Tuple[color]): The stroke color for the shape, expressed as an
iterable of integer values from 0 to 255 representing an RGB or RGBA color.
Defaults to 'None' if the shape has no stroke.
stroke_width (int): The stroke width for the in pixels. Defaults to '0' if the
shape has no stroke.
stroke_alignment (int): The stroke alignment for the shape (inner, center, or
outer). Defaults to '1' (STROKE_INNER) if the shape has no stroke.
fill_color (None or Tuple[color]): The fill color for the shape, expressed as an
iterable of integer values from 0 to 255 representing an RGB or RGBA color.
Defaults to 'None' if the shape has no fill.
opacity (int): The opacity of the shape, expressed as an integer from 0 (fully
transparent) to 255 (fully opaque).
object_width (int): The width of the shape in pixels.
object_height (int): The height of the shape in pixels.
surface_width (int): The width of the draw surface in pixels. At minimum two
pixels wider than the object_width (if no stroke or stroke is inner aligned),
at maximum (2 + 2*stroke_width) pixels wider than object width (if stroke is
outer aligned).
surface_height (int): The height of the draw surface in pixels. At minimum two
pixels wider than the object_height (if no stroke or stroke is inner aligned),
at maximum (2 + 2*stroke_height) pixels wider than object height (if stroke is
outer aligned).
surface (:obj:`aggdraw.Draw`): The aggdraw context on which the shape is drawn.
When a shape is drawn to the surface, it is immediately applied to the canvas.
canvas (:obj:`PIL.Image.Image`): The Image object that contains the shape of the
Drawbject before opacity has been applied. Initialized upon creation with a
size of (surface_width x surface_height).
rendered (None or :obj:`numpy.array`): The rendered surface containing the shape,
which is created using the render() method. If the Drawbject has not yet been
rendered, this attribute will be 'None'.
rotation (int): The rotation of the shape in degrees. Will be equal to 0 if no
rotation is set.
"""
transparent_brush = Brush((255, 0, 0), 0)
def __init__(self, width, height, stroke, fill, rotation=0):
super(Drawbject, self).__init__()
self.surface = None
self.canvas = None
self.rendered = None
self.__stroke = None
self.stroke_width = 0
self.stroke_color = None
self.stroke_alignment = STROKE_OUTER
self.stroke = stroke
self.__fill = None
self.fill_color = None
self.fill = fill
self.__dimensions = None
self.object_width = width
self.object_height = height
self.rotation = rotation
self._init_surface()
def __str__(self):
properties = [
self.__name__, self.surface_width, self.surface_height,
hex(id(self))
]
return "klibs.Drawbject.{0} ({1} x {2}) at {3}".format(*properties)
def _init_surface(self):
self._update_dimensions()
self.rendered = None # Clear any existing rendered texture
if self.fill_color:
if self.stroke_color and self.fill_color[3] == 255:
col = self.stroke_color
else:
col = self.fill_color
elif self.stroke_color:
col = self.stroke_color
else:
col = (0, 0, 0)
self.canvas = Image.new("RGBA", self.dimensions,
(col[0], col[1], col[2], 0))
self.surface = Draw(self.canvas)
self.surface.setantialias(True)
def render(self):
"""Pre-renders the shape so it can be drawn to the screen using
:func:`~klibs.KLGraphics.blit`. Although it is not necessary to pre-render
shapes before drawing them to the screen, it will make the initial blit faster
and is recommended wherever possible.
Once a Drawbject has been rendered, it will not need to be rendered again unless
any of its properties (e.g. stroke, fill, rotation) are changed.
Returns:
:obj:`~numpy.ndarray`: A numpy array of the rendered shape.
"""
self._init_surface()
self.draw()
self.rendered = asarray(self.canvas)
return self.rendered
def _update_dimensions(self):
pts = self._draw_points(outline=True)
if pts != None:
self.__dimensions = canvas_size_from_points(pts, flat=True)
else:
if self.stroke_alignment == STROKE_OUTER:
stroke_w = self.stroke_width * 2
elif self.stroke_alignment == STROKE_CENTER:
stroke_w = self.stroke_width
else:
stroke_w = 0
w, h = [self.object_width, self.object_height]
self.__dimensions = [
int(ceil(w + stroke_w)) + 2,
int(ceil(h + stroke_w)) + 2
]
@property
def dimensions(self):
"""List[int, int]: The height and width of the internal surface on which the shape
is drawn.
"""
return self.__dimensions
@property
def surface_width(self):
return self.__dimensions[0]
@property
def surface_height(self):
return self.__dimensions[1]
@property
def stroke(self):
"""None or :obj:`aggdraw.Pen`: An aggdraw Pen object set to the specified stroke width
and color, or None if the Drawbject has no stroke.
Raises:
ValueError: If an invalid stroke alignment value is passed to the stroke setter.
Valid values are 1 (STROKE_INNER), 2 (STROKE_CENTER), or 3 (STROKE_OUTER).
For the sake of clarity, it is recommended that you define stroke alignment
using the variable names provided in KLConstants (in brackets above).
"""
return self.__stroke
@stroke.setter
def stroke(self, style):
if not style:
self.stroke_width = 0
self.stroke_color = None
self.stroke_alignment = STROKE_OUTER
return self
try:
width, color, alignment = style
except ValueError:
width, color = style
alignment = STROKE_OUTER
if alignment in [STROKE_INNER, STROKE_CENTER, STROKE_OUTER]:
self.stroke_alignment = alignment
else:
raise ValueError(
"Invalid stroke alignment, see KLConstants for accepted values"
)
color = list(color)
if len(color) == 3:
color += [255]
self.stroke_color = color
self.stroke_width = width
self.__stroke = Pen(tuple(color[:3]), width, color[3])
if self.surface: # don't call this when initializing the Drawbject for the first time
self._init_surface()
return self
@property
def stroke_offset(self):
if self.stroke_alignment == STROKE_OUTER:
return self.stroke_width * 0.5
if self.stroke_alignment == STROKE_INNER:
return self.stroke_width * -0.5
else:
return 0
@property
def fill(self):
"""None or :obj:`aggdraw.Brush`: An aggdraw Brush object set to the specified fill
color, or None if the Drawbject has no fill.
"""
return self.__fill
@fill.setter
def fill(self, color):
if not color:
self.fill_color = None
return self
color = list(color)
if len(color) == 3:
color += [255]
self.fill_color = color
self.__fill = Brush(tuple(color[:3]), color[3])
if self.surface: # don't call this when initializing the Drawbject for the first time
self._init_surface()
return self
@abc.abstractmethod
def _draw_points(self, outline=False):
return None
@abc.abstractmethod
def draw(self):
pts = self._draw_points()
dx = self.surface_width / 2.0
dy = self.surface_height / 2.0
pts = translate_points(pts, delta=(dx, dy), flat=True)
self.surface.polygon(pts, self.stroke, self.fill)
self.surface.flush()
return self.canvas
@abc.abstractproperty
def __name__(self):
pass
from PIL import Image
from aggdraw import Draw, Brush
transBlack = (0, 0, 0, 0) # shows your example with visible edges
solidBlack = (0, 0, 0, 255) # shows shape on a black background
transWhite = (255, 255, 255, 0)
solidWhite = (255, 255, 255, 255)
im = Image.new("RGBA", (600, 600), solidBlack)
draw = Draw(im)
brush = Brush("yellow")
draw.polygon(
(
50, 50,
550, 60,
550, 550,
60, 550,
),
None, brush
)
draw.flush()
im.save("squar.png")
im.show()
