我正在尝试使用Bresenham的线算法来计算网格上的视野。我正在使用的代码计算线条没有问题,但我遇到问题,让它总是返回从起点到终点的行。我需要做什么才能使所有返回的行从(x0,y0)运行到(x1,y1)
def bresenham_line(self, x0, y0, x1, y1):
steep = abs(y1 - y0) > abs(x1 - x0)
if steep:
x0, y0 = y0, x0
x1, y1 = y1, x1
if x0 > x1:
x0, x1 = x1, x0
y0, y1 = y1, y0
if y0 < y1:
ystep = 1
else:
ystep = -1
deltax = x1 - x0
deltay = abs(y1 - y0)
error = -deltax / 2
y = y0
line = []
for x in range(x0, x1 + 1):
if steep:
line.append((y,x))
else:
line.append((x,y))
error = error + deltay
if error > 0:
y = y + ystep
error = error - deltax
return line
答案 0 :(得分:4)
请记住是否切换了x0和x1然后反转列表。
if x0 > x1:
x0, x1 = x1, x0
y0, y1 = y1, y0
变为
switched = False
if x0 > x1:
switched = True
x0, x1 = x1, x0
y0, y1 = y1, y0
最后,只需添加:
if switched:
line.reverse()