ProjektAI/venv/Lib/site-packages/pygame/draw_py.py
2020-04-20 22:51:16 +02:00

540 lines
18 KiB
Python

'''Pygame Drawing algorithms written in Python. (Work in Progress)
Implement Pygame's Drawing Algorithms in a Python version for testing
and debugging.
'''
from __future__ import division
import sys
if sys.version_info >= (3, 0, 0):
from math import floor, ceil
else:
# Python2.7
# FIXME : the import of the builtin math module is broken ...
def floor(x):
int_x = int(x)
return int_x if (x == int_x or x > 0) else int_x - 1
def ceil(x):
int_x = int(x)
return int_x if (int_x == x or x < 0) else int_x + 1
# H E L P E R F U N C T I O N S #
# fractional part of x
def frac(x):
'''return fractional part of x'''
return x - floor(x)
def inv_frac(x):
'''return inverse fractional part of x'''
return 1 - (x - floor(x)) # eg, 1 - frac(x)
# L O W L E V E L D R A W F U N C T I O N S #
# (They are too low-level to be translated into python, right?)
def set_at(surf, x, y, color):
surf.set_at((x, y), color)
def draw_pixel(surf, x, y, color, bright, blend=True):
'''draw one blended pixel with given brightness.'''
try:
other_col = surf.get_at((x, y)) if blend else (0, 0, 0, 0)
except IndexError: # pixel outside the surface
return
new_color = tuple((bright * col + (1 - bright) * pix)
for col, pix in zip(color, other_col))
# FIXME what should happen if only one, color or surf_col, has alpha?
surf.set_at((x, y), new_color)
def _drawhorzline(surf, color, x_from, y, x_to):
if x_from == x_to:
surf.set_at((x_from, y), color)
return
start, end = (x_from, x_to) if x_from <= x_to else (x_to, x_from)
for x in range(start, end + 1):
surf.set_at((x, y), color)
def _drawvertline(surf, color, x, y_from, y_to):
if y_from == y_to:
surf.set_at((x, y_from), color)
return
start, end = (y_from, y_to) if y_from <= y_to else (y_to, y_from)
for y in range(start, end + 1):
surf.set_at((x, y), color)
# I N T E R N A L D R A W L I N E F U N C T I O N S #
def _clip_and_draw_horzline(surf, color, x_from, y, x_to):
'''draw clipped horizontal line.'''
# check Y inside surf
clip = surf.get_clip()
if y < clip.y or y >= clip.y + clip.h:
return
x_from = max(x_from, clip.x)
x_to = min(x_to, clip.x + clip.w - 1)
# check any x inside surf
if x_to < clip.x or x_from >= clip.x + clip.w:
return
_drawhorzline(surf, color, x_from, y, x_to)
def _clip_and_draw_vertline(surf, color, x, y_from, y_to):
'''draw clipped vertical line.'''
# check X inside surf
clip = surf.get_clip()
if x < clip.x or x >= clip.x + clip.w:
return
y_from = max(y_from, clip.y)
y_to = min(y_to, clip.y + clip.h - 1)
# check any y inside surf
if y_to < clip.y or y_from >= clip.y + clip.h:
return
_drawvertline(surf, color, x, y_from, y_to)
# These constans xxx_EDGE are "outside-the-bounding-box"-flags
LEFT_EDGE = 0x1
RIGHT_EDGE = 0x2
BOTTOM_EDGE = 0x4
TOP_EDGE = 0x8
def encode(x, y, left, top, right, bottom):
'''returns a code that defines position with respect to a bounding box'''
# we use the fact that python interprets booleans (the inqualities)
# as 0/1, and then multiply them with the xxx_EDGE flags
return ((x < left) * LEFT_EDGE +
(x > right) * RIGHT_EDGE +
(y < top) * TOP_EDGE +
(y > bottom) * BOTTOM_EDGE)
INSIDE = lambda a: not a
ACCEPT = lambda a, b: not (a or b)
REJECT = lambda a, b: a and b
def clip_line(line, left, top, right, bottom, use_float=False):
'''Algorithm to calculate the clipped line.
We calculate the coordinates of the part of the line segment within the
bounding box (defined by left, top, right, bottom). The we write
the coordinates of the line segment into "line", much like the C-algorithm.
With `use_float` True, clip_line is usable for float-clipping.
Returns: true if the line segment cuts the bounding box (false otherwise)
'''
assert isinstance(line, list)
x1, y1, x2, y2 = line
dtype = float if use_float else int
while True:
# the coordinates are progressively modified with the codes,
# until they are either rejected or correspond to the final result.
code1 = encode(x1, y1, left, top, right, bottom)
code2 = encode(x2, y2, left, top, right, bottom)
if ACCEPT(code1, code2):
# write coordinates into "line" !
line[:] = x1, y1, x2, y2
return True
if REJECT(code1, code2):
return False
# We operate on the (x1, y1) point, and swap if it is inside the bbox:
if INSIDE(code1):
x1, x2 = x2, x1
y1, y2 = y2, y1
code1, code2 = code2, code1
if (x2 != x1):
m = (y2 - y1) / float(x2 - x1)
else:
m = 1.0
# Each case, if true, means that we are outside the border:
# calculate x1 and y1 to be the "first point" inside the bbox...
if code1 & LEFT_EDGE:
y1 += dtype((left - x1) * m)
x1 = left
elif code1 & RIGHT_EDGE:
y1 += dtype((right - x1) * m)
x1 = right
elif code1 & BOTTOM_EDGE:
if x2 != x1:
x1 += dtype((bottom - y1) / m)
y1 = bottom
elif code1 & TOP_EDGE:
if x2 != x1:
x1 += dtype((top - y1) / m)
y1 = top
def _draw_line(surf, color, x1, y1, x2, y2):
'''draw a non-horizontal line (without anti-aliasing).'''
# Variant of https://en.wikipedia.org/wiki/Bresenham's_line_algorithm
#
# This strongly differs from craw.c implementation, because we use a
# "slope" variable (instead of delta_x and delta_y) and a "error" variable.
# And we can not do pointer-arithmetic with "BytesPerPixel", like in
# the C-algorithm.
if x1 == x2:
# This case should not happen...
raise ValueError
slope = abs((y2 - y1) / (x2 - x1))
error = 0.0
if slope < 1:
# Here, it's a rather horizontal line
# 1. check in which octants we are & set init values
if x2 < x1:
x1, x2 = x2, x1
y1, y2 = y2, y1
y = y1
dy_sign = 1 if (y1 < y2) else -1
# 2. step along x coordinate
for x in range(x1, x2 + 1):
set_at(surf, x, y, color)
error += slope
if error >= 0.5:
y += dy_sign
error -= 1
else:
# Case of a rather vertical line
# 1. check in which octants we are & set init values
if y1 > y2:
x1, x2 = x2, x1
y1, y2 = y2, y1
x = x1
slope = 1 / slope
dx_sign = 1 if (x1 < x2) else -1
# 2. step along y coordinate
for y in range(y1, y2 + 1):
set_at(surf, x, y, color)
error += slope
if error >= 0.5:
x += dx_sign
error -= 1
def _draw_aaline(surf, color, from_x, from_y, to_x, to_y, blend):
'''draw an anti-aliased line.
The algorithm yields identical results with _draw_line for horizontal,
vertical or diagonal lines, and results changes smoothly when changing
any of the endpoint coordinates.
Note that this yields strange results for very short lines, eg
a line from (0, 0) to (0, 1) will draw 2 pixels, and a line from
(0, 0) to (0, 1.1) will blend 10 % on the pixel (0, 2).
'''
# The different requirements that we have on an antialiasing algorithm
# implies to make some compromises:
# 1. We want smooth evolution wrt to the 4 endpoint coordinates
# (this means also that we want a smooth evolution when the angle
# passes +/- 45°
# 2. We want the same behavior when swapping the endpoints
# 3. We want understandable results for the endpoint values
# (eg we want to avoid half-integer values to draw a simple plain
# horizontal or vertical line between two integer l endpoints)
#
# This implies to somehow make the line artificially 1 pixel longer
# and to draw a full pixel when we have the endpoints are identical.
dx = to_x - from_x
dy = to_y - from_y
if dx == 0 and dy == 0:
# For smoothness reasons, we could also do some blending here,
# but it seems overshoot...
set_at(surf, int(from_x), int(from_y), color)
return
if abs(dx) >= abs(dy):
if from_x > to_x:
from_x, to_x = to_x, from_x
from_y, to_y = to_y, from_y
dx = -dx
dy = -dy
slope = dy / dx
def draw_two_pixel(x, float_y, factor):
y = floor(float_y)
draw_pixel(surf, x, y, color, factor * inv_frac(float_y), blend)
draw_pixel(surf, x, y + 1, color, factor * frac(float_y), blend)
# A and G are respectively left and right to the "from" point, but
# with integer-x-coordinate, (and only if from_x is not integer).
# Hence they appear in following order on the line in general case:
# A from-pt G . . . to-pt S
# |------*-------|--- . . . ---|-----*------|-
G_x = ceil(from_x)
G_y = from_y + (G_x - from_x) * slope
# 1. Draw start of the segment if we have a non-integer-part
if from_x < G_x:
# this corresponds to the point "A"
draw_two_pixel(floor(from_x), G_y - slope, inv_frac(from_x))
# 2. Draw end of the segment: we add one pixel for homogenity reasons
rest = frac(to_x)
S_x = ceil(to_x)
if rest > 0:
# Again we draw only if we have a non-integer-part
S_y = from_y + slope * (dx + 1 - rest)
draw_two_pixel(S_x, S_y, rest)
else:
S_x += 1
# 3. loop for other points
for x in range(G_x, S_x):
y = G_y + slope * (x - G_x)
draw_two_pixel(x, y, 1)
else:
if from_y > to_y:
from_x, to_x = to_x, from_x
from_y, to_y = to_y, from_y
dx = -dx
dy = -dy
slope = dx / dy
def draw_two_pixel(float_x, y, factor):
x = floor(float_x)
draw_pixel(surf, x, y, color, factor * inv_frac(float_x), blend)
draw_pixel(surf, x + 1, y, color, factor * frac(float_x), blend)
G_y = ceil(from_y)
G_x = from_x + (G_y - from_y) * slope
# 1. Draw start of the segment
if from_y < G_y:
draw_two_pixel(G_x - slope, floor(from_y), inv_frac(from_y))
# 2. Draw end of the segment
rest = frac(to_y)
S_y = ceil(to_y)
if rest > 0:
S_x = from_x + slope * (dy + 1 - rest)
draw_two_pixel(S_x, S_y, rest)
else:
S_y += 1
# 3. loop for other points
for y in range(G_y, S_y):
x = G_x + slope * (y - G_y)
draw_two_pixel(x, y, 1)
# C L I P A N D D R A W L I N E F U N C T I O N S #
def _clip_and_draw_line(surf, rect, color, pts):
'''clip the line into the rectangle and draw if needed.
Returns true if anything has been drawn, else false.'''
# "pts" is a list with the four coordinates of the two endpoints
# of the line to be drawn : pts = x1, y1, x2, y2.
# The data format is like that to stay closer to the C-algorithm.
if not clip_line(pts, rect.x, rect.y, rect.x + rect.w - 1,
rect.y + rect.h - 1):
# The line segment defined by "pts" is not crossing the rectangle
return 0
if pts[1] == pts[3]: # eg y1 == y2
_drawhorzline(surf, color, pts[0], pts[1], pts[2])
elif pts[0] == pts[2]: # eg x1 == x2
_drawvertline(surf, color, pts[0], pts[1], pts[3])
else:
_draw_line(surf, color, pts[0], pts[1], pts[2], pts[3])
return 1
def _clip_and_draw_line_width(surf, rect, color, line, width):
yinc = xinc = 0
if abs(line[0] - line[2]) > abs(line[1] - line[3]):
yinc = 1
else:
xinc = 1
newpts = line[:]
if _clip_and_draw_line(surf, rect, color, newpts):
anydrawn = 1
frame = newpts[:]
else:
anydrawn = 0
frame = [10000, 10000, -10000, -10000]
for loop in range(1, width // 2 + 1):
newpts[0] = line[0] + xinc * loop
newpts[1] = line[1] + yinc * loop
newpts[2] = line[2] + xinc * loop
newpts[3] = line[3] + yinc * loop
if _clip_and_draw_line(surf, rect, color, newpts):
anydrawn = 1
frame[0] = min(newpts[0], frame[0])
frame[1] = min(newpts[1], frame[1])
frame[2] = max(newpts[2], frame[2])
frame[3] = max(newpts[3], frame[3])
if loop * 2 < width:
newpts[0] = line[0] - xinc * loop
newpts[1] = line[1] - yinc * loop
newpts[2] = line[2] - xinc * loop
newpts[3] = line[3] - yinc * loop
if _clip_and_draw_line(surf, rect, color, newpts):
anydrawn = 1
frame[0] = min(newpts[0], frame[0])
frame[1] = min(newpts[1], frame[1])
frame[2] = max(newpts[2], frame[2])
frame[3] = max(newpts[3], frame[3])
return anydrawn
def _clip_and_draw_aaline(surf, rect, color, line, blend):
'''draw anti-aliased line between two endpoints.'''
if not clip_line(line, rect.x - 1, rect.y -1, rect.x + rect.w,
rect.y + rect.h, use_float=True):
return # TODO Rect(rect.x, rect.y, 0, 0)
_draw_aaline(surf, color, line[0], line[1], line[2], line[3], blend)
return # TODO Rect(-- affected area --)
# D R A W L I N E F U N C T I O N S #
def draw_aaline(surf, color, from_point, to_point, blend=True):
'''draw anti-aliased line between two endpoints.'''
line = [from_point[0], from_point[1], to_point[0], to_point[1]]
return _clip_and_draw_aaline(surf, surf.get_clip(), color, line, blend)
def draw_line(surf, color, from_point, to_point, width=1):
'''draw anti-aliased line between two endpoints.'''
line = [from_point[0], from_point[1], to_point[0], to_point[1]]
return _clip_and_draw_line_width(surf, surf.get_clip(), color, line, width)
# M U L T I L I N E F U N C T I O N S #
def _multi_lines(surf, color, closed, points, width=1, blend=False, aaline=False):
'''draw several lines, either anti-aliased or not.'''
# The code for anti-aliased or not is almost identical, so it's factorized
length = len(points)
if length <= 2:
raise TypeError
line = [0] * 4 # store x1, y1 & x2, y2 of the lines to be drawn
xlist = [pt[0] for pt in points]
ylist = [pt[1] for pt in points]
left = right = line[0] = xlist[0]
top = bottom = line[1] = ylist[0]
for x, y in points[1:]:
left = min(left, x)
right = max(right, x)
top = min(top, y)
bottom = max(right, x)
rect = surf.get_clip()
for loop in range(1, length):
line[0] = xlist[loop - 1]
line[1] = ylist[loop - 1]
line[2] = xlist[loop]
line[3] = ylist[loop]
if aaline:
_clip_and_draw_aaline(surf, rect, color, line, blend)
else:
_clip_and_draw_line_width(surf, rect, color, line, width)
if closed:
line[0] = xlist[length - 1]
line[1] = ylist[length - 1]
line[2] = xlist[0]
line[3] = ylist[0]
if aaline:
_clip_and_draw_aaline(surf, rect, color, line, blend)
else:
_clip_and_draw_line_width(surf, rect, color, line, width)
return # TODO Rect(...)
def draw_lines(surf, color, closed, points, width=1):
'''draw several lines connected through the points.'''
return _multi_lines(surf, color, closed, points, width, aaline=False)
def draw_aalines(surf, color, closed, points, blend=True):
'''draw several anti-aliased lines connected through the points.'''
return _multi_lines(surf, color, closed, points, blend=blend, aaline=True)
def draw_polygon(surface, color, points, width):
if width:
draw_lines(surface, color, 1, points, width)
return # TODO Rect(...)
num_points = len(points)
point_x = [x for x, y in points]
point_y = [y for x, y in points]
miny = min(point_y)
maxy = max(point_y)
if miny == maxy:
minx = min(point_x)
maxx = max(point_x)
_clip_and_draw_horzline(surface, color, minx, miny, maxx)
return # TODO Rect(...)
for y in range(miny, maxy + 1):
x_intersect = []
for i in range(num_points):
i_prev = i - 1 if i else num_points - 1
y1 = point_y[i_prev]
y2 = point_y[i]
if y1 < y2:
x1 = point_x[i_prev]
x2 = point_x[i]
elif y1 > y2:
y2 = point_y[i_prev]
y1 = point_y[i]
x2 = point_x[i_prev]
x1 = point_x[i]
else: # special case handled below
continue
if ( ((y >= y1) and (y < y2)) or ((y == maxy) and (y <= y2))) :
x_sect = (y - y1) * (x2 - x1) // (y2 - y1) + x1
x_intersect.append(x_sect)
x_intersect.sort()
for i in range(0, len(x_intersect), 2):
_clip_and_draw_horzline(surface, color, x_intersect[i], y,
x_intersect[i + 1])
# special case : horizontal border lines
for i in range(num_points):
i_prev = i - 1 if i else num_points - 1
y = point_y[i]
if miny < y == point_y[i_prev] < maxy:
_clip_and_draw_horzline(surface, color, point_x[i], y, point_x[i_prev])
return # TODO Rect(...)