修改Bresenham的线算法

时间:2010-04-29 04:05:20

标签: python algorithm

我正在尝试使用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

1 个答案:

答案 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()